com.cybermkd.plugin.zbus.ZbusPlugin Maven / Gradle / Ivy
The newest version!
package com.cybermkd.plugin.zbus;
import com.cybermkd.common.Plugin;
import com.cybermkd.common.util.Stringer;
import com.cybermkd.log.Logger;
import com.cybermkd.plugin.zbus.annotation.MqHandler;
import com.cybermkd.plugin.zbus.annotation.TopicHandler;
import com.cybermkd.plugin.zbus.handler.TMsgHandler;
import com.cybermkd.plugin.zbus.sender.Sender;
import org.reflections.Reflections;
import org.zbus.broker.Broker;
import org.zbus.broker.BrokerConfig;
import org.zbus.broker.SingleBroker;
import org.zbus.mq.Consumer;
import org.zbus.mq.Producer;
import org.zbus.mq.Protocol.MqMode;
import java.io.IOException;
import java.util.*;
public class ZbusPlugin implements Plugin {
/**
* 日志
*/
private static final Logger LOG = Logger.getLogger("ZbusPlugin");
/**
* MQ消费者配置Map
*/
private final Map> mqNameTMsgHandlerMap = new HashMap>();
/**
* Topic消费者配置Map mp - topic - TMsgHandler
*/
private final Map>> mqNamePubSubTMsgHandlerMap = new HashMap>>();
/**
* 消费者列表
*/
private final List consumerList = new ArrayList();
/**
* 发送器列表
*/
private final static List> senderList = new ArrayList>();
/**
* zbusServer地址
*/
private String brokerAddress = null;
/**
* broker对象
*/
private static Broker broker = null;
/**
* 简单Broker配置
*/
private BrokerConfig brokerConfig = null;
private final String scanRootPackage;
/**
* 默认构造函数，使用127.0.0.1:15555地址
*/
public ZbusPlugin() {
this.brokerAddress = "127.0.0.1:15555";
this.scanRootPackage = null;
}
/**
* @return Producer
* @throws InterruptedException
* @throws IOException
* @Title: createProducer
* @Description: 创建一个Producer
* @since V1.0.0
*/
public static Producer createProducer(Sender sender, String mq, MqMode mqMode) throws IOException, InterruptedException {
Producer producer = new Producer(broker, mq, mqMode);
producer.createMQ();
//这册这个发送器
senderList.add(sender);
return producer;
}
/**
* 默认构造函数,可指定brokerAddress
*/
public ZbusPlugin(String brokerAddress) {
this(brokerAddress, null);
}
public ZbusPlugin(String brokerAddress, String scanRootPackage) {
this.brokerAddress = brokerAddress;
this.scanRootPackage = scanRootPackage;
//创建broker
ensureBroker();
//加载注解
autoLoadByAnnotation();
}
/**
* 构造函数 使用BrokerConfig config构建
*/
public ZbusPlugin(BrokerConfig config) {
this(config, null);
}
public ZbusPlugin(BrokerConfig config, String scanRootPackage) {
this.brokerConfig = config;
this.scanRootPackage = scanRootPackage;
//创建broker
ensureBroker();
//加载注解
autoLoadByAnnotation();
}
/**
* @param mq MQ名
* @param msgHandler 消息到达回调接口
* @Title: registerMqMsgHandler
* @Description: 注册Mq的消息回调接口
* @since V1.0.0
*/
public void registerMqMsgHandler(String mq, TMsgHandler msgHandler) {
if (mqNameTMsgHandlerMap.containsKey(mq)) {
LOG.warn("(mq=" + mq + ")对应的消息处理器已存在!");
}
mqNameTMsgHandlerMap.put(mq, msgHandler);
}
/**
* @param topic Topic对象
* @param msgHandler 消息到达回调接口
* @Title: registerTopicMsgHandler
* @Description: 注册Topic的消息回调接口
* @since V1.0.0
*/
public void registerTopicMsgHandler(Topic topic, TMsgHandler msgHandler) {
this.registerTopicMsgHandler(topic.getMqName(), topic.getTopicName(), msgHandler);
}
/**
* @param mq MQ名
* @param topic 主题名
* @param msgHandler 消息到达回调接口
* @Title: registerTopicMsgHandler
* @Description: 注册Topic的消息回调接口
* @since V1.0.0
*/
public void registerTopicMsgHandler(String mq, String topic, TMsgHandler msgHandler) {
// 依据mq获得 topic－TMsgHandler映射map
Map> tmc = this.mqNamePubSubTMsgHandlerMap.get(mq);
if (null == tmc) {
tmc = new HashMap>();
}
if (tmc.containsKey(topic)) {
LOG.warn("(mq=" + mq + ",topic=" + topic + ")对应的消息处理器已存在!");
}
tmc.put(topic, msgHandler);
this.mqNamePubSubTMsgHandlerMap.put(mq, tmc);
}
/**
* @throws Exception
* @Title: ensureBroker
* @Description: 确保broker可用
* @since V1.0.0
*/
private void ensureBroker() {
if (broker == null) {
synchronized (this) {
if (broker == null) {
if (this.brokerAddress != null) {
this.brokerConfig = new BrokerConfig();
this.brokerConfig.setServerAddress(brokerAddress);
}
try {
broker = new SingleBroker(this.brokerConfig);
LOG.info("创建broker成功(brokerAddress=" + this.brokerAddress + ")");
} catch (IOException e) {
throw new RuntimeException(e.getMessage(), e);
}
}
}
}
}
@Override
public boolean start() {
try {
//确保创建
ensureBroker();
// 创建Mq消费者
for (Map.Entry> entry : this.mqNameTMsgHandlerMap.entrySet()) {
String mq = entry.getKey();
Consumer c = new Consumer(broker, mq, MqMode.MQ);
c.onMessage(entry.getValue());
c.start();
consumerList.add(c);
LOG.info("创建MQ消费者成功(mq=" + mq + ")");
}
// 创建topic消费者
for (Map.Entry>> mqConfig : this.mqNamePubSubTMsgHandlerMap
.entrySet()) {
String mq = mqConfig.getKey();
// topic <－> TMsgHandler 映射map
Map> tmt = mqConfig.getValue();
for (Map.Entry> topicConfig : tmt.entrySet()) {
String topic = topicConfig.getKey();
Consumer c = new Consumer(broker, mq, MqMode.PubSub);
c.setTopic(topic);
c.onMessage(topicConfig.getValue());
c.start();
consumerList.add(c);
LOG.info("创建Topic消费者成功 (mq=" + mq + ",topic=" + topic + ")");
}
}
return true;
} catch (Exception e) {
LOG.error(e.getMessage(), e);
throw new RuntimeException(e.getMessage(), e);
}
}
@Override
public boolean stop() {
try {
// 关闭消费者
for (Consumer c : consumerList) {
c.close();
c = null;
}
consumerList.clear();
//关闭所有发送器
for (Sender sender : senderList) {
sender.close();
}
senderList.clear();
// 关闭broker
if (broker != null) {
broker.close();
broker = null;
}
return true;
} catch (IOException e) {
LOG.error(e.getMessage(), e);
return false;
}
}
private void autoLoadByAnnotation() {
if (Stringer.isBlank(this.scanRootPackage)) {
return;
}
try {
//自动扫描相关类库检查
Class.forName("org.reflections.Reflections");
Class.forName("com.google.common.collect.Sets");
Class.forName("javassist.bytecode.annotation.Annotation");
} catch (Exception e) {
throw new RuntimeException("Zbus开启自动扫描加载消息处理器需要Reflections、Guava、Javassist类库，请导入相应的jar包");
}
//首先检查是否具有相关的库
Reflections reflections = new Reflections(this.scanRootPackage);
Set> mqHandlerClasses = reflections.getTypesAnnotatedWith(MqHandler.class);
for (Class mc : mqHandlerClasses) {
if (!TMsgHandler.class.isAssignableFrom(mc)) {
throw new RuntimeException(mc.getName() + " 必须继承自 TMsgHandler");
}
MqHandler mh = mc.getAnnotation(MqHandler.class);
try {
TMsgHandler hander = (TMsgHandler) mc.newInstance();
this.registerMqMsgHandler(mh.value(), hander);
LOG.info("通过注解自动加载MQ消息处理器( mq=" + mh.value() + ",handler=" + mc.getName() + " )");
} catch (Exception e) {
throw new RuntimeException(e.getMessage(), e);
}
}
Set> topicHandlerClasses = reflections.getTypesAnnotatedWith(TopicHandler.class);
for (Class mc : topicHandlerClasses) {
if (!TMsgHandler.class.isAssignableFrom(mc)) {
throw new RuntimeException(mc.getName() + " 必须继承自 TMsgHandler");
}
TopicHandler th = mc.getAnnotation(TopicHandler.class);
try {
TMsgHandler hander = (TMsgHandler) mc.newInstance();
this.registerTopicMsgHandler(th.mq(), th.topic(), hander);
LOG.info("通过注解自动加载Topic消息处理器( mq=" + th.mq() + ",topic=" + th.topic() + ",handler=" + mc.getName() + " )");
} catch (Exception e) {
throw new RuntimeException(e.getMessage(), e);
}
}
}
}