启动、暂停应用功能实现

go.mod

@@ -8,7 +8,7 @@ require (
 	github.com/robfig/cron/v3 v3.0.1
 	github.com/sirupsen/logrus v1.9.3
 	github.com/zeromicro/go-zero v1.6.0
-	gitlink.org.cn/jcce-pcm/pcm-coordinator v0.1.17
+	gitlink.org.cn/jcce-pcm/pcm-coordinator v0.1.18
 	gitlink.org.cn/jcce-pcm/utils v0.0.2
 	google.golang.org/grpc v1.60.0
 	google.golang.org/protobuf v1.31.0

go.sum

@@ -198,8 +198,8 @@ github.com/yuin/gopher-lua v1.1.0 h1:BojcDhfyDWgU2f2TOzYK/g5p2gxMrku8oupLDqlnSqE
 github.com/yuin/gopher-lua v1.1.0/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw=
 github.com/zeromicro/go-zero v1.6.0 h1:UwSOR1lGZ2g7L0S07PM8RoneAcubtd5x//EfbuNucQ0=
 github.com/zeromicro/go-zero v1.6.0/go.mod h1:E9GCFPb0SwsTKFBcFr9UynGvXiDMmfc6fI5F15vqvAQ=
-gitlink.org.cn/jcce-pcm/pcm-coordinator v0.1.17 h1:c/uV63RuQul3QrY1DaLOUNYIx/KPSSGFH8pYwDVX1so=
-gitlink.org.cn/jcce-pcm/pcm-coordinator v0.1.17/go.mod h1:0dHxKCTjH3ud1qRQZjE6EqXSs3NTOpiHWTpaip4mrWE=
+gitlink.org.cn/jcce-pcm/pcm-coordinator v0.1.18 h1:XBr95J1xv834u93hJX2MW3syAiViq9V6KAlP7Y5i5dE=
+gitlink.org.cn/jcce-pcm/pcm-coordinator v0.1.18/go.mod h1:0dHxKCTjH3ud1qRQZjE6EqXSs3NTOpiHWTpaip4mrWE=
 gitlink.org.cn/jcce-pcm/utils v0.0.2 h1:Stif8W9C9TOCS2hw4g+OlOywDrsVYNrkiyKfBrWkT0w=
 gitlink.org.cn/jcce-pcm/utils v0.0.2/go.mod h1:u8PTlBpzUyOlbQJgfSiutq91q/JtrJIQiPNDe4S/pGs=
 go.etcd.io/etcd/api/v3 v3.5.11 h1:B54KwXbWDHyD3XYAwprxNzTe7vlhR69LuBgZnMVvS7E=

internal/cron/applycron.go

@@ -37,3 +37,23 @@ func RestartList(svc *svc.ServiceContext) {
 		pkg.Restart(info.Name, info.Namespace, info.Kind, pkg.KClients[info.ParticipantName].ClientSet, pkg.KClients[info.ParticipantName].DynamicClient)
 	}
 }
+
+func PauseList(svc *svc.ServiceContext) {
+	listResp, err := svc.ParticipantRpc.PauseList(context.Background(), &participantservice.ApplyListReq{})
+	if err != nil {
+		return
+	}
+	for _, info := range listResp.InfoList {
+		pkg.Pause(info.Name, info.Namespace, info.Kind, pkg.KClients[info.ParticipantName].ClientSet, pkg.KClients[info.ParticipantName].DynamicClient)
+	}
+}
+
+func StartList(svc *svc.ServiceContext) {
+	listResp, err := svc.ParticipantRpc.StartList(context.Background(), &participantservice.ApplyListReq{})
+	if err != nil {
+		return
+	}
+	for _, info := range listResp.InfoList {
+		pkg.Start(info.Name, info.Namespace, info.Kind, pkg.KClients[info.ParticipantName].ClientSet, pkg.KClients[info.ParticipantName].DynamicClient)
+	}
+}

internal/cron/cron.go

@@ -31,5 +31,7 @@ func AddCronGroup(svc *svc.ServiceContext) {
 		ApplyList(svc)
 		DeleteList(svc)
 		RestartList(svc)
+		PauseList(svc)
+		StartList(svc)
 	})
 }

internal/pkg/pause.go

@@ -0,0 +1,39 @@
+package pkg
+
+import (
+	"context"
+	"github.com/sirupsen/logrus"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/dynamic"
+	kubernetes2 "k8s.io/client-go/kubernetes"
+)
+
+func Pause(name, namespace, kind string, clientSet *kubernetes2.Clientset, dynamicClient dynamic.Interface) {
+	// 缩放副本数量为0
+	rs := int32(0)
+	switch kind {
+	case "Deployment":
+		deployment, err := clientSet.AppsV1().Deployments(namespace).Get(context.Background(), name, metav1.GetOptions{})
+		if err == nil {
+			deployment.Spec.Replicas = &rs
+			annotations := deployment.GetAnnotations()
+			annotations["deploy.cloud.sealos.io/pause"] = `{"target":"","value":""}`
+			_, err = clientSet.AppsV1().Deployments(namespace).Update(context.Background(), deployment, metav1.UpdateOptions{})
+			if err != nil {
+				logrus.Error(err)
+			}
+		}
+	case "StatefulSet":
+		sts, err := clientSet.AppsV1().StatefulSets(namespace).Get(context.Background(), name, metav1.GetOptions{})
+		if err == nil {
+			sts.Spec.Replicas = &rs
+			annotations := sts.GetAnnotations()
+			annotations["deploy.cloud.sealos.io/pause"] = `{"target":"","value":""}`
+			_, err = clientSet.AppsV1().StatefulSets(namespace).Update(context.Background(), sts, metav1.UpdateOptions{})
+
+			if err != nil {
+				logrus.Error(err)
+			}
+		}
+	}
+}

internal/pkg/restart.go

@@ -12,10 +12,9 @@ import (
 )
 
 func Restart(name, namespace, kind string, clientSet *kubernetes2.Clientset, dynamicClient dynamic.Interface) {
-	fmt.Println("重启应用成功")
 	switch kind {
 	case "Deployment":
-		//实现kubectl rollout restart StatefulSets功能
+		//实现kubectl rollout restart Deployment功能
 		patchOpt := metav1.PatchOptions{FieldManager: "kubectl-rollout"}
 
 		dt := time.Now()

internal/pkg/start.go

@@ -0,0 +1,46 @@
+package pkg
+
+import (
+	"context"
+	"github.com/sirupsen/logrus"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/dynamic"
+	kubernetes2 "k8s.io/client-go/kubernetes"
+	"strconv"
+)
+
+func Start(name, namespace, kind string, clientSet *kubernetes2.Clientset, dynamicClient dynamic.Interface) {
+	switch kind {
+	case "Deployment":
+		deployment, err := clientSet.AppsV1().Deployments(namespace).Get(context.Background(), name, metav1.GetOptions{})
+		minReplicas := deployment.Annotations["deploy.cloud.sealos.io/minReplicas"]
+		if minReplicas != "" {
+			rs := StringToInt32(minReplicas)
+			deployment.Spec.Replicas = &rs
+		}
+		annotations := deployment.GetAnnotations()
+		delete(annotations, "deploy.cloud.sealos.io/pause")
+		deployment, err = clientSet.AppsV1().Deployments(namespace).Update(context.Background(), deployment, metav1.UpdateOptions{})
+		if err != nil {
+			logrus.Error(err)
+		}
+	case "StatefulSet":
+		sts, err := clientSet.AppsV1().StatefulSets(namespace).Get(context.Background(), name, metav1.GetOptions{})
+		minReplicas := sts.Annotations["deploy.cloud.sealos.io/minReplicas"]
+		if minReplicas != "" {
+			rs := StringToInt32(minReplicas)
+			sts.Spec.Replicas = &rs
+		}
+		annotations := sts.GetAnnotations()
+		delete(annotations, "deploy.cloud.sealos.io/pause")
+		sts, err = clientSet.AppsV1().StatefulSets(namespace).Update(context.Background(), sts, metav1.UpdateOptions{})
+		if err != nil {
+			logrus.Error(err)
+		}
+	}
+}
+
+func StringToInt32(str string) int32 {
+	j, _ := strconv.ParseInt(str, 10, 32)
+	return int32(j)
+}