diff --git a/riis-system/src/main/java/com/yfd/platform/modules/algorithm/controller/AlgorithmArrangeController.java b/riis-system/src/main/java/com/yfd/platform/modules/algorithm/controller/AlgorithmArrangeController.java
index f0ada01..f8e0bd6 100644
--- a/riis-system/src/main/java/com/yfd/platform/modules/algorithm/controller/AlgorithmArrangeController.java
+++ b/riis-system/src/main/java/com/yfd/platform/modules/algorithm/controller/AlgorithmArrangeController.java
@@ -10,14 +10,12 @@ import com.yfd.platform.modules.algorithm.service.IAlgorithmArrangeService;
import com.yfd.platform.utils.SecurityUtils;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.time.LocalDateTime;
+import java.util.List;
/**
*
@@ -49,9 +47,10 @@ public class AlgorithmArrangeController {
Page arrangePage = algorithmArrangeService.page(page, queryWrapper);
return ResponseResult.successData(arrangePage);
}
+
@GetMapping("/getAlgorithmArrangeById")
@ApiOperation("根据Id查询算法布点")
- public ResponseResult getAlgorithmArrangeById (String id){
+ public ResponseResult getAlgorithmArrangeById(String id) {
AlgorithmArrange algorithmArrange = algorithmArrangeService.getById(id);
return ResponseResult.successData(algorithmArrange);
}
@@ -94,7 +93,8 @@ public class AlgorithmArrangeController {
public ResponseResult updateAlgorithmArrange(AlgorithmArrange algorithmArrange, MultipartFile file) {
// 判断是否存在相同的方案
LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.ne(AlgorithmArrange::getId, algorithmArrange.getId()).eq(AlgorithmArrange::getArrangeName, algorithmArrange.getArrangeName());
+ queryWrapper.ne(AlgorithmArrange::getId, algorithmArrange.getId()).eq(AlgorithmArrange::getArrangeName,
+ algorithmArrange.getArrangeName());
if (algorithmArrangeService.getOne(queryWrapper) != null) {
return ResponseResult.error("方案名称已存在");
}
@@ -118,12 +118,33 @@ public class AlgorithmArrangeController {
return ResponseResult.success();
}
+ @PostMapping("/deleteAlgorithmArrange")
+ @ApiOperation("删除布点数据")
+ public ResponseResult deleteAlgorithmArrange(String id) {
+ boolean b = algorithmArrangeService.removeById(id);
+ if (b) {
+ return ResponseResult.success();
+ } else {
+ return ResponseResult.error();
+ }
+ }
+
+ @PostMapping("/deleteAlgorithmArrange")
+ @ApiOperation("删除布点数据")
+ public ResponseResult deleteAlgorithmArrange(@RequestBody List ids) {
+ boolean b = algorithmArrangeService.removeByIds(ids);
+ if (b) {
+ return ResponseResult.success();
+ } else {
+ return ResponseResult.error();
+ }
+ }
+
@PostMapping("/callAlgorithmArrange")
@ApiOperation("调用算法布点API")
public ResponseResult callAlgorithmArrange(String id) {
- AlgorithmArrange algorithmArrange=algorithmArrangeService.callAlgorithmArrange(id);
+ AlgorithmArrange algorithmArrange = algorithmArrangeService.callAlgorithmArrange(id);
return ResponseResult.successData(algorithmArrange);
}
-
}