| package com.dy.pipIrrProject.divide; | 
|   | 
| import com.alibaba.excel.EasyExcel; | 
| import com.alibaba.excel.support.ExcelTypeEnum; | 
| import com.dy.common.aop.SsoAop; | 
| import com.dy.common.webUtil.BaseResponse; | 
| import com.dy.common.webUtil.BaseResponseUtils; | 
| import com.dy.common.webUtil.QueryResultVo; | 
| import com.dy.common.webUtil.ResultCodeMsg; | 
| import com.dy.pipIrrGlobal.pojoBa.BaClient; | 
| import com.dy.pipIrrGlobal.pojoPr.PrDivide; | 
| import com.dy.pipIrrGlobal.voPr.VoDivide; | 
| import com.dy.pipIrrGlobal.voPr.VoDivideDetails; | 
| import com.dy.pipIrrGlobal.voSe.VoActiveCard; | 
| import com.dy.pipIrrProject.result.ProjectResultCode; | 
| import io.swagger.v3.oas.annotations.Operation; | 
| import io.swagger.v3.oas.annotations.media.Content; | 
| import io.swagger.v3.oas.annotations.media.Schema; | 
| import io.swagger.v3.oas.annotations.responses.ApiResponse; | 
| import io.swagger.v3.oas.annotations.responses.ApiResponses; | 
| import io.swagger.v3.oas.annotations.tags.Tag; | 
| import jakarta.servlet.http.HttpServletResponse; | 
| import jakarta.validation.Valid; | 
| import lombok.RequiredArgsConstructor; | 
| import lombok.SneakyThrows; | 
| import lombok.extern.slf4j.Slf4j; | 
| import org.springframework.http.MediaType; | 
| import org.springframework.transaction.annotation.Transactional; | 
| import org.springframework.validation.BindingResult; | 
| import org.springframework.web.bind.annotation.*; | 
| import org.springframework.web.multipart.MultipartFile; | 
|   | 
| import java.io.IOException; | 
| import java.io.UnsupportedEncodingException; | 
| import java.net.URLEncoder; | 
| import java.util.*; | 
|   | 
| /** | 
|  * @author ZhuBaoMin | 
|  * @date 2023/12/20 20:11 | 
|  * @LastEditTime 2023/12/20 20:11 | 
|  * @Description | 
|  */ | 
|   | 
| @Slf4j | 
| @Tag(name = "分水房管理", description = "分水房操作") | 
| @RestController | 
| @RequestMapping(path="divide") | 
| @RequiredArgsConstructor | 
| public class DivideCtrl { | 
|     private final DivideSv divideSv; | 
|   | 
|     /** | 
|      * 分页获取分水房记录 | 
|      * @param vo | 
|      * @return | 
|      */ | 
|     @Operation(summary = "获得一页分水房记录", description = "返回一页分水房数据") | 
|     @ApiResponses(value = { | 
|             @ApiResponse( | 
|                     responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, | 
|                     description = "返回一页开卡数据(BaseResponse.content:QueryResultVo[{}])", | 
|                     content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, | 
|                             schema = @Schema(implementation = VoActiveCard.class))} | 
|             ) | 
|     }) | 
|     @GetMapping(path = "/getDivides") | 
|     @SsoAop() | 
|     public BaseResponse<QueryResultVo<List<VoDivide>>> getDivides(QueryVo vo){ | 
|         try { | 
|             //QueryResultVo<List<VoDivide>> res = divideSv.getDivides(vo); | 
|             //if(res.itemTotal != null && res.itemTotal > 0) { | 
|             //    return BaseResponseUtils.buildSuccess(res); | 
|             //}else { | 
|             //    return BaseResponseUtils.buildSuccess(ProjectResultCode.NO_DIVIDES.getMessage()); | 
|             //} | 
|             return BaseResponseUtils.buildSuccess(divideSv.getDivides(vo)); | 
|         } catch (Exception e) { | 
|             log.error("获取开卡记录异常", e); | 
|             return BaseResponseUtils.buildException(e.getMessage()) ; | 
|         } | 
|     } | 
|   | 
|     /** | 
|      * 根据分水房主键获取分水房对象 | 
|      * @param id | 
|      * @return | 
|      */ | 
|     @Operation(summary = "获得一个分水房", description = "返回一个分水房数据") | 
|     @ApiResponses(value = { | 
|             @ApiResponse( | 
|                     responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, | 
|                     description = "返回一页农户数据(BaseResponse.content:QueryResultVo[{}])", | 
|                     content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, | 
|                             schema = @Schema(implementation = BaClient.class))} | 
|             ) | 
|     }) | 
|     @GetMapping(path = "/getone/{id}") | 
|     @SsoAop() | 
|     //public BaseResponse<PrDivide> getOneDivide(@PathVariable("id") Long id){ | 
|     public BaseResponse<VoDivideDetails> getOneDivide(@PathVariable("id") Long id){ | 
|         try { | 
|             //PrDivide res = divideSv.getOneDivide(id); | 
|             //if(res != null) { | 
|             //    return BaseResponseUtils.buildSuccess(res); | 
|             //}else { | 
|             //    return BaseResponseUtils.buildErrorMsg(ProjectResultCode.NO_DIVIDES.getMessage()); | 
|             //} | 
|             return BaseResponseUtils.buildSuccess(divideSv.getDivideDetails(id)); | 
|         } catch (Exception e) { | 
|             log.error("查询农户异常", e); | 
|             return BaseResponseUtils.buildException(e.getMessage()) ; | 
|         } | 
|     } | 
|   | 
|     /** | 
|      * 添加分水房记录 | 
|      * @param po | 
|      * @param bindingResult | 
|      * @return | 
|      */ | 
|     @Operation(summary = "添加分水房记录", description = "添加分水房记录") | 
|     @ApiResponses(value = { | 
|             @ApiResponse( | 
|                     responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, | 
|                     description = "操作结果:true:成功,false:失败(BaseResponse.content)", | 
|                     content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, | 
|                             schema = @Schema(implementation = Boolean.class))} | 
|             ) | 
|     }) | 
|     @PostMapping(path = "add", consumes = MediaType.APPLICATION_JSON_VALUE) | 
|     @Transactional(rollbackFor = Exception.class) | 
|     @SsoAop() | 
|     public BaseResponse<Boolean> add(@RequestBody @Valid DtoDivide po, BindingResult bindingResult){ | 
|         if(bindingResult != null && bindingResult.hasErrors()){ | 
|             return BaseResponseUtils.buildErrorMsg(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage()); | 
|         } | 
|         // 接收村编号(主键) | 
|         Long villageId = po.getVillageId(); | 
|   | 
|         PrDivide prDivide = DtoToDividePojo.INSTANCT.po2vo(po); | 
|   | 
|         if (villageId!=null){ | 
|             /** | 
|              * 获取5级行政区划信息 | 
|              */ | 
|             Map map_districts = Optional.ofNullable(divideSv.getDistrictsByVillageId(villageId)).orElse(new HashMap()); | 
|             if(map_districts.size() <= 0) { | 
|                 return BaseResponseUtils.buildErrorMsg("区划信息有误"); | 
|             } | 
|             Long countryId = Long.parseLong(map_districts.get("countryId").toString()); | 
|             Long townId = Long.parseLong(map_districts.get("townId").toString()); | 
|             prDivide.setCountyid(countryId); | 
|             prDivide.setTownid(townId); | 
|         } | 
|   | 
|         Date operateTime = new Date(); | 
|         prDivide.setOperatedt(operateTime); | 
|         prDivide.setDeleted((byte)0); | 
|         Integer rec = Optional.ofNullable(divideSv.addDivide(prDivide)).orElse(0); | 
|         if(rec == 0) { | 
|             return BaseResponseUtils.buildErrorMsg(ProjectResultCode.DIVIDE_FAIL.getMessage()); | 
|         } | 
|         return BaseResponseUtils.buildSuccess(true) ; | 
|     } | 
|   | 
|     /** | 
|      * 根据分水房ID逻辑删除分水房 | 
|      * @param map | 
|      * @return | 
|      */ | 
|     @Operation(summary = "删除一个分水房", description = "删除一个分水房数据") | 
|     @ApiResponses(value = { | 
|             @ApiResponse( | 
|                     responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, | 
|                     description = "返回一页农户数据(BaseResponse.content:QueryResultVo[{}])", | 
|                     content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, | 
|                             schema = @Schema(implementation = BaClient.class))} | 
|             ) | 
|     }) | 
|   | 
|     //@GetMapping(path = "/delone/{id}") | 
|     @PostMapping(path = "delete", consumes = MediaType.APPLICATION_JSON_VALUE) | 
|     @SsoAop() | 
|     public BaseResponse<Boolean> deleteDivideById(@RequestBody Map map){ | 
|         Long id = Optional.ofNullable(Long.parseLong(map.get("id").toString())).orElse(0L); | 
|         if(id == 0) | 
|             return BaseResponseUtils.buildException(ProjectResultCode.DELETE_DIVIDE_FAIL.getMessage()); | 
|   | 
|         try { | 
|             Integer res = Optional.ofNullable(divideSv.deleteDivideById(id)).orElse(0); | 
|             if(res == 0) | 
|                 return BaseResponseUtils.buildErrorMsg(ProjectResultCode.DELETE_DIVIDE_FAIL.getMessage()); | 
|   | 
|             return BaseResponseUtils.buildSuccess(true); | 
|         } catch (Exception e) { | 
|             log.error("删除分水房异常", e); | 
|             return BaseResponseUtils.buildException(e.getMessage()) ; | 
|         } | 
|     } | 
|   | 
|     /** | 
|      * 修改分水房 | 
|      * @param po | 
|      * @param bindingResult | 
|      * @return | 
|      */ | 
|     @Operation(summary = "修改分水房记录", description = "修改分水房记录") | 
|     @ApiResponses(value = { | 
|             @ApiResponse( | 
|                     responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, | 
|                     description = "操作结果:true:成功,false:失败(BaseResponse.content)", | 
|                     content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, | 
|                             schema = @Schema(implementation = Boolean.class))} | 
|             ) | 
|     }) | 
|     @PostMapping(path = "update", consumes = MediaType.APPLICATION_JSON_VALUE) | 
|     @Transactional(rollbackFor = Exception.class) | 
|     @SsoAop() | 
|     public BaseResponse<Boolean> update(@RequestBody @Valid DtoDivide po, BindingResult bindingResult){ | 
|         if(bindingResult != null && bindingResult.hasErrors()){ | 
|             return BaseResponseUtils.buildErrorMsg(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage()); | 
|         } | 
|   | 
|         // 接收村编号(主键) | 
|         Long villageId = po.getVillageId(); | 
|   | 
|         PrDivide prDivide = DtoToDividePojo.INSTANCT.po2vo(po); | 
|   | 
|         if (villageId!=null){ | 
|             /** | 
|              * 获取5级行政区划信息 | 
|              */ | 
|             Map map_districts = Optional.ofNullable(divideSv.getDistrictsByVillageId(villageId)).orElse(new HashMap()); | 
|             if(map_districts.size() <= 0) { | 
|                 return BaseResponseUtils.buildErrorMsg("区划信息有误"); | 
|             } | 
|             Long countryId = Long.parseLong(map_districts.get("countryId").toString()); | 
|             Long townId = Long.parseLong(map_districts.get("townId").toString()); | 
|             prDivide.setCountyid(countryId); | 
|             prDivide.setTownid(townId); | 
|         } | 
|   | 
|         Date operateTime = new Date(); | 
|         prDivide.setOperatedt(operateTime); | 
|   | 
|         Integer rec = Optional.ofNullable(divideSv.updateByPrimaryKey(prDivide)).orElse(0); | 
|         if(rec == 0) { | 
|             return BaseResponseUtils.buildErrorMsg("分水房修改失败"); | 
|         } | 
|         return BaseResponseUtils.buildSuccess(true) ; | 
|     } | 
|   | 
|     /** | 
|      * 导出分水房列表 | 
|      * @param response | 
|      * @param vo | 
|      */ | 
|     @SneakyThrows(IOException.class) | 
|     @RequestMapping(value = "/export", method = RequestMethod.GET) | 
|     public void export(HttpServletResponse response, QueryVo vo) { | 
|         setExcelRespProp(response, "分水房列表"); | 
|         //List<VoDivide> memberList = LocalJsonUtil.getListFromJson("json/members.json", VoDivide.class); | 
|         List<VoDivide> memberList = divideSv.exportDivides(vo); | 
|   | 
|         EasyExcel.write(response.getOutputStream()) | 
|                 .head(VoDivide.class) | 
|                 .excelType(ExcelTypeEnum.XLSX) | 
|                 .sheet("分水房列表") | 
|                 .doWrite(memberList); | 
|     } | 
|   | 
|     @SneakyThrows | 
|     @RequestMapping(value = "/importDivides", method = RequestMethod.POST) | 
|     @ResponseBody | 
|     public BaseResponse<Boolean> importDivides(@RequestPart("file") MultipartFile file) { | 
|         List<VoDivide> memberList = EasyExcel.read(file.getInputStream()) | 
|                 .head(VoDivide.class) | 
|                 .sheet() | 
|                 .doReadSync(); | 
|         return BaseResponseUtils.buildSuccess(memberList) ; | 
|     } | 
|   | 
|     /** | 
|      * 设置excel下载响应头属性 | 
|      */ | 
|     private void setExcelRespProp(HttpServletResponse response, String rawFileName) throws UnsupportedEncodingException { | 
|         response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); | 
|         response.setCharacterEncoding("utf-8"); | 
|         String fileName = URLEncoder.encode(rawFileName, "UTF-8").replaceAll("\\+", "%20"); | 
|         response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx"); | 
|     } | 
| } |