feat: 添加节点自动更新功能，支持手动触发更新

atsf_server/service/agent.go

@@ -49,6 +49,7 @@ type AgentSettings struct {
 	SyncInterval      int    `json:"sync_interval"`
 	AutoUpdate        bool   `json:"auto_update"`
 	UpdateRepo        string `json:"update_repo"`
+	UpdateNow         bool   `json:"update_now"`
 }
 
 type HeartbeatResponse struct {
@@ -63,6 +64,8 @@ type NodeView struct {
 	IP                 string     `json:"ip"`
 	AgentToken         string     `json:"agent_token"`
 	Pending            bool       `json:"pending"`
+	AutoUpdateEnabled  bool       `json:"auto_update_enabled"`
+	UpdateRequested    bool       `json:"update_requested"`
 	AgentVersion       string     `json:"agent_version"`
 	NginxVersion       string     `json:"nginx_version"`
 	Status             string     `json:"status"`
@@ -87,8 +90,10 @@ func HeartbeatNode(node *model.Node, payload AgentNodePayload) (*HeartbeatRespon
 	if err := validateAgentNodePayload(payload); err != nil {
 		return nil, err
 	}
+	updateNow := node.UpdateRequested
 	applyNodeRuntime(node, payload, true)
-	if err := model.DB.Model(node).Select("ip", "agent_version", "nginx_version", "status", "current_version", "last_seen_at", "last_error").Updates(node).Error; err != nil {
+	node.UpdateRequested = false
+	if err := model.DB.Model(node).Select("ip", "agent_version", "nginx_version", "status", "current_version", "last_seen_at", "last_error", "update_requested").Updates(node).Error; err != nil {
 		return nil, err
 	}
 	return &HeartbeatResponse{
@@ -96,8 +101,9 @@ func HeartbeatNode(node *model.Node, payload AgentNodePayload) (*HeartbeatRespon
 		AgentSettings: &AgentSettings{
 			HeartbeatInterval: common.AgentHeartbeatInterval,
 			SyncInterval:      common.AgentSyncInterval,
-			AutoUpdate:        common.AgentAutoUpdate,
+			AutoUpdate:        node.AutoUpdateEnabled,
 			UpdateRepo:        common.AgentUpdateRepo,
+			UpdateNow:         updateNow,
 		},
 	}, nil
 }

atsf_server/service/node.go

@@ -11,7 +11,8 @@ import (
 )
 
 type NodeInput struct {
-	Name string `json:"name"`
+	Name              string `json:"name"`
+	AutoUpdateEnabled bool   `json:"auto_update_enabled"`
 }
 
 type NodeBootstrapView struct {
@@ -30,11 +31,12 @@ func CreateNode(input NodeInput) (*NodeView, error) {
 		return nil, errors.New("节点名不能为空")
 	}
 	node := &model.Node{
-		Name:         name,
-		IP:           "",
-		AgentVersion: "",
-		NginxVersion: "",
-		Status:       NodeStatusPending,
+		Name:              name,
+		IP:                "",
+		AgentVersion:      "",
+		NginxVersion:      "",
+		Status:            NodeStatusPending,
+		AutoUpdateEnabled: input.AutoUpdateEnabled,
 	}
 	var err error
 	node.NodeID, err = newServerNodeID()
@@ -65,6 +67,7 @@ func UpdateNode(id uint, input NodeInput) (*NodeView, error) {
 		return nil, err
 	}
 	node.Name = name
+	node.AutoUpdateEnabled = input.AutoUpdateEnabled
 	if err = node.Update(); err != nil {
 		return nil, err
 	}
@@ -81,6 +84,19 @@ func DeleteNode(id uint) error {
 	return node.Delete()
 }
 
+func RequestNodeAgentUpdate(id uint) (*NodeView, error) {
+	node, err := model.GetNodeByID(id)
+	if err != nil {
+		return nil, err
+	}
+	node.UpdateRequested = true
+	if err = model.DB.Model(node).Update("update_requested", true).Error; err != nil {
+		return nil, err
+	}
+	common.SysLog("agent manual update requested: node_id=" + node.NodeID + " name=" + node.Name)
+	return buildNodeView(node), nil
+}
+
 func AuthenticateAgentToken(token string) (*model.Node, error) {
 	token = strings.TrimSpace(token)
 	if token == "" {
@@ -149,20 +165,22 @@ func RotateGlobalDiscoveryToken() (*NodeBootstrapView, error) {
 func buildNodeView(node *model.Node) *NodeView {
 	status := computeNodeStatus(node)
 	view := &NodeView{
-		ID:             node.ID,
-		NodeID:         node.NodeID,
-		Name:           node.Name,
-		IP:             node.IP,
-		AgentToken:     node.AgentToken,
-		AgentVersion:   node.AgentVersion,
-		NginxVersion:   node.NginxVersion,
-		Status:         status,
-		CurrentVersion: node.CurrentVersion,
-		LastSeenAt:     node.LastSeenAt,
-		LastError:      node.LastError,
-		CreatedAt:      node.CreatedAt,
-		UpdatedAt:      node.UpdatedAt,
-		Pending:        status == NodeStatusPending,
+		ID:                node.ID,
+		NodeID:            node.NodeID,
+		Name:              node.Name,
+		IP:                node.IP,
+		AgentToken:        node.AgentToken,
+		AgentVersion:      node.AgentVersion,
+		NginxVersion:      node.NginxVersion,
+		Status:            status,
+		CurrentVersion:    node.CurrentVersion,
+		LastSeenAt:        node.LastSeenAt,
+		LastError:         node.LastError,
+		CreatedAt:         node.CreatedAt,
+		UpdatedAt:         node.UpdatedAt,
+		Pending:           status == NodeStatusPending,
+		AutoUpdateEnabled: node.AutoUpdateEnabled,
+		UpdateRequested:   node.UpdateRequested,
 	}
 	return view
 }