优化代码

web-ui.service/main.go

@@ -4,8 +4,10 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"rtu_linux_services/servicelib"
 
 	"github.com/kardianos/service"
+	"hnyfkj.com.cn/rtu/linux/baseapp"
 )
 
 var (
@@ -14,6 +16,7 @@ var (
 )
 
 type program struct {
+	name  string
 	webUI *WebUI
 }
 
@@ -49,7 +52,7 @@ func main() {
 	svcFlag := flag.String("service", "", "Control the yfkj-webui service.")
 	flag.Parse()
 
-	cfg := &service.Config{
+	svcConfig := &service.Config{
 		Name:        "yfkj-webui",
 		DisplayName: "yfkj-webui",
 		Description: "Yunfei Device Web Management UI",
@@ -58,9 +61,16 @@ func main() {
 		},
 	}
 
-	svc, err := service.New(&program{}, cfg)
+	servicelib.WriteVersionFile(baseapp.EXEC_DIR, Version) //-> for 升级
+
+	prg := &program{
+		name: "WebUI",
+	}
+
+	svc, err := service.New(prg, svcConfig)
 	if err != nil {
-		fmt.Println(err)
+		fmt.Printf("%s.service creation failed: %v\n",
+			svcConfig.Name, err)
 		os.Exit(1)
 	}