init...

core/dao/cron_job_dao.go

@@ -0,0 +1,99 @@
+package dao
+
+import (
+	"cutego/core/api/v1/request"
+	"cutego/core/entity"
+	"cutego/pkg/common"
+	"cutego/pkg/page"
+	"github.com/druidcaesa/gotool"
+	"github.com/go-xorm/xorm"
+)
+
+type CronJobDao struct {
+}
+
+func (d CronJobDao) sql(session *xorm.Session) *xorm.Session {
+	return session.Table("sys_cron_job")
+}
+
+// SelectPage 分页查询数据
+func (d CronJobDao) SelectPage(query request.CronJobQuery) ([]entity.SysCronJob, int64) {
+	configs := make([]entity.SysCronJob, 0)
+	session := d.sql(SqlDB.NewSession())
+	if gotool.StrUtils.HasNotEmpty(query.JobName) {
+		session.And("job_name like concat('%', ?, '%')", query.JobName)
+	}
+	total, _ := page.GetTotal(session.Clone())
+	err := session.Limit(query.PageSize, page.StartSize(query.PageNum, query.PageSize)).Find(&configs)
+	if err != nil {
+		common.ErrorLog(err)
+		return nil, 0
+	}
+	return configs, total
+}
+
+// Insert 添加数据
+func (d CronJobDao) Insert(config entity.SysCronJob) int64 {
+	session := SqlDB.NewSession()
+	session.Begin()
+	insert, err := session.Insert(&config)
+	if err != nil {
+		common.ErrorLog(err)
+		session.Rollback()
+		return 0
+	}
+	session.Commit()
+	return insert
+}
+
+// SelectById 查询数据
+func (d CronJobDao) SelectById(id int64) *entity.SysCronJob {
+	config := entity.SysCronJob{}
+	session := d.sql(SqlDB.NewSession())
+	_, err := session.Where("job_id = ?", id).Get(&config)
+	if err != nil {
+		common.ErrorLog(err)
+		return nil
+	}
+	return &config
+}
+
+// Update 修改数据
+func (d CronJobDao) Update(config entity.SysCronJob) int64 {
+	//session := SqlDB.NewSession()
+	//session.Begin()
+	//update, err := session.Where("job_id = ?", config.JobId).Update(&config)
+	//if err != nil {
+	//	common.ErrorLog(err)
+	//	session.Rollback()
+	//	return 0
+	//}
+	//session.Commit()
+	return CustomUpdateById("job_id", config.JobId, config)
+}
+
+// Remove 删除数据
+func (d CronJobDao) Delete(list []int64) bool {
+	session := SqlDB.NewSession()
+	session.Begin()
+	_, err := session.In("config_id", list).Delete(&entity.SysCronJob{})
+	if err != nil {
+		common.ErrorLog(err)
+		session.Rollback()
+		return false
+	}
+	session.Commit()
+	return true
+}
+
+// 通过方法别名获取任务详情
+func (d CronJobDao) SelectByFuncAlias(funcAlias string) *entity.SysCronJob {
+	config := entity.SysCronJob{}
+	session := d.sql(SqlDB.NewSession())
+	_, err := session.Where("func_alias = ?", funcAlias).Get(&config)
+	if err != nil {
+		common.ErrorLog(err)
+		return nil
+	}
+	return &config
+}