com.jeesuite.amqp.MQServiceRegistryBean 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!
/*
 * Copyright 2016-2020 www.jeesuite.com.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.jeesuite.amqp;

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

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.Ordered;
import org.springframework.core.PriorityOrdered;
import org.springframework.data.redis.core.StringRedisTemplate;

import com.jeesuite.amqp.memoryqueue.MemoryQueueProducerAdapter;
import com.jeesuite.amqp.qcloud.cmq.CMQConsumerAdapter;
import com.jeesuite.amqp.qcloud.cmq.CMQProducerAdapter;
import com.jeesuite.amqp.redis.RedisConsumerAdapter;
import com.jeesuite.amqp.redis.RedisProducerAdapter;
import com.jeesuite.amqp.rocketmq.RocketProducerAdapter;
import com.jeesuite.amqp.rocketmq.RocketmqConsumerAdapter;
import com.jeesuite.common.JeesuiteBaseException;
import com.jeesuite.spring.InstanceFactory;
import com.jeesuite.spring.helper.SpringAopHelper;

public class MQServiceRegistryBean implements InitializingBean,DisposableBean,ApplicationContextAware,PriorityOrdered {

	protected  static final Logger logger = LoggerFactory.getLogger("com.jeesuite.amqp");
	
	private ApplicationContext applicationContext;
	private MQConsumer consumer;
	private MQProducer producer;
	
	@Autowired(required = false)
	private StringRedisTemplate redisTemplate;
	
	@Override
	public void afterPropertiesSet() throws Exception {
		
		String providerName = MQContext.getProviderName();
		//
		if(MQContext.isProducerEnabled()){			
			startProducer(providerName);
			MQInstanceDelegate.setProducer(producer);
		}
		//
		if(MQContext.isConsumerEnabled()){			
			startConsumer(providerName);
		}
	}
	
    private void startProducer(String providerName) throws Exception{
    	
		if("rocketmq".equals(providerName)){
			producer = new RocketProducerAdapter();
		}else if("cmq".equals(providerName)){
			producer = new CMQProducerAdapter();
		}else if("memoryqueue".equals(providerName)){
			producer = new MemoryQueueProducerAdapter();
		}else if("redis".equals(providerName)){
			producer = new RedisProducerAdapter(redisTemplate);
		}else{
			throw new JeesuiteBaseException("NOT_SUPPORT[providerName]:" + providerName);
		}
		producer.start();
		logger.info("MQ_PRODUCER started -> groupName:{},providerName:{}",MQContext.getGroupName(),providerName);
	}
    
    private void startConsumer(String providerName) throws Exception{
		Map messageHanlders = applicationContext.getBeansOfType(MessageHandler.class);
		if(messageHanlders != null && !messageHanlders.isEmpty()){
			Map messageHandlerMaps = new HashMap<>(); 
			messageHanlders.values().forEach(e -> {
				Object origin = e;
				try {origin = SpringAopHelper.getTarget(e);} catch (Exception ex) {ex.printStackTrace();}
				MQTopicRef topicRef = origin.getClass().getAnnotation(MQTopicRef.class);
				String topicName = MQContext.rebuildWithNamespace(topicRef.value());
				messageHandlerMaps.put(topicName, e);
				logger.info("ADD MQ_COMSUMER_HANDLER ->topic:{},handlerClass:{} ",topicName,e.getClass().getName());
			});
			
			if("rocketmq".equals(providerName)){
				consumer = new RocketmqConsumerAdapter(messageHandlerMaps);
			}else if("cmq".equals(providerName)){
				consumer = new CMQConsumerAdapter(messageHandlerMaps);
			}else if("memoryqueue".equals(providerName)){
				MemoryQueueProducerAdapter.setMessageHandlers(messageHandlerMaps);
			}else if("redis".equals(providerName)){
				consumer = new RedisConsumerAdapter(redisTemplate, messageHandlerMaps);
			}else{
				throw new JeesuiteBaseException("NOT_SUPPORT[providerName]:" + providerName);
			}
			
			consumer.start();
			logger.info("MQ_COMSUMER started -> groupName:{},providerName:{}",MQContext.getGroupName(),providerName);
		}
	}

	@Override
	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
		this.applicationContext = applicationContext;
		InstanceFactory.setApplicationContext(applicationContext);
	}
	
	@Override
	public void destroy() throws Exception {
		if(consumer != null) {
			consumer.shutdown();
		}
		if(producer != null) {
			producer.shutdown();
		}
		MQContext.close();
	}

	@Override
	public int getOrder() {
		return Ordered.LOWEST_PRECEDENCE;
	}

}