Message-Push-Nest/models/cron_messages.go

package models

import (
	"errors"
	"fmt"
	"gorm.io/gorm"
	"message-nest/pkg/util"
)

type CronMessages struct {
	UUIDModel

	Name    string `json:"name" gorm:"type:varchar(200) ;default:'';"`
	TaskID  string `json:"task_id" gorm:"type:varchar(36) ;default:'';"`
	Cron    string `json:"cron" gorm:"type:varchar(4096) ;default:'';"`
	Title   string `json:"title" gorm:"type:varchar(1000) ;default:'';"`
	Content string `json:"content" gorm:"type:varchar(4096) ;default:'';"`
	//MarkDown string `json:"markdown" gorm:"type:varchar(4096) ;default:'';"`
	Url    string `json:"url" gorm:"type:varchar(4096) ;default:'';"`
	Enable int    `json:"enable" gorm:"type:int ;default:1;"`
}

func GenerateMsgUniqueID() string {
	newUUID := util.GenerateUniqueID()
	return fmt.Sprintf("CM%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{}) (int64, error) {
	var (
		err   error
		total int64
	)
	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).Take(&msg).Error
	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
		return msg, err
	}
	return msg, nil
}