增加健康检查定时任务/健康检查可以发送消息

internal/db/models/message_dao.go

@@ -0,0 +1,128 @@
+package models
+
+import (
+	"crypto/md5"
+	"fmt"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"time"
+)
+
+const (
+	MessageStateEnabled  = 1 // 已启用
+	MessageStateDisabled = 0 // 已禁用
+
+	MessageLevelInfo    = "info"
+	MessageLevelWarning = "warning"
+	MessageLevelError   = "error"
+)
+
+type MessageType = string
+
+const (
+	MessageTypeHealthCheckFail MessageType = "HealthCheckFail"
+)
+
+type MessageDAO dbs.DAO
+
+func NewMessageDAO() *MessageDAO {
+	return dbs.NewDAO(&MessageDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeMessages",
+			Model:  new(Message),
+			PkName: "id",
+		},
+	}).(*MessageDAO)
+}
+
+var SharedMessageDAO *MessageDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedMessageDAO = NewMessageDAO()
+	})
+}
+
+// 启用条目
+func (this *MessageDAO) EnableMessage(id int64) error {
+	_, err := this.Query().
+		Pk(id).
+		Set("state", MessageStateEnabled).
+		Update()
+	return err
+}
+
+// 禁用条目
+func (this *MessageDAO) DisableMessage(id int64) error {
+	_, err := this.Query().
+		Pk(id).
+		Set("state", MessageStateDisabled).
+		Update()
+	return err
+}
+
+// 查找启用中的条目
+func (this *MessageDAO) FindEnabledMessage(id int64) (*Message, error) {
+	result, err := this.Query().
+		Pk(id).
+		Attr("state", MessageStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*Message), err
+}
+
+// 创建集群消息
+func (this *MessageDAO) CreateClusterMessage(clusterId int64, messageType MessageType, level string, body string, paramsJSON []byte) error {
+	_, err := this.createMessage(clusterId, 0, messageType, level, body, paramsJSON)
+	return err
+}
+
+// 删除某天之前的消息
+func (this *MessageDAO) DeleteMessagesBeforeDay(dayTime time.Time) error {
+	day := timeutil.Format("Ymd", dayTime)
+	_, err := this.Query().
+		Where("day<:day").
+		Param("day", day).
+		Delete()
+	return err
+}
+
+// 创建消息
+func (this *MessageDAO) createMessage(clusterId int64, nodeId int64, messageType MessageType, level string, body string, paramsJSON []byte) (int64, error) {
+	h := md5.New()
+	h.Write([]byte(body))
+	h.Write(paramsJSON)
+	hash := fmt.Sprintf("%x", h.Sum(nil))
+
+	// TODO 检查同样的消息最近是否发送过
+
+	// 创建新消息
+	op := NewMessageOperator()
+	op.AdminId = 0 // TODO
+	op.UserId = 0  // TODO
+	op.ClusterId = clusterId
+	op.NodeId = nodeId
+	op.Type = messageType
+	op.Level = level
+	op.Body = body
+	if len(paramsJSON) > 0 {
+		op.Params = paramsJSON
+	}
+	op.IsRead = false
+	op.State = MessageStateEnabled
+	op.CreatedAt = time.Now().Unix()
+	op.Day = timeutil.Format("Ymd")
+	op.Hash = hash
+
+	_, err := this.Save(op)
+	if err != nil {
+		return 0, err
+	}
+	return types.Int64(op.Id), nil
+}

internal/db/models/message_dao_test.go

@@ -0,0 +1,25 @@
+package models
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"testing"
+	"time"
+)
+
+func TestMessageDAO_CreateClusterMessage(t *testing.T) {
+	dao := NewMessageDAO()
+	err := dao.CreateClusterMessage(1, "test", "error", "123", []byte("456"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}
+
+func TestMessageDAO_DeleteMessagesBeforeDay(t *testing.T) {
+	dao := NewMessageDAO()
+	err := dao.DeleteMessagesBeforeDay(time.Now())
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}

internal/db/models/message_model.go

@@ -0,0 +1,40 @@
+package models
+
+// 消息通知
+type Message struct {
+	Id        uint64 `field:"id"`        // ID
+	AdminId   uint32 `field:"adminId"`   // 管理员ID
+	UserId    uint32 `field:"userId"`    // 用户ID
+	ClusterId uint32 `field:"clusterId"` // 集群ID
+	NodeId    uint32 `field:"nodeId"`    // 节点ID
+	Level     string `field:"level"`     // 级别
+	Body      string `field:"body"`      // 内容
+	Type      string `field:"type"`      // 消息类型
+	Params    string `field:"params"`    // 额外的参数
+	IsRead    uint8  `field:"isRead"`    // 是否已读
+	State     uint8  `field:"state"`     // 状态
+	CreatedAt uint64 `field:"createdAt"` // 创建时间
+	Day       string `field:"day"`       // 日期YYYYMMDD
+	Hash      string `field:"hash"`      // 消息内容的Hash
+}
+
+type MessageOperator struct {
+	Id        interface{} // ID
+	AdminId   interface{} // 管理员ID
+	UserId    interface{} // 用户ID
+	ClusterId interface{} // 集群ID
+	NodeId    interface{} // 节点ID
+	Level     interface{} // 级别
+	Body      interface{} // 内容
+	Type      interface{} // 消息类型
+	Params    interface{} // 额外的参数
+	IsRead    interface{} // 是否已读
+	State     interface{} // 状态
+	CreatedAt interface{} // 创建时间
+	Day       interface{} // 日期YYYYMMDD
+	Hash      interface{} // 消息内容的Hash
+}
+
+func NewMessageOperator() *MessageOperator {
+	return &MessageOperator{}
+}

internal/db/models/message_model_ext.go

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

internal/db/models/sys_setting_dao.go

@@ -5,6 +5,7 @@ import (
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
 )
 
 type SysSettingDAO dbs.DAO
@@ -80,9 +81,9 @@ func (this *SysSettingDAO) UpdateSetting(codeFormat string, valueJSON []byte, co
 }
 
 // 读取配置
-func (this *SysSettingDAO) ReadSetting(code string, args ...interface{}) (valueJSON []byte, err error) {
-	if len(args) > 0 {
-		code = fmt.Sprintf(code, args...)
+func (this *SysSettingDAO) ReadSetting(code string, codeFormatArgs ...interface{}) (valueJSON []byte, err error) {
+	if len(codeFormatArgs) > 0 {
+		code = fmt.Sprintf(code, codeFormatArgs...)
 	}
 	col, err := this.Query().
 		Attr("code", code).
@@ -90,3 +91,19 @@ func (this *SysSettingDAO) ReadSetting(code string, args ...interface{}) (valueJ
 		FindStringCol("")
 	return []byte(col), err
 }
+
+// 对比配置中的数字大小
+func (this *SysSettingDAO) CompareInt64Setting(code string, anotherValue int64) (int8, error) {
+	valueJSON, err := this.ReadSetting(code)
+	if err != nil {
+		return 0, err
+	}
+	value := types.Int64(string(valueJSON))
+	if value > anotherValue {
+		return 1, nil
+	}
+	if value < anotherValue {
+		return -1, nil
+	}
+	return 0, nil
+}

internal/db/models/sys_setting_dao_test.go

@@ -6,12 +6,12 @@ import (
 )
 
 func TestSysSettingDAO_UpdateSetting(t *testing.T) {
-	err := SharedSysSettingDAO.UpdateSetting("hello", []byte(`"world"`))
+	err := NewSysSettingDAO().UpdateSetting("hello", []byte(`"world"`))
 	if err != nil {
 		t.Fatal(err)
 	}
 
-	value, err := SharedSysSettingDAO.ReadSetting("hello")
+	value, err := NewSysSettingDAO().ReadSetting("hello")
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -19,14 +19,22 @@ func TestSysSettingDAO_UpdateSetting(t *testing.T) {
 }
 
 func TestSysSettingDAO_UpdateSetting_Args(t *testing.T) {
-	err := SharedSysSettingDAO.UpdateSetting("hello %d", []byte(`"world 123"`), 123)
+	err := NewSysSettingDAO().UpdateSetting("hello %d", []byte(`"world 123"`), 123)
 	if err != nil {
 		t.Fatal(err)
 	}
 
-	value, err := SharedSysSettingDAO.ReadSetting("hello %d", 123)
+	value, err := NewSysSettingDAO().ReadSetting("hello %d", 123)
 	if err != nil {
 		t.Fatal(err)
 	}
 	t.Log("value:", string(value))
 }
+
+func TestSysSettingDAO_CompareInt64Setting(t *testing.T) {
+	i, err := NewSysSettingDAO().CompareInt64Setting("int64", 1024)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("result:", i)
+}