🔄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/EmitLogTopic.java

@@ -1,47 +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.nio.charset.StandardCharsets;
-import java.util.Scanner;
-import java.util.concurrent.TimeoutException;
-
-public class EmitLogTopic {
-
-    private static final String EXCHANGE_NAME = "topic_logs";
-
-    public static void main(String[] args) {
-        ConnectionFactory factory = new ConnectionFactory();
-
-        try (Connection connection = factory.newConnection();
-             Channel channel = connection.createChannel();
-             Scanner scanner = new Scanner(System.in)) {
-
-            // Init exchange
-            channel.exchangeDeclare(EXCHANGE_NAME, "topic");
-
-            // Send message
-            while (true) {
-                System.out.println("[#] Facility ?");
-                String input_facility = scanner.nextLine();
-                System.out.println("[#] Severity ?");
-                String input_severity = scanner.nextLine();
-                if (input_facility.equals("exit") || input_severity.equals("exit")) {
-                    break;
-                }
-
-                System.out.println("[#] Message ?");
-                String input_message = scanner.nextLine();
-                channel.basicPublish(EXCHANGE_NAME, input_facility + '.' + input_severity, null, input_message.getBytes(StandardCharsets.UTF_8));
-                System.out.println("🔶🔶🔶");
-            }
-
-            // Init queue
-        } catch (IOException | TimeoutException e) {
-            System.err.println("=== #Error ===");
-        }
-    }
-}

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

@@ -1,46 +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 ReceiveLogsTopic {
-
-    private static final String EXCHANGE_NAME = "topic_logs";
-
-    public static void main(String[] args) throws Exception {
-        ConnectionFactory factory = new ConnectionFactory();
-
-        Connection connection = factory.newConnection();
-        Channel channel = connection.createChannel();
-
-        channel.exchangeDeclare(EXCHANGE_NAME, "topic");
-
-        // Queue
-        String queue_name = channel.queueDeclare().getQueue();
-
-        if (args.length < 1) {
-            System.err.println("[Null args]");
-        }
-
-        for (String bindingKey : args) {
-            channel.queueBind(queue_name, EXCHANGE_NAME, bindingKey);
-        }
-
-        System.out.println("🎉 Go Go Go");
-
-        DeliverCallback callback = (consumerTag, delivery) -> {
-            String output_message = new String(delivery.getBody(), StandardCharsets.UTF_8);
-            String output_facility_severity = delivery.getEnvelope().getRoutingKey();
-            System.out.printf("✅ [%s] %s\n", output_facility_severity, output_message);
-        };
-
-        channel.basicConsume(queue_name, true, callback, consumerTag -> {});
-
-    }
-}

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);
+        }
+    }
+}