com.jeesuite.amqp.memoryqueue.MemoryQueueProducerAdapter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-amqp-adapter Show documentation
MQ适配器，支持各种云厂商云MQ，已经原生kafka,RocketMQ,RabbitMQ等
The newest version!
package com.jeesuite.amqp.memoryqueue;

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

import org.apache.commons.lang3.concurrent.BasicThreadFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.lmax.disruptor.EventFactory;
import com.lmax.disruptor.EventHandler;
import com.lmax.disruptor.EventTranslatorOneArg;
import com.lmax.disruptor.RingBuffer;
import com.lmax.disruptor.YieldingWaitStrategy;
import com.lmax.disruptor.dsl.Disruptor;
import com.lmax.disruptor.dsl.ProducerType;
import com.jeesuite.amqp.MQMessage;
import com.jeesuite.amqp.MQProducer;
import com.jeesuite.amqp.MessageHandler;

/**
 * 
 * 

 * Class Name : MemoryQueueProducer
 *
 * @author jiangwei
 * @version 1.0.0
 * @date 2020年3月24日
 */
public class MemoryQueueProducerAdapter implements MQProducer {

	private final static Logger logger = LoggerFactory.getLogger("com.jeesuite.amqp");

	private final static Translator TRANSLATOR = new Translator();
	private static EventHandler eventHandler = new MQMessageEventHandler();
	private static Map messageHandlers = new HashMap<>(); 
	
	private Disruptor disruptor;
	
	@Override
	public void start() throws Exception {
		EventFactory eventFactory = new MQMessageEventFactory();
		int ringBufferSize = 1024 * 1024;
		BasicThreadFactory factory = new BasicThreadFactory.Builder()
				      .namingPattern("workerthread-%d")
				      .daemon(true)
				      .priority(Thread.MAX_PRIORITY)
				      .build();
		disruptor = new Disruptor(eventFactory,
		                ringBufferSize, factory, ProducerType.SINGLE,
		                new YieldingWaitStrategy());
		
		disruptor.handleEventsWith(eventHandler);
		
		disruptor.start();
	}

	@Override
	public String sendMessage(MQMessage message, boolean async) {
		RingBuffer ringBuffer = disruptor.getRingBuffer();
	    ringBuffer.publishEvent(TRANSLATOR, message);
		return null;
	}

	@Override
	public void shutdown() {
		disruptor.shutdown();
	}
	
	public static void setMessageHandlers(Map messageHandlers) {
		MemoryQueueProducerAdapter.messageHandlers = messageHandlers;
	}

	static class MQMessageEvent {
		private MQMessage value;
		public void set(MQMessage value) {
			this.value = value;
		}
		public MQMessage get() {
			return value;
		}
	}

	static class MQMessageEventFactory implements EventFactory {
		public MQMessageEvent newInstance() {
			return new MQMessageEvent();
		}
	}

	static class MQMessageEventHandler implements EventHandler {
		public void onEvent(MQMessageEvent event, long sequence, boolean endOfBatch) {
			MQMessage message = event.get();
			MessageHandler handler = messageHandlers.get(message.getTopic());
			if(handler != null){
				try {
					messageHandlers.get(message.getTopic()).process(message);
					if(logger.isDebugEnabled())logger.debug("MQ_MESSAGE_CONSUME_SUCCESS ->topic:{},message:{}",message.getTopic(),sequence);
				} catch (Exception e) {
					logger.error(String.format("MQ_MESSAGE_CONSUME_ERROR ->topic:%s,msgId:%s", message.getTopic(),sequence),e);
				}				
			}
		}
	}

	static class Translator implements EventTranslatorOneArg {
		@Override
		public void translateTo(MQMessageEvent event, long sequence, MQMessage data) {
			event.set(data);
		}
	}

}