|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; | 
|---|
|  |  |  | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; | 
|---|
|  |  |  | import com.zy.acs.common.utils.GsonUtils; | 
|---|
|  |  |  | import com.zy.acs.common.constant.CommonConstant; | 
|---|
|  |  |  | 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.common.domain.PageParam; | 
|---|
|  |  |  | import com.zy.acs.manager.common.domain.RouteExcel; | 
|---|
|  |  |  | import com.zy.acs.manager.common.utils.ExcelUtil; | 
|---|
|  |  |  | import com.zy.acs.manager.manager.entity.Code; | 
|---|
|  |  |  | import com.zy.acs.manager.manager.entity.Route; | 
|---|
|  |  |  | import com.zy.acs.manager.manager.service.CodeGapService; | 
|---|
|  |  |  | import com.zy.acs.manager.manager.service.CodeService; | 
|---|
|  |  |  | import com.zy.acs.manager.manager.service.RouteService; | 
|---|
|  |  |  | 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; | 
|---|
|  |  |  | 
|---|
|  |  |  | private RouteService routeService; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private CodeService codeService; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private CodeGapService codeGapService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @PreAuthorize("hasAuthority('manager:route:list')") | 
|---|
|  |  |  | @PostMapping("/route/page") | 
|---|
|  |  |  | 
|---|
|  |  |  | @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)) { | 
|---|
|  |  |  | 
|---|
|  |  |  | @PreAuthorize("hasAuthority('manager:route:save')") | 
|---|
|  |  |  | @PostMapping("/route/import") | 
|---|
|  |  |  | public R importBatch(@RequestBody List<Map<String, Object>> list) { | 
|---|
|  |  |  | Date now = new Date();  Long userId = getLoginUserId(); | 
|---|
|  |  |  | Long userId = getLoginUserId(); | 
|---|
|  |  |  | String regex = "([a-zA-Z]+)\\("; | 
|---|
|  |  |  | Pattern pattern = Pattern.compile(regex); | 
|---|
|  |  |  | for (Map<String, Object> map : list) { | 
|---|
|  |  |  | 
|---|
|  |  |  | key = Utils.toCamelCase(key); | 
|---|
|  |  |  | one.put(key, entry.getValue()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Route route = Cools.conver(one, Route.class); | 
|---|
|  |  |  | System.out.println(GsonUtils.toJson(route)); | 
|---|
|  |  |  | RouteExcel excelDto = Cools.conver(one, RouteExcel.class); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | Code code0 = codeService.selectByData(Utils.zeroFill(excelDto.getStartCode(), CommonConstant.QR_CODE_LEN)); | 
|---|
|  |  |  | Code code1 = codeService.selectByData(Utils.zeroFill(excelDto.getEndCode(), CommonConstant.QR_CODE_LEN)); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (null == code0 || null == code1) { continue; } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | routeService.createRouteByCode(code0, code1, excelDto.getDirection(), userId); | 
|---|
|  |  |  | codeGapService.createCodeGapByCode(code0, code1, userId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return R.ok(); | 
|---|
|  |  |  | } | 
|---|