增加DNS同步任务状态显示

internal/db/models/dns/dns_task_dao.go

@@ -0,0 +1,153 @@
+package dns
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/maps"
+	"time"
+)
+
+type DNSTaskType = string
+
+const (
+	DNSTaskTypeClusterChange DNSTaskType = "clusterChange"
+	DNSTaskTypeNodeChange    DNSTaskType = "nodeChange"
+	DNSTaskTypeServerChange  DNSTaskType = "serverChange"
+	DNSTaskTypeDomainChange  DNSTaskType = "domainChange"
+)
+
+type DNSTaskDAO dbs.DAO
+
+func NewDNSTaskDAO() *DNSTaskDAO {
+	return dbs.NewDAO(&DNSTaskDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeDNSTasks",
+			Model:  new(DNSTask),
+			PkName: "id",
+		},
+	}).(*DNSTaskDAO)
+}
+
+var SharedDNSTaskDAO *DNSTaskDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedDNSTaskDAO = NewDNSTaskDAO()
+	})
+}
+
+// 生成任务
+func (this *DNSTaskDAO) CreateDNSTask(tx *dbs.Tx, clusterId int64, serverId int64, nodeId int64, domainId int64, taskType string) error {
+	if clusterId <= 0 && serverId <= 0 && nodeId <= 0 {
+		return nil
+	}
+	err := this.Query(tx).InsertOrUpdateQuickly(maps.Map{
+		"clusterId": clusterId,
+		"serverId":  serverId,
+		"nodeId":    nodeId,
+		"domainId":  domainId,
+		"updatedAt": time.Now().Unix(),
+		"type":      taskType,
+		"isDone":    false,
+		"isOk":      false,
+		"error":     "",
+	}, maps.Map{
+		"updatedAt": time.Now().Unix(),
+		"isDone":    false,
+		"isOk":      false,
+		"error":     "",
+	})
+	return err
+}
+
+// 生成集群任务
+func (this *DNSTaskDAO) CreateClusterTask(tx *dbs.Tx, clusterId int64, taskType DNSTaskType) error {
+	return this.CreateDNSTask(tx, clusterId, 0, 0, 0, taskType)
+}
+
+// 生成节点任务
+func (this *DNSTaskDAO) CreateNodeTask(tx *dbs.Tx, nodeId int64, taskType DNSTaskType) error {
+	return this.CreateDNSTask(tx, 0, 0, nodeId, 0, taskType)
+}
+
+// 生成服务任务
+func (this *DNSTaskDAO) CreateServerTask(tx *dbs.Tx, serverId int64, taskType DNSTaskType) error {
+	return this.CreateDNSTask(tx, 0, serverId, 0, 0, taskType)
+}
+
+// 生成域名更新任务
+func (this *DNSTaskDAO) CreateDomainTask(tx *dbs.Tx, domainId int64, taskType DNSTaskType) error {
+	return this.CreateDNSTask(tx, 0, 0, 0, domainId, taskType)
+}
+
+// 查找所有正在执行的任务
+func (this *DNSTaskDAO) FindAllDoingTasks(tx *dbs.Tx) (result []*DNSTask, err error) {
+	_, err = this.Query(tx).
+		Attr("isDone", 0).
+		AscPk().
+		Slice(&result).
+		FindAll()
+	return
+}
+
+// 查找正在执行的和错误的任务
+func (this *DNSTaskDAO) FindAllDoingOrErrorTasks(tx *dbs.Tx) (result []*DNSTask, err error) {
+	_, err = this.Query(tx).
+		Where("(isDone=0 OR (isDone=1 AND isOk=0))").
+		AscPk().
+		Slice(&result).
+		FindAll()
+	return
+}
+
+// 检查是否有正在执行的任务
+func (this *DNSTaskDAO) ExistDoingTasks(tx *dbs.Tx) (bool, error) {
+	return this.Query(tx).
+		Attr("isDone", 0).
+		Exist()
+}
+
+// 检查是否有错误的任务
+func (this *DNSTaskDAO) ExistErrorTasks(tx *dbs.Tx) (bool, error) {
+	return this.Query(tx).
+		Attr("isDone", 1).
+		Attr("isOk", 0).
+		Exist()
+}
+
+// 删除任务
+func (this *DNSTaskDAO) DeleteDNSTask(tx *dbs.Tx, taskId int64) error {
+	_, err := this.Query(tx).
+		Pk(taskId).
+		Delete()
+	return err
+}
+
+// 设置任务错误
+func (this *DNSTaskDAO) UpdateDNSTaskError(tx *dbs.Tx, taskId int64, err string) error {
+	if taskId <= 0 {
+		return errors.New("invalid taskId")
+	}
+	op := NewDNSTaskOperator()
+	op.Id = taskId
+	op.IsDone = true
+	op.Error = err
+	op.IsOk = false
+	return this.Save(tx, op)
+}
+
+// 设置任务完成
+func (this *DNSTaskDAO) UpdateDNSTaskDone(tx *dbs.Tx, taskId int64) error {
+	if taskId <= 0 {
+		return errors.New("invalid taskId")
+	}
+	op := NewDNSTaskOperator()
+	op.Id = taskId
+	op.IsDone = true
+	op.IsOk = true
+	op.Error = ""
+	return this.Save(tx, op)
+}

internal/db/models/dns/dns_task_dao_test.go

@@ -0,0 +1,17 @@
+package dns
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/TeaGo/dbs"
+	"testing"
+)
+
+func TestDNSTaskDAO_CreateDNSTask(t *testing.T) {
+	dbs.NotifyReady()
+	err := SharedDNSTaskDAO.CreateDNSTask(nil, 1, 2, 3, "taskType")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}

internal/db/models/dns/dns_task_model.go

@@ -0,0 +1,32 @@
+package dns
+
+// DNS更新任务
+type DNSTask struct {
+	Id        uint64 `field:"id"`        // ID
+	ClusterId uint32 `field:"clusterId"` // 集群ID
+	ServerId  uint32 `field:"serverId"`  // 服务ID
+	NodeId    uint32 `field:"nodeId"`    // 节点ID
+	DomainId  uint32 `field:"domainId"`  // 域名ID
+	Type      string `field:"type"`      // 任务类型
+	UpdatedAt uint64 `field:"updatedAt"` // 更新时间
+	IsDone    uint8  `field:"isDone"`    // 是否已完成
+	IsOk      uint8  `field:"isOk"`      // 是否成功
+	Error     string `field:"error"`     // 错误信息
+}
+
+type DNSTaskOperator struct {
+	Id        interface{} // ID
+	ClusterId interface{} // 集群ID
+	ServerId  interface{} // 服务ID
+	NodeId    interface{} // 节点ID
+	DomainId  interface{} // 域名ID
+	Type      interface{} // 任务类型
+	UpdatedAt interface{} // 更新时间
+	IsDone    interface{} // 是否已完成
+	IsOk      interface{} // 是否成功
+	Error     interface{} // 错误信息
+}
+
+func NewDNSTaskOperator() *DNSTaskOperator {
+	return &DNSTaskOperator{}
+}

internal/db/models/dns/dns_task_model_ext.go

@@ -0,0 +1 @@
+package dns