perf: install

Author: y1jiong

internal/client/public.go

@@ -32,8 +32,7 @@ type AsyncServiceCallback func(enabledServices common.Enable, ipv4, ipv6 string)
 
 func Install() (err error) {
 	if common.IsWindows() {
-		err = errors.New("windows 暂不支持安装到系统")
-		return
+		return errors.New("windows 暂不支持安装到系统")
 	}
 	// 注册系统服务
 	if Client.CheckCycleMinutes == 0 {
@@ -42,7 +41,7 @@ func Install() (err error) {
 	}
 	wd, err := os.Getwd()
 	if err != nil {
-		return err
+		return
 	}
 	serviceContent := []byte(
 		"[Unit]\n" +
@@ -58,24 +57,23 @@ func Install() (err error) {
 			"WantedBy=multi-user.target\n")
 	err = os.WriteFile(installPath, serviceContent, 0600)
 	if err != nil {
-		return err
+		return
 	}
 	log.Println("可以使用 systemctl 管理 " + ProjName + " 服务了")
 	return
 }
 
 func Uninstall() (err error) {
 	if common.IsWindows() {
-		err = errors.New("windows 暂不支持安装到系统")
-		return
+		return errors.New("windows 暂不支持安装到系统")
 	}
 	wd, err := os.Getwd()
 	if err != nil {
-		return err
+		return
 	}
 	err = os.Remove(installPath)
 	if err != nil {
-		return err
+		return
 	}
 	log.Println("卸载服务成功")
 	log.Println("若要完全删除，请移步到 " + wd + " 和 " + ConfDirectoryName + " 完全删除")

internal/server/public.go

@@ -188,49 +188,46 @@ func GetClientIP(req *http.Request) (ipAddr string) {
 
 func Install() (err error) {
 	if common.IsWindows() {
-		err = errors.New("windows 暂不支持安装到系统")
-	} else {
-		// 注册系统服务
-		wd, err := os.Getwd()
-		if err != nil {
-			return err
-		}
-		serviceContent := []byte(
-			"[Unit]\n" +
-				"Description=" + RunningName + " Service\n" +
-				"After=network-online.target\n\n" +
-				"[Service]\n" +
-				"Type=simple\n" +
-				"WorkingDirectory=" + wd +
-				"\nExecStart=" + wd + "/" + RunningName + " -c " + ConfDirectoryName +
-				"\nRestart=on-failure\n" +
-				"RestartSec=2\n\n" +
-				"[Install]\n" +
-				"WantedBy=multi-user.target\n")
-		err = os.WriteFile(InstallPath, serviceContent, 0600)
-		if err != nil {
-			return err
-
-		}
-		log.Println("可以使用 systemctl 控制 " + RunningName + " 服务了")
+		return errors.New("windows 暂不支持安装到系统")
+	}
+	// 注册系统服务
+	wd, err := os.Getwd()
+	if err != nil {
+		return
+	}
+	serviceContent := []byte(
+		"[Unit]\n" +
+			"Description=" + RunningName + " Service\n" +
+			"After=network-online.target\n\n" +
+			"[Service]\n" +
+			"Type=simple\n" +
+			"WorkingDirectory=" + wd +
+			"\nExecStart=" + wd + "/" + RunningName + " -c " + ConfDirectoryName +
+			"\nRestart=on-failure\n" +
+			"RestartSec=2\n\n" +
+			"[Install]\n" +
+			"WantedBy=multi-user.target\n")
+	err = os.WriteFile(InstallPath, serviceContent, 0600)
+	if err != nil {
+		return
 	}
+	log.Println("可以使用 systemctl 控制 " + RunningName + " 服务了")
 	return
 }
 
 func Uninstall() (err error) {
 	if common.IsWindows() {
-		err = errors.New("windows 暂不支持安装到系统")
-	} else {
-		wd, err := os.Getwd()
-		if err != nil {
-			return err
-		}
-		err = os.Remove(InstallPath)
-		if err != nil {
-			return err
-		}
-		log.Println("卸载服务成功")
-		log.Println("若要完全删除，请移步到 " + wd + " 和 " + ConfDirectoryName + " 完全删除")
+		return errors.New("windows 暂不支持安装到系统")
+	}
+	wd, err := os.Getwd()
+	if err != nil {
+		return
+	}
+	err = os.Remove(InstallPath)
+	if err != nil {
+		return
 	}
+	log.Println("卸载服务成功")
+	log.Println("若要完全删除，请移步到 " + wd + " 和 " + ConfDirectoryName + " 完全删除")
 	return
 }