ActiveMQ源码分析之Producer发送消息

AMQ源码分析基于5.15.6版本，先来段Producer连接Broker的demo。

public class Sender {
	
    private static final int SEND_NUMBER = 1;

    public static void main(String[] args) {
        ConnectionFactory connectionFactory;
        Connection connection = null;
        Session session;
        Destination destination;
        MessageProducer producer;
        connectionFactory = new ActiveMQConnectionFactory(
                ActiveMQConnection.DEFAULT_USER,
                ActiveMQConnection.DEFAULT_PASSWORD,
                "tcp://localhost:61616");
        try {
            connection = connectionFactory.createConnection();
            connection.start();
            session = connection.createSession(Boolean.TRUE,
                    Session.AUTO_ACKNOWLEDGE);
            destination = session.createQueue("FirstQueue");
            producer = session.createProducer(destination);
            producer.setDeliveryMode(DeliveryMode.PERSISTENT);
            sendMessage(session, producer);
            session.commit();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != connection)
                    connection.close();
            } catch (Throwable ignore) {
            }
        }
    }
    public static void sendMessage(Session session, MessageProducer producer)
            throws Exception {
        for (int i = 1; i <= SEND_NUMBER; i++) {
            TextMessage message = session
                    .createTextMessage("ActiveMq 发送的消息" + i);
            System.out.println("发送消息：" + message);
            producer.send(message);
        }
    }

首先需要创建ActiveMQConnectionFactory工厂对象，这个对象的username和password属性可以被Broker用作简单的安全和权限校验方式，如果校验不通过的话则不允许客户端和Broker建立网络连接。下面来分析发送消息的几个步骤：

1、建立和Broker的网络连接createConnection流程

创建连接的过程主要使用了责任链设计模式，使整个初始化流程开起来很清晰。

ActiveMQConnectionFactory------>ResponseCorrelator------>MutexTransport------>WireFormatNegotiator

------>InactivityMonitor------>AbstractInactivityMonitor------>TcpTransport------>TransportThreadSupport

首先ActiveMQConnectionFactory创建OpenWireFormat对象，该对象封装了和Broker之间通信的协议版本，是否缓存等属性，也是客户端和服务端通信发送的第一条数据。接着创建TcpTransport对象，该对象是对socket的一个封装，用于连接broker，Connect to a remote Broker。同时这个步骤还会组装一个任务链，链中的每个节点都继承了TransportFilter类，这个类中有个Transport 类型的next属性，指向链中的下一个节点，最终成链。任务链的最后一个节点是TcpTransport，ActiveMQConnection在实例化的时候会将TcpTransport加到链尾。

ResponseCorrelator类主要用于接收同步请求的响应，关于同步请求和异步请求的响应后续会详细说明。

MutexTransport类主要用于保证请求的线程安全，保证同一时间只能有一个请求。Thread safe Transport Filter that serializes calls to and from the Transport Stack。

WireFormatNegotiator类在Producer第一次启动的时候会向broker发送一条WireFromatInfo信息，主要是获取broker使用的协议版本，后续Producer接收消息部分会进行说明。

InactivityMonitor类实现了连接成功后启动心跳检查机制，客户端每10秒发送一次心跳信息，服务端每30秒读一次心跳信息，如果没有读到则会断开连接，心跳检测是相互的，客户端也会每30秒读取服务端发送来的心跳信息，如果没有读到也一样会断开连接。AMQ的心跳机制实现后面专门讲任务的部分会详细说明。

TcpTransport类需要重点关注，我们知道责任链模式的重点在于末尾节点，一般重要逻辑都会放在末尾节点进行处理。而TcpTransport刚好就是末尾节点，它的connect方法设置socket的连接属性，如connectionTimeout设置为30秒，receiveBufferSize设置为65535 ，sendBufferSize设置为65535等，这几个属性都是与socket通信紧密相关的。设置完socket属性之后与Broker建立连接，并且初始化对应的input和output对应的封装对象，为后面的socket数据读写做好准备。紧接着调用super.doStart()方法，TransportThreadSupport是TcpTransport的父类，start方法中开起了一个线程，由于TransportThreadSupport类并没有实现线程的run方法，所以实际执行的是TcpTransport中的run方法。run方法主要是读取broker返回的数据，reads packets from a Socket ，从网络流中读取数据并进行处理。

TcpTransport的run方法启动之后，整个初始化过程算是结束了。

接着创建Session，实例为ActiveMQSession。SessionInfo封装了sessionId和connectionId，AMQ的客户端和服务端通信的数据都是以info结尾的对象。

JMSSessionStatsImpl对象主要针对session进行统计操作，producers 和consumers 数据单独处理messageCount ，pendingMessageCount ，expiredMessageCount ，durableSubscriptionCount 这类消息数量统一由CountStatisticImpl 处理messageWaitTime ，messageRateTime 这类跟消息耗时相关的统计统一由TimeStatisticImpl 处理。

ActiveMQSessionExecutor类消息发送流程不会涉及，但是在消息消费流程中很关键。

 ActiveMQSessionExecutor(ActiveMQSession session) {
        this.session = session;
        if (this.session.connection != null && this.session.connection.isMessagePrioritySupported()) {
           this.messageQueue = new SimplePriorityMessageDispatchChannel();
        }else {
            this.messageQueue = new FifoMessageDispatchChannel();
        }
    }

类中messageQueue有两种实现方式，一种是SimplePriorityMessageDispatchChannel ，其中的消息具有优先级。另一种是FifoMessageDispatchChannel ，先进先出，根据消息进入先后顺序出队列。这块在消息消费的时候会详细说明。

Session创建好之后接着创建Destination和Producer，因这两个流程逻辑比较简单，所以此处不再啰嗦，直接进入消息发送流程。

消息发送之前会开启事务，然后创建TransactionInfo对象，该对象封装了transactionId，connectionId及事务所处在阶段BEGIN并异步发送给Broker。然后设置消息的优先级priority，目标地址destination，过期时间expiration等属性。如果消息设置了存活时间timeToLive，还会计算到期时间expiration = timeToLive + 当前时间。最后异步发送ActiveMQTextMessage 到Broker。

最后一个阶段是事务提交阶段。该阶段主要将事务所处状态COMMIT_ONE_PHASE和transactionId，connectionId封装成TransactionInfo提交到Broker。Broker对Producer的请求处理统一会在Broker模块详细分析。

下一节Producer消息接收和处理流程。打完收工