#

vincentlu

2 天以前 b0d8223aa90f5a812ecbd6bab9eaf4382b62f5b2

 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/RouteController.java

@@ -107,6 +107,10 @@
    @OperationLog("Update Route")
    @PostMapping("/route/update")
    public R update(@RequestBody Route route) {
        Route origin = routeService.getById(route.getId());
        if (origin == null) {
            return R.error("Update Fail");
        }
        if (route.getStartCode().equals(route.getEndCode())) {
            return R.error("Update Fail");
        }
@@ -127,6 +131,20 @@
        if (!routeService.updateById(route)) {
            return R.error("Update Fail");
        }
        List<Long> affectedCodeIds = new ArrayList<>();
        if (origin.getStartCode() != null) {
            affectedCodeIds.add(origin.getStartCode());
        }
        if (origin.getEndCode() != null) {
            affectedCodeIds.add(origin.getEndCode());
        }
        if (route.getStartCode() != null) {
            affectedCodeIds.add(route.getStartCode());
        }
        if (route.getEndCode() != null) {
            affectedCodeIds.add(route.getEndCode());
        }
        codeService.refreshCornerByCodeIds(affectedCodeIds);
        return R.ok("Update Success").add(route);
    }

@@ -134,9 +152,20 @@
    @OperationLog("Delete Route")
    @PostMapping("/route/remove/{ids}")
    public R remove(@PathVariable Long[] ids) {
        List<Route> routes = routeService.listByIds(Arrays.asList(ids));
        List<Long> affectedCodeIds = new ArrayList<>();
        for (Route route : routes) {
            if (route.getStartCode() != null) {
                affectedCodeIds.add(route.getStartCode());
            }
            if (route.getEndCode() != null) {
                affectedCodeIds.add(route.getEndCode());
            }
        }
        if (!routeService.removeByIds(Arrays.asList(ids))) {
            return R.error("Delete Fail");
        }
        codeService.refreshCornerByCodeIds(affectedCodeIds);
        return R.ok("Delete Success").add(ids);
    }