|  |  |  | 
|---|
|  |  |  | import com.zy.acs.common.utils.Utils; | 
|---|
|  |  |  | import com.zy.acs.framework.common.Cools; | 
|---|
|  |  |  | import com.zy.acs.framework.common.R; | 
|---|
|  |  |  | import com.zy.acs.framework.exception.CoolException; | 
|---|
|  |  |  | import com.zy.acs.manager.common.annotation.OperationLog; | 
|---|
|  |  |  | import com.zy.acs.manager.common.domain.BaseParam; | 
|---|
|  |  |  | import com.zy.acs.manager.common.domain.KeyValVo; | 
|---|
|  |  |  | 
|---|
|  |  |  | import com.zy.acs.manager.system.controller.BaseController; | 
|---|
|  |  |  | import org.springframework.beans.factory.annotation.Autowired; | 
|---|
|  |  |  | import org.springframework.security.access.prepost.PreAuthorize; | 
|---|
|  |  |  | import org.springframework.transaction.annotation.Transactional; | 
|---|
|  |  |  | import org.springframework.web.bind.annotation.*; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import javax.servlet.http.HttpServletResponse; | 
|---|
|  |  |  | 
|---|
|  |  |  | @PreAuthorize("hasAuthority('manager:route:save')") | 
|---|
|  |  |  | @OperationLog("Create Route") | 
|---|
|  |  |  | @PostMapping("/route/save") | 
|---|
|  |  |  | @Transactional | 
|---|
|  |  |  | public R save(@RequestBody Route route) { | 
|---|
|  |  |  | route.setCreateBy(getLoginUserId()); | 
|---|
|  |  |  | route.setCreateTime(new Date()); | 
|---|
|  |  |  | route.setUpdateBy(getLoginUserId()); | 
|---|
|  |  |  | route.setUpdateTime(new Date()); | 
|---|
|  |  |  | if (!routeService.save(route)) { | 
|---|
|  |  |  | if (route.getStartCode() == null || route.getEndCode() == null) { | 
|---|
|  |  |  | return R.error("Save Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (route.getStartCode().equals(route.getEndCode())) { | 
|---|
|  |  |  | return R.error("Save Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (routeService.count(new LambdaQueryWrapper<Route>() | 
|---|
|  |  |  | .eq(Route::getStartCode, route.getStartCode()) | 
|---|
|  |  |  | .eq(Route::getEndCode, route.getEndCode())) > 0 || | 
|---|
|  |  |  | routeService.count(new LambdaQueryWrapper<Route>() | 
|---|
|  |  |  | .eq(Route::getStartCode, route.getEndCode()) | 
|---|
|  |  |  | .eq(Route::getEndCode, route.getStartCode())) > 0 | 
|---|
|  |  |  | ) { | 
|---|
|  |  |  | return R.error("Save Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | Code startCode = codeService.getById(route.getStartCode()); | 
|---|
|  |  |  | Code endCode = codeService.getById(route.getEndCode()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (null == startCode || null == endCode) { | 
|---|
|  |  |  | return R.error("Save Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | Route result = routeService.createRouteByCode(startCode, endCode, route.getDirection(), getLoginUserId()); | 
|---|
|  |  |  | codeGapService.createCodeGapByCode(startCode, endCode, getLoginUserId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (null == result) { | 
|---|
|  |  |  | throw new CoolException("Save Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return R.ok("Save Success").add(route); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | @OperationLog("Update Route") | 
|---|
|  |  |  | @PostMapping("/route/update") | 
|---|
|  |  |  | public R update(@RequestBody Route route) { | 
|---|
|  |  |  | if (route.getStartCode().equals(route.getEndCode())) { | 
|---|
|  |  |  | return R.error("Update Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (routeService.count(new LambdaQueryWrapper<Route>() | 
|---|
|  |  |  | .eq(Route::getStartCode, route.getStartCode()) | 
|---|
|  |  |  | .eq(Route::getEndCode, route.getEndCode()) | 
|---|
|  |  |  | .ne(Route::getId, route.getId())) > 0 || | 
|---|
|  |  |  | routeService.count(new LambdaQueryWrapper<Route>() | 
|---|
|  |  |  | .eq(Route::getStartCode, route.getEndCode()) | 
|---|
|  |  |  | .eq(Route::getEndCode, route.getStartCode()) | 
|---|
|  |  |  | .ne(Route::getId, route.getId())) > 0 | 
|---|
|  |  |  | ) { | 
|---|
|  |  |  | return R.error("Update Fail"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | route.setUpdateBy(getLoginUserId()); | 
|---|
|  |  |  | route.setUpdateTime(new Date()); | 
|---|
|  |  |  | if (!routeService.updateById(route)) { | 
|---|