新增服务网络建立流程图文档，优化边缘节点和代理服务的注册逻辑

web/handlers/edge.go

@@ -1,20 +1,23 @@
 package handlers
 
 import (
-	"github.com/gofiber/fiber/v2"
-	"gorm.io/gen/field"
+	edge2 "platform/web/domains/edge"
 	proxy2 "platform/web/domains/proxy"
 	g "platform/web/globals"
 	m "platform/web/models"
 	q "platform/web/queries"
+
+	"github.com/gofiber/fiber/v2"
+	"gorm.io/gen/field"
+	"gorm.io/gorm/clause"
 )
 
 type RegisterEdgeReq struct {
-	Name    int    `json:"name" validate:"required"`
-	Version int    `json:"version" validate:"required"`
-	ISP     int    `json:"isp" validate:"required"`
-	Prov    string `json:"prov" validate:"required"`
-	City    string `json:"city" validate:"required"`
+	Name    string    `json:"name" validate:"required"`
+	Version int       `json:"version" validate:"required"`
+	ISP     edge2.ISP `json:"isp"`
+	Prov    string    `json:"prov"`
+	City    string    `json:"city"`
 }
 
 type RegisterEdgeResp struct {
@@ -44,13 +47,19 @@ func OnlineEdge(c *fiber.Ctx) (err error) {
 	}
 
 	// 保存节点信息
-	err = q.Edge.Save(&m.Edge{
+	err = q.Edge.Clauses(clause.OnConflict{
+		UpdateAll: true,
+		Columns:   []clause.Column{{Name: "name"}},
+	}).Create(&m.Edge{
+		Name:    req.Name,
 		Version: int32(req.Version),
-		Host:    c.IP(),
+		Host:    c.Context().RemoteIP().String(),
 		Isp:     int32(req.ISP),
 		Prov:    req.Prov,
 		City:    req.City,
 		ProxyID: fwd.ID,
+		Type:    int32(edge2.TypeSelfHosted),
+		Status:  1,
 	})
 	if err != nil {
 		return err

web/handlers/proxy.go

@@ -1,14 +1,15 @@
 package handlers
 
 import (
-	"github.com/gofiber/fiber/v2"
-	"gorm.io/gorm/clause"
 	"log/slog"
 	auth2 "platform/web/auth"
 	proxy2 "platform/web/domains/proxy"
 	g "platform/web/globals"
 	m "platform/web/models"
 	q "platform/web/queries"
+
+	"github.com/gofiber/fiber/v2"
+	"gorm.io/gorm/clause"
 )
 
 // region OnlineProxy
@@ -36,7 +37,8 @@ func OnlineProxy(c *fiber.Ctx) (err error) {
 	}
 
 	// 创建代理
-	slog.Debug("注册转发服务", "ip", c.IP())
+	ip := c.Context().RemoteIP()
+	slog.Debug("注册转发服务", "ip", ip)
 	err = q.Proxy.
 		Clauses(clause.OnConflict{
 			UpdateAll: true,
@@ -47,7 +49,7 @@ func OnlineProxy(c *fiber.Ctx) (err error) {
 		Create(&m.Proxy{
 			Name:    req.Name,
 			Version: int32(req.Version),
-			Host:    c.IP(),
+			Host:    ip.String(),
 			Type:    int32(proxy2.TypeSelfHosted),
 			Status:  1,
 		})

web/router.go

@@ -71,4 +71,9 @@ func ApplyRouters(app *fiber.App) {
 	// 节点
 	edge := api.Group("/edge")
 	edge.Post("/online", handlers.OnlineEdge)
+
+	// 临时
+	app.Get("/test", func(c *fiber.Ctx) error {
+		return c.JSON(c.GetReqHeaders())
+	})
 }