feat: Implement node management features including creation, updating, and deletion

- Added CreateNode, UpdateNode, and DeleteNode functions in the controller for managing nodes.
- Introduced NodeInput struct for input validation during node creation and updates.
- Enhanced the Node model to include DiscoveryToken and Pending status.
- Updated the AgentRegister and AgentHeartbeat functions to utilize the new node management logic.
- Refactored the API router to include new routes for node management.
- Improved the frontend Node component to support node creation, editing, and deletion with appropriate UI feedback.
- Added tests to ensure the new functionality works as expected.
This commit is contained in:
ryan
2026-03-10 15:33:06 +08:00
parent 861d759f97
commit 05e75549d2
16 changed files with 916 additions and 296 deletions
+21 -10
View File
@@ -1,7 +1,7 @@
package middleware
import (
"gin-template/common"
"gin-template/service"
"github.com/gin-gonic/gin"
"net/http"
)
@@ -9,15 +9,8 @@ import (
func AgentAuth() func(c *gin.Context) {
return func(c *gin.Context) {
token := c.GetHeader("X-Agent-Token")
if common.AgentToken == "" {
c.JSON(http.StatusUnauthorized, gin.H{
"success": false,
"message": "Agent Token 未配置",
})
c.Abort()
return
}
if token == "" || token != common.AgentToken {
node, err := service.AuthenticateAgentToken(token)
if err != nil {
c.JSON(http.StatusUnauthorized, gin.H{
"success": false,
"message": "无权进行此操作,Agent Token 无效",
@@ -25,6 +18,24 @@ func AgentAuth() func(c *gin.Context) {
c.Abort()
return
}
c.Set("agent_node", node)
c.Next()
}
}
func AgentDiscoveryAuth() func(c *gin.Context) {
return func(c *gin.Context) {
token := c.GetHeader("X-Agent-Token")
node, err := service.AuthenticateDiscoveryToken(token)
if err != nil {
c.JSON(http.StatusUnauthorized, gin.H{
"success": false,
"message": "无权进行此操作,Discovery Token 无效",
})
c.Abort()
return
}
c.Set("discovery_node", node)
c.Next()
}
}