阶段性提交

internal/db/models/server_dao.go

@@ -1,9 +1,11 @@
 package models
 
 import (
+	"errors"
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
 )
 
 const (
@@ -53,3 +55,67 @@ func (this *ServerDAO) FindEnabledServer(id uint32) (*Server, error) {
 	}
 	return result.(*Server), err
 }
+
+// 创建服务
+func (this *ServerDAO) CreateServer(adminId int64, userId int64, clusterId int64, configJSON string, includeNodesJSON string, excludeNodesJSON string) (serverId int64, err error) {
+	op := NewServerOperator()
+	op.UserId = userId
+	op.AdminId = adminId
+	op.ClusterId = clusterId
+	if len(configJSON) > 0 {
+		op.Config = configJSON
+	}
+	if len(includeNodesJSON) > 0 {
+		op.IncludeNodes = includeNodesJSON
+	}
+	if len(excludeNodesJSON) > 0 {
+		op.ExcludeNodes = excludeNodesJSON
+	}
+	op.GroupIds = "[]"
+	op.Version = 1
+	op.IsOn = 1
+	op.State = ServerStateEnabled
+	_, err = this.Save(op)
+	return types.Int64(op.Id), err
+}
+
+// 修改服务
+func (this *ServerDAO) UpdateServer(serverId int64, clusterId int64, configJSON string, includeNodesJSON string, excludeNodesJSON string) error {
+	if serverId <= 0 {
+		return errors.New("serverId should not be smaller than 0")
+	}
+	op := NewServerOperator()
+	op.Id = serverId
+	op.ClusterId = clusterId
+	if len(configJSON) > 0 {
+		op.Config = configJSON
+	}
+	if len(includeNodesJSON) > 0 {
+		op.IncludeNodes = includeNodesJSON
+	}
+	if len(excludeNodesJSON) > 0 {
+		op.ExcludeNodes = excludeNodesJSON
+	}
+	op.Version = dbs.SQL("version=version+1")
+	_, err := this.Save(op)
+	return err
+}
+
+// 计算所有可用服务数量
+func (this *ServerDAO) CountAllEnabledServers() (int64, error) {
+	return this.Query().
+		State(ServerStateEnabled).
+		Count()
+}
+
+// 列出单页的服务
+func (this *ServerDAO) ListEnabledServers(offset int64, size int64) (result []*Server, err error) {
+	_, err = this.Query().
+		State(ServerStateEnabled).
+		Offset(offset).
+		Limit(size).
+		DescPk().
+		Slice(&result).
+		FindAll()
+	return
+}