From 8f8f0ca83d8f3aadb0b5507e085c95ca01ddc767 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Thu, 28 Mar 2024 23:46:32 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81rabbitmq?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                       |  4 ++++
 .../service/CoalParameterDefService.java      | 23 +++++++++++++++++--
 .../service/OrganizationService.java          |  9 ++++++--
 .../coal/rabbitmq/RabbitMQConfiguration.java  | 20 ++++++++++++++++
 .../coal/rabbitmq/RabbitMQService.java        | 20 ++++++++++++++++
 src/main/resources/application.yaml           |  6 +++++
 6 files changed, 78 insertions(+), 4 deletions(-)
 create mode 100644 src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQConfiguration.java
 create mode 100644 src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQService.java

diff --git a/pom.xml b/pom.xml
index 9e9b0b2f..dd242eac 100644
--- a/pom.xml
+++ b/pom.xml
@@ -51,6 +51,10 @@
     </pluginRepositories>
     <dependencies>
 
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-amqp</artifactId>
+        </dependency>
 
         <dependency>
             <groupId>org.reflections</groupId>
diff --git a/src/main/java/cn/lihongjie/coal/coalParameterDef/service/CoalParameterDefService.java b/src/main/java/cn/lihongjie/coal/coalParameterDef/service/CoalParameterDefService.java
index 56b18893..af0fd0ff 100644
--- a/src/main/java/cn/lihongjie/coal/coalParameterDef/service/CoalParameterDefService.java
+++ b/src/main/java/cn/lihongjie/coal/coalParameterDef/service/CoalParameterDefService.java
@@ -27,6 +27,9 @@ import org.apache.commons.lang3.StringUtils;
 import org.jgrapht.graph.DefaultDirectedGraph;
 import org.jgrapht.graph.DefaultEdge;
 import org.jgrapht.traverse.TopologicalOrderIterator;
+import org.springframework.amqp.rabbit.annotation.Exchange;
+import org.springframework.amqp.rabbit.annotation.QueueBinding;
+import org.springframework.amqp.rabbit.annotation.RabbitListener;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.convert.ConversionService;
 import org.springframework.core.io.ClassPathResource;
@@ -43,8 +46,8 @@ import java.util.Optional;
 @Service
 @Slf4j
 @Transactional
-public
-class CoalParameterDefService extends BaseService<CoalParameterDefEntity, CoalParameterDefRepository> {
+public class CoalParameterDefService
+        extends BaseService<CoalParameterDefEntity, CoalParameterDefRepository> {
 
     @Autowired CoalParameterDefRepository repository;
 
@@ -177,6 +180,22 @@ class CoalParameterDefService extends BaseService<CoalParameterDefEntity, CoalPa
                 .forEach(organizationEntity -> initDefault(organizationEntity.getId()));
     }
 
+    @RabbitListener(
+            bindings = {
+                @QueueBinding(
+                        value =
+                                @org.springframework.amqp.rabbit.annotation.Queue(
+                                        value = "coalParameterDef.init",
+                                        durable = "true"),
+                        exchange =
+                                @org.springframework.amqp.rabbit.annotation.Exchange(
+                                        value = "sysExchange", declare = Exchange.FALSE),
+                        key = "organization.create")
+            })
+    public void onOrganizationCreate(String orgId) {
+        initDefault(orgId);
+    }
+
     @SneakyThrows
     public void initDefault(String orgId) {
 
diff --git a/src/main/java/cn/lihongjie/coal/organization/service/OrganizationService.java b/src/main/java/cn/lihongjie/coal/organization/service/OrganizationService.java
index 09b61aa2..db6e939c 100644
--- a/src/main/java/cn/lihongjie/coal/organization/service/OrganizationService.java
+++ b/src/main/java/cn/lihongjie/coal/organization/service/OrganizationService.java
@@ -10,6 +10,7 @@ import cn.lihongjie.coal.organization.dto.UpdateOrganizationDto;
 import cn.lihongjie.coal.organization.entity.OrganizationEntity;
 import cn.lihongjie.coal.organization.mapper.OrganizationMapper;
 import cn.lihongjie.coal.organization.repository.OrganizationRepository;
+import cn.lihongjie.coal.rabbitmq.RabbitMQService;
 import cn.lihongjie.coal.user.dto.CreateOrgAdminDto;
 import cn.lihongjie.coal.user.service.UserService;
 
@@ -17,6 +18,7 @@ import jakarta.annotation.PostConstruct;
 
 import lombok.extern.slf4j.Slf4j;
 
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.convert.ConversionService;
 import org.springframework.data.domain.Page;
@@ -42,10 +44,13 @@ class OrganizationService extends BaseService<OrganizationEntity, OrganizationRe
 
     @Autowired CoalParameterDefService coalParameterDefService;
 
+    @Autowired RabbitMQService rabbitMQService;
 
     @PostConstruct
     public void init() {}
 
+    @Autowired RabbitTemplate rabbitTemplate;
+
     public OrganizationDto create(CreateOrganizationDto request) {
 
         OrganizationEntity entity = mapper.toEntity(request);
@@ -57,7 +62,7 @@ class OrganizationService extends BaseService<OrganizationEntity, OrganizationRe
         dto.setUsername(request.getOrgAdminUserName());
         dto.setPassword(request.getOrgAdminPassword());
         userService.createOrgAdmin(dto);
-        coalParameterDefService.initDefault(entity.getId());
+        rabbitMQService.sendToSysExchange("organization.create", entity.getId());
         return getById(entity.getId());
     }
 
@@ -66,7 +71,7 @@ class OrganizationService extends BaseService<OrganizationEntity, OrganizationRe
         this.mapper.updateEntity(entity, request);
 
         this.repository.save(entity);
-
+        rabbitMQService.sendToSysExchange("organization.update", entity.getId());
         return getById(entity.getId());
     }
 
diff --git a/src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQConfiguration.java b/src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQConfiguration.java
new file mode 100644
index 00000000..ba9a67dd
--- /dev/null
+++ b/src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQConfiguration.java
@@ -0,0 +1,20 @@
+package cn.lihongjie.coal.rabbitmq;
+
+import org.springframework.amqp.core.TopicExchange;
+import org.springframework.amqp.rabbit.annotation.EnableRabbit;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import java.util.*;
+
+@EnableRabbit
+@Configuration
+public class RabbitMQConfiguration {
+
+
+    @Bean
+    public TopicExchange topicExchange() {
+        return new TopicExchange("sysExchange", true, false);
+    }
+
+}
diff --git a/src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQService.java b/src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQService.java
new file mode 100644
index 00000000..bf16de81
--- /dev/null
+++ b/src/main/java/cn/lihongjie/coal/rabbitmq/RabbitMQService.java
@@ -0,0 +1,20 @@
+package cn.lihongjie.coal.rabbitmq;
+
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+@Service
+public class RabbitMQService {
+
+    @Autowired RabbitTemplate rabbitTemplate;
+
+    public void send(String exchange, String routingKey, String data){
+
+        rabbitTemplate.convertAndSend(exchange, routingKey, data);
+    }
+
+    public void sendToSysExchange(String routingKey, String data) {
+        send("sysExchange", routingKey, data);
+    }
+}
diff --git a/src/main/resources/application.yaml b/src/main/resources/application.yaml
index 9e2cc6d0..66f05ad6 100644
--- a/src/main/resources/application.yaml
+++ b/src/main/resources/application.yaml
@@ -40,6 +40,12 @@ management:
 
 
 spring:
+  rabbitmq:
+    host: ${RABBITMQ_HOST:localhost}
+    port: ${RABBITMQ_PORT:5672}
+    username: ${RABBITMQ_USER:coal}
+    password: ${RABBITMQ_PASSWORD:coal}
+    virtual-host: ${RABBITMQ_VHOST:/coal}
   quartz:
     jdbc:
       initialize-schema: NEVER