package com.vincent.rsf.server.system.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.vincent.rsf.framework.common.Cools;
|
import com.vincent.rsf.framework.common.R;
|
import com.vincent.rsf.framework.common.SnowflakeIdWorker;
|
import com.vincent.rsf.server.ai.service.diagnosis.AiDiagnosisPlanRunnerService;
|
import com.vincent.rsf.server.common.annotation.OperationLog;
|
import com.vincent.rsf.server.common.domain.BaseParam;
|
import com.vincent.rsf.server.common.domain.PageParam;
|
import com.vincent.rsf.server.common.utils.ExcelUtil;
|
import com.vincent.rsf.server.system.entity.AiDiagnosisPlan;
|
import com.vincent.rsf.server.system.service.AiDiagnosisPlanService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.util.Arrays;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.Map;
|
|
@RestController
|
public class AiDiagnosisPlanController extends BaseController {
|
|
@Autowired
|
private AiDiagnosisPlanService aiDiagnosisPlanService;
|
@Autowired
|
private AiDiagnosisPlanRunnerService aiDiagnosisPlanRunnerService;
|
@Autowired
|
private ThreadPoolTaskScheduler taskScheduler;
|
@Autowired
|
private SnowflakeIdWorker snowflakeIdWorker;
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:list')")
|
@PostMapping("/aiDiagnosisPlan/page")
|
public R page(@RequestBody Map<String, Object> map) {
|
BaseParam baseParam = buildParam(map, BaseParam.class);
|
PageParam<AiDiagnosisPlan, BaseParam> pageParam = new PageParam<>(baseParam, AiDiagnosisPlan.class);
|
com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<AiDiagnosisPlan> wrapper = pageParam.buildWrapper(true);
|
wrapper.eq("tenant_id", getTenantId());
|
return R.ok().add(aiDiagnosisPlanService.page(pageParam, wrapper));
|
}
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:list')")
|
@GetMapping("/aiDiagnosisPlan/{id}")
|
public R get(@PathVariable("id") Long id) {
|
AiDiagnosisPlan plan = aiDiagnosisPlanService.getTenantPlan(getTenantId(), id);
|
if (plan == null) {
|
return R.error("plan not found");
|
}
|
return R.ok().add(plan);
|
}
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:save')")
|
@OperationLog("Create AiDiagnosisPlan")
|
@PostMapping("/aiDiagnosisPlan/save")
|
public R save(@RequestBody AiDiagnosisPlan plan) {
|
if (Cools.isEmpty(plan.getPlanName()) || Cools.isEmpty(plan.getCronExpr())) {
|
return R.error("计划名称和Cron表达式不能为空");
|
}
|
if (!aiDiagnosisPlanService.validateCron(plan.getCronExpr())) {
|
return R.error("Cron表达式不合法");
|
}
|
Date now = new Date();
|
plan.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
|
plan.setTenantId(getTenantId());
|
plan.setSceneCode(Cools.isEmpty(plan.getSceneCode()) ? "system_diagnose" : plan.getSceneCode());
|
plan.setRunningFlag(0);
|
plan.setStatus(plan.getStatus() == null ? 1 : plan.getStatus());
|
plan.setNextRunTime(Integer.valueOf(1).equals(plan.getStatus())
|
? aiDiagnosisPlanService.calculateNextRunTime(plan.getCronExpr(), now)
|
: null);
|
plan.setCreateBy(getLoginUserId());
|
plan.setCreateTime(now);
|
plan.setUpdateBy(getLoginUserId());
|
plan.setUpdateTime(now);
|
if (!aiDiagnosisPlanService.save(plan)) {
|
return R.error("Save Fail");
|
}
|
return R.ok("Save Success").add(plan);
|
}
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:update')")
|
@OperationLog("Update AiDiagnosisPlan")
|
@PostMapping("/aiDiagnosisPlan/update")
|
public R update(@RequestBody AiDiagnosisPlan plan) {
|
AiDiagnosisPlan existed = aiDiagnosisPlanService.getTenantPlan(getTenantId(), plan.getId());
|
if (existed == null) {
|
return R.error("plan not found");
|
}
|
if (Cools.isEmpty(plan.getPlanName()) || Cools.isEmpty(plan.getCronExpr())) {
|
return R.error("计划名称和Cron表达式不能为空");
|
}
|
if (!aiDiagnosisPlanService.validateCron(plan.getCronExpr())) {
|
return R.error("Cron表达式不合法");
|
}
|
plan.setTenantId(getTenantId());
|
plan.setSceneCode(Cools.isEmpty(plan.getSceneCode()) ? existed.getSceneCode() : plan.getSceneCode());
|
plan.setRunningFlag(existed.getRunningFlag());
|
plan.setLastResult(existed.getLastResult());
|
plan.setLastDiagnosisId(existed.getLastDiagnosisId());
|
plan.setLastRunTime(existed.getLastRunTime());
|
plan.setLastMessage(existed.getLastMessage());
|
plan.setNextRunTime(Integer.valueOf(1).equals(plan.getStatus())
|
? aiDiagnosisPlanService.calculateNextRunTime(plan.getCronExpr(), new Date())
|
: null);
|
plan.setCreateBy(existed.getCreateBy());
|
plan.setCreateTime(existed.getCreateTime());
|
plan.setUpdateBy(getLoginUserId());
|
plan.setUpdateTime(new Date());
|
if (!aiDiagnosisPlanService.updateById(plan)) {
|
return R.error("Update Fail");
|
}
|
return R.ok("Update Success").add(plan);
|
}
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:remove')")
|
@OperationLog("Delete AiDiagnosisPlan")
|
@PostMapping("/aiDiagnosisPlan/remove/{ids}")
|
public R remove(@PathVariable Long[] ids) {
|
List<Long> idList = Arrays.asList(ids);
|
List<AiDiagnosisPlan> plans = aiDiagnosisPlanService.list(new LambdaQueryWrapper<AiDiagnosisPlan>()
|
.eq(AiDiagnosisPlan::getTenantId, getTenantId())
|
.in(AiDiagnosisPlan::getId, idList));
|
if (plans.size() != idList.size() || !aiDiagnosisPlanService.removeByIds(idList)) {
|
return R.error("Delete Fail");
|
}
|
return R.ok("Delete Success").add(ids);
|
}
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:list')")
|
@PostMapping("/aiDiagnosisPlan/export")
|
public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
|
ExcelUtil.build(ExcelUtil.create(aiDiagnosisPlanService.list(new LambdaQueryWrapper<AiDiagnosisPlan>()
|
.eq(AiDiagnosisPlan::getTenantId, getTenantId())), AiDiagnosisPlan.class), response);
|
}
|
|
@PreAuthorize("hasAuthority('system:aiDiagnosisPlan:update')")
|
@OperationLog("Run AiDiagnosisPlan")
|
@PostMapping("/ai/diagnosis-plan/run")
|
public R run(@RequestBody Map<String, Object> map) {
|
Long id = Long.valueOf(String.valueOf(map.get("id")));
|
AiDiagnosisPlan plan = aiDiagnosisPlanService.getTenantPlan(getTenantId(), id);
|
if (plan == null) {
|
return R.error("plan not found");
|
}
|
if (Integer.valueOf(1).equals(plan.getRunningFlag())) {
|
return R.error("计划正在执行中");
|
}
|
Date nextRunTime = Integer.valueOf(1).equals(plan.getStatus())
|
? aiDiagnosisPlanService.calculateNextRunTime(plan.getCronExpr(), new Date())
|
: null;
|
boolean acquired = aiDiagnosisPlanService.acquireForExecution(
|
plan.getId(),
|
getLoginUserId(),
|
"手动执行中",
|
nextRunTime
|
);
|
if (!acquired) {
|
return R.error("计划正在执行中");
|
}
|
taskScheduler.execute(() -> aiDiagnosisPlanRunnerService.runPlan(plan.getId(), true));
|
return R.ok();
|
}
|
}
|
