| | |
| | | package com.consum.base.controller; |
| | | |
| | | import java.io.IOException; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | |
| | | import org.springframework.web.bind.annotation.PostMapping; |
| | | import org.springframework.web.bind.annotation.RequestMapping; |
| | | import org.springframework.web.bind.annotation.RestController; |
| | | import org.springframework.web.multipart.MultipartFile; |
| | | |
| | | import com.alibaba.excel.EasyExcel; |
| | | import com.alibaba.excel.context.AnalysisContext; |
| | | import com.alibaba.excel.event.AnalysisEventListener; |
| | | import com.alibaba.excel.exception.ExcelDataConvertException; |
| | | import com.consum.base.BaseController; |
| | | import com.consum.base.core.utils.CommonUtil; |
| | | import com.consum.base.core.utils.MapUtils; |
| | | import com.consum.base.core.utils.MapperUtil; |
| | | import com.consum.base.pojo.BaseGoodsTemplateParam; |
| | | import com.consum.base.pojo.excel.ImportGoodsInfoTemplate; |
| | | import com.consum.base.pojo.response.GoodsTemplateVO; |
| | | import com.consum.base.service.BaseGoodsTemplateService; |
| | | import com.consum.base.service.BaseWarehouseService; |
| | | import com.consum.base.service.FinSysTenantService; |
| | | import com.consum.base.service.LWhGoodsService; |
| | | import com.consum.model.po.BaseGoodsModels; |
| | | import com.consum.model.po.BaseGoodsTemplate; |
| | | import com.consum.model.po.BaseWarehouse; |
| | | import com.consum.model.po.FinSysTenant; |
| | | import com.consum.model.po.FinSysTenantUser; |
| | | import com.consum.model.po.SDictData; |
| | | import com.consum.model.vo.BaseGoodsTemplateVo; |
| | | import com.walker.db.page.GenericPager; |
| | | import com.walker.infrastructure.utils.DateUtils; |
| | | import com.walker.infrastructure.utils.StringUtils; |
| | | import com.walker.web.ResponseValue; |
| | | |
| | |
| | | |
| | | @Autowired |
| | | private BaseWarehouseService baseWarehouseService; |
| | | |
| | | @Autowired |
| | | private FinSysTenantService finSysTenantService; |
| | | |
| | | /** |
| | | * @Description 新增物品模板 |
| | |
| | | return ResponseValue.success("查询成功!", resultList); |
| | | } |
| | | |
| | | @PostMapping("/import") |
| | | public ResponseValue upload(MultipartFile file) throws IOException { |
| | | String originalFilename = file.getOriginalFilename(); |
| | | if (!originalFilename.endsWith(".xls")) { |
| | | return ResponseValue.error("文件格式有误!"); |
| | | } |
| | | FinSysTenantUser sysInfo = this.getSysInfo(); |
| | | if (sysInfo == null) { |
| | | return ResponseValue.error("当前登录用户为空"); |
| | | } |
| | | EasyExcel.read(file.getInputStream(), ImportGoodsInfoTemplate.class, |
| | | new AnalysisEventListener<ImportGoodsInfoTemplate>() { |
| | | List<BaseGoodsTemplateParam> list = Lists.newArrayList(); |
| | | |
| | | @Override |
| | | public void invoke(ImportGoodsInfoTemplate data, AnalysisContext analysisContext) { |
| | | String categoryOne = data.getCategoryOne(); |
| | | String categoryTwo = data.getCategoryTwo(); |
| | | String categoryThree = data.getCategoryThree(); |
| | | String goodName = data.getGoodName(); |
| | | String goodModel = data.getGoodModel(); |
| | | String unit = data.getUnit(); |
| | | String type = data.getType(); |
| | | String agencyName = data.getAgencyName(); |
| | | |
| | | list.stream().forEach(item -> { |
| | | if (!item.getCategoryId().equals(categoryThree)) { |
| | | BaseGoodsTemplateParam baseGoodsTemplate = new BaseGoodsTemplateParam(); |
| | | baseGoodsTemplate.setCategoryId(Long.valueOf(categoryThree)); |
| | | baseGoodsTemplate.setGoodsName(goodName); |
| | | baseGoodsTemplate.setStates(1); |
| | | FinSysTenant finSysTenant = finSysTenantService.selectByName(agencyName); |
| | | baseGoodsTemplate.setAgencyId(finSysTenant.getId()); |
| | | |
| | | List<BaseGoodsModels> models = Lists.newArrayList(); |
| | | BaseGoodsModels baseGoodsModels = new BaseGoodsModels(); |
| | | baseGoodsModels.setModelName(goodModel); |
| | | baseGoodsModels.setStates(1); |
| | | baseGoodsModels.setGoodsTemplatesId(item.getId()); |
| | | baseGoodsModels.setUnit(unit); |
| | | baseGoodsModels.setCreateTime(DateUtils.getDateTimeNumber(System.currentTimeMillis())); |
| | | models.add(baseGoodsModels); |
| | | baseGoodsTemplate.setModels(models); |
| | | } else { |
| | | List<BaseGoodsModels> models = item.getModels(); |
| | | BaseGoodsModels baseGoodsModels = new BaseGoodsModels(); |
| | | baseGoodsModels.setModelName(goodModel); |
| | | baseGoodsModels.setStates(1); |
| | | baseGoodsModels.setGoodsTemplatesId(item.getId()); |
| | | baseGoodsModels.setUnit(unit); |
| | | baseGoodsModels.setCreateTime(DateUtils.getDateTimeNumber(System.currentTimeMillis())); |
| | | models.add(baseGoodsModels); |
| | | } |
| | | }); |
| | | |
| | | } |
| | | |
| | | @Override |
| | | public void doAfterAllAnalysed(AnalysisContext analysisContext) { |
| | | for (BaseGoodsTemplateParam a : list) { |
| | | baseGoodsTemplateService.add(a, sysInfo); |
| | | } |
| | | } |
| | | |
| | | @Override |
| | | public void onException(Exception exception, AnalysisContext analysisContext) throws Exception { |
| | | if (exception instanceof ExcelDataConvertException) { |
| | | ExcelDataConvertException excelDataConvertException = (ExcelDataConvertException)exception; |
| | | Integer row = excelDataConvertException.getRowIndex() + 1; |
| | | Integer column = excelDataConvertException.getColumnIndex() + 1; |
| | | throw new RuntimeException("第" + row + "行,第" + column + "列解析异常,请正确填写"); |
| | | } |
| | | } |
| | | }).sheet(0).doRead(); |
| | | |
| | | return ResponseValue.success("导入成功!", 1); |
| | | } |
| | | } |