新增win服务管理

services/index.go

@@ -0,0 +1,115 @@
+package services
+
+import (
+	"go-process-guardian/config"
+	"log"
+	"os"
+	"path/filepath"
+	"time"
+
+	"golang.org/x/sys/windows"
+	"golang.org/x/sys/windows/svc"
+	"golang.org/x/sys/windows/svc/eventlog"
+	"golang.org/x/sys/windows/svc/mgr"
+)
+
+// 服务管理 注册Windows服务 卸载服务
+
+func InstallService() {
+	appConfig, _ := config.LoadConfig()
+
+	// 连接系统的服务管理
+	serviceMessage, err := mgr.Connect()
+	if err != nil {
+		log.Println("连接电脑服务失败，请使用管理员权限运行")
+		return
+	}
+	defer serviceMessage.Disconnect()
+	// 打开服务 如果打开失败则创建服务
+	service, err := serviceMessage.OpenService(appConfig.ServiceName)
+	if err != nil {
+		// 若错误码为 ERROR_SERVICE_DOES_NOT_EXIST，表示服务不存在
+		if errCode, ok := err.(windows.Errno); ok && errCode == windows.ERROR_SERVICE_DOES_NOT_EXIST {
+			createService(serviceMessage)
+		}
+	}
+	if service != nil {
+		defer service.Close()
+	}
+}
+
+func createService(serviceMessage *mgr.Mgr) {
+	exePath, err := filepath.Abs(os.Args[0])
+	if err != nil {
+		log.Fatal(err)
+	}
+	// 创建服务
+	mgrConfig := mgr.Config{
+		DisplayName: "PMG Windows",
+		Description: "进程守护",
+		StartType:   mgr.StartAutomatic,
+	}
+	appConfig, _ := config.LoadConfig()
+	service, err := serviceMessage.CreateService(appConfig.ServiceName, exePath, mgrConfig)
+	if err != nil {
+		log.Fatal("CreateService failed:", err)
+	}
+	defer service.Close()
+	// 设置恢复策略
+	recoveryActions := []mgr.RecoveryAction{
+		{Type: mgr.ServiceRestart, Delay: 60 * time.Second},
+		{Type: mgr.ServiceRestart, Delay: 120 * time.Second},
+		{Type: mgr.NoAction},
+	}
+
+	if err := service.SetRecoveryActions(recoveryActions, 86400); err != nil {
+		log.Println("Warning: failed to set recovery actions:", err)
+	}
+	// 创建事件日志源
+	if err := eventlog.InstallAsEventCreate(appConfig.ServiceName, eventlog.Error|eventlog.Warning|eventlog.Info); err != nil {
+		log.Println("Warning: failed to create event source:", err)
+	}
+
+	log.Println("Service installed successfully")
+}
+
+func UninstallService() {
+	appConfig, _ := config.LoadConfig()
+	// 连接系统的服务管理
+	serviceMessage, err := mgr.Connect()
+	if err != nil {
+		log.Println("连接电脑服务失败")
+		log.Println(err)
+		return
+	}
+	defer serviceMessage.Disconnect()
+	// 打开服务
+	service, err := serviceMessage.OpenService(appConfig.ServiceName)
+	if err != nil {
+		log.Fatal("Service not installed:", err)
+		return
+	}
+	defer service.Close()
+	// 停止服务
+	if status, err := service.Control(svc.Stop); err == nil {
+		timeout := time.Now().Add(10 * time.Second)
+		for status.State != svc.Stopped && time.Now().Before(timeout) {
+			time.Sleep(500 * time.Millisecond)
+			status, err = service.Query()
+			if err != nil {
+				break
+			}
+		}
+	}
+	// 删除服务
+	if err := service.Delete(); err != nil {
+		log.Fatal("DeleteService failed:", err)
+	}
+
+	// 移除事件日志源
+	if err := eventlog.Remove(appConfig.ServiceName); err != nil {
+		log.Println("Warning: failed to remove event source:", err)
+	}
+
+	log.Println("Service uninstalled successfully")
+}