package tech.powerjob.server.web.controller;
|
|
import org.apache.commons.lang3.StringUtils;
|
import tech.powerjob.common.request.http.SaveJobInfoRequest;
|
import tech.powerjob.common.response.ResultDTO;
|
import tech.powerjob.server.auth.Permission;
|
import tech.powerjob.server.auth.RoleScope;
|
import tech.powerjob.server.auth.interceptor.ApiPermission;
|
import tech.powerjob.common.enums.SwitchableStatus;
|
import tech.powerjob.server.persistence.PageResult;
|
import tech.powerjob.server.persistence.remote.model.JobInfoDO;
|
import tech.powerjob.server.persistence.remote.repository.JobInfoRepository;
|
import tech.powerjob.server.core.service.JobService;
|
import tech.powerjob.server.web.request.QueryJobInfoRequest;
|
import tech.powerjob.server.web.response.JobInfoVO;
|
import com.google.common.collect.Lists;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.data.domain.Page;
|
import org.springframework.data.domain.PageRequest;
|
import org.springframework.data.domain.Sort;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
import java.util.Optional;
|
import java.util.stream.Collectors;
|
|
/**
|
* 任务信息管理 Controller
|
*
|
* @author tjq
|
* @since 2020/3/30
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("/job")
|
public class JobController {
|
|
@Resource
|
private JobService jobService;
|
@Resource
|
private JobInfoRepository jobInfoRepository;
|
|
@PostMapping("/save")
|
@ApiPermission(name = "Job-Save", roleScope = RoleScope.APP, requiredPermission = Permission.WRITE)
|
public ResultDTO<Void> saveJobInfo(@RequestBody SaveJobInfoRequest request) {
|
jobService.saveJob(request);
|
return ResultDTO.success(null);
|
}
|
|
@PostMapping("/copy")
|
@ApiPermission(name = "Job-Copy", roleScope = RoleScope.APP, requiredPermission = Permission.WRITE)
|
public ResultDTO<JobInfoVO> copyJob(String jobId) {
|
return ResultDTO.success(JobInfoVO.from(jobService.copyJob(Long.valueOf(jobId))));
|
}
|
|
@GetMapping("/export")
|
@ApiPermission(name = "Job-Export", roleScope = RoleScope.APP, requiredPermission = Permission.READ)
|
public ResultDTO<SaveJobInfoRequest> exportJob(String jobId) {
|
return ResultDTO.success(jobService.exportJob(Long.valueOf(jobId)));
|
}
|
|
@GetMapping("/disable")
|
@ApiPermission(name = "Job-Disable", roleScope = RoleScope.APP, requiredPermission = Permission.WRITE)
|
public ResultDTO<Void> disableJob(String jobId) {
|
jobService.disableJob(Long.valueOf(jobId));
|
return ResultDTO.success(null);
|
}
|
|
@GetMapping("/delete")
|
@ApiPermission(name = "Job-Delete", roleScope = RoleScope.APP, requiredPermission = Permission.WRITE)
|
public ResultDTO<Void> deleteJob(String jobId) {
|
jobService.deleteJob(Long.valueOf(jobId));
|
return ResultDTO.success(null);
|
}
|
|
@GetMapping("/run")
|
@ApiPermission(name = "Job-Copy", roleScope = RoleScope.APP, requiredPermission = Permission.OPS)
|
public ResultDTO<Long> runImmediately(String appId, String jobId, @RequestParam(required = false) String instanceParams) {
|
return ResultDTO.success(jobService.runJob(Long.valueOf(appId), Long.valueOf(jobId), instanceParams, 0L));
|
}
|
|
@PostMapping("/list")
|
@ApiPermission(name = "Job-Copy", roleScope = RoleScope.APP, requiredPermission = Permission.READ)
|
public ResultDTO<PageResult<JobInfoVO>> listJobs(@RequestBody QueryJobInfoRequest request) {
|
|
Sort sort = Sort.by(Sort.Direction.ASC, "id");
|
PageRequest pageRequest = PageRequest.of(request.getIndex(), request.getPageSize(), sort);
|
Page<JobInfoDO> jobInfoPage;
|
|
// 无查询条件，查询全部
|
if (request.getJobId() == null && StringUtils.isEmpty(request.getKeyword())) {
|
jobInfoPage = jobInfoRepository.findByAppIdAndStatusNot(request.getAppId(), SwitchableStatus.DELETED.getV(), pageRequest);
|
return ResultDTO.success(convertPage(jobInfoPage));
|
}
|
|
// 有 jobId，直接精确查询
|
if (request.getJobId() != null) {
|
|
Optional<JobInfoDO> jobInfoOpt = jobInfoRepository.findById(request.getJobId());
|
|
PageResult<JobInfoVO> result = new PageResult<>();
|
|
if (!jobInfoOpt.isPresent()) {
|
result.setTotalPages(0);
|
result.setTotalItems(0);
|
result.setData(Lists.newLinkedList());
|
return ResultDTO.success(result);
|
}
|
|
if (!jobInfoOpt.get().getAppId().equals(request.getAppId())){
|
return ResultDTO.failed("请输入该app下的jobId");
|
}
|
|
result.setIndex(0);
|
result.setPageSize(request.getPageSize());
|
|
result.setTotalItems(1);
|
result.setTotalPages(1);
|
result.setData(Lists.newArrayList(JobInfoVO.from(jobInfoOpt.get())));
|
|
return ResultDTO.success(result);
|
}
|
|
// 模糊查询
|
String condition = "%" + request.getKeyword() + "%";
|
jobInfoPage = jobInfoRepository.findByAppIdAndJobNameLikeAndStatusNot(request.getAppId(), condition, SwitchableStatus.DELETED.getV(), pageRequest);
|
return ResultDTO.success(convertPage(jobInfoPage));
|
}
|
|
|
private static PageResult<JobInfoVO> convertPage(Page<JobInfoDO> jobInfoPage) {
|
List<JobInfoVO> jobInfoVOList = jobInfoPage.getContent().stream().map(JobInfoVO::from).collect(Collectors.toList());
|
|
PageResult<JobInfoVO> pageResult = new PageResult<>(jobInfoPage);
|
pageResult.setData(jobInfoVOList);
|
return pageResult;
|
}
|
|
}
|
