升级后删除旧的压缩包，修改版本对比文件

internal/logic/system/system.go

@@ -1,8 +1,11 @@
 package system
 
 import (
+	"context"
+
 	"github.com/ayflying/p2p/internal/service"
 	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcron"
 	"github.com/gogf/gf/v2/os/gctx"
 )
 
@@ -17,6 +20,14 @@ func init() {
 
 	getDev, _ := g.Cfg().GetWithEnv(gctx.New(), "dev")
 	if !getDev.Bool() {
+		// 每天0点检查更新
+		gcron.Add(gctx.New(), "0 0 0 * * *", func(ctx context.Context) {
+			err := service.System().CheckUpdate()
+			if err != nil {
+				g.Log().Errorf(ctx, "检查更新失败：%v", err)
+			}
+		})
+
 		err := service.System().CheckUpdate()
 		if err != nil {
 			g.Log().Errorf(gctx.New(), "检查更新失败：%v", err)
@@ -26,4 +37,4 @@ func init() {
 	}
 }
 
-func (system *sSystem) Init() {}
+func (s *sSystem) Init() {}

internal/logic/system/update.go

@@ -17,12 +17,15 @@ import (
 	"github.com/ayflying/p2p/internal/service"
 	"github.com/gogf/gf/v2/encoding/gcompress"
 	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
 	"github.com/gogf/gf/v2/os/gcmd"
 	"github.com/gogf/gf/v2/os/gctx"
 	"github.com/gogf/gf/v2/os/gfile"
 )
 
 // 本地版本号（建议从编译参数注入，如 -ldflags "-X main.version=v0.1.3"）
+const versionFile = "version.txt"
+
 var localVersion = "v0.0.0"
 
 // 对应 GitHub API 响应的核心字段（按需精简）
@@ -58,6 +61,8 @@ func (s *sSystem) Update(ctx context.Context, gzFile string) (err error) {
 	if gzFile == "" {
 		gzFile = path.Join("download", platform+".gz")
 	}
+	//结束后删除压缩包
+	defer gfile.RemoveFile(gzFile)
 
 	ext := gfile.Ext(gzFile)
 	if ext == ".zip" {
@@ -84,6 +89,16 @@ func (s *sSystem) Update(ctx context.Context, gzFile string) (err error) {
 
 // RestartSelf 实现 Windows 平台下的程序自重启
 func (s *sSystem) RestartSelf() error {
+	ctx := gctx.New()
+	// 判断是否为linux平台
+	if runtime.GOOS == "linux" {
+		err := ghttp.RestartAllServer(ctx, os.Args[0])
+		if err != nil {
+			g.Log().Errorf(ctx, "重启失败：%v", err)
+		}
+		return err
+	}
+
 	// 1. 获取当前程序的绝对路径
 	exePath, err := os.Executable()
 	if err != nil {
@@ -187,7 +202,7 @@ func (s *sSystem) CheckUpdate() (err error) {
 		return
 	}
 
-	localVersion = gfile.GetContents("download/version.txt")
+	localVersion = gfile.GetContents(versionFile)
 
 	if s.isNewVersion(localVersion, latestVersion) {
 		g.Log().Printf(ctx, "发现新版本：%s（当前版本：%s）", latestVersion, localVersion)
@@ -212,7 +227,7 @@ func (s *sSystem) CheckUpdate() (err error) {
 					return
 				}
 				// 保存最新版本号到文件
-				gfile.PutContents("download/version.txt", latestVersion)
+				gfile.PutContents(versionFile, latestVersion)
 				break
 			}
 		}