zd-parents-v2/zd-modules/zd-chemical/src/main/java/com/zd/chemical/ZdChemicalApplication.java

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 <HxpChemicalClassify> hxpChemicalClassifies =  hxpChemicalClassifyService.selectHxpChemicalClassifyList(hxpChemicalClassify);
        Optional.ofNullable(hxpChemicalClassifies).orElseGet(Collections::emptyList).stream()
                .forEach(a->{
                    HxpClassifyConfig hxpClassifyConfigSel = new HxpClassifyConfig();
                    hxpClassifyConfigSel.setJoinClassifyId(Integer.parseInt(a.getId()+""));
                    List<HxpClassifyConfig> 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);
                    }
                });
    }
}