tgtools.notify.rabbitmq.websocket.ConsumerMap Maven / Gradle / Ivy

Go to download
package tgtools.notify.rabbitmq.websocket;


import com.rabbitmq.client.Channel;
import org.springframework.amqp.core.AcknowledgeMode;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.core.ChannelAwareMessageListener;
import org.springframework.amqp.rabbit.listener.AbstractMessageListenerContainer;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import tgtools.notify.rabbitmq.service.RabbitMqService;
import tgtools.util.LogHelper;

import java.io.Closeable;
import java.io.IOException;

/**
 * @author 田径
 * @Title
 * @Description
 * @date 19:07
 */

public class ConsumerMap implements Closeable {
    protected java.util.concurrent.ConcurrentHashMap mContainers = new java.util.concurrent.ConcurrentHashMap();
    protected AbstractClientWebSocketHandler mWebSocketHandler;
    protected RabbitMqService mRabbitMqService;

    public ConsumerMap(AbstractClientWebSocketHandler pWebSocketHandler) {
        mWebSocketHandler = pWebSocketHandler;
        if (null != mWebSocketHandler) {
            mRabbitMqService = new RabbitMqService(mWebSocketHandler.getRabbitAdmin());
        }
    }

    public void createConsumer(String pLoginName) {
        try {
            SimpleMessageListenerContainer container = mRabbitMqService.createUserConsumer(pLoginName, new MessageListenerImpl(pLoginName), AcknowledgeMode.MANUAL);
            container.start();
            mContainers.put(pLoginName, container);
            LogHelper.info("", "createConsumer name:" + pLoginName, "ConsumerMap");
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    public void removeConsumer(String pLoginName) {
        if (mContainers.containsKey(pLoginName)) {
            AbstractMessageListenerContainer container = mContainers.get(pLoginName);
            if (null != container) {
                mContainers.remove(pLoginName);
                container.stop();
                container = null;
            }
        }
    }

    @Override
    public void close() {
        for (AbstractMessageListenerContainer item : mContainers.values()) {
            try {
                item.stop();
            } catch (Exception e) {
            }
        }
        mContainers.clear();
        mContainers=null;
    }


    public class MessageListenerImpl implements ChannelAwareMessageListener {
        private String mLoginName;

        public MessageListenerImpl(String pLoginName) {
            mLoginName = pLoginName;
        }

        @Override
        public void onMessage(Message pMsg, Channel pChannel) throws Exception {
            try {
                System.out.println(mLoginName + " onMessage:" + new String(pMsg.getBody(), "UTF-8"));
                ConsumerMap.this.mWebSocketHandler.sendMessage(mLoginName, new String(pMsg.getBody(), "UTF-8"));
            } catch (Exception e) {
                LogHelper.error("", "onMessage Error", "MessageListenerImpl.onMessage", e);
            }

            try {
                pChannel.basicAck(pMsg.getMessageProperties().getDeliveryTag(), false);
            } catch (IOException e) {
                LogHelper.error("", "消息 ack 出错", "MessageListenerImpl.onMessage", e);
            }
        }

    }

}