package com.skua.modules.safe.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.context.BaseContextHandler; import com.skua.core.query.QueryGenerator; import com.skua.core.service.ISequenceService; import com.skua.core.util.ConvertUtils; import com.skua.modules.safe.entity.EpibolyCompany; import com.skua.modules.safe.service.IEpibolyCompanyService; 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 com.skua.modules.safe.vo.EpibolyCompanyVO; import lombok.extern.slf4j.Slf4j; import org.apache.shiro.SecurityUtils; 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; /** * <pre> * 外协单位表 * </pre> * @author 开发者姓名 * @version V0.1, 开发时间 */ @Slf4j @Api(tags="外协单位表") @RestController("webEpibolyCompanyController") @RequestMapping("/safe/epiboly/epibolyCompany") public class EpibolyCompanyController { @Autowired private IEpibolyCompanyService epibolyCompanyService; /** * 根据部门编号获取,部门下的所有外协单位集合 * * @return */ @AutoLog(value = "根据部门编号获取,部门下的所有外协单位集合") @ApiOperation(value="外协单位表-列表查询", notes="外协单位表-列表查询") @RequestMapping(value = "/queryList", method = RequestMethod.GET) public Result<List<EpibolyCompany>> queryTreeList(String departId) { Result<List<EpibolyCompany>> result = new Result<>(); try { List<EpibolyCompany> list = epibolyCompanyService.queryListByDepartId(departId); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * <pre> * 分页列表查询 * </pre> * @param epibolyCompany * @param pageNo * @param pageSize * @param req * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @AutoLog(value = "外协单位表-分页列表查询") @ApiOperation(value="外协单位表-分页列表查询", notes="外协单位表-分页列表查询") @GetMapping(value = "/list") public Result<IPage<EpibolyCompany>> queryPageList(EpibolyCompany epibolyCompany, @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, HttpServletRequest req) { Result<IPage<EpibolyCompany>> result = new Result<IPage<EpibolyCompany>>(); QueryWrapper<EpibolyCompany> queryWrapper = QueryGenerator.initQueryWrapper(epibolyCompany, req.getParameterMap()); Page<EpibolyCompany> page = new Page<EpibolyCompany>(pageNo, pageSize); queryWrapper.eq("depart_id",epibolyCompany.getDepartId()); IPage<EpibolyCompany> pageList = epibolyCompanyService.page(page, queryWrapper); result.setSuccess(true); result.setResult(pageList); return result; } /** * <pre> * 添加 * </pre> * @param epibolyCompany * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @AutoLog(value = "外协单位表-添加") @ApiOperation(value="外协单位表-添加", notes="外协单位表-添加") @PostMapping(value = "/add") public Result<EpibolyCompany> add(@RequestBody EpibolyCompany epibolyCompany) { Result<EpibolyCompany> result = new Result<EpibolyCompany>(); try { epibolyCompany.setCreateBy(BaseContextHandler.getUserName()); epibolyCompany.setCreateDept( BaseContextHandler.getRealDepartId() ); epibolyCompany.setCreateTime(new Date()); epibolyCompanyService.save(epibolyCompany); result.setResult( epibolyCompany); result.success("添加成功!"); } catch (Exception e) { log.error(e.getMessage(),e); result.error500("操作失败"); } return result; } /** * <pre> * 编辑 * </pre> * @param epibolyCompany * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @AutoLog(value = "外协单位表-编辑") @ApiOperation(value="外协单位表-编辑", notes="外协单位表-编辑") @PutMapping(value = "/edit") public Result<EpibolyCompany> edit(@RequestBody EpibolyCompany epibolyCompany) { Result<EpibolyCompany> result = new Result<EpibolyCompany>(); EpibolyCompany epibolyCompanyEntity = epibolyCompanyService.getById(epibolyCompany.getId()); if(epibolyCompanyEntity==null) { result.error500("未找到对应实体"); }else { epibolyCompany.setUpdateBy(BaseContextHandler.getUserName()); epibolyCompany.setUpdateTime(new Date()); boolean ok = epibolyCompanyService.updateById(epibolyCompany); result.setResult( epibolyCompany ); //TODO 返回false说明什么? if(ok) { result.success("修改成功!"); } } return result; } /** * <pre> * 通过id删除 * </pre> * @param id * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @AutoLog(value = "外协单位表-通过id删除") @ApiOperation(value="外协单位表-通过id删除", notes="外协单位表-通过id删除") @DeleteMapping(value = "/delete") public Result<?> delete(@RequestParam(name="id",required=true) String id) { try { epibolyCompanyService.delById(id); } catch (Exception e) { log.error("删除失败",e.getMessage()); return Result.error("删除失败!"); } return Result.ok("删除成功!"); } /** * <pre> * 批量删除 * </pre> * @param ids * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @AutoLog(value = "外协单位表-批量删除") @ApiOperation(value="外协单位表-批量删除", notes="外协单位表-批量删除") @DeleteMapping(value = "/deleteBatch") public Result<EpibolyCompany> deleteBatch(@RequestParam(name="ids",required=true) String ids) { Result<EpibolyCompany> result = new Result<EpibolyCompany>(); if(ids==null || "".equals(ids.trim())) { result.error500("参数不识别!"); }else { this.epibolyCompanyService.removeByIds(Arrays.asList(ids.split(","))); result.success("删除成功!"); } return result; } /** * <pre> * 通过id查询 * </pre> * @param id * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @AutoLog(value = "外协单位表-通过id查询") @ApiOperation(value="外协单位表-通过id查询", notes="外协单位表-通过id查询") @GetMapping(value = "/queryById") public Result<EpibolyCompanyVO> queryById(@RequestParam(name="id",required=true) String id) { Result<EpibolyCompanyVO> result = new Result<EpibolyCompanyVO>(); EpibolyCompanyVO epibolyCompany = epibolyCompanyService.queryByCompanyId(id); if(epibolyCompany==null) { result.error500("未找到对应实体"); }else { result.setResult(epibolyCompany); result.setSuccess(true); } return result; } /** * <pre> * 导出excel * </pre> * @param request * @param response * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @RequestMapping(value = "/exportXls") public ModelAndView exportXls(HttpServletRequest request, HttpServletResponse response) { // Step.1 组装查询条件 QueryWrapper<EpibolyCompany> queryWrapper = null; try { String paramsStr = request.getParameter("paramsStr"); if (ConvertUtils.isNotEmpty(paramsStr)) { String deString = URLDecoder.decode(paramsStr, "UTF-8"); EpibolyCompany epibolyCompany = JSON.parseObject(deString, EpibolyCompany.class); queryWrapper = QueryGenerator.initQueryWrapper(epibolyCompany, request.getParameterMap()); } } catch (UnsupportedEncodingException e) { e.printStackTrace(); } //Step.2 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); List<EpibolyCompany> pageList = epibolyCompanyService.list(queryWrapper); //导出文件名称 mv.addObject(NormalExcelConstants.FILE_NAME, "外协单位表列表"); mv.addObject(NormalExcelConstants.CLASS, EpibolyCompany.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("外协单位表列表数据", "导出人:Jeecg", "导出信息")); mv.addObject(NormalExcelConstants.DATA_LIST, pageList); return mv; } /** * <pre> * 通过excel导入数据 * </pre> * @param request * @param response * @return * @author 开发者姓名, 开发时间 * @Description: TODO(这里描述这个方法的需求变更情况) */ @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<EpibolyCompany> listEpibolyCompanys = ExcelImportUtil.importExcel(file.getInputStream(), EpibolyCompany.class, params); epibolyCompanyService.saveBatch(listEpibolyCompanys); return Result.ok("文件导入成功!数据行数:" + listEpibolyCompanys.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("文件导入失败!"); } }