package com.dy.pipIrrProject.intake;
|
|
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.pojoPr.PrIntake;
|
import com.dy.pipIrrGlobal.voPr.VoIntake;
|
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.Parameter;
|
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.time.format.DateTimeFormatter;
|
import java.util.*;
|
|
/**
|
* @author WuZeYu
|
* @date 2023/12/26 20:11
|
* @LastEditTime 2023/12/20 20:11
|
* @Description
|
*/
|
|
@Slf4j
|
@Tag(name = "取水口管理", description = "取水口操作")
|
@RestController
|
@RequestMapping(path = "intake")
|
@RequiredArgsConstructor
|
public class IntakeCtrl {
|
private final IntakeSv intakeSv;
|
|
/**
|
* 客户端请求得到所有取水口名字
|
*
|
* @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 = PrIntake.class))}
|
)
|
})
|
@GetMapping(path = "all")
|
@SsoAop()
|
public BaseResponse<QueryResultVo<List<PrIntake>>> all() {
|
try {
|
QueryResultVo<List<PrIntake>> res = this.intakeSv.selectAll();
|
if (res == null) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.NO_INTAKES.getMessage());
|
} else {
|
return BaseResponseUtils.buildSuccess(res);
|
}
|
} catch (Exception e) {
|
log.error("查询取水口异常", e);
|
return BaseResponseUtils.buildException(e.getMessage());
|
}
|
}
|
|
/**
|
* 得到一个取水口数据
|
*
|
* @return 一个取水口数据
|
*/
|
@Operation(summary = "一个取水口", description = "得到一个取水口数据")
|
@ApiResponses(value = {
|
@ApiResponse(
|
responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE,
|
description = "返回一个取水口数据（BaseResponse.content:{}）",
|
content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE,
|
schema = @Schema(implementation = PrIntake.class))}
|
)
|
})
|
@GetMapping(path = "one/{id}")
|
@SsoAop()
|
public BaseResponse<PrIntake> one(@PathVariable("id") Long id) {
|
if (this.intakeSv.selectById(id) == null) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.NO_INTAKES.getMessage());
|
} else {
|
return BaseResponseUtils.buildSuccess(this.intakeSv.selectById(id));
|
}
|
}
|
|
@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 = "getIntakes")
|
@SsoAop()
|
public BaseResponse<QueryResultVo<List<VoIntake>>> getIntakes(QueryVo vo) {
|
try {
|
QueryResultVo<List<VoIntake>> res = intakeSv.getIntakes(vo);
|
if (res == null) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.NO_INTAKES.getMessage());
|
}
|
return BaseResponseUtils.buildSuccess(res);
|
} catch (Exception e) {
|
log.error("获取取水口记录异常", e);
|
return BaseResponseUtils.buildException(e.getMessage());
|
}
|
}
|
|
@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 PrIntake po, @Parameter(hidden = true) BindingResult bindingResult) {
|
DateTimeFormatter dtf = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
|
if (bindingResult != null && bindingResult.hasErrors()) {
|
return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
|
}
|
|
Date operateTime = new Date();
|
po.setOperateDt(operateTime);
|
po.setDeleted((byte)0);
|
Integer rec = Optional.ofNullable(intakeSv.addIntake(po)).orElse(0);
|
if (rec == 0) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.ADD_INTAKE_FAIL.getMessage());
|
}
|
return BaseResponseUtils.buildSuccess(true);
|
}
|
|
/**
|
* 编辑修改取水口
|
*
|
* @param po 保存取水口form表单对象
|
* @return 是否成功
|
*/
|
@Operation(summary = "编辑修改取水口", description = "提交取水口数据（form表单），进行修改")
|
@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)
|
@SsoAop()
|
public BaseResponse<Boolean> update(@RequestBody @Parameter(description = "form表单json数据", required = true) @Valid PrIntake po, @Parameter(hidden = true) BindingResult bindingResult) {
|
if (bindingResult != null && bindingResult.hasErrors()) {
|
return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
|
}
|
int count;
|
Date operateTime = new Date();
|
po.setOperateDt(operateTime);
|
try {
|
count = this.intakeSv.update(po);
|
} catch (Exception e) {
|
log.error("保存分水口异常", e);
|
return BaseResponseUtils.buildException(e.getMessage());
|
}
|
if (count <= 0) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.UPDATE_INTAKE.getMessage());
|
} else {
|
return BaseResponseUtils.buildSuccess(true);
|
}
|
}
|
|
/**
|
* 删除取水口
|
*
|
* @param map 取水口ID
|
* @return 是否成功
|
*/
|
@Operation(summary = "删除分水口", description = "提交取水口ID，进行逻辑删除")
|
@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 = "delete")
|
@SsoAop()
|
public BaseResponse<Boolean> delete(@RequestBody Map map) {
|
if (map == null || map.size() <= 0) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.PLEASE_INPUT_INTAKE_ID.getMessage());
|
}
|
Long id = Long.parseLong(map.get("id").toString());
|
try {
|
//取水口ID
|
Integer recordCount = Optional.ofNullable(intakeSv.delete(id)).orElse(0);
|
if (recordCount == 0) {
|
return BaseResponseUtils.buildFail(ProjectResultCode.DELETE_INTAKE_FAIL.getMessage());
|
} else {
|
return BaseResponseUtils.buildSuccess(true);
|
}
|
} catch (Exception e) {
|
log.error("保存分水口异常", e);
|
return BaseResponseUtils.buildException(e.getMessage());
|
}
|
|
|
}
|
|
/**
|
* 导出取水口列表
|
*
|
* @param response
|
* @param vo
|
*/
|
@SneakyThrows(IOException.class)
|
@GetMapping(value = "exportIntakes")
|
public void exportIntakes(HttpServletResponse response, QueryVo vo) {
|
setExcelRespProp(response, "取水口列表");
|
//List<VoDivide> memberList = LocalJsonUtil.getListFromJson("json/members.json", VoDivide.class);
|
List<VoIntake> memberList = intakeSv.exportIntakes(vo);
|
|
EasyExcel.write(response.getOutputStream())
|
.head(VoIntake.class)
|
.excelType(ExcelTypeEnum.XLSX)
|
.sheet("取水口列表")
|
.doWrite(memberList);
|
}
|
|
/**
|
* 导入取水口列表
|
*
|
* @param file
|
*/
|
@SneakyThrows
|
@PostMapping(value = "importIntakes")
|
@ResponseBody
|
public BaseResponse<Boolean> importIntakes(@RequestPart("file") MultipartFile file) {
|
List<VoIntake> memberList = EasyExcel.read(file.getInputStream())
|
.head(VoIntake.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");
|
}
|
}
|
