package com.zt.common.message.consumer;
|
|
import cn.hutool.core.map.MapUtil;
|
import com.alibaba.fastjson.JSON;
|
import com.zt.common.message.handler.MessageHandler;
|
import com.zt.common.message.model.Message;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.context.ApplicationContext;
|
import org.springframework.context.ApplicationListener;
|
import org.springframework.context.event.ContextRefreshedEvent;
|
|
import java.util.Collection;
|
import java.util.Collections;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
* 抽象公共消息队列
|
*
|
* @author wacxhs
|
*/
|
public abstract class AbstractMessageQueue implements ApplicationListener<ContextRefreshedEvent> {
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(AbstractMessageQueue.class);
|
|
private ApplicationContext applicationContext;
|
|
/**
|
* 注册消息处理器
|
*/
|
private Map<String, MessageHandler> registerJmsHandler = Collections.emptyMap();
|
|
@Override
|
public void onApplicationEvent(ContextRefreshedEvent event) {
|
if (event.getApplicationContext().getParent() == null) {
|
LOGGER.debug("准备加载MessageHandler实现");
|
this.applicationContext = event.getApplicationContext();
|
Map<String, MessageHandler> jmsHandlerMap = applicationContext.getBeansOfType(MessageHandler.class);
|
LOGGER.debug("加载MessageHandler实现共{}条", jmsHandlerMap.size());
|
|
if (MapUtil.isNotEmpty(jmsHandlerMap)) {
|
Collection<MessageHandler> jmsHandlers = jmsHandlerMap.values();
|
this.registerJmsHandler = new HashMap<>(jmsHandlers.size(), 1);
|
for (MessageHandler jmsHandler : jmsHandlers) {
|
this.registerJmsHandler.put(jmsHandler.getType().getKey(), jmsHandler);
|
}
|
}
|
containerInitialCompleteAfter();
|
}
|
}
|
|
/**
|
* 容器初始化完成后
|
*/
|
protected void containerInitialCompleteAfter() {
|
}
|
|
protected ApplicationContext getApplicationContext() {
|
return applicationContext;
|
}
|
|
/**
|
* 发送消息
|
*
|
* @param message
|
* 消息传输对象
|
*/
|
@SuppressWarnings("unchecked")
|
public void handleMessage(Message<?> message) {
|
MessageHandler handler = getJmsHandler(message.getType());
|
if (handler == null) {
|
LOGGER.warn("未找到({})消息处理器", message.getType());
|
return;
|
}
|
try {
|
handler.handlerMessage(message);
|
} catch (Exception e) {
|
LOGGER.error("处理消息({})出错, 参数入参：{}", message.getType(), JSON.toJSONString(message), e);
|
}
|
}
|
|
/**
|
* 获取消息处理实现
|
*
|
* @return 消息处理实现
|
*/
|
protected MessageHandler getJmsHandler(String type) {
|
return this.registerJmsHandler.get(type);
|
}
|
}
|
