package com.skua.modules.ajh.controller; import java.util.Arrays; import java.util.List; import java.util.Map; import java.io.IOException; import java.io.UnsupportedEncodingException; import java.net.URLDecoder; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import com.skua.core.api.vo.Result; import com.skua.core.aspect.annotation.AutoLog; import com.skua.core.query.QueryGenerator; import com.skua.modules.ajh.entity.AjhEnvironmentYear; import com.skua.modules.ajh.service.IAjhEnvironmentYearService; import java.util.Date; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.jeecgframework.poi.excel.ExcelImportUtil; import org.jeecgframework.poi.excel.def.NormalExcelConstants; import org.jeecgframework.poi.excel.entity.ExportParams; import org.jeecgframework.poi.excel.entity.ImportParams; import org.jeecgframework.poi.excel.view.JeecgEntityExcelView; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import org.springframework.web.servlet.ModelAndView; import com.alibaba.fastjson.JSON; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; /** * 安质环保年度总结 */ @Slf4j @Api(tags="安质环保年度总结") @RestController @RequestMapping("/ajh/ajhEnvironmentYear") public class AjhEnvironmentYearController { @Autowired private IAjhEnvironmentYearService ajhEnvironmentYearService; /** * 分页列表查询 * @param ajhEnvironmentYear * @param pageNo * @param pageSize * @param req * @return */ @AutoLog(value = "安质环保年度总结-分页列表查询") @ApiOperation(value="安质环保年度总结-分页列表查询", notes="安质环保年度总结-分页列表查询") @GetMapping(value = "/list") public Result<IPage<AjhEnvironmentYear>> queryPageList(AjhEnvironmentYear ajhEnvironmentYear, @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, HttpServletRequest req) { Result<IPage<AjhEnvironmentYear>> result = new Result<IPage<AjhEnvironmentYear>>(); QueryWrapper<AjhEnvironmentYear> queryWrapper = QueryGenerator.initQueryWrapper(ajhEnvironmentYear, req.getParameterMap()); Page<AjhEnvironmentYear> page = new Page<AjhEnvironmentYear>(pageNo, pageSize); IPage<AjhEnvironmentYear> pageList = ajhEnvironmentYearService.page(page, queryWrapper); result.setSuccess(true); result.setResult(pageList); return result; } /** * 添加 * @param ajhEnvironmentYear * @return */ @AutoLog(value = "安质环保年度总结-添加") @ApiOperation(value="安质环保年度总结-添加", notes="安质环保年度总结-添加") @PostMapping(value = "/add") public Result<AjhEnvironmentYear> add(@RequestBody AjhEnvironmentYear ajhEnvironmentYear) { Result<AjhEnvironmentYear> result = new Result<AjhEnvironmentYear>(); try { ajhEnvironmentYearService.save(ajhEnvironmentYear); result.success("添加成功!"); } catch (Exception e) { log.error(e.getMessage(),e); result.error500("操作失败"); } return result; } /** * 编辑 * @param ajhEnvironmentYear * @return */ @AutoLog(value = "安质环保年度总结-编辑") @ApiOperation(value="安质环保年度总结-编辑", notes="安质环保年度总结-编辑") @PutMapping(value = "/edit") public Result<AjhEnvironmentYear> edit(@RequestBody AjhEnvironmentYear ajhEnvironmentYear) { Result<AjhEnvironmentYear> result = new Result<AjhEnvironmentYear>(); AjhEnvironmentYear ajhEnvironmentYearEntity = ajhEnvironmentYearService.getById(ajhEnvironmentYear.getId()); if(ajhEnvironmentYearEntity==null) { result.error500("未找到对应实体"); }else { boolean ok = ajhEnvironmentYearService.updateById(ajhEnvironmentYear); //TODO 返回false说明什么? if(ok) { result.success("修改成功!"); } } return result; } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "安质环保年度总结-通过id删除") @ApiOperation(value="安质环保年度总结-通过id删除", notes="安质环保年度总结-通过id删除") @DeleteMapping(value = "/delete") public Result<?> delete(@RequestParam(name="id",required=true) String id) { try { ajhEnvironmentYearService.removeById(id); } catch (Exception e) { log.error("删除失败",e.getMessage()); return Result.error("删除失败!"); } return Result.ok("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "安质环保年度总结-批量删除") @ApiOperation(value="安质环保年度总结-批量删除", notes="安质环保年度总结-批量删除") @DeleteMapping(value = "/deleteBatch") public Result<AjhEnvironmentYear> deleteBatch(@RequestParam(name="ids",required=true) String ids) { Result<AjhEnvironmentYear> result = new Result<AjhEnvironmentYear>(); if(ids==null || "".equals(ids.trim())) { result.error500("参数不识别!"); }else { this.ajhEnvironmentYearService.removeByIds(Arrays.asList(ids.split(","))); result.success("删除成功!"); } return result; } /** * 通过id查询 * @param id * @return */ @AutoLog(value = "安质环保年度总结-通过id查询") @ApiOperation(value="安质环保年度总结-通过id查询", notes="安质环保年度总结-通过id查询") @GetMapping(value = "/queryById") public Result<AjhEnvironmentYear> queryById(@RequestParam(name="id",required=true) String id) { Result<AjhEnvironmentYear> result = new Result<AjhEnvironmentYear>(); AjhEnvironmentYear ajhEnvironmentYear = ajhEnvironmentYearService.getById(id); if(ajhEnvironmentYear==null) { result.error500("未找到对应实体"); }else { result.setResult(ajhEnvironmentYear); result.setSuccess(true); } return result; } /** * 导出excel * * @param request * @param response */ @RequestMapping(value = "/exportXls") public ModelAndView exportXls(HttpServletRequest request, HttpServletResponse response) { // Step.1 组装查询条件 QueryWrapper<AjhEnvironmentYear> queryWrapper = null; try { String paramsStr = request.getParameter("paramsStr"); if (StringUtils.isNotEmpty(paramsStr)) { String deString = URLDecoder.decode(paramsStr, "UTF-8"); AjhEnvironmentYear ajhEnvironmentYear = JSON.parseObject(deString, AjhEnvironmentYear.class); queryWrapper = QueryGenerator.initQueryWrapper(ajhEnvironmentYear, request.getParameterMap()); } } catch (UnsupportedEncodingException e) { e.printStackTrace(); } //Step.2 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); List<AjhEnvironmentYear> pageList = ajhEnvironmentYearService.list(queryWrapper); //导出文件名称 mv.addObject(NormalExcelConstants.FILE_NAME, "安质环保年度总结列表"); mv.addObject(NormalExcelConstants.CLASS, AjhEnvironmentYear.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("安质环保年度总结列表数据", "导出人:Jeecg", "导出信息")); mv.addObject(NormalExcelConstants.DATA_LIST, pageList); return mv; } /** * 通过excel导入数据 * * @param request * @param response * @return */ @RequestMapping(value = "/importExcel", method = RequestMethod.POST) public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; Map<String, MultipartFile> fileMap = multipartRequest.getFileMap(); for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) { MultipartFile file = entity.getValue();// 获取上传文件对象 ImportParams params = new ImportParams(); params.setTitleRows(2); params.setHeadRows(1); params.setNeedSave(true); try { List<AjhEnvironmentYear> listAjhEnvironmentYears = ExcelImportUtil.importExcel(file.getInputStream(), AjhEnvironmentYear.class, params); ajhEnvironmentYearService.saveBatch(listAjhEnvironmentYears); return Result.ok("文件导入成功!数据行数:" + listAjhEnvironmentYears.size()); } catch (Exception e) { log.error(e.getMessage(),e); return Result.error("文件导入失败:"+e.getMessage()); } finally { try { file.getInputStream().close(); } catch (IOException e) { e.printStackTrace(); } } } return Result.ok("文件导入失败!"); } }