lanhu/platform
2
0
Fork 0
Code Issues Actions Packages Releases 13 Activity

实现云端控制的动态节点分配逻辑

Browse Source
This commit is contained in:
luorijun
2025-03-28 10:03:29 +08:00
parent e337a9c08e
commit e61f0bef32
16 changed files with 1313 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
web/queries/channel.gen.go
View File

@@ -29,12 +29,14 @@ func newChannel(db *gorm.DB, opts ...gen.DOOption) channel {
_channel.ALL = field.NewAsterisk(tableName)
_channel.ID = field.NewInt32(tableName, "id")
_channel.UserID = field.NewInt32(tableName, "user_id")
_channel.ProxyID = field.NewInt32(tableName, "proxy_id")
_channel.NodeID = field.NewInt32(tableName, "node_id")
_channel.ProxyPort = field.NewInt32(tableName, "proxy_port")
_channel.Protocol = field.NewString(tableName, "protocol")
_channel.UserHost = field.NewString(tableName, "user_host")
_channel.NodePort = field.NewInt32(tableName, "node_port")
_channel.NodeHost = field.NewString(tableName, "node_host")
_channel.AuthIP = field.NewBool(tableName, "auth_ip")
_channel.AuthPass = field.NewBool(tableName, "auth_pass")
_channel.Protocol = field.NewString(tableName, "protocol")
_channel.Username = field.NewString(tableName, "username")
_channel.Password = field.NewString(tableName, "password")
_channel.Expiration = field.NewTime(tableName, "expiration")
@@ -53,12 +55,14 @@ type channel struct {
ALL field.Asterisk
ID field.Int32 // 通道ID
UserID field.Int32 // 用户ID
ProxyID field.Int32 // 代理ID
NodeID field.Int32 // 节点ID
ProxyPort field.Int32 // 转发端口
Protocol field.String // 协议
UserHost field.String // 用户地址
NodePort field.Int32 // 节点端口
NodeHost field.String // 节点地址
AuthIP field.Bool // IP认证
AuthPass field.Bool // 密码认证
Protocol field.String // 协议
Username field.String // 用户名
Password field.String // 密码
Expiration field.Time // 过期时间
@@ -83,12 +87,14 @@ func (c *channel) updateTableName(table string) *channel {
c.ALL = field.NewAsterisk(table)
c.ID = field.NewInt32(table, "id")
c.UserID = field.NewInt32(table, "user_id")
c.ProxyID = field.NewInt32(table, "proxy_id")
c.NodeID = field.NewInt32(table, "node_id")
c.ProxyPort = field.NewInt32(table, "proxy_port")
c.Protocol = field.NewString(table, "protocol")
c.UserHost = field.NewString(table, "user_host")
c.NodePort = field.NewInt32(table, "node_port")
c.NodeHost = field.NewString(table, "node_host")
c.AuthIP = field.NewBool(table, "auth_ip")
c.AuthPass = field.NewBool(table, "auth_pass")
c.Protocol = field.NewString(table, "protocol")
c.Username = field.NewString(table, "username")
c.Password = field.NewString(table, "password")
c.Expiration = field.NewTime(table, "expiration")
@@ -111,15 +117,17 @@ func (c *channel) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
}
func (c *channel) fillFieldMap() {
c.fieldMap = make(map[string]field.Expr, 14)
c.fieldMap = make(map[string]field.Expr, 16)
c.fieldMap["id"] = c.ID
c.fieldMap["user_id"] = c.UserID
c.fieldMap["proxy_id"] = c.ProxyID
c.fieldMap["node_id"] = c.NodeID
c.fieldMap["proxy_port"] = c.ProxyPort
c.fieldMap["protocol"] = c.Protocol
c.fieldMap["user_host"] = c.UserHost
c.fieldMap["node_port"] = c.NodePort
c.fieldMap["node_host"] = c.NodeHost
c.fieldMap["auth_ip"] = c.AuthIP
c.fieldMap["auth_pass"] = c.AuthPass
c.fieldMap["protocol"] = c.Protocol
c.fieldMap["username"] = c.Username
c.fieldMap["password"] = c.Password
c.fieldMap["expiration"] = c.Expiration

8
web/queries/gen.go
View File

@@ -28,6 +28,7 @@ var (
Node *node
Permission *permission
Product *product
Proxy *proxy
Refund *refund
Resource *resource
ResourcePps *resourcePps
@@ -54,6 +55,7 @@ func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
Node = &Q.Node
Permission = &Q.Permission
Product = &Q.Product
Proxy = &Q.Proxy
Refund = &Q.Refund
Resource = &Q.Resource
ResourcePps = &Q.ResourcePps
@@ -81,6 +83,7 @@ func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
Node: newNode(db, opts...),
Permission: newPermission(db, opts...),
Product: newProduct(db, opts...),
Proxy: newProxy(db, opts...),
Refund: newRefund(db, opts...),
Resource: newResource(db, opts...),
ResourcePps: newResourcePps(db, opts...),
@@ -109,6 +112,7 @@ type Query struct {
Node node
Permission permission
Product product
Proxy proxy
Refund refund
Resource resource
ResourcePps resourcePps
@@ -138,6 +142,7 @@ func (q *Query) clone(db *gorm.DB) *Query {
Node: q.Node.clone(db),
Permission: q.Permission.clone(db),
Product: q.Product.clone(db),
Proxy: q.Proxy.clone(db),
Refund: q.Refund.clone(db),
Resource: q.Resource.clone(db),
ResourcePps: q.ResourcePps.clone(db),
@@ -174,6 +179,7 @@ func (q *Query) ReplaceDB(db *gorm.DB) *Query {
Node: q.Node.replaceDB(db),
Permission: q.Permission.replaceDB(db),
Product: q.Product.replaceDB(db),
Proxy: q.Proxy.replaceDB(db),
Refund: q.Refund.replaceDB(db),
Resource: q.Resource.replaceDB(db),
ResourcePps: q.ResourcePps.replaceDB(db),
@@ -200,6 +206,7 @@ type queryCtx struct {
Node *nodeDo
Permission *permissionDo
Product *productDo
Proxy *proxyDo
Refund *refundDo
Resource *resourceDo
ResourcePps *resourcePpsDo
@@ -226,6 +233,7 @@ func (q *Query) WithContext(ctx context.Context) *queryCtx {
Node: q.Node.WithContext(ctx),
Permission: q.Permission.WithContext(ctx),
Product: q.Product.WithContext(ctx),
Proxy: q.Proxy.WithContext(ctx),
Refund: q.Refund.WithContext(ctx),
Resource: q.Resource.WithContext(ctx),
ResourcePps: q.ResourcePps.WithContext(ctx),

38
web/queries/node.gen.go
View File

@@ -28,12 +28,17 @@ func newNode(db *gorm.DB, opts ...gen.DOOption) node {
tableName := _node.nodeDo.TableName()
_node.ALL = field.NewAsterisk(tableName)
_node.ID = field.NewInt32(tableName, "id")
_node.Name = field.NewString(tableName, "name")
_node.Version = field.NewInt32(tableName, "version")
_node.FwdPort = field.NewInt32(tableName, "fwd_port")
_node.Name = field.NewString(tableName, "name")
_node.Host = field.NewString(tableName, "host")
_node.Isp = field.NewString(tableName, "isp")
_node.Prov = field.NewString(tableName, "prov")
_node.City = field.NewString(tableName, "city")
_node.ProxyID = field.NewInt32(tableName, "proxy_id")
_node.ProxyPort = field.NewInt32(tableName, "proxy_port")
_node.Status = field.NewInt32(tableName, "status")
_node.Rtt = field.NewInt32(tableName, "rtt")
_node.Loss = field.NewInt32(tableName, "loss")
_node.CreatedAt = field.NewTime(tableName, "created_at")
_node.UpdatedAt = field.NewTime(tableName, "updated_at")
_node.DeletedAt = field.NewField(tableName, "deleted_at")
@@ -48,12 +53,17 @@ type node struct {
ALL field.Asterisk
ID field.Int32 // 节点ID
Name field.String // 节点名称
Version field.Int32 // 节点版本
FwdPort field.Int32 // 转发端口
Name field.String // 节点名称
Host field.String // 节点地址
Isp field.String // 运营商
Prov field.String // 省份
City field.String // 城市
ProxyID field.Int32 // 代理ID
ProxyPort field.Int32 // 代理端口
Status field.Int32 // 节点状态：1-正常，0-离线
Rtt field.Int32 // 延迟
Loss field.Int32 // 丢包率
CreatedAt field.Time // 创建时间
UpdatedAt field.Time // 更新时间
DeletedAt field.Field // 删除时间
@@ -74,12 +84,17 @@ func (n node) As(alias string) *node {
func (n *node) updateTableName(table string) *node {
n.ALL = field.NewAsterisk(table)
n.ID = field.NewInt32(table, "id")
n.Name = field.NewString(table, "name")
n.Version = field.NewInt32(table, "version")
n.FwdPort = field.NewInt32(table, "fwd_port")
n.Name = field.NewString(table, "name")
n.Host = field.NewString(table, "host")
n.Isp = field.NewString(table, "isp")
n.Prov = field.NewString(table, "prov")
n.City = field.NewString(table, "city")
n.ProxyID = field.NewInt32(table, "proxy_id")
n.ProxyPort = field.NewInt32(table, "proxy_port")
n.Status = field.NewInt32(table, "status")
n.Rtt = field.NewInt32(table, "rtt")
n.Loss = field.NewInt32(table, "loss")
n.CreatedAt = field.NewTime(table, "created_at")
n.UpdatedAt = field.NewTime(table, "updated_at")
n.DeletedAt = field.NewField(table, "deleted_at")
@@ -99,14 +114,19 @@ func (n *node) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
}
func (n *node) fillFieldMap() {
n.fieldMap = make(map[string]field.Expr, 10)
n.fieldMap = make(map[string]field.Expr, 15)
n.fieldMap["id"] = n.ID
n.fieldMap["name"] = n.Name
n.fieldMap["version"] = n.Version
n.fieldMap["fwd_port"] = n.FwdPort
n.fieldMap["name"] = n.Name
n.fieldMap["host"] = n.Host
n.fieldMap["isp"] = n.Isp
n.fieldMap["prov"] = n.Prov
n.fieldMap["city"] = n.City
n.fieldMap["proxy_id"] = n.ProxyID
n.fieldMap["proxy_port"] = n.ProxyPort
n.fieldMap["status"] = n.Status
n.fieldMap["rtt"] = n.Rtt
n.fieldMap["loss"] = n.Loss
n.fieldMap["created_at"] = n.CreatedAt
n.fieldMap["updated_at"] = n.UpdatedAt
n.fieldMap["deleted_at"] = n.DeletedAt

347
web/queries/proxy.gen.go Normal file
View File

@@ -0,0 +1,347 @@
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
package queries
import (
"context"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"platform/web/models"
)
func newProxy(db *gorm.DB, opts ...gen.DOOption) proxy {
_proxy := proxy{}
_proxy.proxyDo.UseDB(db, opts...)
_proxy.proxyDo.UseModel(&models.Proxy{})
tableName := _proxy.proxyDo.TableName()
_proxy.ALL = field.NewAsterisk(tableName)
_proxy.ID = field.NewInt32(tableName, "id")
_proxy.Version = field.NewInt32(tableName, "version")
_proxy.Name = field.NewString(tableName, "name")
_proxy.Host = field.NewString(tableName, "host")
_proxy.Type = field.NewInt32(tableName, "type")
_proxy.CreatedAt = field.NewTime(tableName, "created_at")
_proxy.UpdatedAt = field.NewTime(tableName, "updated_at")
_proxy.DeletedAt = field.NewField(tableName, "deleted_at")
_proxy.fillFieldMap()
return _proxy
}
type proxy struct {
proxyDo
ALL field.Asterisk
ID field.Int32 // 代理服务ID
Version field.Int32 // 代理服务版本
Name field.String // 代理服务名称
Host field.String // 代理服务地址
Type field.Int32 // 代理服务类型：0-自有，1-三方
CreatedAt field.Time // 创建时间
UpdatedAt field.Time // 更新时间
DeletedAt field.Field // 删除时间
fieldMap map[string]field.Expr
}
func (p proxy) Table(newTableName string) *proxy {
p.proxyDo.UseTable(newTableName)
return p.updateTableName(newTableName)
}
func (p proxy) As(alias string) *proxy {
p.proxyDo.DO = *(p.proxyDo.As(alias).(*gen.DO))
return p.updateTableName(alias)
}
func (p *proxy) updateTableName(table string) *proxy {
p.ALL = field.NewAsterisk(table)
p.ID = field.NewInt32(table, "id")
p.Version = field.NewInt32(table, "version")
p.Name = field.NewString(table, "name")
p.Host = field.NewString(table, "host")
p.Type = field.NewInt32(table, "type")
p.CreatedAt = field.NewTime(table, "created_at")
p.UpdatedAt = field.NewTime(table, "updated_at")
p.DeletedAt = field.NewField(table, "deleted_at")
p.fillFieldMap()
return p
}
func (p *proxy) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := p.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (p *proxy) fillFieldMap() {
p.fieldMap = make(map[string]field.Expr, 8)
p.fieldMap["id"] = p.ID
p.fieldMap["version"] = p.Version
p.fieldMap["name"] = p.Name
p.fieldMap["host"] = p.Host
p.fieldMap["type"] = p.Type
p.fieldMap["created_at"] = p.CreatedAt
p.fieldMap["updated_at"] = p.UpdatedAt
p.fieldMap["deleted_at"] = p.DeletedAt
}
func (p proxy) clone(db *gorm.DB) proxy {
p.proxyDo.ReplaceConnPool(db.Statement.ConnPool)
return p
}
func (p proxy) replaceDB(db *gorm.DB) proxy {
p.proxyDo.ReplaceDB(db)
return p
}
type proxyDo struct{ gen.DO }
func (p proxyDo) Debug() *proxyDo {
return p.withDO(p.DO.Debug())
}
func (p proxyDo) WithContext(ctx context.Context) *proxyDo {
return p.withDO(p.DO.WithContext(ctx))
}
func (p proxyDo) ReadDB() *proxyDo {
return p.Clauses(dbresolver.Read)
}
func (p proxyDo) WriteDB() *proxyDo {
return p.Clauses(dbresolver.Write)
}
func (p proxyDo) Session(config *gorm.Session) *proxyDo {
return p.withDO(p.DO.Session(config))
}
func (p proxyDo) Clauses(conds ...clause.Expression) *proxyDo {
return p.withDO(p.DO.Clauses(conds...))
}
func (p proxyDo) Returning(value interface{}, columns ...string) *proxyDo {
return p.withDO(p.DO.Returning(value, columns...))
}
func (p proxyDo) Not(conds ...gen.Condition) *proxyDo {
return p.withDO(p.DO.Not(conds...))
}
func (p proxyDo) Or(conds ...gen.Condition) *proxyDo {
return p.withDO(p.DO.Or(conds...))
}
func (p proxyDo) Select(conds ...field.Expr) *proxyDo {
return p.withDO(p.DO.Select(conds...))
}
func (p proxyDo) Where(conds ...gen.Condition) *proxyDo {
return p.withDO(p.DO.Where(conds...))
}
func (p proxyDo) Order(conds ...field.Expr) *proxyDo {
return p.withDO(p.DO.Order(conds...))
}
func (p proxyDo) Distinct(cols ...field.Expr) *proxyDo {
return p.withDO(p.DO.Distinct(cols...))
}
func (p proxyDo) Omit(cols ...field.Expr) *proxyDo {
return p.withDO(p.DO.Omit(cols...))
}
func (p proxyDo) Join(table schema.Tabler, on ...field.Expr) *proxyDo {
return p.withDO(p.DO.Join(table, on...))
}
func (p proxyDo) LeftJoin(table schema.Tabler, on ...field.Expr) *proxyDo {
return p.withDO(p.DO.LeftJoin(table, on...))
}
func (p proxyDo) RightJoin(table schema.Tabler, on ...field.Expr) *proxyDo {
return p.withDO(p.DO.RightJoin(table, on...))
}
func (p proxyDo) Group(cols ...field.Expr) *proxyDo {
return p.withDO(p.DO.Group(cols...))
}
func (p proxyDo) Having(conds ...gen.Condition) *proxyDo {
return p.withDO(p.DO.Having(conds...))
}
func (p proxyDo) Limit(limit int) *proxyDo {
return p.withDO(p.DO.Limit(limit))
}
func (p proxyDo) Offset(offset int) *proxyDo {
return p.withDO(p.DO.Offset(offset))
}
func (p proxyDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *proxyDo {
return p.withDO(p.DO.Scopes(funcs...))
}
func (p proxyDo) Unscoped() *proxyDo {
return p.withDO(p.DO.Unscoped())
}
func (p proxyDo) Create(values ...*models.Proxy) error {
if len(values) == 0 {
return nil
}
return p.DO.Create(values)
}
func (p proxyDo) CreateInBatches(values []*models.Proxy, batchSize int) error {
return p.DO.CreateInBatches(values, batchSize)
}
// Save : !!! underlying implementation is different with GORM
// The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
func (p proxyDo) Save(values ...*models.Proxy) error {
if len(values) == 0 {
return nil
}
return p.DO.Save(values)
}
func (p proxyDo) First() (*models.Proxy, error) {
if result, err := p.DO.First(); err != nil {
return nil, err
} else {
return result.(*models.Proxy), nil
}
}
func (p proxyDo) Take() (*models.Proxy, error) {
if result, err := p.DO.Take(); err != nil {
return nil, err
} else {
return result.(*models.Proxy), nil
}
}
func (p proxyDo) Last() (*models.Proxy, error) {
if result, err := p.DO.Last(); err != nil {
return nil, err
} else {
return result.(*models.Proxy), nil
}
}
func (p proxyDo) Find() ([]*models.Proxy, error) {
result, err := p.DO.Find()
return result.([]*models.Proxy), err
}
func (p proxyDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*models.Proxy, err error) {
buf := make([]*models.Proxy, 0, batchSize)
err = p.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
defer func() { results = append(results, buf...) }()
return fc(tx, batch)
})
return results, err
}
func (p proxyDo) FindInBatches(result *[]*models.Proxy, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return p.DO.FindInBatches(result, batchSize, fc)
}
func (p proxyDo) Attrs(attrs ...field.AssignExpr) *proxyDo {
return p.withDO(p.DO.Attrs(attrs...))
}
func (p proxyDo) Assign(attrs ...field.AssignExpr) *proxyDo {
return p.withDO(p.DO.Assign(attrs...))
}
func (p proxyDo) Joins(fields ...field.RelationField) *proxyDo {
for _, _f := range fields {
p = *p.withDO(p.DO.Joins(_f))
}
return &p
}
func (p proxyDo) Preload(fields ...field.RelationField) *proxyDo {
for _, _f := range fields {
p = *p.withDO(p.DO.Preload(_f))
}
return &p
}
func (p proxyDo) FirstOrInit() (*models.Proxy, error) {
if result, err := p.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*models.Proxy), nil
}
}
func (p proxyDo) FirstOrCreate() (*models.Proxy, error) {
if result, err := p.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*models.Proxy), nil
}
}
func (p proxyDo) FindByPage(offset int, limit int) (result []*models.Proxy, count int64, err error) {
result, err = p.Offset(offset).Limit(limit).Find()
if err != nil {
return
}
if size := len(result); 0 < limit && 0 < size && size < limit {
count = int64(size + offset)
return
}
count, err = p.Offset(-1).Limit(-1).Count()
return
}
func (p proxyDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = p.Count()
if err != nil {
return
}
err = p.Offset(offset).Limit(limit).Scan(result)
return
}
func (p proxyDo) Scan(result interface{}) (err error) {
return p.DO.Scan(result)
}
func (p proxyDo) Delete(models ...*models.Proxy) (result gen.ResultInfo, err error) {
return p.DO.Delete(models)
}
func (p *proxyDo) withDO(do gen.Dao) *proxyDo {
p.DO = *do.(*gen.DO)
return p
}