初步完成高防相关管理对象

This commit is contained in:
GoEdgeLab
2023-02-22 17:34:05 +08:00
parent 13975fe6f8
commit da41495b7b
31 changed files with 580 additions and 7 deletions

View File

@@ -0,0 +1,71 @@
package models
import (
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
)
const (
ADNetworkStateEnabled = 1 // 已启用
ADNetworkStateDisabled = 0 // 已禁用
)
type ADNetworkDAO dbs.DAO
func NewADNetworkDAO() *ADNetworkDAO {
return dbs.NewDAO(&ADNetworkDAO{
DAOObject: dbs.DAOObject{
DB: Tea.Env,
Table: "edgeADNetworks",
Model: new(ADNetwork),
PkName: "id",
},
}).(*ADNetworkDAO)
}
var SharedADNetworkDAO *ADNetworkDAO
func init() {
dbs.OnReady(func() {
SharedADNetworkDAO = NewADNetworkDAO()
})
}
// EnableADNetwork 启用条目
func (this *ADNetworkDAO) EnableADNetwork(tx *dbs.Tx, id uint32) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADNetworkStateEnabled).
Update()
return err
}
// DisableADNetwork 禁用条目
func (this *ADNetworkDAO) DisableADNetwork(tx *dbs.Tx, id int64) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADNetworkStateDisabled).
Update()
return err
}
// FindEnabledADNetwork 查找启用中的条目
func (this *ADNetworkDAO) FindEnabledADNetwork(tx *dbs.Tx, id int64) (*ADNetwork, error) {
result, err := this.Query(tx).
Pk(id).
State(ADNetworkStateEnabled).
Find()
if result == nil {
return nil, err
}
return result.(*ADNetwork), err
}
// FindADNetworkName 根据主键查找名称
func (this *ADNetworkDAO) FindADNetworkName(tx *dbs.Tx, id uint32) (string, error) {
return this.Query(tx).
Pk(id).
Result("name").
FindStringCol("")
}

View File

@@ -0,0 +1,6 @@
package models_test
import (
_ "github.com/go-sql-driver/mysql"
_ "github.com/iwind/TeaGo/bootstrap"
)

View File

@@ -0,0 +1,24 @@
package models
// ADNetwork 高防线路
type ADNetwork struct {
Id uint32 `field:"id"` // ID
IsOn bool `field:"isOn"` // 是否启用
Name string `field:"name"` // 名称
Description string `field:"description"` // 描述
Order uint32 `field:"order"` // 排序
State uint8 `field:"state"` // 状态
}
type ADNetworkOperator struct {
Id any // ID
IsOn any // 是否启用
Name any // 名称
Description any // 描述
Order any // 排序
State any // 状态
}
func NewADNetworkOperator() *ADNetworkOperator {
return &ADNetworkOperator{}
}

View File

@@ -0,0 +1 @@
package models

View File

@@ -0,0 +1,71 @@
package models
import (
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
)
const (
ADPackageStateEnabled = 1 // 已启用
ADPackageStateDisabled = 0 // 已禁用
)
type ADPackageDAO dbs.DAO
func NewADPackageDAO() *ADPackageDAO {
return dbs.NewDAO(&ADPackageDAO{
DAOObject: dbs.DAOObject{
DB: Tea.Env,
Table: "edgeADPackages",
Model: new(ADPackage),
PkName: "id",
},
}).(*ADPackageDAO)
}
var SharedADPackageDAO *ADPackageDAO
func init() {
dbs.OnReady(func() {
SharedADPackageDAO = NewADPackageDAO()
})
}
// EnableADPackage 启用条目
func (this *ADPackageDAO) EnableADPackage(tx *dbs.Tx, id uint32) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADPackageStateEnabled).
Update()
return err
}
// DisableADPackage 禁用条目
func (this *ADPackageDAO) DisableADPackage(tx *dbs.Tx, id int64) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADPackageStateDisabled).
Update()
return err
}
// FindEnabledADPackage 查找启用中的条目
func (this *ADPackageDAO) FindEnabledADPackage(tx *dbs.Tx, id int64) (*ADPackage, error) {
result, err := this.Query(tx).
Pk(id).
State(ADPackageStateEnabled).
Find()
if result == nil {
return nil, err
}
return result.(*ADPackage), err
}
// FindADPackageName 根据主键查找名称
func (this *ADPackageDAO) FindADPackageName(tx *dbs.Tx, id uint32) (string, error) {
return this.Query(tx).
Pk(id).
Result("name").
FindStringCol("")
}

View File

@@ -0,0 +1,6 @@
package models_test
import (
_ "github.com/go-sql-driver/mysql"
_ "github.com/iwind/TeaGo/bootstrap"
)

View File

@@ -0,0 +1,54 @@
package models
import (
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
)
const (
ADPackageInstanceStateEnabled = 1 // 已启用
ADPackageInstanceStateDisabled = 0 // 已禁用
)
type ADPackageInstanceDAO dbs.DAO
func NewADPackageInstanceDAO() *ADPackageInstanceDAO {
return dbs.NewDAO(&ADPackageInstanceDAO{
DAOObject: dbs.DAOObject{
DB: Tea.Env,
Table: "edgeADPackageInstances",
Model: new(ADPackageInstance),
PkName: "id",
},
}).(*ADPackageInstanceDAO)
}
var SharedADPackageInstanceDAO *ADPackageInstanceDAO
func init() {
dbs.OnReady(func() {
SharedADPackageInstanceDAO = NewADPackageInstanceDAO()
})
}
// EnableADPackageInstance 启用条目
func (this *ADPackageInstanceDAO) EnableADPackageInstance(tx *dbs.Tx, id int64) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADPackageInstanceStateEnabled).
Update()
return err
}
// FindEnabledADPackageInstance 查找启用中的条目
func (this *ADPackageInstanceDAO) FindEnabledADPackageInstance(tx *dbs.Tx, id int64) (*ADPackageInstance, error) {
result, err := this.Query(tx).
Pk(id).
State(ADPackageInstanceStateEnabled).
Find()
if result == nil {
return nil, err
}
return result.(*ADPackageInstance), err
}

View File

@@ -0,0 +1,6 @@
package models_test
import (
_ "github.com/go-sql-driver/mysql"
_ "github.com/iwind/TeaGo/bootstrap"
)

View File

@@ -0,0 +1,36 @@
package models
import "github.com/iwind/TeaGo/dbs"
// ADPackageInstance 高防实例
type ADPackageInstance struct {
Id uint32 `field:"id"` // ID
IsOn bool `field:"isOn"` // 是否启用
PackageId uint32 `field:"packageId"` // 规格ID
ClusterId uint32 `field:"clusterId"` // 集群ID
NodeIds dbs.JSON `field:"nodeIds"` // 节点ID
IpAddresses dbs.JSON `field:"ipAddresses"` // IP地址
UserId uint64 `field:"userId"` // 用户ID
UserDayTo string `field:"userDayTo"` // 用户有效期YYYYMMDD
UserInstanceId uint64 `field:"userInstanceId"` // 用户实例ID
State uint8 `field:"state"` // 状态
ObjectCodes dbs.JSON `field:"objectCodes"` // 防护对象
}
type ADPackageInstanceOperator struct {
Id any // ID
IsOn any // 是否启用
PackageId any // 规格ID
ClusterId any // 集群ID
NodeIds any // 节点ID
IpAddresses any // IP地址
UserId any // 用户ID
UserDayTo any // 用户有效期YYYYMMDD
UserInstanceId any // 用户实例ID
State any // 状态
ObjectCodes any // 防护对象
}
func NewADPackageInstanceOperator() *ADPackageInstanceOperator {
return &ADPackageInstanceOperator{}
}

View File

@@ -0,0 +1 @@
package models

View File

@@ -0,0 +1,32 @@
package models
// ADPackage 高防产品规格
type ADPackage struct {
Id uint32 `field:"id"` // ID
IsOn bool `field:"isOn"` // 是否启用
NetworkId uint32 `field:"networkId"` // 线路ID
ProtectionBandwidthSize uint32 `field:"protectionBandwidthSize"` // 防护带宽尺寸
ProtectionBandwidthUnit string `field:"protectionBandwidthUnit"` // 防护带宽单位
ProtectionBandwidthBits uint64 `field:"protectionBandwidthBits"` // 防护带宽比特
ServerBandwidthSize uint32 `field:"serverBandwidthSize"` // 业务带宽尺寸
ServerBandwidthUnit string `field:"serverBandwidthUnit"` // 业务带宽单位
ServerBandwidthBits uint64 `field:"serverBandwidthBits"` // 业务带宽比特
State uint8 `field:"state"` // 状态
}
type ADPackageOperator struct {
Id any // ID
IsOn any // 是否启用
NetworkId any // 线路ID
ProtectionBandwidthSize any // 防护带宽尺寸
ProtectionBandwidthUnit any // 防护带宽单位
ProtectionBandwidthBits any // 防护带宽比特
ServerBandwidthSize any // 业务带宽尺寸
ServerBandwidthUnit any // 业务带宽单位
ServerBandwidthBits any // 业务带宽比特
State any // 状态
}
func NewADPackageOperator() *ADPackageOperator {
return &ADPackageOperator{}
}

View File

@@ -0,0 +1 @@
package models

View File

@@ -0,0 +1,63 @@
package models
import (
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
)
const (
ADPackagePeriodStateEnabled = 1 // 已启用
ADPackagePeriodStateDisabled = 0 // 已禁用
)
type ADPackagePeriodDAO dbs.DAO
func NewADPackagePeriodDAO() *ADPackagePeriodDAO {
return dbs.NewDAO(&ADPackagePeriodDAO{
DAOObject: dbs.DAOObject{
DB: Tea.Env,
Table: "edgeADPackagePeriods",
Model: new(ADPackagePeriod),
PkName: "id",
},
}).(*ADPackagePeriodDAO)
}
var SharedADPackagePeriodDAO *ADPackagePeriodDAO
func init() {
dbs.OnReady(func() {
SharedADPackagePeriodDAO = NewADPackagePeriodDAO()
})
}
// EnableADPackagePeriod 启用条目
func (this *ADPackagePeriodDAO) EnableADPackagePeriod(tx *dbs.Tx, id uint32) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADPackagePeriodStateEnabled).
Update()
return err
}
// DisableADPackagePeriod 禁用条目
func (this *ADPackagePeriodDAO) DisableADPackagePeriod(tx *dbs.Tx, id int64) error {
_, err := this.Query(tx).
Pk(id).
Set("state", ADPackagePeriodStateDisabled).
Update()
return err
}
// FindEnabledADPackagePeriod 查找启用中的条目
func (this *ADPackagePeriodDAO) FindEnabledADPackagePeriod(tx *dbs.Tx, id int64) (*ADPackagePeriod, error) {
result, err := this.Query(tx).
Pk(id).
State(ADPackagePeriodStateEnabled).
Find()
if result == nil {
return nil, err
}
return result.(*ADPackagePeriod), err
}

View File

@@ -0,0 +1,6 @@
package models_test
import (
_ "github.com/go-sql-driver/mysql"
_ "github.com/iwind/TeaGo/bootstrap"
)

View File

@@ -0,0 +1,24 @@
package models
// ADPackagePeriod 高防产品有效期
type ADPackagePeriod struct {
Id uint32 `field:"id"` // ID
IsOn bool `field:"isOn"` // 是否启用
Count uint32 `field:"count"` // 数量
Unit string `field:"unit"` // 单位month, year
Months uint32 `field:"months"` // 月数
State uint8 `field:"state"` // 状态
}
type ADPackagePeriodOperator struct {
Id any // ID
IsOn any // 是否启用
Count any // 数量
Unit any // 单位month, year
Months any // 月数
State any // 状态
}
func NewADPackagePeriodOperator() *ADPackagePeriodOperator {
return &ADPackagePeriodOperator{}
}

View File

@@ -0,0 +1 @@
package models

View File

@@ -0,0 +1,28 @@
package models
import (
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
)
type ADPackagePriceDAO dbs.DAO
func NewADPackagePriceDAO() *ADPackagePriceDAO {
return dbs.NewDAO(&ADPackagePriceDAO{
DAOObject: dbs.DAOObject{
DB: Tea.Env,
Table: "edgeADPackagePrices",
Model: new(ADPackagePrice),
PkName: "id",
},
}).(*ADPackagePriceDAO)
}
var SharedADPackagePriceDAO *ADPackagePriceDAO
func init() {
dbs.OnReady(func() {
SharedADPackagePriceDAO = NewADPackagePriceDAO()
})
}

View File

@@ -0,0 +1,6 @@
package models_test
import (
_ "github.com/go-sql-driver/mysql"
_ "github.com/iwind/TeaGo/bootstrap"
)

View File

@@ -0,0 +1,22 @@
package models
// ADPackagePrice 流量包价格
type ADPackagePrice struct {
Id uint32 `field:"id"` // ID
PackageId uint32 `field:"packageId"` // 高防产品ID
PeriodId uint32 `field:"periodId"` // 有效期ID
Price float64 `field:"price"` // 价格
DiscountPrice float64 `field:"discountPrice"` // 折后价格
}
type ADPackagePriceOperator struct {
Id any // ID
PackageId any // 高防产品ID
PeriodId any // 有效期ID
Price any // 价格
DiscountPrice any // 折后价格
}
func NewADPackagePriceOperator() *ADPackagePriceOperator {
return &ADPackagePriceOperator{}
}

View File

@@ -0,0 +1 @@
package models

View File

@@ -40,6 +40,7 @@ type NodeCluster struct {
GlobalServerConfig dbs.JSON `field:"globalServerConfig"` // 全局服务配置
AutoRemoteStart bool `field:"autoRemoteStart"` // 自动远程启动
AutoInstallNftables bool `field:"autoInstallNftables"` // 自动安装nftables
IsAD bool `field:"isAD"` // 是否为高防集群
}
type NodeClusterOperator struct {
@@ -79,6 +80,7 @@ type NodeClusterOperator struct {
GlobalServerConfig any // 全局服务配置
AutoRemoteStart any // 自动远程启动
AutoInstallNftables any // 自动安装nftables
IsAD any // 是否为高防集群
}
func NewNodeClusterOperator() *NodeClusterOperator {

View File

@@ -12,6 +12,7 @@ import (
"github.com/TeaOSLab/EdgeAPI/internal/utils/numberutils"
"github.com/TeaOSLab/EdgeAPI/internal/utils/sizes"
"github.com/TeaOSLab/EdgeAPI/internal/utils/ttlcache"
"github.com/TeaOSLab/EdgeAPI/internal/zero"
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
@@ -971,12 +972,32 @@ func (this *NodeDAO) ComposeNodeConfig(tx *dbs.Tx, nodeId int64, cacheMap *utils
}
config.AllowedIPs = append(config.AllowedIPs, apiNodeIPs...)
// 所属集群
var primaryClusterId = int64(node.ClusterId)
var clusterIds = []int64{primaryClusterId}
clusterIds = append(clusterIds, node.DecodeSecondaryClusterIds()...)
// 获取所有的服务
servers, err := SharedServerDAO.FindAllEnabledServersWithNode(tx, int64(node.Id))
if err != nil {
return nil, err
}
// 获取集群上的其他服务
var serverIdMap = map[int64]zero.Zero{}
for _, server := range servers {
serverIdMap[int64(server.Id)] = zero.Zero{}
}
for _, clusterId := range clusterIds {
clusterServers, err := this.loadServersFromCluster(tx, clusterId, serverIdMap)
if err != nil {
return nil, err
}
for _, clusterServer := range clusterServers {
servers = append(servers, clusterServer)
}
}
for _, server := range servers {
serverConfig, err := SharedServerDAO.ComposeServerConfig(tx, server, false, cacheMap, true, false)
if err != nil {
@@ -1016,9 +1037,6 @@ func (this *NodeDAO) ComposeNodeConfig(tx *dbs.Tx, nodeId int64, cacheMap *utils
config.GlobalConfig = globalConfig
}
var primaryClusterId = int64(node.ClusterId)
var clusterIds = []int64{primaryClusterId}
clusterIds = append(clusterIds, node.DecodeSecondaryClusterIds()...)
var clusterIndex = 0
config.WebPImagePolicies = map[int64]*nodeconfigs.WebPImagePolicy{}
config.UAMPolicies = map[int64]*nodeconfigs.UAMPolicy{}

View File

@@ -6,10 +6,15 @@ package models
import (
"github.com/TeaOSLab/EdgeAPI/internal/utils"
"github.com/TeaOSLab/EdgeAPI/internal/zero"
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
"github.com/iwind/TeaGo/dbs"
)
func (this *NodeDAO) loadServersFromCluster(tx *dbs.Tx, clusterId int64, serverIdMap map[int64]zero.Zero) ([]*Server, error) {
return nil, nil
}
func (this *NodeDAO) composeExtConfig(tx *dbs.Tx, config *nodeconfigs.NodeConfig, clusterIds []int64, cacheMap *utils.CacheMap) error {
return nil
}

View File

@@ -154,7 +154,6 @@ func (this *NodeTaskDAO) ExtractNodeClusterTask(tx *dbs.Tx, clusterId int64, use
_, err = this.Query(tx).
Attr("role", nodeconfigs.NodeRoleNode).
Attr("clusterId", clusterId).
Param("clusterIdString", types.String(clusterId)).
Where("nodeId> 0").
Attr("type", taskType).
Delete()

View File

@@ -121,7 +121,7 @@ func (this *ServerDAO) FindEnabledServerBasic(tx *dbs.Tx, serverId int64) (*Serv
result, err := this.Query(tx).
Pk(serverId).
State(ServerStateEnabled).
Result("id", "name", "description", "isOn", "type", "clusterId").
Result("id", "name", "description", "isOn", "type", "clusterId", "userId").
Find()
if result == nil {
return nil, err

View File

@@ -0,0 +1,42 @@
package models
import (
_ "github.com/go-sql-driver/mysql"
"github.com/iwind/TeaGo/Tea"
"github.com/iwind/TeaGo/dbs"
)
const (
UserADInstanceStateEnabled = 1 // 已启用
UserADInstanceStateDisabled = 0 // 已禁用
)
type UserADInstanceDAO dbs.DAO
func NewUserADInstanceDAO() *UserADInstanceDAO {
return dbs.NewDAO(&UserADInstanceDAO{
DAOObject: dbs.DAOObject{
DB: Tea.Env,
Table: "edgeUserADInstances",
Model: new(UserADInstance),
PkName: "id",
},
}).(*UserADInstanceDAO)
}
var SharedUserADInstanceDAO *UserADInstanceDAO
func init() {
dbs.OnReady(func() {
SharedUserADInstanceDAO = NewUserADInstanceDAO()
})
}
// EnableUserADInstance 启用条目
func (this *UserADInstanceDAO) EnableUserADInstance(tx *dbs.Tx, id uint64) error {
_, err := this.Query(tx).
Pk(id).
Set("state", UserADInstanceStateEnabled).
Update()
return err
}

View File

@@ -0,0 +1,6 @@
package models_test
import (
_ "github.com/go-sql-driver/mysql"
_ "github.com/iwind/TeaGo/bootstrap"
)

View File

@@ -0,0 +1,40 @@
package models
import "github.com/iwind/TeaGo/dbs"
// UserADInstance 高防实例
type UserADInstance struct {
Id uint64 `field:"id"` // ID
AdminId uint32 `field:"adminId"` // 管理员ID
UserId uint64 `field:"userId"` // 用户ID
InstanceId uint32 `field:"instanceId"` // 高防实例ID
PeriodId uint32 `field:"periodId"` // 有效期
PeriodCount uint32 `field:"periodCount"` // 有效期数量
PeriodUnit string `field:"periodUnit"` // 有效期单位
DayFrom string `field:"dayFrom"` // 开始日期
DayTo string `field:"dayTo"` // 结束日期
MaxObjects uint32 `field:"maxObjects"` // 最多防护对象数
ObjectCodes dbs.JSON `field:"objectCodes"` // 防护对象
CreatedAt uint64 `field:"createdAt"` // 创建时间
State uint8 `field:"state"` // 状态
}
type UserADInstanceOperator struct {
Id any // ID
AdminId any // 管理员ID
UserId any // 用户ID
InstanceId any // 高防实例ID
PeriodId any // 有效期
PeriodCount any // 有效期数量
PeriodUnit any // 有效期单位
DayFrom any // 开始日期
DayTo any // 结束日期
MaxObjects any // 最多防护对象数
ObjectCodes any // 防护对象
CreatedAt any // 创建时间
State any // 状态
}
func NewUserADInstanceOperator() *UserADInstanceOperator {
return &UserADInstanceOperator{}
}

View File

@@ -0,0 +1 @@
package models

View File

@@ -134,7 +134,7 @@ func (this *UserDAO) FindEnabledBasicUser(tx *dbs.Tx, id int64) (*User, error) {
result, err := this.Query(tx).
Pk(id).
Attr("state", UserStateEnabled).
Result("id", "fullname", "username").
Result("id", "fullname", "username", "isOn").
Find()
if result == nil {
return nil, err

File diff suppressed because one or more lines are too long