refactor 规则引擎

V0.5.x
jay 2023-05-31 08:56:21 +08:00
parent bb924b513f
commit 7eee581dbc
11 changed files with 824 additions and 173 deletions

View File

@ -9,14 +9,25 @@
*/
package cc.iotkit.manager.controller;
import cc.iotkit.common.api.PageRequest;
import cc.iotkit.common.api.Request;
import cc.iotkit.common.enums.ErrCode;
import cc.iotkit.common.exception.BizException;
import cc.iotkit.common.satoken.utils.AuthUtil;
import cc.iotkit.common.utils.ReflectUtil;
import cc.iotkit.data.manager.IRuleInfoData;
import cc.iotkit.data.manager.ITaskInfoData;
import cc.iotkit.manager.dto.bo.ruleinfo.RuleInfoBo;
import cc.iotkit.manager.dto.bo.ruleinfo.RuleLogBo;
import cc.iotkit.manager.dto.bo.taskinfo.TaskInfoBo;
import cc.iotkit.manager.dto.bo.taskinfo.TaskLogBo;
import cc.iotkit.manager.dto.vo.ruleinfo.RuleInfoVo;
import cc.iotkit.manager.dto.vo.ruleinfo.RuleLogVo;
import cc.iotkit.manager.dto.vo.taskinfo.TaskInfoVo;
import cc.iotkit.manager.dto.vo.taskinfo.TaskLogVo;
import cc.iotkit.manager.service.DataOwnerService;
import cc.iotkit.common.api.Paging;
import cc.iotkit.manager.service.IRuleEngineService;
import cc.iotkit.model.rule.RuleInfo;
import cc.iotkit.model.rule.RuleLog;
import cc.iotkit.model.rule.TaskInfo;
@ -26,10 +37,13 @@ import cc.iotkit.ruleengine.task.TaskManager;
import cc.iotkit.temporal.IRuleLogData;
import cc.iotkit.temporal.ITaskLogData;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiModelProperty;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.quartz.SchedulerException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import java.util.List;
@ -42,214 +56,115 @@ import java.util.UUID;
public class RuleEngineController {
@Autowired
private ITaskInfoData taskInfoData;
IRuleEngineService ruleEngineService;
@Autowired
private IRuleInfoData ruleInfoData;
@Autowired
private IRuleLogData ruleLogData;
@Autowired
private DataOwnerService dataOwnerService;
@Autowired
private TaskManager taskManager;
@Autowired
private RuleManager ruleManager;
@Autowired
private ITaskLogData taskLogData;
@PostMapping("/rules/{type}/{size}/{page}")
public Paging<RuleInfo> rules(
@PathVariable("type") String type,
@PathVariable("size") int size,
@PathVariable("page") int page
@ApiOperation("规则列表")
@PostMapping("/list")
public Paging<RuleInfoVo> rules(
@Validated @RequestBody
PageRequest<RuleInfoBo> request
) {
RuleInfo ruleInfo = new RuleInfo();
ruleInfo.setType(type);
if (AuthUtil.isAdmin()) {
return ruleInfoData.findByType(type, page, size);
} else {
return ruleInfoData.findByUidAndType(AuthUtil.getUserId(), type, page, size);
}
return ruleEngineService.selectPageList(request);
}
@PostMapping("/rule/save")
public void saveRule(@RequestBody RuleInfo rule) {
if (StringUtils.isBlank(rule.getId())) {
rule.setId(UUID.randomUUID().toString());
rule.setState(RuleInfo.STATE_STOPPED);
rule.setCreateAt(System.currentTimeMillis());
rule.setUid(AuthUtil.getUserId());
ruleInfoData.save(rule);
ruleManager.add(rule);
} else {
RuleInfo ruleInfo = ruleInfoData.findById(rule.getId());
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
if (RuleInfo.STATE_RUNNING.equals(ruleInfo.getState())) {
throw new BizException(ErrCode.RULE_ALREADY_RUNNING);
@ApiOperation("规则编辑")
@PostMapping("/edit")
public boolean saveRule(@RequestBody @Validated RuleInfoBo ruleInfoBo) {
return ruleEngineService.saveRule(ruleInfoBo);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfo.setListeners(rule.getListeners());
ruleInfo.setFilters(rule.getFilters());
ruleInfo.setActions(rule.getActions());
ruleInfo.setName(rule.getName());
ruleInfo.setDesc(rule.getDesc());
ruleInfoData.save(ruleInfo);
}
@ApiOperation("暂停规则")
@PostMapping("/pause")
public boolean pauseRule(@Validated @RequestBody Request<String> request) {
String ruleId = request.getData();
return ruleEngineService.pauseRule(ruleId);
}
@PostMapping("/rule/{ruleId}/pause")
public void pauseRule(@PathVariable("ruleId") String ruleId) {
RuleInfo ruleInfo = ruleInfoData.findById(ruleId);
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfo.setState(RuleInfo.STATE_STOPPED);
ruleInfoData.save(ruleInfo);
ruleManager.pause(ruleInfo.getId());
@ApiOperation("恢复规则")
@PostMapping("/resume")
public boolean resumeRule(@Validated @RequestBody Request<String> request) {
String ruleId = request.getData();
return ruleEngineService.resumeRule(ruleId);
}
@PostMapping("/rule/{ruleId}/resume")
public void resumeRule(@PathVariable("ruleId") String ruleId) {
RuleInfo ruleInfo = ruleInfoData.findById(ruleId);
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfo.setState(RuleInfo.STATE_RUNNING);
ruleInfoData.save(ruleInfo);
ruleManager.resume(ruleInfo);
@ApiOperation("删除规则")
@DeleteMapping("/delete")
public boolean deleteRule(@Validated @RequestBody Request<String> request) {
String ruleId = request.getData();
return ruleEngineService.deleteRule(ruleId);
}
@DeleteMapping("/rule/{ruleId}/delete")
public void deleteRule(@PathVariable("ruleId") String ruleId) {
RuleInfo ruleInfo = ruleInfoData.findById(ruleId);
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfoData.deleteById(ruleInfo.getId());
ruleManager.remove(ruleInfo.getId());
ruleLogData.deleteByRuleId(ruleId);
}
@PostMapping("/rule/{ruleId}/logs/{size}/{page}")
public Paging<RuleLog> getRuleLogs(
@PathVariable("ruleId") String ruleId,
@PathVariable("size") int size,
@PathVariable("page") int page
@ApiOperation("规则日志")
@PostMapping("/ruleLog/list")
public Paging<RuleLogVo> getRuleLogs(
@Validated @RequestBody PageRequest<RuleLogBo> request
) {
RuleLog ruleLog = new RuleLog();
ruleLog.setRuleId(ruleId);
return ruleLogData.findByRuleId(ruleId, page, size);
return ruleEngineService.selectRuleLogPageList(request);
}
@DeleteMapping("/rule/{ruleId}/logs/clear")
public void clearRuleLogs(@PathVariable("ruleId") String ruleId) {
ruleLogData.deleteByRuleId(ruleId);
@ApiOperation("清理日志")
@DeleteMapping("/ruleLog//clear")
public boolean clearRuleLogs(@Validated @RequestBody Request<String> request) {
String ruleId = request.getData();
return ruleEngineService.clearRuleLogs(ruleId);
}
@PostMapping("/tasks")
public List<TaskInfo> tasks() {
if (AuthUtil.isAdmin()) {
return taskInfoData.findAll();
}
return taskInfoData.findByUid(AuthUtil.getUserId());
@ApiOperation("定时任务列表")
@PostMapping("/tasks/list")
public Paging<TaskInfoVo> tasks(@Validated @RequestBody PageRequest<TaskInfoBo> request) {
return ruleEngineService.selectTaskPageList(request);
}
@PostMapping("/saveTask")
public void saveTask(@RequestBody TaskInfo taskInfo) {
if (StringUtils.isBlank(taskInfo.getId())) {
taskInfo.setId(UUID.randomUUID().toString());
taskInfo.setUid(AuthUtil.getUserId());
taskInfo.setCreateAt(System.currentTimeMillis());
taskInfo.setState(TaskInfo.STATE_STOP);
} else {
TaskInfo oldTask = taskInfoData.findById(taskInfo.getId());
if (oldTask == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
taskInfo = ReflectUtil.copyNoNulls(taskInfo, oldTask);
dataOwnerService.checkOwner(taskInfo);
@ApiOperation("定时任务编辑")
@PostMapping("/task/save")
public boolean saveTask(@Validated @RequestBody TaskInfoBo taskInfo) {
return ruleEngineService.saveTask(taskInfo);
}
taskInfoData.save(taskInfo);
@ApiOperation("停止定时任务")
@PostMapping("/task/pause")
public boolean pauseTask(@Validated @RequestBody Request<String> request) {
String taskId = request.getData();
return ruleEngineService.pauseTask(taskId);
}
@PostMapping("/task/{taskId}/pause")
public void pauseTask(@PathVariable("taskId") String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
taskManager.pauseTask(taskId, "stop by " + AuthUtil.getUserId());
@ApiOperation("恢复定时任务")
@PostMapping("/task/resume")
public boolean resumeTask(@Validated @RequestBody Request<String> request) {
return ruleEngineService.resumeTask(request.getData());
}
@PostMapping("/task/{taskId}/resume")
public void resumeTask(@PathVariable("taskId") String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
taskManager.resumeTask(taskId, "resume by " + AuthUtil.getUserId());
}
@PostMapping("/task/renew")
public boolean renewTask(@Validated @RequestBody Request<String> request) {
String taskId = request.getData();
return ruleEngineService.renewTask(taskId);
@PostMapping("/task/{taskId}/renew")
public void renewTask(@PathVariable("taskId") String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
try {
taskManager.renewTask(taskInfo);
taskManager.updateTaskState(taskId, TaskInfo.STATE_RUNNING, "renew by " + AuthUtil.getUserId());
} catch (SchedulerException e) {
log.error("renew task error", e);
throw new BizException(ErrCode.RENEW_TASK_ERROR);
}
}
@DeleteMapping("/task/{taskId}/delete")
public void deleteTask(@PathVariable("taskId") String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
@DeleteMapping("/task/delete")
public boolean deleteTask(@Validated @RequestBody Request<String> request) {
String taskId = request.getData();
return ruleEngineService.deleteTask(taskId);
}
dataOwnerService.checkOwner(taskInfo);
taskManager.deleteTask(taskId, "delete by " + AuthUtil.getUserId());
taskInfoData.deleteById(taskId);
taskLogData.deleteByTaskId(taskId);
}
@PostMapping("/task/{taskId}/logs/{size}/{page}")
public Paging<TaskLog> getTaskLogs(
@PathVariable("taskId") String taskId,
@PathVariable("size") int size,
@PathVariable("page") int page
@PostMapping("/taskLogs/list")
public Paging<TaskLogVo> getTaskLogs(
@Validated @RequestBody PageRequest<TaskLogBo> request
) {
TaskLog taskLog = new TaskLog();
taskLog.setTaskId(taskId);
return taskLogData.findByTaskId(taskId, page, size);
return ruleEngineService.selectTaskLogPageList(request);
}
@DeleteMapping("/task/{taskId}/logs/clear")
public void clearTaskLogs(@PathVariable("taskId") String taskId) {
taskLogData.deleteByTaskId(taskId);
@DeleteMapping("/taskLogs/clear")
public boolean clearTaskLogs( @Validated @RequestBody PageRequest<String> request) {
return ruleEngineService.clearTaskLogs(request.getData());
}
}

View File

@ -0,0 +1,72 @@
package cc.iotkit.manager.dto.bo.ruleinfo;
import cc.iotkit.model.rule.RuleInfo;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonFormat;
import org.springframework.format.annotation.DateTimeFormat;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import javax.validation.constraints.Size;
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
import java.io.Serializable;
import java.util.Date;
import java.math.BigDecimal;
import cc.iotkit.common.api.BaseDto;
import cc.iotkit.common.validate.AddGroup;
import cc.iotkit.common.validate.EditGroup;
import io.github.linpeilie.annotations.AutoMapper;
import lombok.Data;
import lombok.EqualsAndHashCode;
@ApiModel(value = "RuleInfoBo")
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = RuleInfo.class, reverseConvertGenerate = false)
public class RuleInfoBo extends BaseDto {
private static final long serialVersionUID = -1L;
@ApiModelProperty(value="id")
private String id;
@ApiModelProperty(value="动作")
@Size(max = 65535, message = "动作长度不正确")
private String actions;
@ApiModelProperty(value="创建时间")
private Long createAt;
@ApiModelProperty(value="描述")
@Size(max = 255, message = "描述长度不正确")
private String desc;
@ApiModelProperty(value="过滤器")
@Size(max = 65535, message = "过滤器长度不正确")
private String filters;
@ApiModelProperty(value="监听器")
@Size(max = 65535, message = "监听器长度不正确")
private String listeners;
@ApiModelProperty(value="规则名称")
@Size(max = 255, message = "规则名称长度不正确")
private String name;
@ApiModelProperty(value="状态")
@Size(max = 255, message = "状态长度不正确")
private String state;
@ApiModelProperty(value="规则类型")
@Size(max = 255, message = "规则类型长度不正确")
private String type;
@ApiModelProperty(value="用户id")
@Size(max = 255, message = "用户id长度不正确")
private String uid;
}

View File

@ -0,0 +1,40 @@
package cc.iotkit.manager.dto.bo.ruleinfo;
import cc.iotkit.common.api.BaseDto;
import cc.iotkit.model.rule.RuleInfo;
import io.github.linpeilie.annotations.AutoMapper;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import javax.validation.constraints.Size;
@ApiModel(value = "RuleLogBo")
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = RuleInfo.class, reverseConvertGenerate = false)
public class RuleLogBo extends BaseDto {
private static final long serialVersionUID = -1L;
@ApiModelProperty(value="时间")
private Long time;
@ApiModelProperty(value="规则id")
private String ruleId;
@ApiModelProperty(value="状态")
private String state1;
@ApiModelProperty(value="内容")
private String content;
@ApiModelProperty(value="是否成功")
private Boolean success;
}

View File

@ -0,0 +1,70 @@
package cc.iotkit.manager.dto.bo.taskinfo;
import cc.iotkit.model.rule.TaskInfo;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonFormat;
import org.springframework.format.annotation.DateTimeFormat;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import javax.validation.constraints.Size;
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
import java.io.Serializable;
import java.util.Date;
import java.math.BigDecimal;
import cc.iotkit.common.api.BaseDto;
import cc.iotkit.common.validate.AddGroup;
import cc.iotkit.common.validate.EditGroup;
import io.github.linpeilie.annotations.AutoMapper;
import lombok.Data;
import lombok.EqualsAndHashCode;
@ApiModel(value = "TaskInfoBo")
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = TaskInfo.class, reverseConvertGenerate = false)
public class TaskInfoBo extends BaseDto {
private static final long serialVersionUID = -1L;
@ApiModelProperty(value="任务输出")
@Size(max = 65535, message = "任务输出长度不正确")
private String actions;
@ApiModelProperty(value="创建时间")
private Long createAt;
@ApiModelProperty(value="描述")
@Size(max = 255, message = "描述长度不正确")
private String desc;
@ApiModelProperty(value="表达式")
@Size(max = 255, message = "表达式长度不正确")
private String expression;
@ApiModelProperty(value="任务名称")
@Size(max = 255, message = "任务名称长度不正确")
private String name;
@ApiModelProperty(value="操作备注")
@Size(max = 255, message = "操作备注长度不正确")
private String reason;
@ApiModelProperty(value="任务状态")
@Size(max = 255, message = "任务状态长度不正确")
private String state;
@ApiModelProperty(value="任务类型")
@Size(max = 255, message = "任务类型长度不正确")
private String type;
@ApiModelProperty(value="创建者")
@Size(max = 255, message = "创建者长度不正确")
private String uid;
}

View File

@ -0,0 +1,34 @@
/*
* +----------------------------------------------------------------------
* | Copyright (c) 2021-2022 All rights reserved.
* +----------------------------------------------------------------------
* | Licensed
* +----------------------------------------------------------------------
* | Author: xw2sy@163.com
* +----------------------------------------------------------------------
*/
package cc.iotkit.manager.dto.bo.taskinfo;
import cc.iotkit.common.api.BaseDto;
import cc.iotkit.model.rule.TaskInfo;
import cc.iotkit.model.rule.TaskLog;
import io.github.linpeilie.annotations.AutoMapper;
import io.swagger.annotations.ApiModel;
import lombok.*;
@ApiModel(value = "TaskLogBo")
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = TaskLog.class, reverseConvertGenerate = false)
public class TaskLogBo extends BaseDto {
private String id;
private String taskId;
private String content;
private Boolean success;
private Long logAt;
}

View File

@ -0,0 +1,68 @@
package cc.iotkit.manager.dto.vo.ruleinfo;
import cc.iotkit.model.rule.RuleInfo;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.ToString;
import java.util.Date;
import java.io.Serializable;
import java.math.BigDecimal;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
import com.alibaba.excel.annotation.ExcelProperty;
import io.github.linpeilie.annotations.AutoMapper;
@ApiModel(value = "RuleInfoVo")
@Data
@ExcelIgnoreUnannotated
@AutoMapper(target = RuleInfo.class)
public class RuleInfoVo implements Serializable {
private static final long serialVersionUID = -1L;
@ApiModelProperty(value="规则id")
@ExcelProperty(value = "规则id")
private String id;
@ApiModelProperty(value="动作")
@ExcelProperty(value = "动作")
private String actions;
@ApiModelProperty(value="创建时间")
@ExcelProperty(value = "创建时间")
private Long createAt;
@ApiModelProperty(value="描述")
@ExcelProperty(value = "描述")
private String desc;
@ApiModelProperty(value="过滤器")
@ExcelProperty(value = "过滤器")
private String filters;
@ApiModelProperty(value="监听器")
@ExcelProperty(value = "监听器")
private String listeners;
@ApiModelProperty(value="规则名称")
@ExcelProperty(value = "规则名称")
private String name;
@ApiModelProperty(value="状态")
@ExcelProperty(value = "状态")
private String state;
@ApiModelProperty(value="规则类型")
@ExcelProperty(value = "规则类型")
private String type;
@ApiModelProperty(value="用户id")
@ExcelProperty(value = "用户id")
private String uid;
}

View File

@ -0,0 +1,38 @@
package cc.iotkit.manager.dto.vo.ruleinfo;
import cc.iotkit.common.api.BaseDto;
import cc.iotkit.model.rule.RuleInfo;
import io.github.linpeilie.annotations.AutoMapper;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
@ApiModel(value = "RuleLogBo")
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = RuleInfo.class, reverseConvertGenerate = false)
public class RuleLogVo extends BaseDto {
private static final long serialVersionUID = -1L;
@ApiModelProperty(value="时间")
private Long time;
@ApiModelProperty(value="规则id")
private String ruleId;
@ApiModelProperty(value="状态")
private String state1;
@ApiModelProperty(value="内容")
private String content;
@ApiModelProperty(value="是否成功")
private Boolean success;
}

View File

@ -0,0 +1,68 @@
package cc.iotkit.manager.dto.vo.taskinfo;
import cc.iotkit.model.rule.TaskInfo;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.ToString;
import java.util.Date;
import java.io.Serializable;
import java.math.BigDecimal;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
import com.alibaba.excel.annotation.ExcelProperty;
import io.github.linpeilie.annotations.AutoMapper;
@ApiModel(value = "TaskInfoVo")
@Data
@ExcelIgnoreUnannotated
@AutoMapper(target = TaskInfo.class)
public class TaskInfoVo implements Serializable {
private static final long serialVersionUID = -1L;
@ApiModelProperty(value="主键")
@ExcelProperty(value = "主键")
private String id;
@ApiModelProperty(value="任务输出")
@ExcelProperty(value = "任务输出")
private String actions;
@ApiModelProperty(value="创建时间")
@ExcelProperty(value = "创建时间")
private Long createAt;
@ApiModelProperty(value="描述")
@ExcelProperty(value = "描述")
private String desc;
@ApiModelProperty(value="表达式")
@ExcelProperty(value = "表达式")
private String expression;
@ApiModelProperty(value="任务名称")
@ExcelProperty(value = "任务名称")
private String name;
@ApiModelProperty(value="操作备注")
@ExcelProperty(value = "操作备注")
private String reason;
@ApiModelProperty(value="任务状态")
@ExcelProperty(value = "任务状态")
private String state;
@ApiModelProperty(value="任务类型")
@ExcelProperty(value = "任务类型")
private String type;
@ApiModelProperty(value="创建者")
@ExcelProperty(value = "创建者")
private String uid;
}

View File

@ -0,0 +1,32 @@
/*
* +----------------------------------------------------------------------
* | Copyright (c) 2021-2022 All rights reserved.
* +----------------------------------------------------------------------
* | Licensed
* +----------------------------------------------------------------------
* | Author: xw2sy@163.com
* +----------------------------------------------------------------------
*/
package cc.iotkit.manager.dto.vo.taskinfo;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
public class TaskLogVo {
private String id;
private String taskId;
private String content;
private Boolean success;
private Long logAt;
}

View File

@ -0,0 +1,50 @@
package cc.iotkit.manager.service;
import cc.iotkit.common.api.PageRequest;
import cc.iotkit.common.api.Paging;
import cc.iotkit.manager.dto.bo.ruleinfo.RuleInfoBo;
import cc.iotkit.manager.dto.bo.ruleinfo.RuleLogBo;
import cc.iotkit.manager.dto.bo.taskinfo.TaskInfoBo;
import cc.iotkit.manager.dto.bo.taskinfo.TaskLogBo;
import cc.iotkit.manager.dto.vo.ruleinfo.RuleInfoVo;
import cc.iotkit.manager.dto.vo.ruleinfo.RuleLogVo;
import cc.iotkit.manager.dto.vo.taskinfo.TaskInfoVo;
import cc.iotkit.manager.dto.vo.taskinfo.TaskLogVo;
/**
* @Author: jay
* @Date: 2023/5/30 18:14
* @Version: V1.0
* @Description:
*/
public interface IRuleEngineService {
Paging<RuleInfoVo> selectPageList(PageRequest<RuleInfoBo> request);
boolean saveRule(RuleInfoBo ruleInfoBo);
boolean pauseRule(String ruleId);
boolean resumeRule(String ruleId);
boolean deleteRule(String ruleId);
Paging<RuleLogVo> selectRuleLogPageList(PageRequest<RuleLogBo> request);
boolean clearRuleLogs(String ruleId);
Paging<TaskInfoVo> selectTaskPageList(PageRequest<TaskInfoBo> request);
boolean saveTask(TaskInfoBo taskInfo);
boolean pauseTask(String taskId);
boolean resumeTask(String data);
boolean renewTask(String taskId);
boolean deleteTask(String taskId);
Paging<TaskLogVo> selectTaskLogPageList(PageRequest<TaskLogBo> request);
boolean clearTaskLogs(String taskId);
}

View File

@ -0,0 +1,264 @@
package cc.iotkit.manager.service.impl;
import cc.iotkit.common.api.PageRequest;
import cc.iotkit.common.api.Paging;
import cc.iotkit.common.enums.ErrCode;
import cc.iotkit.common.exception.BizException;
import cc.iotkit.common.satoken.utils.AuthUtil;
import cc.iotkit.common.utils.MapstructUtils;
import cc.iotkit.common.utils.ReflectUtil;
import cc.iotkit.data.manager.IRuleInfoData;
import cc.iotkit.data.manager.ITaskInfoData;
import cc.iotkit.manager.dto.bo.ruleinfo.RuleInfoBo;
import cc.iotkit.manager.dto.bo.ruleinfo.RuleLogBo;
import cc.iotkit.manager.dto.bo.taskinfo.TaskInfoBo;
import cc.iotkit.manager.dto.bo.taskinfo.TaskLogBo;
import cc.iotkit.manager.dto.vo.ruleinfo.RuleInfoVo;
import cc.iotkit.manager.dto.vo.ruleinfo.RuleLogVo;
import cc.iotkit.manager.dto.vo.taskinfo.TaskInfoVo;
import cc.iotkit.manager.dto.vo.taskinfo.TaskLogVo;
import cc.iotkit.manager.service.DataOwnerService;
import cc.iotkit.manager.service.IRuleEngineService;
import cc.iotkit.model.rule.RuleInfo;
import cc.iotkit.model.rule.RuleLog;
import cc.iotkit.model.rule.TaskInfo;
import cc.iotkit.model.rule.TaskLog;
import cc.iotkit.ruleengine.rule.RuleManager;
import cc.iotkit.ruleengine.task.TaskManager;
import cc.iotkit.temporal.IRuleLogData;
import cc.iotkit.temporal.ITaskLogData;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.quartz.SchedulerException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.UUID;
/**
* @Author: jay
* @Date: 2023/5/30 18:15
* @Version: V1.0
* @Description:
*/
@Slf4j
@RequiredArgsConstructor
@Service
public class RuleEngineServiceImpl implements IRuleEngineService {
@Autowired
private ITaskInfoData taskInfoData;
@Autowired
private IRuleInfoData ruleInfoData;
@Autowired
private IRuleLogData ruleLogData;
@Autowired
private DataOwnerService dataOwnerService;
@Autowired
private TaskManager taskManager;
@Autowired
private RuleManager ruleManager;
@Autowired
private ITaskLogData taskLogData;
@Override
public Paging<RuleInfoVo> selectPageList(PageRequest<RuleInfoBo> request) {
RuleInfoBo data = request.getData();
String type = data.getType();
Integer page = request.getPageNum();
Integer size = request.getPageSize();
if (AuthUtil.isAdmin()) {
return MapstructUtils.convert(ruleInfoData.findByType(type, page, size), RuleInfoVo.class);
} else {
return MapstructUtils.convert(ruleInfoData.findByUidAndType(AuthUtil.getUserId(), type, page, size), RuleInfoVo.class);
}
}
@Override
public boolean saveRule(RuleInfoBo ruleInfoBo) {
RuleInfo rule = MapstructUtils.convert(ruleInfoBo, RuleInfo.class);
if (StringUtils.isBlank(rule.getId())) {
rule.setId(UUID.randomUUID().toString());
rule.setState(RuleInfo.STATE_STOPPED);
rule.setCreateAt(System.currentTimeMillis());
rule.setUid(AuthUtil.getUserId());
ruleInfoData.save(rule);
ruleManager.add(rule);
} else {
RuleInfo ruleInfo = ruleInfoData.findById(rule.getId());
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
if (RuleInfo.STATE_RUNNING.equals(ruleInfo.getState())) {
throw new BizException(ErrCode.RULE_ALREADY_RUNNING);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfo.setListeners(rule.getListeners());
ruleInfo.setFilters(rule.getFilters());
ruleInfo.setActions(rule.getActions());
ruleInfo.setName(rule.getName());
ruleInfo.setDesc(rule.getDesc());
ruleInfoData.save(ruleInfo);
}
return true;
}
@Override
public boolean pauseRule(String ruleId) {
RuleInfo ruleInfo = ruleInfoData.findById(ruleId);
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfo.setState(RuleInfo.STATE_STOPPED);
ruleInfoData.save(ruleInfo);
ruleManager.pause(ruleInfo.getId());
return true;
}
@Override
public boolean resumeRule(String ruleId) {
RuleInfo ruleInfo = ruleInfoData.findById(ruleId);
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfo.setState(RuleInfo.STATE_RUNNING);
ruleInfoData.save(ruleInfo);
ruleManager.resume(ruleInfo);
return true;
}
@Override
public boolean deleteRule(String ruleId) {
RuleInfo ruleInfo = ruleInfoData.findById(ruleId);
if (ruleInfo == null) {
throw new BizException(ErrCode.RULE_NOT_FOUND);
}
dataOwnerService.checkOwner(ruleInfo);
ruleInfoData.deleteById(ruleInfo.getId());
ruleManager.remove(ruleInfo.getId());
ruleLogData.deleteByRuleId(ruleId);
return true;
}
@Override
public Paging<RuleLogVo> selectRuleLogPageList(PageRequest<RuleLogBo> request) {
RuleLog ruleLog = request.getData().to(RuleLog.class);
return ruleLogData.findByRuleId(ruleLog.getRuleId(), request.getPageNum(), request.getPageSize()).to(RuleLogVo.class);
}
@Override
public boolean clearRuleLogs(String ruleId) {
ruleLogData.deleteByRuleId(ruleId);
return true;
}
@Override
public Paging<TaskInfoVo> selectTaskPageList(PageRequest<TaskInfoBo> request) {
if (AuthUtil.isAdmin()) {
return taskInfoData.findAll(request.to(TaskInfo.class)).to(TaskInfoVo.class);
}
return taskInfoData.findByUid(AuthUtil.getUserId(), request.getPageNum(), request.getPageSize()).to(TaskInfoVo.class);
}
@Override
public boolean saveTask(TaskInfoBo bo) {
TaskInfo taskInfo = bo.to(TaskInfo.class);
if (StringUtils.isBlank(taskInfo.getId())) {
taskInfo.setId(UUID.randomUUID().toString());
taskInfo.setUid(AuthUtil.getUserId());
taskInfo.setCreateAt(System.currentTimeMillis());
taskInfo.setState(TaskInfo.STATE_STOP);
} else {
TaskInfo oldTask = taskInfoData.findById(taskInfo.getId());
if (oldTask == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
taskInfo = ReflectUtil.copyNoNulls(taskInfo, oldTask);
dataOwnerService.checkOwner(taskInfo);
}
taskInfoData.save(taskInfo);
return true;
}
@Override
public boolean pauseTask(String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
taskManager.pauseTask(taskId, "stop by " + AuthUtil.getUserId());
return true;
}
@Override
public boolean resumeTask(String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
taskManager.resumeTask(taskId, "resume by " + AuthUtil.getUserId());
return true;
}
@Override
public boolean renewTask(String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
try {
taskManager.renewTask(taskInfo);
taskManager.updateTaskState(taskId, TaskInfo.STATE_RUNNING, "renew by " + AuthUtil.getUserId());
} catch (SchedulerException e) {
log.error("renew task error", e);
throw new BizException(ErrCode.RENEW_TASK_ERROR);
}
return true;
}
@Override
public boolean deleteTask(String taskId) {
TaskInfo taskInfo = taskInfoData.findById(taskId);
if (taskInfo == null) {
throw new BizException(ErrCode.TASK_NOT_FOUND);
}
dataOwnerService.checkOwner(taskInfo);
taskManager.deleteTask(taskId, "delete by " + AuthUtil.getUserId());
taskInfoData.deleteById(taskId);
taskLogData.deleteByTaskId(taskId);
return true;
}
@Override
public Paging<TaskLogVo> selectTaskLogPageList(PageRequest<TaskLogBo> request) {
TaskLog taskLog = request.getData().to(TaskLog.class);
Paging<TaskLog> byTaskId = taskLogData.findByTaskId(taskLog.getTaskId(), request.getPageNum(), request.getPageSize());
return byTaskId.to(TaskLogVo.class);
}
@Override
public boolean clearTaskLogs(String taskId) {
taskLogData.deleteByTaskId(taskId);
return true;
}
}