feat: add cron msg

models/cron_messages.go

@@ -0,0 +1,118 @@
+package models
+
+import (
+	"errors"
+	"fmt"
+	"github.com/jinzhu/gorm"
+	"message-nest/pkg/util"
+)
+
+type CronMessages struct {
+	UUIDModel
+
+	Name    string `json:"name" gorm:"type:varchar(200) comment '关联的消息名称';default:'';"`
+	TaskID  string `json:"task_id" gorm:"type:varchar(36) comment '关联的消息ID';default:'';"`
+	Cron    string `json:"cron" gorm:"type:varchar(4096) comment '定时表达式';default:'';"`
+	Title   string `json:"title" gorm:"type:varchar(1000) comment '消息名称';default:'';"`
+	Content string `json:"content" gorm:"type:varchar(4096) comment '消息内容';default:'';"`
+	//MarkDown string `json:"markdown" gorm:"type:varchar(4096) comment 'markdown内容';default:'';"`
+	Url    string `json:"url" gorm:"type:varchar(4096) comment 'url地址';default:'';"`
+	Enable int    `json:"enable" gorm:"type:int comment '开启、暂停状态';default:1;"`
+}
+
+func GenerateMsgUniqueID() string {
+	newUUID := util.GenerateUniqueID()
+	return fmt.Sprintf("C-%s", newUUID)
+}
+
+func AddSendCronMsg(
+	name string,
+	task_id string,
+	cron string,
+	title string,
+	content string,
+	url string,
+	createdBy string,
+) (string, error) {
+	newUUID := GenerateMsgUniqueID()
+	msg := CronMessages{
+		UUIDModel: UUIDModel{
+			ID:         newUUID,
+			CreatedBy:  createdBy,
+			ModifiedBy: createdBy,
+		},
+		Name:    name,
+		TaskID:  task_id,
+		Cron:    cron,
+		Title:   title,
+		Content: content,
+		Url:     url,
+		Enable:  1,
+	}
+	if err := db.Create(&msg).Error; err != nil {
+		return newUUID, err
+	}
+	return newUUID, nil
+}
+
+// GetCronMessages 获取所有任务
+func GetCronMessages(pageNum int, pageSize int, name string, maps interface{}) ([]CronMessages, error) {
+	var (
+		msgs []CronMessages
+		err  error
+	)
+	query := db.Where(maps)
+	if name != "" {
+		query = query.Where("name like ?", fmt.Sprintf("%%%s%%", name))
+	}
+	query = query.Order("created_on DESC")
+	if pageSize > 0 || pageNum > 0 {
+		query = query.Offset(pageNum).Limit(pageSize)
+	}
+	err = query.Find(&msgs).Error
+	if err != nil && err != gorm.ErrRecordNotFound {
+		return nil, err
+	}
+	return msgs, nil
+}
+
+// GetCronMessagesTotal 获取所有任务总数
+func GetCronMessagesTotal(name string, maps interface{}) (int, error) {
+	var (
+		err   error
+		total int
+	)
+	query := db.Model(&CronMessages{}).Where(maps)
+	if name != "" {
+		query = query.Where("name like ?", fmt.Sprintf("%%%s%%", name))
+	}
+
+	err = query.Count(&total).Error
+	if err != nil {
+		return 0, err
+	}
+	return total, nil
+}
+
+func DeleteCronMsg(id string) error {
+	if err := db.Where("id = ?", id).Delete(&CronMessages{}).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+func EditCronMsg(id string, data interface{}) error {
+	if err := db.Model(&CronMessages{}).Where("id = ? ", id).Updates(data).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+func GetCronMsgByID(id string) (CronMessages, error) {
+	var msg CronMessages
+	err := db.Where("id = ? ", id).Find(&msg).Error
+	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
+		return msg, err
+	}
+	return msg, nil
+}

models/send_tasks.go

@@ -174,3 +174,12 @@ func EditSendTask(id string, data interface{}) error {
 
 	return nil
 }
+
+func GetTaskByID(id string) (SendTasks, error) {
+	var task SendTasks
+	err := db.Where("id = ? ", id).Find(&task).Error
+	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
+		return task, err
+	}
+	return task, nil
+}