package com.gunshi.project.hsz.controller; import cn.hutool.core.collection.CollectionUtil; import com.alibaba.excel.EasyExcel; import com.alibaba.excel.write.metadata.style.WriteCellStyle; import com.alibaba.excel.write.style.HorizontalCellStyleStrategy; import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.gunshi.core.result.R; import com.gunshi.project.hsz.entity.so.GateHisPageSo; import com.gunshi.project.hsz.entity.vo.GateStautsVo; import com.gunshi.project.hsz.entity.vo.GateValveOplogVo; import com.gunshi.project.hsz.model.GateValveKey; import com.gunshi.project.hsz.model.GateValveOplog; import com.gunshi.project.hsz.model.GateValveReal; import com.gunshi.project.hsz.service.GateValveRealService; import com.gunshi.project.hsz.validate.markers.Insert; import com.gunshi.project.hsz.validate.markers.Update; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.media.Schema; import io.swagger.v3.oas.annotations.tags.Tag; import jakarta.servlet.http.HttpServletResponse; import org.apache.poi.ss.usermodel.FillPatternType; import org.apache.poi.ss.usermodel.IndexedColors; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import java.io.Serializable; import java.math.BigDecimal; import java.net.URLEncoder; import java.util.*; import java.util.stream.Collectors; /** * 描述: 闸阀开关表 * author: xusan * date: 2024-07-08 17:40:37 */ @Tag(name = "闸阀开关表") @RestController @RequestMapping(value="/gateValveReal") public class GateValveRealController { private static final Logger log = LoggerFactory.getLogger(GateValveRealController.class); @Autowired private GateValveRealService service; @Operation(summary = "新增") @PostMapping("/insert") public R insert(@Validated(Insert.class) @RequestBody GateValveReal dto) { boolean result = service.save(dto); return R.ok(result ? dto : null); } @Operation(summary = "修改") @PostMapping("/update") public R update(@Validated(Update.class) @RequestBody GateValveReal dto) { boolean result = service.updateById(dto); return R.ok(result ? dto : null); } @Operation(summary = "删除") @GetMapping("/del/{id}") public R del(@Schema(name = "id") @PathVariable("id") Serializable id) { return R.ok(service.removeById(id)); } @Operation(summary = "列表") @PostMapping("/list") public R> list() { return R.ok(service.gateStatusList()); } @Operation(summary = "通过闸阀查询实时流量") @GetMapping("/realQ") public R realQ(@RequestParam("valveCode") @Parameter(description = "闸阀编码") String valveCode) { return R.ok(service.realQ(valveCode)); } @Operation(summary = "调节闸阀") @PostMapping("/control") public R control(@RequestBody GateValveKey gateValveKey) { return R.ok(service.control(gateValveKey)); } @Operation(summary = "闸阀操作日志-分页") @PostMapping("/log/page") public R> logPage(@RequestBody GateHisPageSo so) { return R.ok(service.logPage(so)); } @Operation(summary = "闸阀操作日志-导出") @PostMapping("/log/exp") public void logexp(@RequestBody GateHisPageSo so, HttpServletResponse response) { so.getPageSo().setPageSize(1000000); Page gateValveOplogVoPage = service.logPage(so); try { // 设置响应 response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); response.setCharacterEncoding("utf-8"); String fileName = URLEncoder.encode("闸阀操作日志_" + System.currentTimeMillis(), "UTF-8") .replaceAll("\\+", "%20"); response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx"); // 自定义表头样式 WriteCellStyle headStyle = new WriteCellStyle(); headStyle.setFillForegroundColor(IndexedColors.SKY_BLUE.getIndex()); headStyle.setFillPatternType(FillPatternType.SOLID_FOREGROUND); // 自定义内容样式 WriteCellStyle contentStyle = new WriteCellStyle(); contentStyle.setWrapped(true); // 自动换行 List> headlist= Arrays.asList( Arrays.asList("闸阀名称"), Arrays.asList("操作人"), Arrays.asList("操作时间"), Arrays.asList("操作内容"), Arrays.asList("设定开度"), Arrays.asList("操作前开度") ); // 构建导出器 EasyExcel.write(response.getOutputStream()) .head(headlist) // 自定义表头 .registerWriteHandler(new HorizontalCellStyleStrategy(headStyle, contentStyle)) .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) // 自动列宽 .sheet("闸阀操作日志") .doWrite(gateValveOplogVoPage.getRecords()); }catch (Exception e){ log.error("导出异常", e); } } @Operation(summary = "闸阀操作日志-分页") @PostMapping("/log/loglist") public R> loglist(@RequestBody GateHisPageSo so) { List gateStautsVos = service.gateStatusList(); if(CollectionUtil.isEmpty(gateStautsVos)){ return R.ok(null); } List collect = gateStautsVos.stream().map(e -> { GateValveOplogVo vo = new GateValveOplogVo(); GateValveOplog loginfo = service.loginfo(e.getValveCode()); if(Objects.nonNull(loginfo)){ BeanUtils.copyProperties(loginfo, vo); } vo.setValveCode(e.getValveCode()); vo.setValveName(e.getValveName()); return vo; }).collect(Collectors.toList()); return R.ok(collect); } @Operation(summary = "闸阀操作日志-导出") @PostMapping("/log/export") public void logExport(@RequestBody GateHisPageSo so, HttpServletResponse response) { service.logExport(so,response); } @Operation(summary = "预计可供水时间") @GetMapping("/supply/time") public R> supplyTime(@RequestParam(value = "year",required = false) @Parameter(description = "年份") Integer year,@RequestParam(value = "month",required = false) @Parameter(description = "月份") Integer month) { return R.ok(service.supplyTime(year,month)); } @Operation(summary = "预测来水量") @GetMapping("/predict/water") public R predictWater(@RequestParam(value = "year") @Parameter(description = "年份") Integer year, @RequestParam(value = "month") @Parameter(description = "月份") Integer month) { return R.ok(service.predictWater(year,month)); } }