EquipmentApplication.java

package com.devplatform.equipment;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.devplatform.equipment.common.utils.Constants;
import com.devplatform.equipment.common.utils.SpringContextUtil;
import com.devplatform.equipment.common.utils.TestClient;
import com.devplatform.equipment.modules.ewarning.service.EwarningService;
import com.devplatform.equipment.modules.listation.bean.LiStation;
import com.devplatform.equipment.modules.listation.service.LiStationService;
import com.devplatform.equipment.modules.sys.bean.SysSystem;
import com.devplatform.equipment.modules.sys.service.SysSystemService;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.cloud.netflix.eureka.EnableEurekaClient;
import org.springframework.cloud.openfeign.EnableFeignClients;
import org.springframework.context.annotation.Import;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * springboot启动类
 *
 * @author Administrator
 */
@SpringBootApplication
@EnableSwagger2
@MapperScan(basePackages = "com.devplatform.**.dao")
@EnableEurekaClient
@EnableFeignClients
@EnableScheduling
@EnableTransactionManagement
@Import(SpringContextUtil.class)
public class EquipmentApplication {

    public static void main(String[] args) {
        SpringApplication.run(EquipmentApplication.class, args);
    }

    private static void handleMqInit() {
        LiStationService liStationService =
                (LiStationService) SpringContextUtil.getBean(LiStationService.class);
        SysSystemService sysSystemService =
                (SysSystemService) SpringContextUtil.getBean(SysSystemService.class);
        EwarningService ewarningService =
                (EwarningService) SpringContextUtil.getBean(EwarningService.class);

        SysSystem sysSystem = sysSystemService.getOne(new LambdaQueryWrapper<>());
        // 站点
        List<LiStation> lists = null;
        if (Constants.STRING_1.equals(sysSystem.getType())) {
            lists =
                    liStationService.list(
                            new LambdaQueryWrapper<LiStation>()
                                    .eq(LiStation::getDeleted, 0)
                                    .eq(LiStation::getId, sysSystem.getCode())
                                    .orderByAsc(LiStation::getOrderNum));
        } else if (Constants.STRING_2.equals(sysSystem.getType())) {
            //线路
            lists =
                    liStationService.list(
                            new LambdaQueryWrapper<LiStation>()
                                    .eq(LiStation::getDeleted, 0)
                                    .eq(LiStation::getLineId, sysSystem.getCode())
                                    .orderByAsc(LiStation::getOrderNum));
        } else {
            //路网
            lists =
                    liStationService.list(
                            new LambdaQueryWrapper<LiStation>()
                                    .eq(LiStation::getDeleted, 0)
                                    .orderByAsc(LiStation::getLineId, LiStation::getOrderNum));
        }
        if (lists != null && lists.size() > 0) {
            for (int i = 0; i < lists.size(); i++) {
                System.out.println("初始化MQ消费端");
                ewarningService.initMq(lists.get(i).getId());
            }
        }
    }
}