mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-03 16:00:25 +08:00
104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"mayfly-go/internal/db/api/form"
|
|
"mayfly-go/internal/db/api/vo"
|
|
"mayfly-go/internal/db/application"
|
|
"mayfly-go/internal/db/domain/entity"
|
|
"mayfly-go/pkg/biz"
|
|
"mayfly-go/pkg/req"
|
|
"mayfly-go/pkg/utils/stringx"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type DataSyncTask struct {
|
|
DataSyncTaskApp application.DataSyncTask `inject:"DbDataSyncTaskApp"`
|
|
}
|
|
|
|
func (d *DataSyncTask) Tasks(rc *req.Ctx) {
|
|
queryCond, page := req.BindQueryAndPage[*entity.DataSyncTaskQuery](rc, new(entity.DataSyncTaskQuery))
|
|
res, err := d.DataSyncTaskApp.GetPageList(queryCond, page, new([]vo.DataSyncTaskListVO))
|
|
biz.ErrIsNil(err)
|
|
rc.ResData = res
|
|
}
|
|
|
|
func (d *DataSyncTask) Logs(rc *req.Ctx) {
|
|
queryCond, page := req.BindQueryAndPage[*entity.DataSyncLogQuery](rc, new(entity.DataSyncLogQuery))
|
|
res, err := d.DataSyncTaskApp.GetTaskLogList(queryCond, page, new([]vo.DataSyncLogListVO))
|
|
biz.ErrIsNil(err)
|
|
rc.ResData = res
|
|
}
|
|
|
|
func (d *DataSyncTask) SaveTask(rc *req.Ctx) {
|
|
form := &form.DataSyncTaskForm{}
|
|
task := req.BindJsonAndCopyTo[*entity.DataSyncTask](rc, form, new(entity.DataSyncTask))
|
|
|
|
// 解码base64 sql
|
|
sqlBytes, err := base64.StdEncoding.DecodeString(task.DataSql)
|
|
biz.ErrIsNilAppendErr(err, "sql解码失败: %s")
|
|
sql := stringx.TrimSpaceAndBr(string(sqlBytes))
|
|
task.DataSql = sql
|
|
form.DataSql = sql
|
|
|
|
rc.ReqParam = form
|
|
biz.ErrIsNil(d.DataSyncTaskApp.Save(rc.MetaCtx, task))
|
|
}
|
|
|
|
func (d *DataSyncTask) DeleteTask(rc *req.Ctx) {
|
|
taskId := rc.PathParam("taskId")
|
|
rc.ReqParam = taskId
|
|
ids := strings.Split(taskId, ",")
|
|
|
|
for _, v := range ids {
|
|
value, err := strconv.Atoi(v)
|
|
biz.ErrIsNilAppendErr(err, "string类型转换为int异常: %s")
|
|
biz.ErrIsNil(d.DataSyncTaskApp.Delete(rc.MetaCtx, uint64(value)))
|
|
}
|
|
}
|
|
|
|
func (d *DataSyncTask) ChangeStatus(rc *req.Ctx) {
|
|
form := &form.DataSyncTaskStatusForm{}
|
|
task := req.BindJsonAndCopyTo[*entity.DataSyncTask](rc, form, new(entity.DataSyncTask))
|
|
_ = d.DataSyncTaskApp.UpdateById(rc.MetaCtx, task)
|
|
|
|
if task.Status == entity.DataSyncTaskStatusEnable {
|
|
task, err := d.DataSyncTaskApp.GetById(new(entity.DataSyncTask), task.Id)
|
|
biz.ErrIsNil(err, "该任务不存在")
|
|
d.DataSyncTaskApp.AddCronJob(task)
|
|
} else {
|
|
d.DataSyncTaskApp.RemoveCronJobById(task.Id)
|
|
}
|
|
// 记录请求日志
|
|
rc.ReqParam = form
|
|
}
|
|
|
|
func (d *DataSyncTask) Run(rc *req.Ctx) {
|
|
taskId := getTaskId(rc)
|
|
rc.ReqParam = taskId
|
|
_ = d.DataSyncTaskApp.RunCronJob(taskId)
|
|
}
|
|
|
|
func (d *DataSyncTask) Stop(rc *req.Ctx) {
|
|
taskId := getTaskId(rc)
|
|
rc.ReqParam = taskId
|
|
|
|
task := new(entity.DataSyncTask)
|
|
task.Id = taskId
|
|
task.RunningState = entity.DataSyncTaskRunStateStop
|
|
_ = d.DataSyncTaskApp.UpdateById(rc.MetaCtx, task)
|
|
}
|
|
|
|
func (d *DataSyncTask) GetTask(rc *req.Ctx) {
|
|
taskId := getTaskId(rc)
|
|
dbEntity, _ := d.DataSyncTaskApp.GetById(new(entity.DataSyncTask), taskId)
|
|
rc.ResData = dbEntity
|
|
}
|
|
|
|
func getTaskId(rc *req.Ctx) uint64 {
|
|
instanceId := rc.PathParamInt("taskId")
|
|
biz.IsTrue(instanceId > 0, "instanceId 错误")
|
|
return uint64(instanceId)
|
|
}
|