feat: 增加自升级处理逻辑

cmd/agent/config/config.go

@@ -5,7 +5,7 @@ import "time"
 var (
 	WorkDir = "/linkfog/agent/"
 
-	BinDir            = "bin"
+	BackupDir         = "backup"
 	PluginDir         = "plugin"
 	LogDir            = "log"
 	LogFileName       = "agent.log"

cmd/agent/core/agent.go

@@ -26,7 +26,8 @@ var AgentSvc = &Agent{}
 var monitorDaemon *monitor_daemon.MonitorDaemon
 
 type Agent struct {
-	IsStop bool
+	IsStop   bool
+	cmdline0 string
 }
 
 func (a *Agent) Init(env svc.Environment) error {
@@ -46,6 +47,8 @@ func (a *Agent) Init(env svc.Environment) error {
 
 	if option.Opt.EnableMonitorDaemon {
 		config.LogFileName = config.DaemonLogFileName
+	} else {
+		a.cmdline0 = os.Args[0]
 	}
 
 	// 日志设置
@@ -79,7 +82,7 @@ func (a *Agent) Start() error {
 			monitor_daemon.WithMaxInterval(option.Opt.MonitorDefaultMaxInterval),   // 设置重新拉起进程的最大时间间隔，防止频繁重启带来的系统压力
 			monitor_daemon.WithIncrInterval(option.Opt.MonitorDefaultIncrInterval), // 设置重新拉起进程的时间间隔，防止频繁重启带来的系统压力
 			monitor_daemon.WithMaxRetryTimes(option.Opt.MonitorDefaultMaxRetryTimes),
-			monitor_daemon.WithAgentBackupPath(filepath.Join(config.BinDir, option.Opt.MonitorDefaultBackupFileName)),
+			monitor_daemon.WithAgentBackupPath(filepath.Join(config.BackupDir, option.Opt.MonitorDefaultBackupFileName)),
 		}
 		if option.Opt.EnableProcAbnormalCb {
 			opts = append(opts, monitor_daemon.WithIsProcAbnormalCallback(IsProcAbnormal))
@@ -157,6 +160,10 @@ func initSvcCfg() error {
 	}
 	global.SubscribePrefixInfo = global.DeviceSerialNumber + "/publish/"
 
+	// NOTE: 自测使用
+	if config.Edition == "dev" {
+		config.WorkDir = option.Opt.WorkDir
+	}
 	// 时间修正
 	if !iTime.SynchronizeSystemTime() {
 		return fmt.Errorf("synchronization system time failed")

cmd/agent/core/chat_msg.go

 package core
 
 import (
-	"agent/module"
+	"agent/pkg/file"
+	"os"
+	"path/filepath"
+	"strings"
+	"syscall"
 	"time"
 
+	"agent/cmd/agent/config"
 	"agent/cmd/agent/global"
+	"agent/cmd/agent/option"
+	"agent/module"
 	"linkfog.com/public/lib/l"
 )
 
@@ -28,6 +35,8 @@ func (a *Agent) chatMsgProcess(msg *global.Message) {
 
 	switch msg.Key {
 	case global.ConsumerTopicAgentSelfUpgrade:
+		go a.SelfUpgradeProcess()
+		return
 	case global.ConsumerTopicPluginUpgrade:
 		fallthrough
 	case global.ConsumerTopicStartupPlugin:
@@ -40,3 +49,31 @@ func (a *Agent) chatMsgProcess(msg *global.Message) {
 		}
 	}
 }
+
+func (a *Agent) SelfUpgradeProcess() {
+	execFile := a.cmdline0[strings.LastIndex(a.cmdline0, "/")+1:]
+	// 备份
+	err := os.Rename(a.cmdline0, config.BackupDir+"/"+execFile)
+	if err != nil {
+		l.Error(err)
+		return
+	}
+	// 下载
+	err = global.DownloadFile(option.Opt.DownloadURL+"/"+execFile, config.WorkDir, execFile)
+	if err != nil {
+		l.Errorf("download file err:%v", err)
+		return
+	}
+
+	// 增加执行权限
+	err = file.SetFileExecPerm(filepath.Join(config.WorkDir, execFile))
+	if err != nil {
+		l.Error(err)
+	}
+	// TODO md5校验
+	// kill自身
+	err = syscall.Kill(os.Getpid(), syscall.SIGTERM)
+	if err != nil {
+		l.Error(err)
+	}
+}

cmd/agent/global/global.go

@@ -87,14 +87,14 @@ func SetWorkDir() error {
 		return err
 	}
 
-	config.BinDir = path.Join(config.WorkDir, config.BinDir)
-	err = os.MkdirAll(config.BinDir, 0644)
+	config.BackupDir = path.Join(config.WorkDir, config.BackupDir)
+	err = os.MkdirAll(config.BackupDir, 0644)
 	if err != nil {
 		return err
 	}
 
 	config.PluginDir = path.Join(config.WorkDir, config.PluginDir)
-	err = os.MkdirAll(config.BinDir, 0644)
+	err = os.MkdirAll(config.PluginDir, 0644)
 	if err != nil {
 		return err
 	}

cmd/agent/global/service.go

@@ -6,6 +6,8 @@ import (
 	"crypto/sha512"
 	"encoding/hex"
 	"fmt"
+	"io"
+	"net/http"
 	"os"
 	"path"
 	"path/filepath"
@@ -85,3 +87,23 @@ func produceDeviceID() (string, error) {
 
 	return string(snByte), nil
 }
+
+func DownloadFile(url, filepath, name string) error {
+	resp, err := http.Get(url)
+	if err != nil {
+		return fmt.Errorf("download file failed, err:%v", err)
+	}
+	defer resp.Body.Close()
+
+	file, err := os.Create(path.Join(filepath, name))
+	if err != nil {
+		return fmt.Errorf("create file failed, err:%v", err)
+	}
+	defer file.Close()
+
+	_, err = io.Copy(file, resp.Body)
+	if err != nil {
+		return fmt.Errorf("copy file failed, err:%v", err)
+	}
+	return nil
+}

cmd/agent/option/option.go

@@ -43,6 +43,8 @@ type options struct {
 	// 调试信息
 	MQTTBrokerURL  string
 	MQTTBrokerPort int
+	DownloadURL    string
+	WorkDir        string
 }
 
 const (
@@ -73,8 +75,11 @@ const (
 
 func flagSet() {
 	if config.Edition == "dev" {
+		// NOTE: 自测使用
 		flag.StringVar(&Opt.MQTTBrokerURL, "mqtt-url", "tcp://broker.hivemq.com", "MQTT broker url")
 		flag.IntVar(&Opt.MQTTBrokerPort, "mqtt-port", 1883, "MQTT broker port")
+		flag.StringVar(&Opt.DownloadURL, "download-url", "http://fae-cdn.linkfog.cn/tmp", "download url")
+		flag.StringVar(&Opt.WorkDir, "work-dir", "", "work dir")
 	}
 	flag.BoolVar(&Opt.Usage, ArgUsage, false, "custom format usage")
 	flag.BoolVar(&Opt.PrintVersion, ArgVersion, false, "print version and exit")

module/backend/backend.go

@@ -40,6 +40,7 @@ func New(opts ...BackendOpt) *Backend {
 		clientOpts.AddBroker(fmt.Sprintf("ssl://%s", b.BrokerIP))
 	}
 
+	// NOTE: 自测使用
 	if config.Edition == "dev" {
 		clientOpts.AddBroker(fmt.Sprintf("%s:%d", option.Opt.MQTTBrokerURL, option.Opt.MQTTBrokerPort))
 	}
@@ -101,7 +102,7 @@ func (b *Backend) Start() error {
 					b.client.Publish(topic, 1, false, []byte(msg.Payload))
 				}
 			case <-b.signal:
-				l.Info("MQTT publish exit")
+				l.Info("MQTT exit")
 				return
 			}
 		}

pkg/file/file.go

+package file
+
+import (
+	"os"
+)
+
+func SetFileExecPerm(file string) error {
+	fileInfo, err := os.Stat(file)
+	if err != nil {
+		return err
+	}
+	newMode := fileInfo.Mode() | 0111
+	err = os.Chmod(file, newMode)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}