实现域名服务器集群管理

internal/db/models/nameservers/ns_cluster_dao.go

@@ -0,0 +1,114 @@
+package nameservers
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+const (
+	NSClusterStateEnabled  = 1 // 已启用
+	NSClusterStateDisabled = 0 // 已禁用
+)
+
+type NSClusterDAO dbs.DAO
+
+func NewNSClusterDAO() *NSClusterDAO {
+	return dbs.NewDAO(&NSClusterDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeNSClusters",
+			Model:  new(NSCluster),
+			PkName: "id",
+		},
+	}).(*NSClusterDAO)
+}
+
+var SharedNSClusterDAO *NSClusterDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedNSClusterDAO = NewNSClusterDAO()
+	})
+}
+
+// EnableNSCluster 启用条目
+func (this *NSClusterDAO) EnableNSCluster(tx *dbs.Tx, id int64) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSClusterStateEnabled).
+		Update()
+	return err
+}
+
+// DisableNSCluster 禁用条目
+func (this *NSClusterDAO) DisableNSCluster(tx *dbs.Tx, id int64) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSClusterStateDisabled).
+		Update()
+	return err
+}
+
+// FindEnabledNSCluster 查找启用中的条目
+func (this *NSClusterDAO) FindEnabledNSCluster(tx *dbs.Tx, id int64) (*NSCluster, error) {
+	result, err := this.Query(tx).
+		Pk(id).
+		Attr("state", NSClusterStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*NSCluster), err
+}
+
+// CreateCluster 创建集群
+func (this *NSClusterDAO) CreateCluster(tx *dbs.Tx, name string) (int64, error) {
+	op := NewNSClusterOperator()
+	op.Name = name
+	op.IsOn = true
+	op.State = NSClusterStateEnabled
+	return this.SaveInt64(tx, op)
+}
+
+// UpdateCluster 修改集群
+func (this *NSClusterDAO) UpdateCluster(tx *dbs.Tx, clusterId int64, name string, isOn bool) error {
+	if clusterId <= 0 {
+		return errors.New("invalid clusterId")
+	}
+	op := NewNSClusterOperator()
+	op.Id = clusterId
+	op.Name = name
+	op.IsOn = isOn
+	return this.Save(tx, op)
+}
+
+// CountAllEnabledClusters 计算可用集群数量
+func (this *NSClusterDAO) CountAllEnabledClusters(tx *dbs.Tx) (int64, error) {
+	return this.Query(tx).
+		State(NSClusterStateEnabled).
+		Count()
+}
+
+// ListEnabledNSClusters 列出单页集群
+func (this *NSClusterDAO) ListEnabledNSClusters(tx *dbs.Tx, offset int64, size int64) (result []*NSCluster, err error) {
+	_, err = this.Query(tx).
+		State(NSClusterStateEnabled).
+		Offset(offset).
+		Limit(size).
+		DescPk().
+		Slice(&result).
+		FindAll()
+	return
+}
+
+// FindAllEnabledNSClusters 列出所有集群
+func (this *NSClusterDAO) FindAllEnabledNSClusters(tx *dbs.Tx) (result []*NSCluster, err error) {
+	_, err = this.Query(tx).
+		State(NSClusterStateEnabled).
+		DescPk().
+		Slice(&result).
+		FindAll()
+	return
+}

internal/db/models/nameservers/ns_cluster_dao_test.go

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

internal/db/models/nameservers/ns_cluster_model.go

@@ -0,0 +1,20 @@
+package nameservers
+
+// NSCluster 域名服务器集群
+type NSCluster struct {
+	Id    uint32 `field:"id"`    // ID
+	IsOn  uint8  `field:"isOn"`  // 是否启用
+	Name  string `field:"name"`  // 集群名
+	State uint8  `field:"state"` // 状态
+}
+
+type NSClusterOperator struct {
+	Id    interface{} // ID
+	IsOn  interface{} // 是否启用
+	Name  interface{} // 集群名
+	State interface{} // 状态
+}
+
+func NewNSClusterOperator() *NSClusterOperator {
+	return &NSClusterOperator{}
+}

internal/db/models/nameservers/ns_cluster_model_ext.go

@@ -0,0 +1 @@
+package nameservers

internal/db/models/nameservers/ns_domain_dao.go

@@ -0,0 +1,71 @@
+package nameservers
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+const (
+	NSDomainStateEnabled  = 1 // 已启用
+	NSDomainStateDisabled = 0 // 已禁用
+)
+
+type NSDomainDAO dbs.DAO
+
+func NewNSDomainDAO() *NSDomainDAO {
+	return dbs.NewDAO(&NSDomainDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeNSDomains",
+			Model:  new(NSDomain),
+			PkName: "id",
+		},
+	}).(*NSDomainDAO)
+}
+
+var SharedNSDomainDAO *NSDomainDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedNSDomainDAO = NewNSDomainDAO()
+	})
+}
+
+// EnableNSDomain 启用条目
+func (this *NSDomainDAO) EnableNSDomain(tx *dbs.Tx, id uint32) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSDomainStateEnabled).
+		Update()
+	return err
+}
+
+// DisableNSDomain 禁用条目
+func (this *NSDomainDAO) DisableNSDomain(tx *dbs.Tx, id uint32) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSDomainStateDisabled).
+		Update()
+	return err
+}
+
+// FindEnabledNSDomain 查找启用中的条目
+func (this *NSDomainDAO) FindEnabledNSDomain(tx *dbs.Tx, id uint32) (*NSDomain, error) {
+	result, err := this.Query(tx).
+		Pk(id).
+		Attr("state", NSDomainStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*NSDomain), err
+}
+
+// FindNSDomainName 根据主键查找名称
+func (this *NSDomainDAO) FindNSDomainName(tx *dbs.Tx, id uint32) (string, error) {
+	return this.Query(tx).
+		Pk(id).
+		Result("name").
+		FindStringCol("")
+}

internal/db/models/nameservers/ns_domain_dao_test.go

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

internal/db/models/nameservers/ns_domain_model.go

@@ -0,0 +1,26 @@
+package nameservers
+
+// NSDomain DNS域名
+type NSDomain struct {
+	Id        uint32 `field:"id"`        // ID
+	ClusterId uint32 `field:"clusterId"` // 集群ID
+	UserId    uint32 `field:"userId"`    // 用户ID
+	IsOn      uint8  `field:"isOn"`      // 是否启用
+	Name      string `field:"name"`      // 域名
+	CreatedAt uint64 `field:"createdAt"` // 创建时间
+	State     uint8  `field:"state"`     // 状态
+}
+
+type NSDomainOperator struct {
+	Id        interface{} // ID
+	ClusterId interface{} // 集群ID
+	UserId    interface{} // 用户ID
+	IsOn      interface{} // 是否启用
+	Name      interface{} // 域名
+	CreatedAt interface{} // 创建时间
+	State     interface{} // 状态
+}
+
+func NewNSDomainOperator() *NSDomainOperator {
+	return &NSDomainOperator{}
+}

internal/db/models/nameservers/ns_domain_model_ext.go

@@ -0,0 +1 @@
+package nameservers

internal/db/models/nameservers/ns_node_dao.go

@@ -0,0 +1,63 @@
+package nameservers
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+const (
+	NSNodeStateEnabled  = 1 // 已启用
+	NSNodeStateDisabled = 0 // 已禁用
+)
+
+type NSNodeDAO dbs.DAO
+
+func NewNSNodeDAO() *NSNodeDAO {
+	return dbs.NewDAO(&NSNodeDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeNSNodes",
+			Model:  new(NSNode),
+			PkName: "id",
+		},
+	}).(*NSNodeDAO)
+}
+
+var SharedNSNodeDAO *NSNodeDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedNSNodeDAO = NewNSNodeDAO()
+	})
+}
+
+// EnableNSNode 启用条目
+func (this *NSNodeDAO) EnableNSNode(tx *dbs.Tx, id uint32) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSNodeStateEnabled).
+		Update()
+	return err
+}
+
+// DisableNSNode 禁用条目
+func (this *NSNodeDAO) DisableNSNode(tx *dbs.Tx, id uint32) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSNodeStateDisabled).
+		Update()
+	return err
+}
+
+// FindEnabledNSNode 查找启用中的条目
+func (this *NSNodeDAO) FindEnabledNSNode(tx *dbs.Tx, id uint32) (*NSNode, error) {
+	result, err := this.Query(tx).
+		Pk(id).
+		Attr("state", NSNodeStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*NSNode), err
+}

internal/db/models/nameservers/ns_node_dao_test.go

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

internal/db/models/nameservers/ns_node_model.go

@@ -0,0 +1,26 @@
+package nameservers
+
+// NSNode 域名服务器节点
+type NSNode struct {
+	Id        uint32 `field:"id"`        // ID
+	ClusterId uint32 `field:"clusterId"` // 集群ID
+	IsOn      uint8  `field:"isOn"`      // 是否启用
+	Status    string `field:"status"`    // 运行状态
+	UniqueId  string `field:"uniqueId"`  // 节点ID
+	Secret    string `field:"secret"`    // 密钥
+	State     uint8  `field:"state"`     // 状态
+}
+
+type NSNodeOperator struct {
+	Id        interface{} // ID
+	ClusterId interface{} // 集群ID
+	IsOn      interface{} // 是否启用
+	Status    interface{} // 运行状态
+	UniqueId  interface{} // 节点ID
+	Secret    interface{} // 密钥
+	State     interface{} // 状态
+}
+
+func NewNSNodeOperator() *NSNodeOperator {
+	return &NSNodeOperator{}
+}

internal/db/models/nameservers/ns_node_model_ext.go

@@ -0,0 +1 @@
+package nameservers

internal/db/models/nameservers/ns_record_dao.go

@@ -0,0 +1,71 @@
+package nameservers
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+const (
+	NSRecordStateEnabled  = 1 // 已启用
+	NSRecordStateDisabled = 0 // 已禁用
+)
+
+type NSRecordDAO dbs.DAO
+
+func NewNSRecordDAO() *NSRecordDAO {
+	return dbs.NewDAO(&NSRecordDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeNSRecords",
+			Model:  new(NSRecord),
+			PkName: "id",
+		},
+	}).(*NSRecordDAO)
+}
+
+var SharedNSRecordDAO *NSRecordDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedNSRecordDAO = NewNSRecordDAO()
+	})
+}
+
+// EnableNSRecord 启用条目
+func (this *NSRecordDAO) EnableNSRecord(tx *dbs.Tx, id uint64) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSRecordStateEnabled).
+		Update()
+	return err
+}
+
+// DisableNSRecord 禁用条目
+func (this *NSRecordDAO) DisableNSRecord(tx *dbs.Tx, id uint64) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSRecordStateDisabled).
+		Update()
+	return err
+}
+
+// FindEnabledNSRecord 查找启用中的条目
+func (this *NSRecordDAO) FindEnabledNSRecord(tx *dbs.Tx, id uint64) (*NSRecord, error) {
+	result, err := this.Query(tx).
+		Pk(id).
+		Attr("state", NSRecordStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*NSRecord), err
+}
+
+// FindNSRecordName 根据主键查找名称
+func (this *NSRecordDAO) FindNSRecordName(tx *dbs.Tx, id uint64) (string, error) {
+	return this.Query(tx).
+		Pk(id).
+		Result("name").
+		FindStringCol("")
+}

internal/db/models/nameservers/ns_record_dao_test.go

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

internal/db/models/nameservers/ns_record_model.go

@@ -0,0 +1,34 @@
+package nameservers
+
+// NSRecord DNS记录
+type NSRecord struct {
+	Id          uint64 `field:"id"`          // ID
+	DomainId    uint32 `field:"domainId"`    // 域名ID
+	Description string `field:"description"` // 备注
+	Name        string `field:"name"`        // 记录名
+	Type        string `field:"type"`        // 类型
+	Value       string `field:"value"`       // 值
+	Ttl         uint32 `field:"ttl"`         // TTL（秒）
+	Weight      uint32 `field:"weight"`      // 权重
+	Routes      string `field:"routes"`      // 线路
+	CreatedAt   uint64 `field:"createdAt"`   // 创建时间
+	State       uint8  `field:"state"`       // 状态
+}
+
+type NSRecordOperator struct {
+	Id          interface{} // ID
+	DomainId    interface{} // 域名ID
+	Description interface{} // 备注
+	Name        interface{} // 记录名
+	Type        interface{} // 类型
+	Value       interface{} // 值
+	Ttl         interface{} // TTL（秒）
+	Weight      interface{} // 权重
+	Routes      interface{} // 线路
+	CreatedAt   interface{} // 创建时间
+	State       interface{} // 状态
+}
+
+func NewNSRecordOperator() *NSRecordOperator {
+	return &NSRecordOperator{}
+}

internal/db/models/nameservers/ns_record_model_ext.go

@@ -0,0 +1 @@
+package nameservers

internal/db/models/nameservers/ns_route_dao.go

@@ -0,0 +1,71 @@
+package nameservers
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+const (
+	NSRouteStateEnabled  = 1 // 已启用
+	NSRouteStateDisabled = 0 // 已禁用
+)
+
+type NSRouteDAO dbs.DAO
+
+func NewNSRouteDAO() *NSRouteDAO {
+	return dbs.NewDAO(&NSRouteDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeNSRoutes",
+			Model:  new(NSRoute),
+			PkName: "id",
+		},
+	}).(*NSRouteDAO)
+}
+
+var SharedNSRouteDAO *NSRouteDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedNSRouteDAO = NewNSRouteDAO()
+	})
+}
+
+// EnableNSRoute 启用条目
+func (this *NSRouteDAO) EnableNSRoute(tx *dbs.Tx, id uint32) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSRouteStateEnabled).
+		Update()
+	return err
+}
+
+// DisableNSRoute 禁用条目
+func (this *NSRouteDAO) DisableNSRoute(tx *dbs.Tx, id uint32) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", NSRouteStateDisabled).
+		Update()
+	return err
+}
+
+// FindEnabledNSRoute 查找启用中的条目
+func (this *NSRouteDAO) FindEnabledNSRoute(tx *dbs.Tx, id uint32) (*NSRoute, error) {
+	result, err := this.Query(tx).
+		Pk(id).
+		Attr("state", NSRouteStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*NSRoute), err
+}
+
+// FindNSRouteName 根据主键查找名称
+func (this *NSRouteDAO) FindNSRouteName(tx *dbs.Tx, id uint32) (string, error) {
+	return this.Query(tx).
+		Pk(id).
+		Result("name").
+		FindStringCol("")
+}

internal/db/models/nameservers/ns_route_dao_test.go

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

internal/db/models/nameservers/ns_route_model.go

@@ -0,0 +1,26 @@
+package nameservers
+
+// NSRoute DNS线路
+type NSRoute struct {
+	Id        uint32 `field:"id"`        // ID
+	ClusterId uint32 `field:"clusterId"` // 集群ID
+	UserId    uint32 `field:"userId"`    // 用户ID
+	Name      string `field:"name"`      // 名称
+	Conds     string `field:"conds"`     // 条件定义
+	IsOn      uint8  `field:"isOn"`      // 是否启用
+	State     uint8  `field:"state"`     // 状态
+}
+
+type NSRouteOperator struct {
+	Id        interface{} // ID
+	ClusterId interface{} // 集群ID
+	UserId    interface{} // 用户ID
+	Name      interface{} // 名称
+	Conds     interface{} // 条件定义
+	IsOn      interface{} // 是否启用
+	State     interface{} // 状态
+}
+
+func NewNSRouteOperator() *NSRouteOperator {
+	return &NSRouteOperator{}
+}

internal/db/models/nameservers/ns_route_model_ext.go

@@ -0,0 +1 @@
+package nameservers