🔄Update: 队列、消息持久化 & 公平分发

src/main/java/cn/hezhaohui/MqApplicaion.java

@@ -1,11 +1,30 @@
 package cn.hezhaohui;
 
+import cn.hezhaohui.mq.Recv;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 
-@SpringBootApplication
+import java.io.IOException;
+import java.util.concurrent.TimeoutException;
+
 public class MqApplicaion {
-    public static void main(String[] args) {
-        SpringApplication.run(MqApplicaion.class, args);
+    public static void main(String[] args) throws IOException, TimeoutException {
+        new Thread(() -> {
+            try {
+                Recv.work("1");
+            } catch (IOException | TimeoutException e) {
+                throw new RuntimeException(e);
+            }
+
+        }).start();
+
+        new Thread(() -> {
+            try {
+                Recv.work("2");
+            } catch (IOException | TimeoutException e) {
+                throw new RuntimeException(e);
+            }
+
+        }).start();
     }
 }

src/main/java/cn/hezhaohui/mq/Consume.java

@@ -1,30 +0,0 @@
-package cn.hezhaohui.mq;
-
-import com.rabbitmq.client.Channel;
-import com.rabbitmq.client.Connection;
-import com.rabbitmq.client.ConnectionFactory;
-import com.rabbitmq.client.DeliverCallback;
-
-import java.io.IOException;
-import java.nio.charset.StandardCharsets;
-import java.util.concurrent.TimeoutException;
-
-public class Consume {
-    private static final String EXCHANGE_NAME = "logs";
-
-    public static void main(String[] args) throws IOException, TimeoutException {
-        ConnectionFactory factory = new ConnectionFactory();
-        Connection connection = factory.newConnection();
-        Channel channel = connection.createChannel();
-
-        channel.exchangeDeclare(EXCHANGE_NAME, "fanout");
-        String queueName = channel.queueDeclare().getQueue();
-        channel.queueBind(queueName, EXCHANGE_NAME, "");
-
-        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
-            String message = new String(delivery.getBody(), StandardCharsets.UTF_8);
-            System.out.println("[Receive] " + message);
-        };
-        channel.basicConsume(queueName, true, deliverCallback, consumerTag -> {});
-    }
-}

src/main/java/cn/hezhaohui/mq/Publish.java

@@ -1,37 +0,0 @@
-package cn.hezhaohui.mq;
-
-import com.rabbitmq.client.Channel;
-import com.rabbitmq.client.Connection;
-import com.rabbitmq.client.ConnectionFactory;
-
-import java.io.IOException;
-import java.util.Scanner;
-import java.util.concurrent.TimeoutException;
-
-public class Publish {
-    private static final String EXCHANGE_NAME = "logs";
-
-    public static void main(String[] args) throws IOException, TimeoutException {
-        ConnectionFactory factory = new ConnectionFactory();
-
-        try (Connection connection = factory.newConnection();
-             Channel channel = connection.createChannel();
-             Scanner scanner = new Scanner(System.in)) {
-            channel.exchangeDeclare(EXCHANGE_NAME, "fanout");
-            String queue = channel.queueDeclare().getQueue();
-            channel.queueBind(queue, EXCHANGE_NAME, "");
-
-            while (true) {
-                String msg = scanner.nextLine();
-                if (msg.equals("exit")) {
-                    break;
-                }
-                channel.basicPublish(EXCHANGE_NAME, "", null, msg.getBytes());
-                System.out.println("[Sent] " + msg);
-            }
-        }
-
-
-
-    }
-}

src/main/java/cn/hezhaohui/mq/Recv.java

@@ -0,0 +1,42 @@
+package cn.hezhaohui.mq;
+
+import com.rabbitmq.client.Channel;
+import com.rabbitmq.client.Connection;
+import com.rabbitmq.client.ConnectionFactory;
+import com.rabbitmq.client.DeliverCallback;
+
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.concurrent.TimeoutException;
+
+public class Recv {
+    private final static String QUEUE_NAME = "task";
+
+    public static void work(String id) throws IOException, TimeoutException {
+        ConnectionFactory connectionFactory = new ConnectionFactory();
+        Connection connection = connectionFactory.newConnection();
+        Channel channel = connection.createChannel();
+        channel.basicQos(1);
+        System.out.println(" [*] Waiting for message");
+
+        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
+            String message = new String(delivery.getBody(), StandardCharsets.UTF_8);
+            System.out.println(" [x] Received '" + message + "'");
+            try {
+                dowork(message);
+            } catch (InterruptedException e) {
+                throw new RuntimeException(e);
+            } finally {
+                System.out.println(" [x] Done");
+                System.out.println("== FROM " + id + " ==");
+            }
+        };
+        boolean autoAck = true;
+        channel.basicConsume(QUEUE_NAME, autoAck, deliverCallback, consumerTag -> {});
+    }
+
+    private static void dowork(String task) throws InterruptedException {
+        System.out.println(" [Working] " + task);
+        Thread.sleep(5000);
+    }
+}

src/main/java/cn/hezhaohui/mq/Send.java

@@ -0,0 +1,42 @@
+package cn.hezhaohui.mq;
+
+import com.rabbitmq.client.Channel;
+import com.rabbitmq.client.Connection;
+import com.rabbitmq.client.ConnectionFactory;
+import com.rabbitmq.client.MessageProperties;
+
+import java.io.IOException;
+import java.util.Scanner;
+import java.util.concurrent.TimeoutException;
+
+public class Send {
+    private final static String QUEUE_NAME = "task";
+
+    public static void main(String[] args) {
+        ConnectionFactory connectionFactory = new ConnectionFactory();
+        try (Connection connection = connectionFactory.newConnection();
+             Channel channel = connection.createChannel();
+             Scanner scanner = new Scanner(System.in)){
+            boolean durable = true;
+            channel.queueDeclare(QUEUE_NAME, durable, false, false, null);
+            System.out.println(" === Now you can send message ===");
+
+            while (true) {
+                System.out.println("[Send]:");
+                String message = scanner.nextLine();
+                channel.basicPublish("", QUEUE_NAME,
+                        MessageProperties.PERSISTENT_TEXT_PLAIN,
+                        message.getBytes());
+                System.out.println(" [x] Sent '" + message + "'");
+
+                if ("exit".equalsIgnoreCase(message)) {
+                    System.out.println(" [x] Bye");
+                    break;
+                }
+            }
+
+        } catch (IOException | TimeoutException e) {
+            throw new RuntimeException(e);
+        }
+    }
+}