150 lines
3.2 KiB
Go
150 lines
3.2 KiB
Go
package handlers
|
||
|
||
import (
|
||
"platform/web/auth"
|
||
edge2 "platform/web/domains/edge"
|
||
proxy2 "platform/web/domains/proxy"
|
||
g "platform/web/globals"
|
||
m "platform/web/models"
|
||
q "platform/web/queries"
|
||
s "platform/web/services"
|
||
|
||
"github.com/gofiber/fiber/v2"
|
||
"gorm.io/gen/field"
|
||
"gorm.io/gorm/clause"
|
||
)
|
||
|
||
type RegisterEdgeReq struct {
|
||
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 {
|
||
Id int32 `json:"id"`
|
||
Host string `json:"host"`
|
||
}
|
||
|
||
func OnlineEdge(c *fiber.Ctx) (err error) {
|
||
|
||
// 验证请求参数
|
||
var req = new(RegisterEdgeReq)
|
||
err = g.Validator.Validate(c, req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 挑选合适的转发服务
|
||
var fwd *m.Proxy
|
||
fwd, err = q.Proxy.
|
||
LeftJoin(q.Edge, q.Edge.ProxyID.EqCol(q.Proxy.ID), q.Edge.Status.Eq(1)).
|
||
Select(q.Proxy.ALL, q.Edge.ALL.Count().As("count")).
|
||
Where(q.Proxy.Type.Eq(int32(proxy2.TypeSelfHosted))).
|
||
Group(q.Proxy.ID).
|
||
Order(field.NewField("", "count").Desc()).
|
||
First()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 保存节点信息
|
||
var edge = &m.Edge{
|
||
Name: req.Name,
|
||
Version: int32(req.Version),
|
||
Host: c.Context().RemoteIP().String(),
|
||
Isp: int32(req.ISP),
|
||
Prov: req.Prov,
|
||
City: req.City,
|
||
ProxyID: fwd.ID,
|
||
Type: int32(edge2.TypeSelfHosted),
|
||
Status: 1,
|
||
}
|
||
err = q.Edge.Clauses(clause.OnConflict{
|
||
UpdateAll: true,
|
||
Columns: []clause.Column{{Name: "name"}},
|
||
}).Create(edge)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 返回服务地址
|
||
return c.JSON(RegisterEdgeResp{
|
||
Id: edge.ID,
|
||
Host: fwd.Host,
|
||
})
|
||
}
|
||
|
||
type OfflineEdgeReq struct {
|
||
Name string `json:"name" validate:"required"`
|
||
}
|
||
|
||
func OfflineEdge(c *fiber.Ctx) (err error) {
|
||
// 验证请求参数
|
||
var req = new(OfflineEdgeReq)
|
||
err = g.Validator.Validate(c, req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 下线转发服务
|
||
_, err = q.Edge.
|
||
Where(q.Edge.Name.Eq(req.Name)).
|
||
UpdateSimple(q.Edge.Status.Value(0))
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return c.SendStatus(fiber.StatusOK)
|
||
}
|
||
|
||
type AllEdgesAvailableReq struct {
|
||
s.EdgeFilter
|
||
Count int `json:"count"`
|
||
}
|
||
|
||
type AllEdgesAvailableRespItem struct {
|
||
Ip string `json:"ip"` // 节点地址
|
||
Port int32 `json:"port"` // 代理服务端口
|
||
Isp string `json:"isp"` // 运营商
|
||
Prov string `json:"prov"` // 省份
|
||
City string `json:"city"` // 城市
|
||
Status int32 `json:"status"` // 节点状态：0-离线，1-正常
|
||
}
|
||
|
||
func AllEdgesAvailable(c *fiber.Ctx) (err error) {
|
||
// 检查权限
|
||
_, err = auth.NewProtect(c).Payload(auth.PayloadInternalServer).Do()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 验证请求参数
|
||
var req = new(AllEdgesAvailableReq)
|
||
err = g.Validator.Validate(c, req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 获取可用的转发服务
|
||
infos, err := s.Edge.AllEdges(req.Count, req.EdgeFilter)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 返回结果
|
||
var edges = make([]AllEdgesAvailableRespItem, len(infos))
|
||
for i, info := range infos {
|
||
edges[i] = AllEdgesAvailableRespItem{
|
||
Ip: info.Host,
|
||
Port: info.ProxyPort,
|
||
Isp: edge2.ISPToStr(edge2.ISP(info.Isp)),
|
||
Prov: info.Prov,
|
||
City: info.City,
|
||
Status: info.Status,
|
||
}
|
||
}
|
||
return c.JSON(edges)
|
||
}
|