线程池与mq的简单结合使用

本文介绍使用ActiveMQ实现多线程发送和接收消息的方法。通过创建多线程生产者和消费者,演示如何配置连接并发送消息到指定队列,同时在消费者端通过消息监听器实现多线程接收。
本文考虑发送方和接收方有多个线程发布消息和多个线程接收消息的情况: 

1.生产者

package com.activemq3;
import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.MessageProducer;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.naming.InitialContext;

import org.apache.activemq.ActiveMQConnection;
import org.apache.activemq.ActiveMQConnectionFactory;

/**
 * 消息生产者-消息发布者(多线程发送)
 * 
 */
public class JMSProducerThread {

    private static final String USERNAME=ActiveMQConnection.DEFAULT_USER; // 默认的连接用户名
    private static final String PASSWORD=ActiveMQConnection.DEFAULT_PASSWORD; // 默认的连接密码
    private static final String BROKEURL=ActiveMQConnection.DEFAULT_BROKER_URL; // 默认的连接地址
    private static final int SENDNUM=10; // 发送的消息数量
    ConnectionFactory connectionFactory=null; // 连接工厂
    private Connection connection = null;
    private Session session = null;
    private Destination destination=null; // 消息的目的地
    public void init(){
        // 实例化连接工厂
        connectionFactory=new ActiveMQConnectionFactory(JMSProducerThread.USERNAME, JMSProducerThread.PASSWORD, JMSProducerThread.BROKEURL);
        try {
            connection=connectionFactory.createConnection(); // 通过连接工厂获取连接
            connection.start();
        } catch (JMSException e) {
            e.printStackTrace();
        }
    }
    public void produce(){
        try {
            MessageProducer messageProducer; // 消息生产者
            session=connection.createSession(Boolean.TRUE, Session.AUTO_ACKNOWLEDGE); // 创建Session
            destination=session.createQueue("queue1");
            messageProducer=session.createProducer(destination); // 创建消息生产者
            for(int i=0;i<JMSProducerThread.SENDNUM;i++){
                TextMessage message=session.createTextMessage("ActiveMQ中"+Thread.currentThread().getName()+"线程发送的数据"+":"+i);
                System.out.println(Thread.currentThread().getName()+"线程"+"发送消息:"+"ActiveMQ 发布的消息"+":"+i);
                messageProducer.send(message);
                session.commit();
            }
        } catch (JMSException e) {
            e.printStackTrace();
        }
    }


    /**
     * 发送消息
     * @param session
     * @param messageProducer
     * @throws Exception
     */
    public static void sendMessage(Session session,MessageProducer messageProducer)throws Exception{
        for(int i=0;i<JMSProducerThread.SENDNUM;i++){
            TextMessage message=session.createTextMessage("ActiveMQ 发送的消息"+i);
            System.out.println("发送消息:"+"ActiveMQ 发布的消息"+i);
            messageProducer.send(message);
        }
    }

}

2.消费者

package com.activemq3;

import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.naming.InitialContext;

import org.apache.activemq.ActiveMQConnection;
import org.apache.activemq.ActiveMQConnectionFactory;

/**
 * 消息生产者-消息发布者(多线程发送消息)
 *
 */
public class JMSConsumerThread {

    private static final String USERNAME=ActiveMQConnection.DEFAULT_USER; // 默认的连接用户名
    private static final String PASSWORD=ActiveMQConnection.DEFAULT_PASSWORD; // 默认的连接密码
    private static final String BROKEURL=ActiveMQConnection.DEFAULT_BROKER_URL; // 默认的连接地址

    ConnectionFactory connectionFactory=null; // 连接工厂
    private Connection connection = null;
    private Session session = null;
    private Destination destination=null; // 消息的目的地
    public void init(){
        // 实例化连接工厂
        connectionFactory=new ActiveMQConnectionFactory(JMSConsumerThread.USERNAME, JMSConsumerThread.PASSWORD, JMSConsumerThread.BROKEURL);
        try {
            connection=connectionFactory.createConnection(); // 通过连接工厂获取连接
            connection.start();
        } catch (JMSException e) {
            e.printStackTrace();
        }
    }
    public void consumer(){
        MessageConsumer messageConsumer; // 消息的消费者

        try {
            session=connection.createSession(Boolean.FALSE, Session.AUTO_ACKNOWLEDGE); // 创建Session
            destination=session.createQueue("queue1");
            messageConsumer=session.createConsumer(destination); // 创建消息消费者
            messageConsumer.setMessageListener(new Listener3()); // 注册消息监听
        } catch (JMSException e) {
            e.printStackTrace();
        }
    }


}

package com.activemq3;

import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageListener;
import javax.jms.TextMessage;

/**
 * 消息监听(多线程接收)
 *
 */
public class Listener3 implements MessageListener{
    private ExecutorService threadPool =Executors.newFixedThreadPool(8);
    @Override
    public void onMessage(final Message message) {
    threadPool.execute(new Runnable() {

            @Override
            public void run() {
                try {
                    try {
                        Thread.sleep(new Random().nextInt(2)*500);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                    System.out.println("接收线程..."+Thread.currentThread().getName()+"收到的消息是:"+((TextMessage)message).getText());
                } catch (JMSException e) {
                    e.printStackTrace();
                }               
            }
        }); 
    }
}
生产者执行发送的主函数: 

在main方法中创建了一个固定3个 线程的线程池,去处理5个线程任务。

package com.activemq3;

import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
/**
 * 多线程发送消息
 */

public class MainProducer {
          public static void main(String[] args) {

              ExecutorService threadPool=Executors.newFixedThreadPool(3);
              for (int i = 0; i < 5; i++) {
                  threadPool.submit(new Runnable() {

                    @Override
                    public void run() {
                        JMSProducerThread jph=new JMSProducerThread();
                        try {
                            Thread.sleep(new Random().nextInt(5)*500);
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                        jph.init();
                        jph.produce();                      
                    }
                });

            }
             /* 
            for (int i = 0; i <5; i++) {

               new Thread(new Runnable() {

                @Override
                public void run() {
                    JMSProducerThread js=new JMSProducerThread();
                    try {
                        Thread.sleep(new Random().nextInt(5)*500);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                    js.init();
                    js.produce();

                }
            }).start();

            }*/         
            }
}
消费者执行发送的主函数: 

消费者中监听器中创建了固定8个线程的线程池去接收消息。

package com.activemq3;
public class MainConsumer {
          public static void main(String[] args) {
            JMSConsumerThread jch=new JMSConsumerThread();
            jch.init();
            jch.consumer();             
  }
}

### Java 线程池拒绝策略消息队列配合使用线程池设计中,当线程数量达到上限且任务队列已满时,新提交的任务将触发线程池的拒绝策略。为了更好地管理这些情况并确保系统的稳定性,可以引入消息队列作为备用方案之一。 #### 使用 `AbortPolicy` 拒绝策略消息队列结合 在这种情况下,一旦线程池无法接受更多任务,将会抛出异常,并尝试将该任务发送至外部的消息队列系统中保存起来等待后续处理[^1]。 ```java import java.util.concurrent.*; public class ThreadPoolWithMessageQueue { private static final int CORE_POOL_SIZE = 5; private static final int MAXIMUM_POOL_SIZE = 10; private static final long KEEP_ALIVE_TIME = 1L; public static void main(String[] args) throws InterruptedException { BlockingQueue<Runnable> workQueue = new ArrayBlockingQueue<>(5); ThreadPoolExecutor executorService = new ThreadPoolExecutor( CORE_POOL_SIZE, MAXIMUM_POOL_SIZE, KEEP_ALIVE_TIME, TimeUnit.SECONDS, workQueue); // 设置自定义的拒绝策略处理器 executorService.setRejectedExecutionHandler(new CustomRejectPolicy()); for (int i = 0; i < 20; ++i){ try{ System.out.println("Submitting task " + i); executorService.submit(() -> { try { Thread.sleep(100); // Simulate some processing time. System.out.println("Task executed by thread: " + Thread.currentThread().getName()); } catch (InterruptedException e) { throw new RuntimeException(e); } }); }catch (Exception ex){ System.err.println(ex.getMessage()); } Thread.sleep(100); } executorService.shutdown(); executorService.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS); } /** * 自定义拒绝策略实现类, */ static class CustomRejectPolicy implements RejectedExecutionHandler { @Override public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) { // 将未完成的任务推送到Kafka或其他MQ服务端口... System.out.printf("Task %s is rejected and sent to message queue.\n", r.toString()); // 这里仅做简单模拟,实际应用应连接真实的消息中间件 sendToMessageQueue(r); } private void sendToMessageQueue(Runnable runnable){ // 实现向消息队列推送逻辑的地方 // 假设这里有一个方法可以直接把runnable对象序列化成字符串形式存入数据库表或者通过网络传输给其他进程/服务器上的消费者消费。 String serializedTask = serializeRunable(runnable); System.out.println("Serialized Task:" + serializedTask); } private String serializeRunable(Runnable run){ return run.getClass().getSimpleName(); // Simplified serialization method } } } ``` 此代码片段展示了如何设置一个带有自定义拒绝策略的线程池实例,在遇到超出承载能力的情况时能够优雅地处理多余的工作项而不是直接丢弃它们。具体来说就是每当有任务因为资源不足而被拒绝执行的时候,这个定制化的处理器会负责将其转发到另一个持久存储位置——比如分布式消息总线之上以便稍后再作进一步处置[^3]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

老马识途2.0

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值