增加企业版认证相关API

Author: iwind

internal/db/models/authority_key_dao.go

@@ -0,0 +1,82 @@
+package models
+
+import (
+	"encoding/json"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"time"
+)
+
+type AuthorityKeyDAO dbs.DAO
+
+func NewAuthorityKeyDAO() *AuthorityKeyDAO {
+	return dbs.NewDAO(&AuthorityKeyDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeAuthorityKeys",
+			Model:  new(AuthorityKey),
+			PkName: "id",
+		},
+	}).(*AuthorityKeyDAO)
+}
+
+var SharedAuthorityKeyDAO *AuthorityKeyDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedAuthorityKeyDAO = NewAuthorityKeyDAO()
+	})
+}
+
+// UpdateKey 设置Key
+func (this *AuthorityKeyDAO) UpdateKey(tx *dbs.Tx, value string, dayFrom string, dayTo string, hostname string, macAddresses []string) error {
+	one, err := this.Query(tx).
+		AscPk().
+		Find()
+	if err != nil {
+		return err
+	}
+	op := NewAuthorityKeyOperator()
+	if one != nil {
+		op.Id = one.(*AuthorityKey).Id
+	}
+	op.Value = value
+	op.DayFrom = dayFrom
+	op.DayTo = dayTo
+	op.Hostname = hostname
+
+	if len(macAddresses) == 0 {
+		macAddresses = []string{}
+	}
+	macAddressesJSON, err := json.Marshal(macAddresses)
+	if err != nil {
+		return err
+	}
+
+	op.MacAddresses = macAddressesJSON
+	op.UpdatedAt = time.Now().Unix()
+
+	return this.Save(tx, op)
+}
+
+// ReadKey 读取Key
+func (this *AuthorityKeyDAO) ReadKey(tx *dbs.Tx) (key *AuthorityKey, err error) {
+	one, err := this.Query(tx).
+		AscPk().
+		Find()
+	if err != nil {
+		return nil, err
+	}
+	if one == nil {
+		return nil, nil
+	}
+	return one.(*AuthorityKey), nil
+}
+
+// ResetKey 重置Key
+func (this *AuthorityKeyDAO) ResetKey(tx *dbs.Tx) error {
+	_, err := this.Query(tx).
+		Delete()
+	return err
+}

internal/db/models/authority_key_dao_test.go

@@ -0,0 +1,23 @@
+package models
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"testing"
+)
+
+func TestAuthorityKeyDAO_UpdateValue(t *testing.T) {
+	err := NewAuthorityKeyDAO().UpdateValue(nil, "12345678")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf("ok")
+}
+
+func TestAuthorityKeyDAO_ReadValue(t *testing.T) {
+	value, err := NewAuthorityKeyDAO().ReadValue(nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf(value)
+}

internal/db/models/authority_key_model.go

@@ -0,0 +1,26 @@
+package models
+
+// AuthorityKey 企业版认证信息
+type AuthorityKey struct {
+	Id           uint32 `field:"id"`           // ID
+	Value        string `field:"value"`        // Key值
+	DayFrom      string `field:"dayFrom"`      // 开始日期
+	DayTo        string `field:"dayTo"`        // 结束日期
+	Hostname     string `field:"hostname"`     // Hostname
+	MacAddresses string `field:"macAddresses"` // MAC地址
+	UpdatedAt    uint64 `field:"updatedAt"`    // 创建/修改时间
+}
+
+type AuthorityKeyOperator struct {
+	Id           interface{} // ID
+	Value        interface{} // Key值
+	DayFrom      interface{} // 开始日期
+	DayTo        interface{} // 结束日期
+	Hostname     interface{} // Hostname
+	MacAddresses interface{} // MAC地址
+	UpdatedAt    interface{} // 创建/修改时间
+}
+
+func NewAuthorityKeyOperator() *AuthorityKeyOperator {
+	return &AuthorityKeyOperator{}
+}

internal/db/models/authority_key_model_ext.go

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

internal/db/models/message_dao_test.go

@@ -11,7 +11,7 @@ func TestMessageDAO_CreateClusterMessage(t *testing.T) {
 	var tx *dbs.Tx
 
 	dao := NewMessageDAO()
-	err := dao.CreateClusterMessage(tx, 1, "test", "error", "123", []byte("456"))
+	err := dao.CreateClusterMessage(tx, 1, "test", "error", "123", "123", []byte("456"))
 	if err != nil {
 		t.Fatal(err)
 	}