step3-1 pub/sub 기본 예제 (실시간 알림)

villainscode · villainscode · commit e482873a1b6c · 2024-12-04T22:09:47.000+09:00
diff --git a/build.gradle b/build.gradle
@@ -20,6 +20,8 @@ repositories {
 dependencies {
     implementation 'org.springframework.boot:spring-boot-starter-amqp'
     implementation 'org.springframework.boot:spring-boot-starter-web'
+    implementation 'org.springframework.boot:spring-boot-starter-thymeleaf'
+    implementation 'org.springframework.boot:spring-boot-starter-websocket'
     developmentOnly 'org.springframework.boot:spring-boot-devtools'
     testImplementation 'org.springframework.boot:spring-boot-starter-test'
     testImplementation 'org.springframework.amqp:spring-rabbit-test'
diff --git a/src/main/java/net/harunote/hellomessagequeue/step2/RabbitMQConfig.java b/src/main/java/net/harunote/hellomessagequeue/step2/RabbitMQConfig.java
diff --git a/src/main/java/net/harunote/hellomessagequeue/step2/WorkQueuController.java b/src/main/java/net/harunote/hellomessagequeue/step2/WorkQueuController.java
diff --git a/src/main/java/net/harunote/hellomessagequeue/step2/WorkQueueConsumer.java b/src/main/java/net/harunote/hellomessagequeue/step2/WorkQueueConsumer.java
diff --git a/src/main/java/net/harunote/hellomessagequeue/step2/WorkQueueProducer.java b/src/main/java/net/harunote/hellomessagequeue/step2/WorkQueueProducer.java
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/HomeController.java b/src/main/java/net/harunote/hellomessagequeue/step3/HomeController.java
@@ -0,0 +1,15 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.stereotype.Controller;
+import org.springframework.ui.Model;
+import org.springframework.web.bind.annotation.GetMapping;
+
+@Controller
+public class HomeController {
+
+    @GetMapping("/home")
+    public String home(Model model) {
+        model.addAttribute("message", "Welcome to RabbitMQ Sample!");
+        return "home"; // src/main/resources/templates/home.html 파일을 찾음
+    }
+}
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/NotificationController.java b/src/main/java/net/harunote/hellomessagequeue/step3/NotificationController.java
@@ -0,0 +1,23 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/notifications")
+public class NotificationController {
+
+    private final NotificationPublisher publisher;
+
+    public NotificationController(NotificationPublisher publisher) {
+        this.publisher = publisher;
+    }
+
+    @PostMapping
+    public String sendNotification(@RequestBody String message) {
+        publisher.publish(message);
+        return "[#] Notification sent: " + message + "\n";
+    }
+}
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/NotificationMessage.java b/src/main/java/net/harunote/hellomessagequeue/step3/NotificationMessage.java
@@ -0,0 +1,24 @@
+package net.harunote.hellomessagequeue.step3;
+
+public class NotificationMessage {
+    private String message;
+
+    // 기본 생성자 (필수)
+    public NotificationMessage() {
+    }
+
+    // 선택
+    public NotificationMessage(String message) {
+        this.message = message;
+    }
+
+    public String getMessage() {
+        return message;
+    }
+
+    // 필수
+    public void setMessage(String message) {
+        this.message = message;
+    }
+}
+
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/NotificationPublisher.java b/src/main/java/net/harunote/hellomessagequeue/step3/NotificationPublisher.java
@@ -0,0 +1,19 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import org.springframework.stereotype.Component;
+
+@Component
+public class NotificationPublisher {
+
+    private final RabbitTemplate rabbitTemplate;
+
+    public NotificationPublisher(RabbitTemplate rabbitTemplate) {
+        this.rabbitTemplate = rabbitTemplate;
+    }
+
+    public void publish(String message) {
+        rabbitTemplate.convertAndSend(RabbitMQConfig.FANOUT_EXCHANGE, "", message); // Fanout에서 routing key는 무시됨
+        System.out.println("[#] Published Notification: " + message);
+    }
+}
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/NotificationSubscriber.java b/src/main/java/net/harunote/hellomessagequeue/step3/NotificationSubscriber.java
@@ -0,0 +1,27 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.amqp.rabbit.annotation.RabbitListener;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.stereotype.Component;
+
+@Component
+public class NotificationSubscriber {
+
+    public static final String CLIENT_URL = "/topic/notifications";
+    // WebSocket으로 메시지를 전달하기 위한 Spring의 템플릿 클래스.
+    private final SimpMessagingTemplate simpMessagingTemplate;
+
+    public NotificationSubscriber(SimpMessagingTemplate simpMessagingTemplate) {
+        this.simpMessagingTemplate = simpMessagingTemplate;
+    }
+
+    @RabbitListener(queues = RabbitMQConfig.QUEUE_NAME)
+    public void subscriber(String message) {
+        // RabbitMQ Queue에서 메시지 수신
+        // String message = (String) rabbitTemplate.receiveAndConvert(RabbitMQConfig.QUEUE_NAME);
+        System.out.println("Received Notification: " + message);
+        // convertAndSend 를 통해 특정 경로로 메시지를 전달함
+        simpMessagingTemplate.convertAndSend(CLIENT_URL, message); // 클라이언트에 브로드캐스트
+    }
+
+}
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/RabbitMQConfig.java b/src/main/java/net/harunote/hellomessagequeue/step3/RabbitMQConfig.java
@@ -0,0 +1,32 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.amqp.core.Binding;
+import org.springframework.amqp.core.BindingBuilder;
+import org.springframework.amqp.core.FanoutExchange;
+import org.springframework.amqp.core.Queue;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class RabbitMQConfig {
+    public static final String QUEUE_NAME = "notificationQueue";
+    public static final String FANOUT_EXCHANGE = "notificationExchange";
+
+
+    @Bean
+    public Queue notificationQueue() {
+        return new Queue(QUEUE_NAME, false); // 메시지는 volatile로 설정
+    }
+
+    @Bean
+    public FanoutExchange fanoutExchange() {
+        // 메시지를 수신하면 연결된 모든 큐로 브로드캐스트
+        return new FanoutExchange(FANOUT_EXCHANGE);
+    }
+
+    @Bean
+    public Binding bindNotification(Queue notificationQueue, FanoutExchange fanoutExchange) {
+        // BindingBuilder.bind().to() 를 통해 큐와 익스체인지를 연결
+        return BindingBuilder.bind(notificationQueue).to(fanoutExchange);
+    }
+}
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/StompController.java b/src/main/java/net/harunote/hellomessagequeue/step3/StompController.java
@@ -0,0 +1,25 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.stereotype.Controller;
+
+@Controller
+public class StompController {
+
+    private final SimpMessagingTemplate messagingTemplate;
+
+    public StompController(SimpMessagingTemplate messagingTemplate) {
+        this.messagingTemplate = messagingTemplate;
+    }
+
+    @MessageMapping("/send")
+    public void sendMessage(NotificationMessage notificationMessage) {
+        // 수신된 메시지를 브로드캐스팅
+        String message = notificationMessage.getMessage();
+
+        System.out.println("[#] message = " + message);
+        // 클라이언트에 메시지 브로드캐스트
+        messagingTemplate.convertAndSend("/topic/notifications", message);
+    }
+}
diff --git a/src/main/java/net/harunote/hellomessagequeue/step3/WebSocketConfig.java b/src/main/java/net/harunote/hellomessagequeue/step3/WebSocketConfig.java
@@ -0,0 +1,23 @@
+package net.harunote.hellomessagequeue.step3;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+
+@Configuration
+@EnableWebSocketMessageBroker
+public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
+    @Override
+    public void configureMessageBroker(MessageBrokerRegistry config) {
+        config.enableSimpleBroker("/topic"); // 클라이언트 구독 경로
+        config.setApplicationDestinationPrefixes("/app"); // 서버 발행 경로
+    }
+
+    @Override
+    public void registerStompEndpoints(StompEndpointRegistry registry) {
+        registry.addEndpoint("/ws").setAllowedOriginPatterns("*").withSockJS(); // WebSocket 엔드포인트
+    }
+}
diff --git a/src/main/resources/static/css/style.css b/src/main/resources/static/css/style.css
@@ -0,0 +1,8 @@
+body {
+    font-family: Arial, sans-serif;
+    margin: 0;
+    padding: 20px;
+}
+h1 {
+    color: #000000;
+}
diff --git a/src/main/resources/templates/home.html b/src/main/resources/templates/home.html
@@ -0,0 +1,11 @@
+<!DOCTYPE html>
+<html xmlns:th="http://www.thymeleaf.org">
+<head>
+    <title>RabbitMQ Sample</title>
+    <link rel="stylesheet" href="/css/style.css">
+</head>
+<body>
+<h1>RabbitMQ Sample</h1>
+<p th:text="${message}"></p>
+</body>
+</html>
diff --git a/src/main/resources/templates/index.html b/src/main/resources/templates/index.html
