RabbitMessageHandler.java 3.17 KB
package com.bsth.message.handler;

import com.bsth.config.RabbitConfig;
import com.bsth.data.alarm.AlarmCenter;
import com.bsth.data.gpsdata_v2.GpsRealData;
import com.bsth.data.gpsdata_v2.entity.GpsEntity;
import com.bsth.data.passenger_flow.PassengerFlowCenter;
import com.bsth.message.entity.*;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.annotation.RabbitHandler;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.stereotype.Component;

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

/**
 * @author Hill
 */
//@Component
//@ConditionalOnProperty("rabbit.use")
public class RabbitMessageHandler {

    private final static Logger log = LoggerFactory.getLogger(RabbitMessageHandler.class);

    @Autowired
    private ObjectMapper mapper;

    @Autowired
    GpsRealData gpsRealData;

    @Autowired
    private PassengerFlowCenter passengerFlowCenter;

    @RabbitHandler
    @RabbitListener(queues = RabbitConfig.QUEUE_ALARM_SCHEDULE)
    public void onAlarmScheduleMessage(String message) {
        try {
            AlarmVo alarmVo = null;
            Map<String, Object> data = mapper.readValue(message, mapper.getTypeFactory().constructMapType(HashMap.class, String.class, Object.class));
            if ("dsm".equals(data.get("alarmType"))) {
                alarmVo = mapper.readValue(message, AlarmDSMVo.class);
            } else if ("adas".equals(data.get("alarmType"))) {
                alarmVo = mapper.readValue(message, AlarmADASVo.class);
            } else if ("over-speed".equals(data.get("alarmType"))) {
                alarmVo = mapper.readValue(message, AlarmDSMVo.class);
                ((AlarmDSMVo)alarmVo).setType(17);
            }
            if (alarmVo != null) {
                if (alarmVo.getVehicleCode() == null) {
                    log.warn("报警未匹配到车辆,alarmVo: {}", alarmVo);
                    return;
                }
                GpsEntity gps = gpsRealData.getByNbbm(alarmVo.getVehicleCode());
                if (gps == null || gps.getLineId() == null) {
                    alarmVo.setLineCode("249231");
                    log.warn("报警未匹配到线路,alarmVo: {}", alarmVo);
                } else {
                    alarmVo.setLineCode(gps.getLineId());
                }
                AlarmCenter.put(alarmVo);
            }
        } catch (Exception e) {
            log.error("报警监听异常,message: {}", message, e);
        }
    }

    @RabbitHandler
    @RabbitListener(queues = RabbitConfig.QUEUE_PASSENGER_FLOW_SCHEDULE)
    public void onPassengerFlowScheduleMessage(String message) {
        try {
            List<PassengerFlow> passengerFlows = mapper.readValue(message, mapper.getTypeFactory().constructParametricType(List.class, PassengerFlow.class));
            passengerFlowCenter.put(passengerFlows);
        } catch (Exception e) {
            log.error("客流监听异常,message: {}", message, e);
        }
    }
}