146 lines
2.6 KiB
Go
146 lines
2.6 KiB
Go
package handlers
|
|
|
|
import (
|
|
"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
|
|
|
|
type OnlineProxyReq struct {
|
|
Name string `json:"name" validate:"required"`
|
|
Version int `json:"version" validate:"required"`
|
|
}
|
|
|
|
type OnlineProxyResp struct {
|
|
Id int32 `json:"id"`
|
|
}
|
|
|
|
func OnlineProxy(c *fiber.Ctx) (err error) {
|
|
|
|
// 检查接口权限
|
|
_, err = auth2.NewProtect(c).Payload(
|
|
auth2.PayloadInternalServer,
|
|
).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 验证请求参数
|
|
var req = new(OnlineProxyReq)
|
|
err = g.Validator.Validate(c, req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 创建代理
|
|
ip := c.Context().RemoteIP()
|
|
var proxy = &m.Proxy{
|
|
Name: req.Name,
|
|
Version: int32(req.Version),
|
|
Host: ip.String(),
|
|
Type: int32(proxy2.TypeSelfHosted),
|
|
Status: 1,
|
|
}
|
|
err = q.Proxy.
|
|
Clauses(clause.OnConflict{
|
|
UpdateAll: true,
|
|
Columns: []clause.Column{
|
|
{Name: q.Proxy.Name.ColumnName().String()},
|
|
},
|
|
}).
|
|
Create(proxy)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
slog.Debug("注册转发服务", "ip", ip, "id", proxy.ID)
|
|
return c.JSON(&OnlineProxyResp{
|
|
Id: proxy.ID,
|
|
})
|
|
}
|
|
|
|
// endregion
|
|
|
|
// region OfflineProxy
|
|
|
|
type OfflineProxyReq struct {
|
|
Name string `json:"name" validate:"required"`
|
|
}
|
|
|
|
func OfflineProxy(c *fiber.Ctx) (err error) {
|
|
// 检查接口权限
|
|
_, err = auth2.NewProtect(c).Payload(
|
|
auth2.PayloadInternalServer,
|
|
).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 验证请求参数
|
|
var req = new(OfflineProxyReq)
|
|
err = g.Validator.Validate(c, req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 下线转发服务
|
|
_, err = q.Proxy.
|
|
Where(q.Proxy.Name.Eq(req.Name)).
|
|
UpdateSimple(q.Proxy.Status.Value(0))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// endregion
|
|
|
|
// region AssignProxyFwdPort
|
|
|
|
type AssignProxyFwdPortReq struct {
|
|
Proxy int32 `json:"proxy" validate:"required"`
|
|
Edge int32 `json:"edge" validate:"required"`
|
|
Port uint16 `json:"port" validate:"required"`
|
|
}
|
|
|
|
func AssignProxyFwdPort(c *fiber.Ctx) (err error) {
|
|
// 检查接口权限
|
|
_, err = auth2.NewProtect(c).Payload(
|
|
auth2.PayloadInternalServer,
|
|
).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 验证请求参数
|
|
var req = new(AssignProxyFwdPortReq)
|
|
err = g.Validator.Validate(c, req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 更新边缘节点端口分配状态
|
|
_, err = q.Edge.
|
|
Where(q.Edge.ID.Eq(req.Edge)).
|
|
UpdateSimple(
|
|
q.Edge.ProxyID.Value(req.Proxy),
|
|
q.Edge.ProxyPort.Value(int32(req.Port)),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// endregion
|