package com.doumee.cloud.admin; import com.doumee.api.BaseController; import com.doumee.config.annotation.CloudRequiredPermission; import com.doumee.core.annotation.excel.ExcelExporter; import com.doumee.core.annotation.pr.PreventRepeat; import com.doumee.core.model.ApiResponse; import com.doumee.core.model.PageData; import com.doumee.core.model.PageWrap; import com.doumee.core.utils.Constants; import com.doumee.dao.business.model.PlatformWaterGas; import com.doumee.dao.business.vo.PlatformWaterGasForExcelVO; import com.doumee.service.business.PlatformWaterGasService; import io.swagger.annotations.*; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.File; import java.util.ArrayList; import java.util.List; import java.util.Objects; /** * @author 江蹄蹄 * @date 2024/08/26 16:22 */ @Api(tags = "月台_用水用气信息记录表") @RestController @RequestMapping(Constants.CLOUD_SERVICE_URL_INDEX+"/business/platformWaterGas") public class PlatformWaterGasCloudController extends BaseController { @Autowired private PlatformWaterGasService platformWaterGasService; @PreventRepeat @ApiOperation("新建") @PostMapping("/create") @CloudRequiredPermission("business:platformwatergas:create") public ApiResponse create(@RequestBody PlatformWaterGas platformWaterGas,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { platformWaterGas.setLoginUserInfo(getLoginUser(token)); return ApiResponse.success(platformWaterGasService.create(platformWaterGas)); } @ApiOperation("根据ID删除") @GetMapping("/delete/{id}") @CloudRequiredPermission("business:platformwatergas:delete") public ApiResponse deleteById(@PathVariable Integer id,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { platformWaterGasService.deleteById(id,getLoginUser(token)); return ApiResponse.success(null); } @ApiOperation("批量删除") @GetMapping("/delete/batch") @CloudRequiredPermission("business:platformwatergas:delete") public ApiResponse deleteByIdInBatch(@RequestParam String ids,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { String [] idArray = ids.split(","); List idList = new ArrayList<>(); for (String id : idArray) { idList.add(Integer.valueOf(id)); } platformWaterGasService.deleteByIdInBatch(idList,getLoginUser(token)); return ApiResponse.success(null); } @ApiOperation("根据ID修改") @PostMapping("/updateById") @CloudRequiredPermission("business:platformwatergas:update") public ApiResponse updateById(@RequestBody PlatformWaterGas platformWaterGas,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { platformWaterGas.setLoginUserInfo(getLoginUser(token)); platformWaterGasService.updateById(platformWaterGas); return ApiResponse.success(null); } @ApiOperation("分页查询") @PostMapping("/page") @CloudRequiredPermission("business:platformwatergas:query") public ApiResponse> findPage (@RequestBody PageWrap pageWrap,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { return ApiResponse.success(platformWaterGasService.findPage(pageWrap)); } @ApiOperation("导出Excel") @PostMapping("/exportExcel") @CloudRequiredPermission("business:platformwatergas:exportExcel") public void exportExcel (@RequestBody PageWrap pageWrap, HttpServletResponse response,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { List waterGasPageData = platformWaterGasService.findPage(pageWrap).getRecords(); if(Objects.nonNull(pageWrap.getModel()) && Objects.nonNull(pageWrap.getModel().getType()) && !Constants.equalsInteger(pageWrap.getModel().getType(),Constants.TWO)){ List pageData = new ArrayList<>(); for (PlatformWaterGas platformWaterGas:waterGasPageData) { PlatformWaterGasForExcelVO platformWaterGasForExcelVO = new PlatformWaterGasForExcelVO(); BeanUtils.copyProperties(platformWaterGas,platformWaterGasForExcelVO); pageData.add(platformWaterGasForExcelVO); } ExcelExporter.build(PlatformWaterGasForExcelVO.class).export(pageData, Constants.equalsInteger(pageWrap.getModel().getType(),Constants.ZERO)?"月台_用水信息记录表":"月台_用气信息记录表", response); }else{ ExcelExporter.build(PlatformWaterGas.class).export(waterGasPageData, "月台_油耗信息记录表", response); } } @ApiOperation(value = "油耗信息导入") @PostMapping("/importExcel") @ApiImplicitParams({ @ApiImplicitParam(name = "file", value = "file", required = true, paramType = "query", dataType = "file", dataTypeClass = File.class), }) @CloudRequiredPermission("business:platformwatergas:exportExcel") public ApiResponse importExcel (@ApiParam(value = "file") MultipartFile file, @RequestHeader(Constants.HEADER_USER_TOKEN) String token) { return ApiResponse.success(platformWaterGasService.importBatch(file,this.getLoginUser(token))); } @ApiOperation("根据ID查询") @GetMapping("/{id}") @CloudRequiredPermission("business:platformwatergas:query") public ApiResponse findById(@PathVariable Integer id,@RequestHeader(Constants.HEADER_USER_TOKEN) String token) { return ApiResponse.success(platformWaterGasService.findById(id)); } }