package com.zd.chemical; import com.zd.chemical.domain.HxpChemicalClassify; import com.zd.chemical.domain.HxpClassifyConfig; import com.zd.chemical.service.IHxpChemicalClassifyService; import com.zd.chemical.service.IHxpClassifyConfigService; import com.zd.common.core.annotation.EnableCustomConfig; import com.zd.common.core.annotation.EnableZdFeignClients; import com.zd.common.core.launch.ZdStartApplication; import com.zd.model.constant.ApplicationConstants; import com.zd.model.constant.BaseConstants; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.ApplicationArguments; import org.springframework.boot.ApplicationRunner; import org.springframework.boot.autoconfigure.SpringBootApplication; import org.springframework.context.annotation.ComponentScan; import org.springframework.scheduling.annotation.EnableScheduling; import java.util.Collections; import java.util.List; import java.util.Optional; @EnableCustomConfig @EnableZdFeignClients @SpringBootApplication @EnableScheduling @ComponentScan(basePackages = BaseConstants.BASE_PACKAGE) public class ZdChemicalApplication implements ApplicationRunner { @Autowired private IHxpChemicalClassifyService hxpChemicalClassifyService; @Autowired private IHxpClassifyConfigService hxpClassifyConfigService; public static void main(String[] args) { ZdStartApplication.run(ApplicationConstants.CHEMICAL_SERVICE, ZdChemicalApplication.class, args); } @Override public void run(ApplicationArguments args) throws Exception { HxpChemicalClassify hxpChemicalClassify = new HxpChemicalClassify(); List hxpChemicalClassifies = hxpChemicalClassifyService.selectHxpChemicalClassifyList(hxpChemicalClassify); Optional.ofNullable(hxpChemicalClassifies).orElseGet(Collections::emptyList).stream() .forEach(a->{ HxpClassifyConfig hxpClassifyConfigSel = new HxpClassifyConfig(); hxpClassifyConfigSel.setJoinClassifyId(Integer.parseInt(a.getId()+"")); List classifyConfigList = hxpClassifyConfigService.selectHxpClassifyConfigList(hxpClassifyConfigSel); if(classifyConfigList.isEmpty() && a.getHazardLevel()==1){ //初始化危险化学品 HxpClassifyConfig classifyConfig = new HxpClassifyConfig(); classifyConfig.setJoinClassifyId(Integer.parseInt(a.getId()+"")); classifyConfig.initDanger(); hxpClassifyConfigService.insertHxpClassifyConfig(classifyConfig); }else if(classifyConfigList.isEmpty() && a.getHazardLevel()==2){ //初始化普通化学品 HxpClassifyConfig classifyConfig = new HxpClassifyConfig(); classifyConfig.setJoinClassifyId(Integer.parseInt(a.getId()+"")); classifyConfig.initOrdinary(); hxpClassifyConfigService.insertHxpClassifyConfig(classifyConfig); } }); } }