kpt-pasture/module/crontab/other.go

package crontab

import (
	"errors"
	"kpt-pasture/model"
	"kpt-pasture/module/backend"
	"time"

	"gitee.com/xuyiping_admin/pkg/logger/zaplog"
	"go.uber.org/zap"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"

	"gorm.io/gorm"
)

// CreateCrontabLog 生成日志记录
func (e *Entry) CreateCrontabLog(name string) bool {
	currDateTime := time.Now().Format(model.LayoutDate2)
	newCronLog := &model.CronLog{}
	if err := e.DB.Where("name = ?", name).
		Where("date = ?", currDateTime).First(newCronLog).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			e.DB.Create(model.NewCronLog(name))
			return true
		}
	}
	return false
}

// CreatedWorkOrderCalendar 创建当天工单日历记录
func (e *Entry) CreatedWorkOrderCalendar(calendarType pasturePb.CalendarType_Kind, count int32) {
	calendarTypeName := backend.CalendarTypeMap()[calendarType]
	workOrderCalendar := model.NewWorkOrderCalendar(calendarTypeName, calendarType, count)
	if err := e.DB.Model(&model.WorkOrderCalendar{}).Create(workOrderCalendar).Error; err != nil {
		zaplog.Error("CreatedWorkOrderCalendar", zap.Any("err", err), zap.Any("workOrderCalendar", workOrderCalendar))
	}
}