3 Commits

Author SHA1 Message Date
03f762e910 更新完后需要修改文件可执行权限 2025-10-31 10:05:10 +08:00
84723248cd 删除更新的临时压缩包 2025-10-30 19:35:21 +08:00
d3f3687201 使用新的解压targz压缩的方法 2025-10-30 19:34:28 +08:00
2 changed files with 66 additions and 4 deletions

View File

@@ -1,9 +1,12 @@
package system
import (
"archive/tar"
"compress/gzip"
"context"
"encoding/json"
"fmt"
"io"
"log"
"net/http"
"os"
@@ -69,12 +72,14 @@ func (s *sSystem) Update(ctx context.Context, gzFile string) (err error) {
g.Log().Debugf(ctx, "zip解压%v到%v", gzFile, gfile.Dir(runFile))
err = gcompress.UnZipFile(gzFile, gfile.Dir(runFile))
} else {
g.Log().Debugf(ctx, "gzip解压%v到%v", gzFile, runFile)
err = gcompress.UnGzipFile(gzFile, runFile)
g.Log().Debugf(ctx, "gzip解压%v到%v", gzFile, gfile.Dir(runFile))
err = s.UnTarGz(gzFile, gfile.Dir(runFile))
}
if err != nil {
return
}
//修改文件权限为755
err = gfile.Chmod(runFile, 0755)
go func() {
log.Println("5秒后开始重启...")
@@ -87,6 +92,64 @@ func (s *sSystem) Update(ctx context.Context, gzFile string) (err error) {
return
}
// UnTarGz 解压tar.gz文件到指定目录
func (s *sSystem) UnTarGz(tarGzFileName, targetDir string) (err error) {
// 打开tar.gz文件
file, err := os.Open(tarGzFileName)
if err != nil {
return err
}
defer file.Close()
// 创建gzip reader
gzr, err := gzip.NewReader(file)
if err != nil {
return err
}
defer gzr.Close()
// 创建tar reader
tr := tar.NewReader(gzr)
// 遍历tar中的每个文件
for {
hdr, err := tr.Next()
if err == io.EOF {
// 到达文件末尾,退出循环
break
}
if err != nil {
return err
}
// 构建解压后的文件路径
targetPath := targetDir + string(os.PathSeparator) + hdr.Name
// 如果是目录,创建目录
if hdr.Typeflag == tar.TypeDir {
err := os.MkdirAll(targetPath, 0755)
if err != nil {
return err
}
continue
}
// 如果是文件,创建文件并写入内容
outFile, err := os.Create(targetPath)
if err != nil {
return err
}
defer outFile.Close()
_, err = io.Copy(outFile, tr)
if err != nil {
return err
}
}
return
}
// RestartSelf 实现 Windows 平台下的程序自重启
func (s *sSystem) RestartSelf() error {
ctx := gctx.New()

View File

@@ -10,7 +10,6 @@ import (
"github.com/gogf/gf/v2/os/gcmd"
"github.com/gogf/gf/v2/os/gfile"
"github.com/gogf/gf/v2/os/gtime"
//步骤1加载驱动
_ "github.com/gogf/gf/contrib/nosql/redis/v2"
@@ -24,7 +23,7 @@ var (
func main() {
g.Log().Infof(ctx, "启动文件最后修改时间:%v", gtime.New(gfile.MTime(gcmd.GetArg(0).String())).String())
g.Dump("v1.0.0.2")
//g.Dump("v1.0.0.2")
if ok := gfile.Exists("runtime"); !ok {
gfile.Mkdir("runtime")