| | |
| | | import com.alibaba.excel.event.AnalysisEventListener; |
| | | import com.alibaba.fastjson.JSONObject; |
| | | import com.panzhihua.common.exceptions.ServiceException; |
| | | import com.panzhihua.common.model.vos.LoginUserInfoVO; |
| | | import com.panzhihua.common.model.vos.R; |
| | | import com.panzhihua.common.model.vos.community.ComCvtServeExcelVO; |
| | | import com.panzhihua.common.model.vos.community.ComMngVillageServeExcelVO; |
| | | import com.panzhihua.common.service.community.CommunityService; |
| | | import com.panzhihua.common.utlis.Snowflake; |
| | | import lombok.extern.slf4j.Slf4j; |
| | | |
| | | import java.util.ArrayList; |
| | | import java.util.Date; |
| | | import java.util.List; |
| | | |
| | | /** |
| | |
| | | |
| | | private CommunityService communityService; |
| | | |
| | | private Long communityId; |
| | | private LoginUserInfoVO loginUserInfo ; |
| | | |
| | | public ComMngVillageServeExcelListen(CommunityService communityService, Long communityId) { |
| | | public ComMngVillageServeExcelListen(CommunityService communityService, LoginUserInfoVO loginUserInfo) { |
| | | this.communityService = communityService; |
| | | this.communityId = communityId; |
| | | this.loginUserInfo = loginUserInfo; |
| | | } |
| | | |
| | | |
| | |
| | | // 达到BATCH_COUNT了,需要去存储一次数据库,防止数据几万条数据在内存,容易OOM |
| | | if (list.size() >= BATCH_COUNT) { |
| | | log.info("excel导入数据【{}】", JSONObject.toJSONString(list)); |
| | | R r = communityService.listSaveVillageServeExcelVO(list, this.communityId); |
| | | list.forEach(villageServe->{ |
| | | villageServe.setVillageId(Snowflake.getId()); |
| | | villageServe.setCreateAt(new Date()); |
| | | }); |
| | | R r = communityService.listSaveVillageServeExcelVO(list, loginUserInfo.getCommunityId()); |
| | | if (!R.isOk(r)) { |
| | | throw new ServiceException(r.getMsg()); |
| | | } |
| | |
| | | @Override |
| | | public void doAfterAllAnalysed(AnalysisContext analysisContext) { log.info("excel导入数据【{}】", JSONObject.toJSONString(list)); |
| | | //确保最后遗留的数据保存在数据库中 |
| | | R r = this.communityService.listSaveVillageServeExcelVO(list, this.communityId); |
| | | list.forEach(villageServe->{ |
| | | villageServe.setVillageId(Snowflake.getId()); |
| | | villageServe.setCreateAt(new Date()); |
| | | }); |
| | | R r = this.communityService.listSaveVillageServeExcelVO(list, loginUserInfo.getCommunityId()); |
| | | if (!R.isOk(r)) { |
| | | throw new ServiceException(r.getMsg()); |
| | | } |