com.ctrip.infosec.rule.rabbitmq.DispatcherMessageSender.java Source code

Introduction

Here is the source code for com.ctrip.infosec.rule.rabbitmq.DispatcherMessageSender.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.ctrip.infosec.rule.rabbitmq;

import com.ctrip.infosec.common.Constants;
import com.ctrip.infosec.common.model.RiskFact;
import com.ctrip.infosec.configs.Configs;
import com.ctrip.infosec.configs.event.DistributionChannel;
import com.ctrip.infosec.configs.utils.BeanMapper;
import com.ctrip.infosec.sars.monitor.SarsMonitorContext;
import com.ctrip.infosec.sars.util.Collections3;
import com.google.common.collect.Maps;
import com.meidusa.fastjson.JSON;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.springframework.amqp.core.AmqpTemplate;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 * @author zhengby
 */
@Service
public class DispatcherMessageSender {

    private static Logger logger = LoggerFactory.getLogger(DispatcherMessageSender.class);
    @Resource(name = "template_datadispatcher")
    private AmqpTemplate template;
    private final String defaultRoutingKey = "datadispatcher";

    public void sendToDataDispatcher(RiskFact fact) {
        Set<DistributionChannel> channels = Configs.getDistributionChannelsByEventPoint(fact.eventPoint);
        List<String> channelNos = Collections3.extractToList(channels, "channelNo");
        channelNos.add(defaultRoutingKey);
        String routingKey = "." + StringUtils.join(channelNos, ".") + ".";
        logger.info(SarsMonitorContext.getLogPrefix() + "routingKey: " + routingKey);
        boolean withScene = Constants.eventPointsWithScene.contains(fact.eventPoint);
        if (withScene) {
            RiskFact factCopy = BeanMapper.copy(fact, RiskFact.class);
            if (!factCopy.resultsGroupByScene.isEmpty()) {
                Map<String, Object> finalResult = Maps.newHashMap();
                finalResult.put(Constants.riskLevel, 0);
                finalResult.put(Constants.riskMessage, "PASS");
                for (Map<String, Object> rs : factCopy.resultsGroupByScene.values()) {
                    finalResult = compareAndReturn(finalResult, rs);
                }
                factCopy.setFinalResult(Maps.newHashMap(finalResult));
                factCopy.finalResult.remove(Constants.async);
                factCopy.finalResult.remove(Constants.timeUsage);

            }
            //            template.convertAndSend(routingKey, JSON.toJSONString(factCopy));
            template.convertAndSend(defaultRoutingKey, JSON.toJSONString(factCopy));
        } else {
            //            template.convertAndSend(routingKey, JSON.toJSONString(fact));
            template.convertAndSend(defaultRoutingKey, JSON.toJSONString(fact));
        }
    }

    /**
     * finalResult
     */
    Map<String, Object> compareAndReturn(Map<String, Object> oldResult, Map<String, Object> newResult) {
        if (newResult == null) {
            return oldResult;
        }
        if (oldResult == null) {
            return newResult;
        }
        int newRriskLevel = MapUtils.getInteger(newResult, Constants.riskLevel, 0);
        int oldRriskLevel = MapUtils.getInteger(oldResult, Constants.riskLevel, 0);
        if (newRriskLevel > oldRriskLevel) {
            return newResult;
        }
        return oldResult;
    }
}