@@ -1,7 +1,7 @@ | |||
# 项目相关配置 | |||
ruoyi: | |||
# 文件路径 示例( Windows配置D:/poject,Linux配置 /home/poject) | |||
profile: /home/pojectFile | |||
profile: /home/pojectFile/ | |||
# Spring配置 数据源配置 | |||
spring: | |||
# redis 配置 | |||
@@ -52,7 +52,7 @@ spring: | |||
# 国际化资源文件路径 | |||
basename: i18n/messages | |||
profiles: | |||
active: dev | |||
active: test | |||
# 文件上传 | |||
servlet: | |||
multipart: | |||
@@ -4,23 +4,20 @@ import java.util.List; | |||
import javax.servlet.http.HttpServletResponse; | |||
import java.util.Arrays; | |||
import com.alibaba.fastjson2.JSON; | |||
import com.alibaba.fastjson2.JSONObject; | |||
import com.ruoyi.business.domain.bo.AddApplyPlanBO; | |||
import com.ruoyi.business.domain.bo.JobQueryBO; | |||
import com.ruoyi.business.domain.bo.YinDaoCallBackBO; | |||
import com.ruoyi.common.utils.bean.BeanUtils; | |||
import io.swagger.annotations.Api; | |||
import io.swagger.annotations.ApiImplicitParam; | |||
import io.swagger.annotations.ApiImplicitParams; | |||
import io.swagger.annotations.ApiOperation; | |||
import lombok.extern.slf4j.Slf4j; | |||
import org.springframework.security.access.prepost.PreAuthorize; | |||
import org.springframework.beans.factory.annotation.Autowired; | |||
import org.springframework.web.bind.annotation.GetMapping; | |||
import org.springframework.web.bind.annotation.PostMapping; | |||
import org.springframework.web.bind.annotation.PutMapping; | |||
import org.springframework.web.bind.annotation.DeleteMapping; | |||
import org.springframework.web.bind.annotation.PathVariable; | |||
import org.springframework.web.bind.annotation.RequestBody; | |||
import org.springframework.web.bind.annotation.RequestMapping; | |||
import org.springframework.web.bind.annotation.RestController; | |||
import org.springframework.web.bind.annotation.*; | |||
import com.ruoyi.common.annotation.Log; | |||
import com.ruoyi.common.core.controller.BaseController; | |||
import com.ruoyi.common.core.domain.AjaxResult; | |||
@@ -39,10 +36,33 @@ import com.ruoyi.common.core.page.TableDataInfo; | |||
@Api(tags = "应用执行计划管理接口") | |||
@RestController | |||
@RequestMapping("/business/ctApplyPlan") | |||
@Slf4j | |||
public class ApplyPlanController extends BaseController { | |||
@Autowired | |||
private IApplyPlanService applyPlanService; | |||
@PostMapping("/callBack/updatePlanStatus") | |||
public void updatePlanStatus(@RequestParam(required = false) String bodyMd5, @RequestParam(required = false) Long timestamp, | |||
@RequestParam(required = false) String sign, @RequestBody String body) { | |||
log.info("fake callback: bodyMd5:{}", bodyMd5); | |||
log.info("fake callback: sign:{}", sign); | |||
log.info("fake callback: timestamp:{}", timestamp); | |||
log.info("fake callback: body:{}", body); | |||
YinDaoCallBackBO yinDaoCallBackBO = JSONObject.parseObject(body, YinDaoCallBackBO.class); | |||
// 获取应用名称 task_uuid | |||
if (applyPlanService.ydCallBack(yinDaoCallBackBO)) { | |||
log.debug("任务id:" + yinDaoCallBackBO.getJobUuid() + "回调成功,回调状态:" + yinDaoCallBackBO.getStatus()); | |||
}else{ | |||
log.debug("任务id:" + yinDaoCallBackBO.getJobUuid() + "回调失败,回调状态:" + yinDaoCallBackBO.getStatus()); | |||
} | |||
//异步处理业务 | |||
// new Runable() -> { | |||
// do things | |||
// } | |||
} | |||
/** | |||
* 重新运行 | |||
*/ | |||
@@ -90,7 +110,8 @@ public class ApplyPlanController extends BaseController { | |||
@ApiOperation(value = "获取应用执行计划管理详细信息", httpMethod = "GET", response = ApplyPlan.class) | |||
@GetMapping(value = "/{id}") | |||
public AjaxResult getInfo(@PathVariable("id") Long id) { | |||
return success(applyPlanService.getById(id)); | |||
ApplyPlan applyPlan = applyPlanService.getById(id); | |||
return success(applyPlan); | |||
} | |||
/** | |||
@@ -100,6 +121,10 @@ public class ApplyPlanController extends BaseController { | |||
@Log(title = "应用执行计划管理", businessType = BusinessType.INSERT) | |||
@PostMapping | |||
public AjaxResult add(@RequestBody AddApplyPlanBO addApplyPlanBO) { | |||
// 随机分配 | |||
addApplyPlanBO.setPlanType("1"); | |||
addApplyPlanBO.setWaitTimeout("10m"); | |||
addApplyPlanBO.setPriority(0); | |||
return toAjax(applyPlanService.save(addApplyPlanBO)); | |||
} | |||
@@ -3,6 +3,14 @@ package com.ruoyi.business.controller; | |||
import java.util.List; | |||
import javax.servlet.http.HttpServletResponse; | |||
import java.util.Arrays; | |||
import java.util.stream.Collectors; | |||
import com.ruoyi.business.domain.Apply; | |||
import com.ruoyi.business.domain.bo.AddBatchResourceLibraryBO; | |||
import com.ruoyi.business.service.IApplyService; | |||
import com.ruoyi.common.core.domain.entity.SysUser; | |||
import com.ruoyi.common.utils.SecurityUtils; | |||
import com.ruoyi.common.utils.StringUtils; | |||
import io.swagger.annotations.Api; | |||
import io.swagger.annotations.ApiImplicitParam; | |||
import io.swagger.annotations.ApiImplicitParams; | |||
@@ -35,21 +43,44 @@ import com.ruoyi.common.core.page.TableDataInfo; | |||
@Api(tags = "资源库管理接口") | |||
@RestController | |||
@RequestMapping("/business/ctResourceLibrary") | |||
public class ResourceLibraryController extends BaseController | |||
{ | |||
public class ResourceLibraryController extends BaseController { | |||
@Autowired | |||
private IResourceLibraryService resourceLibraryService; | |||
@Autowired | |||
private IApplyService applyService; | |||
/** | |||
* 根据应用查询所需参数 | |||
*/ | |||
@ApiOperation(value = "根据应用查询所需参数", httpMethod = "GET", response = ResourceLibrary.class) | |||
@GetMapping("/listByAppId") | |||
public AjaxResult listByAppId(String appId) { | |||
List<ResourceLibrary> list = resourceLibraryService.lambdaQuery().eq(ResourceLibrary::getAppId, appId).list(); | |||
return AjaxResult.success(list); | |||
} | |||
/** | |||
* 查询资源库管理列表 | |||
*/ | |||
@ApiOperation(value = "查询资源库管理列表", httpMethod = "GET", response = ResourceLibrary.class) | |||
@GetMapping("/list") | |||
public TableDataInfo list(ResourceLibrary resourceLibrary) | |||
{ | |||
public TableDataInfo list(ResourceLibrary resourceLibrary) { | |||
// 是否管理员 | |||
SysUser user = SecurityUtils.getLoginUser().getUser(); | |||
boolean isRole = user.getRoles().stream().filter(e -> e.getRoleId() == 1).count() > 0; | |||
if (!isRole) { | |||
List<String> appIdList = applyService.lambdaQuery().eq(Apply::getDeptId, SecurityUtils.getDeptId()).list().stream().map(Apply::getAppId).collect(Collectors.toList()); | |||
startPage(); | |||
List<ResourceLibrary> list = resourceLibraryService.list(resourceLibrary); | |||
List<ResourceLibrary> list = resourceLibraryService.list(resourceLibrary, appIdList); | |||
return getDataTable(list); | |||
} else { | |||
startPage(); | |||
List<ResourceLibrary> list = resourceLibraryService.list(resourceLibrary, null); | |||
return getDataTable(list); | |||
} | |||
} | |||
/** | |||
@@ -58,31 +89,53 @@ public class ResourceLibraryController extends BaseController | |||
@ApiOperation(value = "导出资源库管理列表", httpMethod = "POST") | |||
@Log(title = "资源库管理", businessType = BusinessType.EXPORT) | |||
@PostMapping("/export") | |||
public void export(HttpServletResponse response, ResourceLibrary resourceLibrary) | |||
{ | |||
List<ResourceLibrary> list = resourceLibraryService.list(resourceLibrary); | |||
public void export(HttpServletResponse response, ResourceLibrary resourceLibrary) { | |||
List<String> appIdList = applyService.lambdaQuery().eq(Apply::getDeptId, SecurityUtils.getDeptId()).list().stream().map(Apply::getAppId).collect(Collectors.toList()); | |||
// 是否管理员 | |||
boolean isRole = Arrays.stream(SecurityUtils.getLoginUser().getUser().getRoleIds()).filter(e -> e == 1).count() > 0; | |||
if (!isRole) { | |||
List<ResourceLibrary> list = resourceLibraryService.list(resourceLibrary, appIdList); | |||
ExcelUtil<ResourceLibrary> util = new ExcelUtil<ResourceLibrary>(ResourceLibrary.class); | |||
util.exportExcel(response, list, "资源库管理数据"); | |||
} else { | |||
List<ResourceLibrary> list = resourceLibraryService.list(resourceLibrary, null); | |||
ExcelUtil<ResourceLibrary> util = new ExcelUtil<ResourceLibrary>(ResourceLibrary.class); | |||
util.exportExcel(response, list, "资源库管理数据"); | |||
} | |||
} | |||
/** | |||
* 获取资源库管理详细信息 | |||
*/ | |||
@ApiOperation(value = "获取资源库管理详细信息", httpMethod = "GET", response = ResourceLibrary.class) | |||
@GetMapping(value = "/{id}") | |||
public AjaxResult getInfo(@PathVariable("id") Long id) | |||
{ | |||
public AjaxResult getInfo(@PathVariable("id") Long id) { | |||
return success(resourceLibraryService.getById(id)); | |||
} | |||
/** | |||
* 批量新增资源信息 | |||
*/ | |||
@ApiOperation(value = "批量新增资源信息", httpMethod = "POST") | |||
@Log(title = "批量新增资源信息", businessType = BusinessType.INSERT) | |||
@PostMapping("/addBatch") | |||
public AjaxResult addBatch(@RequestBody AddBatchResourceLibraryBO addBatchResourceLibraryBO) { | |||
return toAjax(resourceLibraryService.addBatch(addBatchResourceLibraryBO)); | |||
} | |||
/** | |||
* 新增资源库管理 | |||
*/ | |||
@ApiOperation(value = "新增资源库管理", httpMethod = "POST") | |||
@Log(title = "资源库管理", businessType = BusinessType.INSERT) | |||
@PostMapping | |||
public AjaxResult add(@RequestBody ResourceLibrary resourceLibrary) | |||
{ | |||
public AjaxResult add(@RequestBody ResourceLibrary resourceLibrary) { | |||
Integer count = resourceLibraryService.lambdaQuery().eq(ResourceLibrary::getAppId, resourceLibrary.getAppId()).eq(ResourceLibrary::getResourceName, resourceLibrary.getResourceName()).count(); | |||
if (count > 0) { | |||
return AjaxResult.error("已经有相同的资源键,无需重复创建"); | |||
} | |||
return toAjax(resourceLibraryService.save(resourceLibrary)); | |||
} | |||
@@ -92,8 +145,7 @@ public class ResourceLibraryController extends BaseController | |||
@ApiOperation(value = "修改资源库管理", httpMethod = "PUT") | |||
@Log(title = "资源库管理", businessType = BusinessType.UPDATE) | |||
@PutMapping | |||
public AjaxResult edit(@RequestBody ResourceLibrary resourceLibrary) | |||
{ | |||
public AjaxResult edit(@RequestBody ResourceLibrary resourceLibrary) { | |||
return toAjax(resourceLibraryService.updateById(resourceLibrary)); | |||
} | |||
@@ -103,8 +155,7 @@ public class ResourceLibraryController extends BaseController | |||
@ApiOperation(value = "删除资源库管理", httpMethod = "DELETE") | |||
@Log(title = "资源库管理", businessType = BusinessType.DELETE) | |||
@DeleteMapping("/{ids}") | |||
public AjaxResult remove(@PathVariable Long[] ids) | |||
{ | |||
public AjaxResult remove(@PathVariable Long[] ids) { | |||
return toAjax(resourceLibraryService.removeByIds(Arrays.asList(ids))); | |||
} | |||
} |
@@ -107,8 +107,8 @@ public class ApplyPlan extends BaseEntity | |||
/** 执行时间 */ | |||
@JsonFormat(pattern = "yyyy-MM-dd") | |||
@Excel(name = "执行时间", width = 30, dateFormat = "yyyy-MM-dd") | |||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") | |||
@Excel(name = "执行时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss") | |||
@ApiModelProperty(name="excTime",value = "执行时间") | |||
private Date excTime; | |||
@@ -147,15 +147,15 @@ public class ApplyPlan extends BaseEntity | |||
/** 任务开始运行的时间 */ | |||
@JsonFormat(pattern = "yyyy-MM-dd") | |||
@Excel(name = "任务开始运行的时间", width = 30, dateFormat = "yyyy-MM-dd") | |||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") | |||
@Excel(name = "任务开始运行的时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss") | |||
@ApiModelProperty(name="startTime",value = "任务开始运行的时间") | |||
private Date startTime; | |||
/** 任务结束运行的时间 */ | |||
@JsonFormat(pattern = "yyyy-MM-dd") | |||
@Excel(name = "任务结束运行的时间", width = 30, dateFormat = "yyyy-MM-dd") | |||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") | |||
@Excel(name = "任务结束运行的时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss") | |||
@ApiModelProperty(name="endTime",value = "任务结束运行的时间") | |||
private Date endTime; | |||
@@ -3,6 +3,7 @@ package com.ruoyi.business.domain.bo; | |||
import com.baomidou.mybatisplus.annotation.TableField; | |||
import com.baomidou.mybatisplus.annotation.TableName; | |||
import com.fasterxml.jackson.annotation.JsonFormat; | |||
import com.ruoyi.business.domain.ResourceLibrary; | |||
import com.ruoyi.common.annotation.Excel; | |||
import com.ruoyi.common.core.domain.BaseEntity; | |||
import io.swagger.annotations.ApiModel; | |||
@@ -64,7 +65,7 @@ public class AddApplyPlanBO extends BaseEntity | |||
/** appid */ | |||
@Excel(name = "appid") | |||
@ApiModelProperty(name="appId",value = "appid") | |||
private List<String> appId; | |||
private String appId; | |||
/** 应用类型枚举(app:应用 activity:指令) */ | |||
@@ -105,8 +106,8 @@ public class AddApplyPlanBO extends BaseEntity | |||
/** 执行时间 */ | |||
@JsonFormat(pattern = "yyyy-MM-dd") | |||
@Excel(name = "执行时间", width = 30, dateFormat = "yyyy-MM-dd") | |||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") | |||
@Excel(name = "执行时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss") | |||
@ApiModelProperty(name="excTime",value = "执行时间") | |||
private Date excTime; | |||
@@ -115,6 +116,11 @@ public class AddApplyPlanBO extends BaseEntity | |||
@ApiModelProperty(name="cronExpression",value = "执行表达式") | |||
private String cronExpression; | |||
@ApiModelProperty(name="resourceLibrarieList",value = "参数列表") | |||
@TableField(exist = false) | |||
private List<ResourceLibrary> resourceLibrarieList; | |||
@Override | |||
public String toString() { | |||
return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) | |||
@@ -0,0 +1,54 @@ | |||
package com.ruoyi.business.domain.bo; | |||
import com.baomidou.mybatisplus.annotation.TableField; | |||
import com.baomidou.mybatisplus.annotation.TableName; | |||
import com.ruoyi.business.domain.ResourceLibrary; | |||
import com.ruoyi.common.annotation.Excel; | |||
import com.ruoyi.common.core.domain.BaseEntity; | |||
import io.swagger.annotations.ApiModel; | |||
import io.swagger.annotations.ApiModelProperty; | |||
import lombok.AllArgsConstructor; | |||
import lombok.Builder; | |||
import lombok.Data; | |||
import lombok.NoArgsConstructor; | |||
import org.apache.commons.lang3.builder.ToStringBuilder; | |||
import org.apache.commons.lang3.builder.ToStringStyle; | |||
import java.util.List; | |||
/** | |||
* 资源库管理对象 ct_resource_library | |||
* | |||
* @author LiuChengRan | |||
* @date 2024-06-17 | |||
*/ | |||
@Data | |||
@Builder(toBuilder = true) | |||
@NoArgsConstructor | |||
@AllArgsConstructor | |||
@ApiModel(value = "ResourceLibrary", description = "资源库管理") | |||
public class AddBatchResourceLibraryBO extends BaseEntity { | |||
@TableField(exist = false) | |||
private static final long serialVersionUID = 1L; | |||
@Excel(name = "所属应用id") | |||
@ApiModelProperty(name = "appId", value = "所属应用id") | |||
private String appId; | |||
@Excel(name = "参数集合") | |||
@ApiModelProperty(name = "list", value = "参数集合") | |||
private List<ResourceLibrary> list; | |||
@Override | |||
public String toString() { | |||
return new ToStringBuilder(this, ToStringStyle.MULTI_LINE_STYLE) | |||
.append("id", getId()) | |||
.append("createBy", getCreateBy()) | |||
.append("createTime", getCreateTime()) | |||
.append("updateBy", getUpdateBy()) | |||
.append("updateTime", getUpdateTime()) | |||
.append("remark", getRemark()) | |||
.append("deleted", getDeleted()) | |||
.toString(); | |||
} | |||
} |
@@ -65,17 +65,17 @@ public class ApplyStartBO { | |||
@Excel(name = "计划参数") | |||
@ApiModelProperty(name="planParams",value = "计划参数") | |||
private String planParams; | |||
/** | |||
* 应用运行参数 | |||
* <p> | |||
* | |||
* 非必填 | |||
*/ | |||
private List<RobotParam> planParamsList; | |||
private List<RobotParam> params; | |||
/** | |||
* 等待超时时间 | |||
* | |||
* @see com.xybot.api.sdk.enums.WaitTimeoutEnum | |||
* <p> | |||
* 非必填,不填,默认为10分钟 | |||
*/ | |||
@@ -83,8 +83,6 @@ public class ApplyStartBO { | |||
/** | |||
* 优先级 | |||
* | |||
* @see com.xybot.api.sdk.enums.PriorityEnum | |||
* 非必填,不填,默认为middle | |||
*/ | |||
private String priority; | |||
@@ -0,0 +1,69 @@ | |||
package com.ruoyi.business.domain.bo; | |||
import lombok.Data; | |||
import java.util.List; | |||
/** | |||
* 任务结果数据对象 | |||
*/ | |||
@Data | |||
public class YinDaoCallBackBO { | |||
/** | |||
* 参考 DataTypeEnum | |||
*/ | |||
private String dataType; | |||
/** | |||
* 应用运行uuid | |||
*/ | |||
private String jobUuid; | |||
/** | |||
* @see YinDaoCallBackBO | |||
*/ | |||
private String status; | |||
/** | |||
* 运行描述 | |||
*/ | |||
private String msg; | |||
/** | |||
* 应用运行参数 | |||
*/ | |||
private List<RobotParam> result; | |||
/** | |||
* 任务开始运行的时间 | |||
*/ | |||
private String startTime; | |||
/** | |||
* 任务结束运行的时间 | |||
*/ | |||
private String endTime; | |||
/** | |||
* 执行的机器人uuid | |||
*/ | |||
private String robotClientUuid; | |||
/** | |||
* 机器人名称 | |||
*/ | |||
private String robotClientName; | |||
/** | |||
* 应用名 | |||
*/ | |||
private String robotName; | |||
/** | |||
* 应用运行参数 | |||
*/ | |||
@Data | |||
public static class RobotParam { | |||
// 参数名 | |||
private String name; | |||
// 参数值 | |||
private String value; | |||
// 参数类型 | |||
private String type; | |||
} | |||
} |
@@ -3,6 +3,7 @@ package com.ruoyi.business.mapper; | |||
import java.util.List; | |||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; | |||
import com.ruoyi.business.domain.ResourceLibrary; | |||
import org.apache.ibatis.annotations.Param; | |||
/** | |||
* 资源库管理Mapper接口 | |||
@@ -19,6 +20,6 @@ public interface ResourceLibraryMapper extends BaseMapper<ResourceLibrary> | |||
* @param resourceLibrary 资源库管理 | |||
* @return 资源库管理集合 | |||
*/ | |||
List<ResourceLibrary> selectResourceLibraryList(ResourceLibrary resourceLibrary); | |||
List<ResourceLibrary> selectResourceLibraryList(@Param("resourceLibrary") ResourceLibrary resourceLibrary, @Param("appIdList") String appIdList); | |||
} |
@@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.extension.service.IService; | |||
import com.ruoyi.business.domain.bo.AddApplyPlanBO; | |||
import com.ruoyi.business.domain.bo.ApplyStartBO; | |||
import com.ruoyi.business.domain.bo.JobQueryBO; | |||
import com.ruoyi.business.domain.bo.YinDaoCallBackBO; | |||
import com.ruoyi.business.domain.vo.JobQueryVO; | |||
/** | |||
@@ -16,6 +17,14 @@ import com.ruoyi.business.domain.vo.JobQueryVO; | |||
* @date 2024-06-14 | |||
*/ | |||
public interface IApplyPlanService extends IService<ApplyPlan> { | |||
/** | |||
* 影刀回调逻辑 | |||
* | |||
* @param | |||
* @return | |||
*/ | |||
boolean ydCallBack(YinDaoCallBackBO yinDaoCallBackBO) ; | |||
/** | |||
* 重新运行 | |||
@@ -47,7 +56,7 @@ public interface IApplyPlanService extends IService<ApplyPlan> { | |||
* | |||
* @return | |||
*/ | |||
boolean execPlan(ApplyStartBO applyStartBO) throws IllegalAccessException; | |||
boolean execPlan(ApplyStartBO applyStartBO,ApplyPlan applyPlan) throws IllegalAccessException; | |||
/** | |||
* 定时查询指定时间执行的数据 | |||
@@ -1,8 +1,10 @@ | |||
package com.ruoyi.business.service; | |||
import java.util.List; | |||
import com.ruoyi.business.domain.ResourceLibrary; | |||
import com.baomidou.mybatisplus.extension.service.IService; | |||
import com.ruoyi.business.domain.bo.AddBatchResourceLibraryBO; | |||
/** | |||
* 资源库管理Service接口 | |||
@@ -10,8 +12,14 @@ import com.baomidou.mybatisplus.extension.service.IService; | |||
* @author LiuChengRan | |||
* @date 2024-06-17 | |||
*/ | |||
public interface IResourceLibraryService extends IService<ResourceLibrary> | |||
{ | |||
public interface IResourceLibraryService extends IService<ResourceLibrary> { | |||
/** | |||
* 批量添加参数 | |||
* | |||
* @param addBatchResourceLibraryBO | |||
*/ | |||
boolean addBatch(AddBatchResourceLibraryBO addBatchResourceLibraryBO); | |||
/** | |||
* 查询资源库管理列表 | |||
@@ -19,6 +27,6 @@ public interface IResourceLibraryService extends IService<ResourceLibrary> | |||
* @param resourceLibrary 资源库管理 | |||
* @return 资源库管理集合 | |||
*/ | |||
List<ResourceLibrary> list(ResourceLibrary resourceLibrary); | |||
List<ResourceLibrary> list(ResourceLibrary resourceLibrary, List<String> appIdList); | |||
} |
@@ -1,8 +1,10 @@ | |||
package com.ruoyi.business.service.impl; | |||
import java.text.ParseException; | |||
import java.time.ZoneId; | |||
import java.time.ZonedDateTime; | |||
import java.util.*; | |||
import java.util.concurrent.TimeUnit; | |||
import java.util.concurrent.atomic.AtomicBoolean; | |||
import java.util.stream.Collectors; | |||
@@ -23,14 +25,20 @@ import com.ruoyi.business.service.IResourceLibraryService; | |||
import com.ruoyi.business.util.YinDaoHttpUtils; | |||
import com.ruoyi.business.yddoman.BaseDTO; | |||
import com.ruoyi.common.annotation.DataScope; | |||
import com.ruoyi.common.core.redis.RedisCache; | |||
import com.ruoyi.common.enums.ExcTypeStatus; | |||
import com.ruoyi.common.enums.PlanRunStatus; | |||
import com.ruoyi.common.enums.RebotStatus; | |||
import com.ruoyi.common.exception.ServiceException; | |||
import com.ruoyi.common.utils.DateUtils; | |||
import com.ruoyi.common.utils.SecurityUtils; | |||
import com.ruoyi.common.utils.StringUtils; | |||
import com.ruoyi.common.utils.bean.BeanUtils; | |||
import org.quartz.*; | |||
import org.redisson.api.RLock; | |||
import org.redisson.api.RedissonClient; | |||
import org.springframework.beans.factory.annotation.Autowired; | |||
import org.springframework.data.redis.core.RedisTemplate; | |||
import org.springframework.scheduling.support.CronSequenceGenerator; | |||
import org.springframework.stereotype.Service; | |||
import com.ruoyi.business.mapper.ApplyPlanMapper; | |||
@@ -56,6 +64,40 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
@Autowired | |||
private IResourceLibraryService resourceLibraryService; | |||
@Autowired | |||
private RedissonClient redissonClient; | |||
/** | |||
* 影刀回调 | |||
* | |||
* @param yinDaoCallBackBO | |||
* @return | |||
*/ | |||
@Override | |||
public boolean ydCallBack(YinDaoCallBackBO yinDaoCallBackBO) { | |||
ApplyPlan applyPlan = this.lambdaQuery().eq(ApplyPlan::getTaskUuid, yinDaoCallBackBO.getJobUuid()).one(); | |||
applyPlan.setStartTime(DateUtils.parseDate(yinDaoCallBackBO.getStartTime())); | |||
applyPlan.setEndTime(DateUtils.parseDate(yinDaoCallBackBO.getStartTime())); | |||
applyPlan.setTaskStatus(yinDaoCallBackBO.getStatus()); | |||
boolean isUpdate = this.updateById(applyPlan); | |||
// 终态 | |||
List<String> finalState = new ArrayList<>(); | |||
finalState.add(PlanRunStatus.FINISH.getKey()); | |||
finalState.add(PlanRunStatus.STOPPED.getKey()); | |||
finalState.add(PlanRunStatus.ERROR.getKey()); | |||
finalState.add(PlanRunStatus.SKIPPED.getKey()); | |||
finalState.add(PlanRunStatus.CANCEL.getKey()); | |||
// 如果是终态 | |||
if (StringUtils.isNotEmpty(yinDaoCallBackBO.getRobotClientName()) && finalState.contains(yinDaoCallBackBO.getStatus())) { | |||
Rebot rebot = new Rebot(); | |||
rebot.setStatus(RebotStatus.IDLE.getKey()); | |||
return rebotService.lambdaUpdate().eq(Rebot::getRobotClientName, yinDaoCallBackBO.getRobotClientName()).update(rebot); | |||
} else { | |||
return isUpdate; | |||
} | |||
} | |||
/** | |||
* 重新运行 | |||
* | |||
@@ -92,6 +134,11 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
@Override | |||
@Transactional | |||
public boolean runAppBySetTime() throws IllegalAccessException { | |||
RLock lock = redissonClient.getLock("runAppBySetTime"); | |||
try { | |||
// 尝试获取锁,最多等待100秒,锁定之后最多持有锁10秒 | |||
boolean isLocked = lock.tryLock(2, 10, TimeUnit.SECONDS); | |||
if (isLocked) { | |||
List<ApplyPlan> applyPlanList = this.lambdaQuery().eq(ApplyPlan::getExcType, ExcTypeStatus.TWO.getKey()).isNull(ApplyPlan::getTaskUuid).orderByAsc(ApplyPlan::getExcTime).list(); | |||
if (applyPlanList.isEmpty()) { | |||
return false; | |||
@@ -109,8 +156,13 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
// rebotService.syn(new Rebot()); | |||
return false; | |||
} | |||
List<String> ids = new ArrayList<>(); | |||
listRebotVos.forEach(rebot -> { | |||
applyPlanList.forEach(e -> { | |||
log.debug("指定时间执行 执行时间:" + e.getExcTime()); | |||
if (ids.contains(e.getId())) { | |||
return; | |||
} | |||
if (e.getExcTime().getTime() <= System.currentTimeMillis() && PlanRunStatus.AWAIT_CREATE.getKey().equals(e.getTaskStatus())) { | |||
ApplyStartBO applyStartBO = new ApplyStartBO(); | |||
applyStartBO.setRobotUuid(e.getAppId()); | |||
@@ -123,7 +175,7 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
if (StringUtils.isNotEmpty(e.getPlanParams())) { | |||
applyStartBO.setPlanParams(e.getPlanParams()); | |||
List<ApplyStartBO.RobotParam> paramList = JSON.parseArray(e.getPlanParams(), ApplyStartBO.RobotParam.class); | |||
applyStartBO.setPlanParamsList(paramList); | |||
applyStartBO.setParams(paramList); | |||
} | |||
} | |||
try { | |||
@@ -140,7 +192,8 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
updateRebot.setUpdateTime(new Date()); | |||
updateRebot.setStatus(RebotStatus.RUNNING.getKey()); | |||
if (rebotService.lambdaUpdate().eq(Rebot::getRobotClientName, rebot.getRobotClientName()).update(updateRebot)) { | |||
applyPlanList.remove(e); | |||
log.debug("applyPlanList:" + applyPlanList.size()); | |||
ids.add(e.getId()); | |||
} | |||
} catch (IllegalAccessException ex) { | |||
throw new RuntimeException(ex); | |||
@@ -148,7 +201,15 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
} | |||
}); | |||
}); | |||
} | |||
} catch (InterruptedException e) { | |||
e.printStackTrace(); | |||
} finally { | |||
if (lock.isHeldByCurrentThread()) { | |||
// 释放锁 | |||
lock.unlock(); | |||
} | |||
} | |||
return false; | |||
} | |||
@@ -157,8 +218,13 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
*/ | |||
@Override | |||
public boolean runAppByCron() throws IllegalAccessException { | |||
RLock lock = redissonClient.getLock("runAppByCron"); | |||
try { | |||
// 尝试获取锁,最多等待100秒,锁定之后最多持有锁10秒 | |||
boolean isLocked = lock.tryLock(2, 10, TimeUnit.SECONDS); | |||
if (isLocked) { | |||
// 先查计划,避免每次命中第三方接口 | |||
List<ApplyPlan> applyPlanList = this.lambdaQuery().eq(ApplyPlan::getExcType, ExcTypeStatus.TREE.getKey()).list(); | |||
List<ApplyPlan> applyPlanList = this.lambdaQuery().eq(ApplyPlan::getExcType, ExcTypeStatus.TREE.getKey()).ne(ApplyPlan::getTaskStatus, PlanRunStatus.RUNNING.getKey()).ne(ApplyPlan::getTaskStatus, PlanRunStatus.CREATED.getKey()).list(); | |||
if (applyPlanList.isEmpty()) { | |||
return false; | |||
} | |||
@@ -169,15 +235,11 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
CronExpression cron = new CronExpression(e.getCronExpression()); | |||
// 如果该计划没有执行过 | |||
if (Objects.isNull(e.getLastExecTime())) { | |||
// 判断当前时间是否满足表达式 | |||
Date nextDate = cron.getNextValidTimeAfter(new Date()); | |||
if (new Date().compareTo(nextDate) > -1) { | |||
// 执行一次 | |||
execApplyPlan.add(e); | |||
} | |||
} else { | |||
// 根据表达式返回下一个执行时间 | |||
Date nextDate = cron.getNextValidTimeAfter(e.getNextExecTime()); | |||
if (new Date().compareTo(nextDate) > -1) { | |||
if (new Date().compareTo(e.getNextExecTime()) > -1) { | |||
execApplyPlan.add(e); | |||
} | |||
} | |||
@@ -203,8 +265,19 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
// rebotService.syn(new Rebot()); | |||
return false; | |||
} | |||
listRebotVos.forEach(rebot -> { | |||
List<String> ids = new ArrayList<>(); | |||
execApplyPlan.forEach(e -> { | |||
if (listRebotVos.isEmpty()) { | |||
// 刷新机器人数据 | |||
// rebotService.syn(new Rebot()); | |||
return; | |||
} | |||
ListRebotVO rebot = listRebotVos.get(0); | |||
log.debug("机器人表达式执行调度日志" + rebot.getRobotClientName() + "调度" + e.getAppName() + "应用,排除的应用id为" + ids.toString()); | |||
if (ids.contains(e.getId())) { | |||
return; | |||
} | |||
log.debug("机器人表达式成功调度日志" + rebot.getRobotClientName() + "调度" + e.getAppName() + "应用,排除的应用id为" + ids.toString()); | |||
// 该计划满足执行条件 | |||
ApplyStartBO applyStartBO = new ApplyStartBO(); | |||
applyStartBO.setRobotUuid(e.getAppId()); | |||
@@ -219,16 +292,10 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
if (StringUtils.isNotEmpty(e.getPlanParams())) { | |||
applyStartBO.setPlanParams(e.getPlanParams()); | |||
List<ApplyStartBO.RobotParam> paramList = JSON.parseArray(e.getPlanParams(), ApplyStartBO.RobotParam.class); | |||
applyStartBO.setPlanParamsList(paramList); | |||
applyStartBO.setParams(paramList); | |||
} | |||
} | |||
String planParams = e.getPlanParams(); | |||
if (StringUtils.isNotEmpty(planParams)) { | |||
JSONArray jsonArray = JSON.parseArray(planParams); | |||
// 直接转换成List | |||
List<ApplyStartBO.RobotParam> list = jsonArray.toJavaList(ApplyStartBO.RobotParam.class); | |||
applyStartBO.setPlanParamsList(list); | |||
} | |||
try { | |||
JobStartVO jobStartVO = YinDaoHttpUtils.appStart(applyStartBO); | |||
ApplyPlan applyPlan = new ApplyPlan(); | |||
@@ -236,36 +303,55 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
applyPlan.setUpdateBy("系统修改"); | |||
applyPlan.setUpdateTime(new Date()); | |||
applyPlan.setTaskUuid(jobStartVO.getJobUuid()); | |||
applyPlan.setTaskStatus(PlanRunStatus.CREATED.getKey()); | |||
applyPlan.setTaskStatus(PlanRunStatus.RUNNING.getKey()); | |||
// 根据表达式返回下一个执行时间 | |||
CronExpression cron = new CronExpression(e.getCronExpression()); | |||
log.debug("表达式:" + e.getCronExpression()); | |||
// 设置时区,例如设置为中国时区 | |||
TimeZone timeZone = TimeZone.getTimeZone(ZoneId.of("Asia/Shanghai").toString()); | |||
final CronExpression cron = new CronExpression(e.getCronExpression()); | |||
Date nextDate = null; | |||
if (Objects.isNull(e.getLastExecTime())) { | |||
nextDate = cron.getNextValidTimeAfter(new Date()); | |||
} else { | |||
nextDate = cron.getNextValidTimeAfter(e.getLastExecTime()); | |||
} | |||
Date date = new Date(); | |||
log.debug("当前时间:" + date); | |||
nextDate = cron.getNextValidTimeAfter(date);//下次执行时间 | |||
log.debug("下次时间:" + nextDate); | |||
applyPlan.setLastExecTime(date); | |||
applyPlan.setNextExecTime(nextDate); | |||
} else { | |||
applyPlan.setLastExecTime(e.getNextExecTime()); | |||
log.debug("当前时间:" + e.getNextExecTime()); | |||
nextDate = cron.getNextValidTimeAfter(e.getNextExecTime());//下次执行时间 | |||
log.debug("下次时间:" + nextDate); | |||
applyPlan.setNextExecTime(nextDate); | |||
} | |||
this.updateById(applyPlan); | |||
Rebot updateRebot = new Rebot(); | |||
updateRebot.setUpdateBy("系统修改"); | |||
updateRebot.setUpdateTime(new Date()); | |||
updateRebot.setStatus(RebotStatus.RUNNING.getKey()); | |||
if (rebotService.lambdaUpdate().eq(Rebot::getRobotClientName, rebot.getRobotClientName()).update(updateRebot)) { | |||
applyPlanList.remove(e); | |||
ids.add(e.getId()); | |||
} | |||
} catch (IllegalAccessException | ParseException ex) { | |||
} catch (IllegalAccessException ex) { | |||
throw new RuntimeException(ex); | |||
} catch (ParseException ex) { | |||
throw new RuntimeException(ex); | |||
} | |||
listRebotVos.remove(0); | |||
}); | |||
}); | |||
} | |||
} catch (InterruptedException e) { | |||
e.printStackTrace(); | |||
} finally { | |||
if (lock.isHeldByCurrentThread()) { | |||
// 释放锁 | |||
lock.unlock(); | |||
} | |||
} | |||
return false; | |||
} | |||
/** | |||
* 获取应用执行结果 | |||
* | |||
@@ -283,7 +369,7 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
noStatus.add(PlanRunStatus.SKIPPED.getKey()); | |||
noStatus.add(PlanRunStatus.CANCEL.getKey()); | |||
// 查询任务运行状态非空非终态并且已经创建的任务 | |||
List<ApplyPlan> list = this.lambdaQuery().notIn(ApplyPlan::getTaskStatus, noStatus).isNotNull(ApplyPlan::getTaskStatus).orderByAsc(ApplyPlan::getPriority).list(); | |||
List<ApplyPlan> list = this.lambdaQuery().notIn(ApplyPlan::getTaskStatus, noStatus).isNotNull(ApplyPlan::getTaskUuid).isNotNull(ApplyPlan::getTaskStatus).orderByAsc(ApplyPlan::getPriority).list(); | |||
if (list.isEmpty()) { | |||
log.debug("没有需要更新状态的计划任务"); | |||
return; | |||
@@ -300,7 +386,7 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
if (null != jobQueryVO.getEndTime()) { | |||
updateApplyPlan.setEndTime(jobQueryVO.getEndTime()); | |||
updateApplyPlan.setStartTime(jobQueryVO.getStartTime()); | |||
updateApplyPlan.setEndTime(jobQueryVO.getEndTime()); | |||
if (!Objects.isNull(updateApplyPlan.getEndTime()) && !Objects.isNull(updateApplyPlan.getStartTime())) { | |||
long diff = updateApplyPlan.getEndTime().getTime() - updateApplyPlan.getStartTime().getTime(); | |||
long planTime = diff / (60 * 1000); | |||
updateApplyPlan.setPlanTime(planTime + ""); | |||
@@ -308,6 +394,7 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
updateApplyPlan.setTimeSaving((Long.parseLong(applyPlan.getManualTime()) - planTime) + ""); | |||
} | |||
} | |||
} | |||
updateApplyPlan.setId(applyPlan.getId()); | |||
updateApplyPlan.setUpdateBy("系统修改"); | |||
updateApplyPlan.setUpdateTime(new Date()); | |||
@@ -351,13 +438,19 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
*/ | |||
@Override | |||
@Transactional | |||
public boolean execPlan(ApplyStartBO applyStartBO) throws IllegalAccessException { | |||
public boolean execPlan(ApplyStartBO applyStartBO, ApplyPlan applyPlan) throws IllegalAccessException { | |||
RLock lock = redissonClient.getLock("execPlan"); | |||
int i = 0; | |||
try { | |||
// 尝试获取锁,最多等待100秒,锁定之后最多持有锁10秒 | |||
boolean isLocked = lock.tryLock(2, 20, TimeUnit.SECONDS); | |||
if (isLocked) { | |||
// 先查计划 避免每次直接命中第三方接口 | |||
Map<Long, List<ApplyPlan>> applyPlanMap = this.lambdaQuery().eq(ApplyPlan::getExcType, ExcTypeStatus.ONE.getKey()).isNull(ApplyPlan::getTaskUuid) | |||
.orderByAsc(ApplyPlan::getPriority).list().stream().collect(Collectors.groupingBy(ApplyPlan::getDeptId)); | |||
// 查询所有待执行的任务 | |||
if (null == applyStartBO && applyPlanMap.isEmpty()) { | |||
if (applyPlanMap.isEmpty()) { | |||
log.debug("没有等待执行的计划"); | |||
return false; | |||
} | |||
@@ -370,8 +463,6 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
List<ListRebotVO> listRebotVos = YinDaoHttpUtils.listRebot(listRebotBO); | |||
if (listRebotVos.isEmpty()) { | |||
log.debug("暂无空闲机器人"); | |||
// 刷新机器人数据 | |||
// rebotService.syn(new Rebot()); | |||
return false; | |||
} | |||
@@ -381,37 +472,29 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
v.sort((Comparator.comparingInt(ApplyPlan::getPriority))); | |||
applyPlans.add(v.get(0)); | |||
}); | |||
if (null == applyStartBO && applyPlans.isEmpty()) { | |||
if (applyPlans.isEmpty()) { | |||
log.debug("没有等待执行的计划"); | |||
return false; | |||
} | |||
applyPlans.sort((Comparator.comparingInt(ApplyPlan::getPriority))); | |||
int i = 0; | |||
for (ListRebotVO listRebotVO : listRebotVos) { | |||
ApplyPlan applyPlan = null; | |||
if (applyPlans.isEmpty()) { | |||
log.debug("没有等待执行的计划!"); | |||
continue; | |||
} | |||
if (null == applyStartBO) { | |||
applyPlan = applyPlans.get(0); | |||
applyStartBO = new ApplyStartBO(); | |||
applyStartBO.setRobotUuid(applyPlan.getAppId()); | |||
//high 高 middle 中 low 低 | |||
applyStartBO.setPriority("high"); | |||
} else { | |||
applyPlan = new ApplyPlan(); | |||
BeanUtils.copyBeanProp(applyPlan, applyStartBO); | |||
// 运行的应用上一步已给值,此处无需再给 | |||
} | |||
applyStartBO.setAccountName(listRebotVO.getRobotClientName()); | |||
// 如果应用支持参数 | |||
if (!Objects.isNull(applyPlan.getSupportParam()) && applyPlan.getSupportParam() == 1) { | |||
if (StringUtils.isNotEmpty(applyPlan.getPlanParams())) { | |||
applyStartBO.setPlanParams(applyPlan.getPlanParams()); | |||
List<ApplyStartBO.RobotParam> paramList = JSON.parseArray(applyPlan.getPlanParams(), ApplyStartBO.RobotParam.class); | |||
applyStartBO.setPlanParamsList(paramList); | |||
applyStartBO.setParams(paramList); | |||
} | |||
} | |||
JobStartVO jobStartVO = YinDaoHttpUtils.appStart(applyStartBO); | |||
@@ -428,11 +511,21 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
updateApplyPlan.setRobotName(listRebotVO.getRobotClientName()); | |||
updateApplyPlan.setManualTime(applyPlan.getManualTime()); | |||
updateApplyPlan.setDeptId(applyStartBO.getDeptId()); | |||
applyPlan.setTaskUuid(jobStartVO.getJobUuid()); | |||
if (this.saveOrUpdate(updateApplyPlan)) { | |||
applyPlans.remove(0); | |||
i++; | |||
} | |||
} | |||
} | |||
} catch (InterruptedException e) { | |||
e.printStackTrace(); | |||
} finally { | |||
if (lock.isHeldByCurrentThread()) { | |||
// 释放锁 | |||
lock.unlock(); | |||
} | |||
} | |||
return i > 0; | |||
} | |||
@@ -440,23 +533,41 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
@Override | |||
@Transactional | |||
public boolean save(AddApplyPlanBO addApplyPlanBO) { | |||
// 指定时间执行 | |||
if (addApplyPlanBO.getExcType().equals("1") && null == addApplyPlanBO.getExcTime()) { | |||
throw new ServiceException("指定时间执行时间不能为空"); | |||
} | |||
// 表达式执行 | |||
if (addApplyPlanBO.getExcType().equals("2")) { | |||
if (null == addApplyPlanBO.getCronExpression()) { | |||
throw new ServiceException("周期执行表达式不能为空"); | |||
} | |||
try { | |||
// cron 验证表达式是否正确 | |||
CronExpression.validateExpression(addApplyPlanBO.getCronExpression()); | |||
} catch (Exception ex) { | |||
throw new ServiceException("时间表达式错误"); | |||
} | |||
} | |||
List<ApplyPlan> applyPlanList = new ArrayList<>(); | |||
Map<String, List<Apply>> appMap = applyService.lambdaQuery().in(Apply::getAppId, addApplyPlanBO.getAppId()).list().stream().collect(Collectors.groupingBy(Apply::getAppId)); | |||
// 是否有空闲机器人并且创建计划成功 | |||
boolean isExec = false; | |||
AtomicBoolean insert = new AtomicBoolean(false); | |||
addApplyPlanBO.getAppId().forEach(e -> { | |||
ApplyPlan applyPlan = new ApplyPlan(); | |||
BeanUtils.copyBeanProp(applyPlan, addApplyPlanBO); | |||
Apply apply = appMap.get(e).get(0); | |||
Apply apply = appMap.get(addApplyPlanBO.getAppId()).get(0); | |||
applyPlan.setAppName(apply.getAppName()); | |||
applyPlan.setAppId(apply.getAppId()); | |||
applyPlan.setAppType(apply.getAppType()); | |||
applyPlan.setAppTypeName(apply.getAppTypeName()); | |||
applyPlan.setManualTime(apply.getManualTime()); | |||
if (null != apply.getSupportParam() && apply.getSupportParam() == 1) { | |||
applyPlan.setSupportParam(apply.getSupportParam()); | |||
// 获取参数 每一次执行计划的参数都是独立的 | |||
List<ApplyStartBO.RobotParam> paramList = resourceLibraryService.lambdaQuery().eq(ResourceLibrary::getAppId, apply.getAppId()).list().stream() | |||
List<ApplyStartBO.RobotParam> paramList = addApplyPlanBO.getResourceLibrarieList().stream() | |||
.map(item -> { | |||
ApplyStartBO.RobotParam robotParam = new ApplyStartBO.RobotParam(); | |||
robotParam.setName(item.getResourceName()); | |||
@@ -464,47 +575,55 @@ public class ApplyPlanServiceImpl extends ServiceImpl<ApplyPlanMapper, ApplyPlan | |||
robotParam.setType(item.getResourceType()); | |||
return robotParam; | |||
}).collect(Collectors.toList()); | |||
if (!paramList.isEmpty()) { | |||
resourceLibraryService.saveOrUpdateBatch(addApplyPlanBO.getResourceLibrarieList()); | |||
if (paramList.isEmpty()) { | |||
throw new ServiceException("该应用需要参数,没有检测到参数!"); | |||
} else { | |||
// json集合转字符串 | |||
applyPlan.setPlanParams(JSONArray.toJSONString(paramList)); | |||
} | |||
} | |||
applyPlan.setDeptId(SecurityUtils.getDeptId()); | |||
try { | |||
// 立即执行才执行 | |||
if (addApplyPlanBO.getExcType().equals("0")) { | |||
// 只指定优先级和应用和人工用时 机器人等下一个阶段判读是否有空闲 | |||
ApplyStartBO applyStartBO = new ApplyStartBO(); | |||
try { | |||
applyStartBO.setRobotUuid(applyPlan.getAppId()); | |||
//high 高 middle 中 low 低 | |||
applyStartBO.setPriority("high"); | |||
applyStartBO.setPlanParams(applyPlan.getPlanParams()); | |||
applyStartBO.setManualTime(applyPlan.getManualTime()); | |||
applyStartBO.setPlanName(addApplyPlanBO.getPlanName()); | |||
insert.set(execPlan(applyStartBO)); | |||
ApplyPlan retApplyPlan = new ApplyPlan(); | |||
isExec = execPlan(applyStartBO, retApplyPlan); | |||
applyPlan.setTaskUuid(retApplyPlan.getTaskUuid()); | |||
} catch (Exception ex) { | |||
log.error("分配机器人异常:" + ex.getMessage()); | |||
} | |||
if (!insert.get()) { | |||
} | |||
if (isExec) { | |||
applyPlan.setTaskStatus(PlanRunStatus.CREATED.getKey()); | |||
} else { | |||
applyPlan.setTaskStatus(PlanRunStatus.AWAIT_CREATE.getKey()); | |||
applyPlanList.add(applyPlan); | |||
} | |||
}); | |||
applyPlanList.add(applyPlan); | |||
return this.saveBatch(applyPlanList) || insert.get(); | |||
} | |||
@Override | |||
public boolean update(AddApplyPlanBO addApplyPlanBO) { | |||
List<ApplyPlan> applyPlanList = new ArrayList<>(); | |||
Map<String, List<Apply>> appMap = applyService.lambdaQuery().in(Apply::getAppId, addApplyPlanBO.getAppId()).eq(Apply::getDeptId, SecurityUtils.getDeptId()).list().stream().collect(Collectors.groupingBy(Apply::getAppId)); | |||
addApplyPlanBO.getAppId().forEach(e -> { | |||
Map<String, List<Apply>> appMap = applyService.lambdaQuery().in(Apply::getAppId, addApplyPlanBO.getAppId()).list().stream().collect(Collectors.groupingBy(Apply::getAppId)); | |||
ApplyPlan applyPlan = new ApplyPlan(); | |||
BeanUtils.copyBeanProp(applyPlan, addApplyPlanBO); | |||
Apply apply = appMap.get(e).get(0); | |||
Apply apply = appMap.get(addApplyPlanBO.getAppId()).get(0); | |||
applyPlan.setAppName(apply.getAppName()); | |||
applyPlan.setAppId(apply.getAppId()); | |||
applyPlan.setAppType(apply.getAppType()); | |||
applyPlan.setAppTypeName(apply.getAppTypeName()); | |||
applyPlanList.add(applyPlan); | |||
}); | |||
return super.updateBatchById(applyPlanList); | |||
} | |||
@@ -5,7 +5,9 @@ import java.util.Map; | |||
import java.util.stream.Collectors; | |||
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; | |||
import com.ruoyi.business.domain.ApplyPlan; | |||
import com.ruoyi.business.domain.Rebot; | |||
import com.ruoyi.business.domain.ResourceLibrary; | |||
import com.ruoyi.business.domain.bo.ApplyStartBO; | |||
import com.ruoyi.business.domain.bo.JobQueryBO; | |||
import com.ruoyi.business.domain.bo.ListApplyBO; | |||
@@ -14,6 +16,8 @@ import com.ruoyi.business.domain.vo.JobQueryVO; | |||
import com.ruoyi.business.domain.vo.JobStartVO; | |||
import com.ruoyi.business.domain.vo.ListApplyVO; | |||
import com.ruoyi.business.domain.vo.ListRebotVO; | |||
import com.ruoyi.business.service.IApplyPlanService; | |||
import com.ruoyi.business.service.IResourceLibraryService; | |||
import com.ruoyi.business.util.YinDaoHttpUtils; | |||
import com.ruoyi.common.annotation.DataScope; | |||
import com.ruoyi.common.core.domain.entity.SysDept; | |||
@@ -40,6 +44,13 @@ public class ApplyServiceImpl extends ServiceImpl<ApplyMapper, Apply> implements | |||
@Autowired | |||
private ISysDeptService sysDeptService; | |||
@Autowired | |||
private IResourceLibraryService resourceLibraryService; | |||
@Autowired | |||
private IApplyPlanService applyPlanService; | |||
/** | |||
* 启动应用 | |||
* | |||
@@ -110,11 +121,21 @@ public class ApplyServiceImpl extends ServiceImpl<ApplyMapper, Apply> implements | |||
newApply.setAppCreateTime(listRebotVO.getCreateTime()); | |||
newApply.setAppUpdateTime(listRebotVO.getUpdateTime()); | |||
newApply.setId(apply.getId()); | |||
newApply.setSupportParam(listRebotVO.getSupportParam()); | |||
if (null != sysDepts) { | |||
newApply.setDeptId(sysDepts.get(0).getDeptId()); | |||
newApply.setDeptName(sysDepts.get(0).getDeptName()); | |||
} | |||
// 修改应用名称 | |||
updateById(newApply); | |||
// 修改参数里应用名称 | |||
ResourceLibrary resourceLibrary = new ResourceLibrary(); | |||
resourceLibrary.setAppName(listRebotVO.getAppName()); | |||
resourceLibraryService.lambdaUpdate().eq(ResourceLibrary::getAppId, apply.getId()).update(resourceLibrary); | |||
// 修改已执行的计划应用名称 | |||
ApplyPlan applyPlan = new ApplyPlan(); | |||
applyPlan.setAppName(listRebotVO.getAppName()); | |||
applyPlanService.lambdaUpdate().eq(ApplyPlan::getAppId, apply.getId()).update(applyPlan); | |||
} | |||
}); | |||
} | |||
@@ -1,10 +1,15 @@ | |||
package com.ruoyi.business.service.impl; | |||
import java.util.ArrayList; | |||
import java.util.List; | |||
import java.util.Map; | |||
import java.util.Optional; | |||
import java.util.stream.Collectors; | |||
import com.ruoyi.business.domain.Apply; | |||
import com.ruoyi.business.domain.bo.AddBatchResourceLibraryBO; | |||
import com.ruoyi.business.service.IApplyService; | |||
import com.ruoyi.common.exception.ServiceException; | |||
import com.ruoyi.common.utils.DateUtils; | |||
import org.springframework.beans.factory.annotation.Autowired; | |||
import org.springframework.stereotype.Service; | |||
@@ -20,12 +25,48 @@ import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; | |||
* @date 2024-06-17 | |||
*/ | |||
@Service | |||
public class ResourceLibraryServiceImpl extends ServiceImpl<ResourceLibraryMapper, ResourceLibrary> implements IResourceLibraryService | |||
{ | |||
public class ResourceLibraryServiceImpl extends ServiceImpl<ResourceLibraryMapper, ResourceLibrary> implements IResourceLibraryService { | |||
@Autowired | |||
private IApplyService applyService; | |||
/** | |||
* 批量添加参数 | |||
* | |||
* @param addBatchResourceLibraryBO | |||
*/ | |||
@Override | |||
public boolean addBatch(AddBatchResourceLibraryBO addBatchResourceLibraryBO) { | |||
Apply apply = Optional.ofNullable(applyService.lambdaQuery().eq(Apply::getAppId, addBatchResourceLibraryBO.getAppId()).one()) | |||
.orElseThrow(() -> new RuntimeException("应用不存在")); | |||
// 参数集合 | |||
List<ResourceLibrary> resourceLibraries = new ArrayList<>(); | |||
// stream统计对象里重复的字段 | |||
Map<String, Long> mapList = addBatchResourceLibraryBO.getList().stream().collect(Collectors.groupingBy(ResourceLibrary::getResourceName, Collectors.counting())); | |||
mapList.forEach((k, v) -> { | |||
if (v > 1) { | |||
throw new ServiceException(k + "重复,请重新输入"); | |||
} | |||
if(this.lambdaQuery().eq(ResourceLibrary::getAppId, addBatchResourceLibraryBO.getAppId()).eq(ResourceLibrary::getResourceName,k).count()>0){ | |||
throw new ServiceException(k + "已存在,请重新输入"); | |||
} | |||
}); | |||
addBatchResourceLibraryBO.getList().forEach(e -> { | |||
ResourceLibrary resourceLibrary = new ResourceLibrary(); | |||
resourceLibrary.setAppId(apply.getAppId()); | |||
resourceLibrary.setAppName(apply.getAppName()); | |||
resourceLibrary.setDeptId(apply.getDeptId()); | |||
resourceLibrary.setDeptName(apply.getDeptName()); | |||
resourceLibrary.setResource(e.getResource()); | |||
resourceLibrary.setResourceName(e.getResourceName()); | |||
resourceLibrary.setResourceType(e.getResourceType()); | |||
resourceLibrary.setResourceValue(e.getResourceValue()); | |||
resourceLibraries.add(resourceLibrary); | |||
}); | |||
return super.saveBatch(resourceLibraries); | |||
} | |||
/** | |||
* 查询资源库管理列表 | |||
* | |||
@@ -33,9 +74,9 @@ public class ResourceLibraryServiceImpl extends ServiceImpl<ResourceLibraryMappe | |||
* @return 资源库管理 | |||
*/ | |||
@Override | |||
public List<ResourceLibrary> list(ResourceLibrary resourceLibrary) | |||
{ | |||
return baseMapper.selectResourceLibraryList(resourceLibrary); | |||
public List<ResourceLibrary> list(ResourceLibrary resourceLibrary, List<String> appIdList) { | |||
// appIdList 转,号分隔字符串 | |||
return baseMapper.selectResourceLibraryList(resourceLibrary, null != appIdList ? String.join(",", appIdList) : null); | |||
} | |||
@Override | |||
@@ -2,6 +2,7 @@ package com.ruoyi.business.util; | |||
import cn.hutool.json.JSONUtil; | |||
import com.alibaba.fastjson2.JSON; | |||
import com.alibaba.fastjson2.JSONArray; | |||
import com.ruoyi.business.domain.bo.*; | |||
import com.ruoyi.business.domain.vo.*; | |||
import com.ruoyi.business.yddoman.BaseDTO; | |||
@@ -12,10 +13,8 @@ import com.ruoyi.common.utils.StringUtils; | |||
import com.ruoyi.common.utils.http.HttpClientUtil; | |||
import lombok.extern.slf4j.Slf4j; | |||
import java.util.ArrayList; | |||
import java.util.HashMap; | |||
import java.util.List; | |||
import java.util.Map; | |||
import java.util.*; | |||
import java.util.stream.Collectors; | |||
/** | |||
* 影刀请求工具类 | |||
@@ -44,14 +43,28 @@ public class YinDaoHttpUtils { | |||
private static final String QUERY_APP_START_RESULT = REQUEST_PREFIX + "/dispatch/v2/job/query"; | |||
// 查询应用运行日志(异常时可以查询原因) | |||
private static final String QUERY_APP_START_RESULT_LOG = REQUEST_PREFIX + "/dispatch/v2/job/list"; | |||
// 服务器ip | |||
private static final String SERVER_URL = "http://192.168.10.214:9000"; | |||
/** | |||
* 启动应用 | |||
*/ | |||
public static JobStartVO appStart(ApplyStartBO applyStartBO) throws IllegalAccessException { | |||
if (!Objects.isNull(applyStartBO.getPlanParams())) { | |||
List<ApplyStartBO.RobotParam> robotParams = JSON.parseArray(applyStartBO.getPlanParams(), ApplyStartBO.RobotParam.class); | |||
List<ApplyStartBO.RobotParam> newParamList = robotParams.stream().peek(e -> { | |||
if ("file".equals(e.getType())) { | |||
e.setValue(SERVER_URL + e.getValue()); | |||
} | |||
}).collect(Collectors.toList()); | |||
applyStartBO.setParams(newParamList); | |||
log.debug("请求的参数:" + applyStartBO.getParams()); | |||
} | |||
// 实体类转map | |||
Map<String, Object> data = BeanToMapUtil.convertEntityToMap(applyStartBO); | |||
BaseDTO baseDTO = sendPost(APP_START, data); | |||
return JSON.parseObject(baseDTO.getData().toString(), JobStartVO.class); | |||
} | |||
@@ -185,7 +198,11 @@ public class YinDaoHttpUtils { | |||
headMap.put("Content-Type", "application/json;charset=utf-8"); | |||
headMap.put("authorization", "Bearer " + getToken(tokenMap)); | |||
log.debug("请求的url:" + url); | |||
log.debug("请求的参数:" + JSONUtil.toJsonStr(map)); | |||
log.debug("请求头:" + JSONUtil.toJsonStr(headMap)); | |||
String returnStr = HttpClientUtil.postJson(url, headMap, JSONUtil.toJsonStr(map), "utf-8"); | |||
if (StringUtils.isEmpty(returnStr)) { | |||
throw new ServiceException("请求影刀接口失败"); | |||
} | |||
@@ -60,14 +60,17 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" | |||
from ct_resource_library | |||
<where> | |||
deleted = 0 | |||
<if test="resource != null and resource != ''"> and resource like concat(#{resource}, '%')</if> | |||
<if test="resourceValue != null and resourceValue != ''"> and resource_value = #{resourceValue}</if> | |||
<if test="resourceType != null and resourceType != ''"> and resource_type = #{resourceType}</if> | |||
<if test="resourceName != null and resourceName != ''"> and resource_name like concat(#{resourceName}, '%')</if> | |||
<if test="appId != null and appId != ''"> and app_id = #{appId}</if> | |||
<if test="appName != null and appName != ''"> and app_name like concat(#{appName}, '%')</if> | |||
<if test="deptId != null "> and dept_id = #{deptId}</if> | |||
<if test="deptName != null and deptName != ''"> and dept_name like concat(#{deptName}, '%')</if> | |||
<if test="resourceLibrary.resource != null and resourceLibrary.resource != ''">and resource like concat(#{resourceLibrary.resource}, '%')</if> | |||
<if test="resourceLibrary.resourceValue != null and resourceLibrary.resourceValue != ''">and resource_value = #{resourceLibrary.resourceValue}</if> | |||
<if test="resourceLibrary.resourceType != null and resourceLibrary.resourceType != ''">and resource_type = #{resourceLibrary.resourceType}</if> | |||
<if test="resourceLibrary.resourceName != null and resourceLibrary.resourceName != ''">and resource_name like concat(#{resourceLibrary.resourceName}, | |||
'%') | |||
</if> | |||
<if test="resourceLibrary.appId != null and resourceLibrary.appId != ''">and app_id = #{resourceLibrary.appId}</if> | |||
<if test="resourceLibrary.appName != null and resourceLibrary.appName != ''">and app_name like concat(#{resourceLibrary.appName}, '%')</if> | |||
<if test="resourceLibrary.deptId != null ">and dept_id = #{resourceLibrary.deptId}</if> | |||
<if test="resourceLibrary.deptName != null and resourceLibrary.deptName != ''">and dept_name like concat(#{resourceLibrary.deptName}, '%')</if> | |||
<if test="appIdList != null">and find_in_set(app_id,#{appIdList})</if> | |||
</where> | |||
</select> | |||
</mapper> |
@@ -163,6 +163,11 @@ | |||
<version>1.18.10</version> | |||
</dependency> | |||
<dependency> | |||
<groupId>org.redisson</groupId> | |||
<artifactId>redisson-spring-boot-starter</artifactId> | |||
<version>3.16.2</version> | |||
</dependency> | |||
</dependencies> | |||
@@ -117,6 +117,8 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter | |||
.antMatchers("/swagger-ui.html", "/swagger-resources/**", "/webjars/**", "/*/api-docs", "/druid/**").permitAll() | |||
// 上传文件 | |||
.antMatchers("/common/upload","/dev-api/profile/upload/**").permitAll() | |||
// 影刀回调 | |||
.antMatchers("/business/ctApplyPlan/callBack/updatePlanStatus").permitAll() | |||
// 除上面外的所有请求全部需要鉴权认证 | |||
.anyRequest().authenticated() | |||
.and() | |||
@@ -37,7 +37,7 @@ public class RyTask { | |||
*/ | |||
public void runAppNow() throws IllegalAccessException { | |||
log.debug("执行顺序启动应用开始"); | |||
applyPlanService.execPlan(null); | |||
applyPlanService.execPlan(null,null); | |||
log.debug("执行顺序启动应用结束"); | |||
} | |||