forked from nudtpc/pcm-kubernetes
53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"time"
|
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-participant-kubernetes/internal/svc"
|
|
"gitlink.org.cn/jcce-pcm/pcm-participant-kubernetes/kubernetes"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type RestartStatefulSetLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
}
|
|
|
|
func NewRestartStatefulSetLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RestartStatefulSetLogic {
|
|
return &RestartStatefulSetLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
// RestartStatefulSet 重启sts
|
|
func (l *RestartStatefulSetLogic) RestartStatefulSet(in *kubernetes.DeploymentDetailReq) (*kubernetes.Resp, error) {
|
|
//实现kubectl rollout restart StatefulSets功能
|
|
patchOpt := metav1.PatchOptions{FieldManager: "kubectl-rollout"}
|
|
|
|
dt := time.Now()
|
|
|
|
patchInfo := fmt.Sprintf(`{"spec":{"template":{"metadata":{"annotations":{"kubectl.kubernetes.io/restartedAt":"%s"}}}}}`, dt.String())
|
|
|
|
_, err := l.svcCtx.ClientSet.AppsV1().StatefulSets(in.Namespace).Patch(l.ctx, in.Name, types.StrategicMergePatchType, []byte(patchInfo), patchOpt)
|
|
if err != nil {
|
|
return &kubernetes.Resp{
|
|
Code: "500",
|
|
Msg: "failed",
|
|
Data: err.Error(),
|
|
}, nil
|
|
}
|
|
|
|
return &kubernetes.Resp{
|
|
Code: "200",
|
|
Msg: "success",
|
|
}, nil
|
|
}
|