修改优化代码

networkd.service/main.go

@@ -141,6 +141,8 @@ func main() {
 		},
 	}
 
+	servicelib.WriteVersionFile(baseapp.EXEC_DIR, Version) //-> for 升级
+
 	prg := &program{
 		name: "NetworkManager",
 	}

servicelib/rpc_utils.go

@@ -3,7 +3,6 @@ package servicelib
 import (
 	"context"
 	"encoding/json"
-	"net"
 
 	"hnyfkj.com.cn/rtu/linux/baseapp"
 	"hnyfkj.com.cn/rtu/linux/utils/jsonrpc2"
@@ -56,12 +55,3 @@ func SaveLogConf(ctx context.Context, req *jsonrpc2.Request) *jsonrpc2.Response
 
 	return jsonrpc2.BuildResponse(req, "success", nil)
 }
-
-func IsAddrInUse(network, addr string) bool {
-	ln, err := net.Listen(network, addr)
-	if err != nil {
-		return true
-	}
-	_ = ln.Close()
-	return false
-}

servicelib/utils.go

@@ -0,0 +1,41 @@
+package servicelib
+
+import (
+	"net"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+func IsAddrInUse(network, addr string) bool {
+	ln, err := net.Listen(network, addr)
+	if err != nil {
+		return true
+	}
+	_ = ln.Close()
+	return false
+}
+
+func WriteVersionFile(dir, version string) error {
+	file := filepath.Join(dir, "version.txt")
+	tmp := file + ".tmp"
+
+	if err := os.MkdirAll(dir, 0755); err != nil {
+		return err
+	}
+
+	if err := os.WriteFile(tmp, []byte(version+"\n"), 0644); err != nil {
+		return err
+	}
+
+	return os.Rename(tmp, file)
+}
+
+func ReadVersionFile(dir string) (string, error) {
+	file := filepath.Join(dir, "version.txt")
+	data, err := os.ReadFile(file)
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSpace(string(data)), nil
+}

timesyncd.service/main.go

@@ -185,6 +185,8 @@ func main() {
 		},
 	}
 
+	servicelib.WriteVersionFile(baseapp.EXEC_DIR, Version) //-> for 升级
+
 	prg := &program{
 		name: "TimeSync",
 	}