From d74032da9b0a6a893a8a03e3fa64d2f0ae2a9f2b Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Sat, 25 Sep 2021 17:19:39 +0800
Subject: [PATCH] add rabbitmq

---
 src/main/java/cc/mrbird/febs/rabbit/consumer/AgentConsumer.java      |   22 ++++
 src/main/java/cc/mrbird/febs/mall/entity/AgentInfo.java              |   44 ++++++++
 src/test/java/cc/mrbird/febs/AgentTest.java                          |   24 ++++
 src/main/resources/application-prod.yml                              |    8 +
 src/main/resources/application-test.yml                              |    8 +
 pom.xml                                                              |   17 +++
 src/main/java/cc/mrbird/febs/rabbit/constants/RouteKeyConstants.java |   10 ++
 src/main/java/cc/mrbird/febs/common/configure/RabbitConfigure.java   |   50 ++++++++++
 src/main/java/cc/mrbird/febs/rabbit/constants/ExchangeConstants.java |   10 ++
 src/main/java/cc/mrbird/febs/rabbit/constants/QueueConstants.java    |   10 ++
 src/main/java/cc/mrbird/febs/rabbit/producter/AgentProducer.java     |   47 +++++++++
 src/main/resources/application-dev.yml                               |    8 +
 src/main/resources/mapper/modules/MallMoneyFlowMapper.xml            |    4 
 13 files changed, 257 insertions(+), 5 deletions(-)

diff --git a/pom.xml b/pom.xml
index 455780a..9e4f7c5 100644
--- a/pom.xml
+++ b/pom.xml
@@ -267,6 +267,23 @@
             <artifactId>mapstruct</artifactId>
             <version>${mapstruct.version}</version>
         </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-amqp</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <scope>test</scope>
+            <exclusions>
+                <exclusion>
+                    <groupId>org.junit.vintage</groupId>
+                    <artifactId>junit-vintage-engine</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
     </dependencies>
 
     <build>
diff --git a/src/main/java/cc/mrbird/febs/common/configure/RabbitConfigure.java b/src/main/java/cc/mrbird/febs/common/configure/RabbitConfigure.java
new file mode 100644
index 0000000..96a523b
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/common/configure/RabbitConfigure.java
@@ -0,0 +1,50 @@
+package cc.mrbird.febs.common.configure;
+
+import cc.mrbird.febs.rabbit.constants.ExchangeConstants;
+import cc.mrbird.febs.rabbit.constants.QueueConstants;
+import cc.mrbird.febs.rabbit.constants.RouteKeyConstants;
+import org.springframework.amqp.core.Binding;
+import org.springframework.amqp.core.BindingBuilder;
+import org.springframework.amqp.core.DirectExchange;
+import org.springframework.amqp.core.Queue;
+import org.springframework.amqp.rabbit.connection.ConnectionFactory;
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import org.springframework.beans.factory.config.ConfigurableBeanFactory;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Scope;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+@Configuration
+public class RabbitConfigure {
+
+    @Resource
+    private ConnectionFactory connectionFactory;
+
+    @Bean
+    @Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE)
+    public RabbitTemplate rabbitTemplate() {
+        return new RabbitTemplate(connectionFactory);
+    }
+
+    @Bean
+    public DirectExchange defaultExchange() {
+        return new DirectExchange(ExchangeConstants.EXCHANGE_DEFAULT);
+    }
+
+    @Bean
+    public Queue defaultQueue() {
+        return new Queue(QueueConstants.QUEUE_DEFAULT);
+    }
+
+    @Bean
+    public Binding defaultBind() {
+        return BindingBuilder.bind(defaultQueue()).to(defaultExchange()).with(RouteKeyConstants.ROUTE_KEY_DEFAULT);
+    }
+}
diff --git a/src/main/java/cc/mrbird/febs/mall/entity/AgentInfo.java b/src/main/java/cc/mrbird/febs/mall/entity/AgentInfo.java
new file mode 100644
index 0000000..86b661e
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/mall/entity/AgentInfo.java
@@ -0,0 +1,44 @@
+package cc.mrbird.febs.mall.entity;
+
+import lombok.Data;
+
+import java.math.BigDecimal;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+@Data
+public class AgentInfo {
+
+    /**
+     * 下单数量要求 1-直推 2-团队
+     */
+    private Integer orderType;
+
+    /**
+     * 下单数量
+     */
+    private Integer orderCnt;
+
+    /**
+     * 需要上一级的数量
+     */
+    private Integer lastCnt;
+
+    /**
+     * 直推收益
+     */
+    private BigDecimal directIncome;
+
+    /**
+     * 团队收益
+     */
+    private BigDecimal TeamIncome;
+
+    /**
+     * 团队收益 1-指定金额 2-比例
+     */
+    private BigDecimal teamIncomeType;
+
+}
diff --git a/src/main/java/cc/mrbird/febs/rabbit/constants/ExchangeConstants.java b/src/main/java/cc/mrbird/febs/rabbit/constants/ExchangeConstants.java
new file mode 100644
index 0000000..aec9654
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/rabbit/constants/ExchangeConstants.java
@@ -0,0 +1,10 @@
+package cc.mrbird.febs.rabbit.constants;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+public class ExchangeConstants {
+
+    public static final String EXCHANGE_DEFAULT = "exchange_default";
+}
diff --git a/src/main/java/cc/mrbird/febs/rabbit/constants/QueueConstants.java b/src/main/java/cc/mrbird/febs/rabbit/constants/QueueConstants.java
new file mode 100644
index 0000000..2656088
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/rabbit/constants/QueueConstants.java
@@ -0,0 +1,10 @@
+package cc.mrbird.febs.rabbit.constants;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+public class QueueConstants {
+
+    public static final String QUEUE_DEFAULT = "queue_default";
+}
diff --git a/src/main/java/cc/mrbird/febs/rabbit/constants/RouteKeyConstants.java b/src/main/java/cc/mrbird/febs/rabbit/constants/RouteKeyConstants.java
new file mode 100644
index 0000000..f9d7c1f
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/rabbit/constants/RouteKeyConstants.java
@@ -0,0 +1,10 @@
+package cc.mrbird.febs.rabbit.constants;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+public class RouteKeyConstants {
+
+    public static final String ROUTE_KEY_DEFAULT = "route_key_default";
+}
diff --git a/src/main/java/cc/mrbird/febs/rabbit/consumer/AgentConsumer.java b/src/main/java/cc/mrbird/febs/rabbit/consumer/AgentConsumer.java
new file mode 100644
index 0000000..79f201f
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/rabbit/consumer/AgentConsumer.java
@@ -0,0 +1,22 @@
+package cc.mrbird.febs.rabbit.consumer;
+
+import cc.mrbird.febs.rabbit.constants.QueueConstants;
+import com.rabbitmq.client.Channel;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.amqp.core.Message;
+import org.springframework.amqp.rabbit.annotation.RabbitListener;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+@Slf4j
+@Component
+public class AgentConsumer {
+
+    @RabbitListener(queues = QueueConstants.QUEUE_DEFAULT)
+    public void agentReturn(Message message, Channel channel) {
+        log.info("消费者:{}", new String(message.getBody()));
+    }
+}
diff --git a/src/main/java/cc/mrbird/febs/rabbit/producter/AgentProducer.java b/src/main/java/cc/mrbird/febs/rabbit/producter/AgentProducer.java
new file mode 100644
index 0000000..8e4f150
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/rabbit/producter/AgentProducer.java
@@ -0,0 +1,47 @@
+package cc.mrbird.febs.rabbit.producter;
+
+import cc.mrbird.febs.rabbit.constants.ExchangeConstants;
+import cc.mrbird.febs.rabbit.constants.RouteKeyConstants;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.amqp.rabbit.connection.CorrelationData;
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.UUID;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+@Slf4j
+@Component
+public class AgentProducer implements RabbitTemplate.ConfirmCallback {
+
+    /**
+     * 配置中配置的RabbitTemplate的是prototype类型,不能直接注入
+     */
+    private RabbitTemplate rabbitTemplate;
+
+    /**
+     * 在构造方法上注入RabbitTemplate
+     *
+     * @param
+     */
+    @Autowired
+    public AgentProducer(RabbitTemplate rabbitTemplate) {
+        this.rabbitTemplate = rabbitTemplate;
+        rabbitTemplate.setConfirmCallback(this);
+    }
+
+    @Override
+    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
+
+    }
+
+    public void sendOrderReturn(Long id) {
+        log.info("发送消息:{}", id);
+        CorrelationData correlationData = new CorrelationData(UUID.randomUUID().toString());
+        rabbitTemplate.convertAndSend(ExchangeConstants.EXCHANGE_DEFAULT, RouteKeyConstants.ROUTE_KEY_DEFAULT, id, correlationData);
+    }
+}
diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml
index ae941c0..0afbd49 100644
--- a/src/main/resources/application-dev.yml
+++ b/src/main/resources/application-dev.yml
@@ -40,4 +40,10 @@
         # 连接池最大阻塞等待时间(使用负值表示没有限制)
         max-wait: 10000
     # 连接超时时间(毫秒)
-    timeout: 5000
\ No newline at end of file
+    timeout: 5000
+  rabbitmq:
+    host: 120.27.238.55
+    port: 5672
+    username: ct_rabbit
+    password: 123456
+    publisher-confirm-type: correlated
\ No newline at end of file
diff --git a/src/main/resources/application-prod.yml b/src/main/resources/application-prod.yml
index 04d0df9..1285e7e 100644
--- a/src/main/resources/application-prod.yml
+++ b/src/main/resources/application-prod.yml
@@ -46,4 +46,10 @@
         # 连接池最大阻塞等待时间(使用负值表示没有限制)
         max-wait: 10000
     # 连接超时时间(毫秒)
-    timeout: 5000
\ No newline at end of file
+    timeout: 5000
+  rabbitmq:
+    host: 120.27.238.55
+    port: 5672
+    username: ct_rabbit
+    password: 123456
+    publisher-confirm-type: correlated
\ No newline at end of file
diff --git a/src/main/resources/application-test.yml b/src/main/resources/application-test.yml
index 394582a..a5d65d6 100644
--- a/src/main/resources/application-test.yml
+++ b/src/main/resources/application-test.yml
@@ -40,4 +40,10 @@
         # 连接池最大阻塞等待时间(使用负值表示没有限制)
         max-wait: 10000
     # 连接超时时间(毫秒)
-    timeout: 5000
\ No newline at end of file
+    timeout: 5000
+  rabbitmq:
+    host: 120.27.238.55
+    port: 5672
+    username: ct_rabbit
+    password: 123456
+    publisher-confirm-type: correlated
\ No newline at end of file
diff --git a/src/main/resources/mapper/modules/MallMoneyFlowMapper.xml b/src/main/resources/mapper/modules/MallMoneyFlowMapper.xml
index e01b963..bbe021b 100644
--- a/src/main/resources/mapper/modules/MallMoneyFlowMapper.xml
+++ b/src/main/resources/mapper/modules/MallMoneyFlowMapper.xml
@@ -14,10 +14,10 @@
         from mall_money_flow a
             inner join mall_member b on a.member_id=b.id
         <where>
-            <if test="record.type == 2">
+            <if test="record.type == 3">
                 and a.amount > 0
             </if>
-            <if test="record.type == 3">
+            <if test="record.type == 2">
                 and 0 > a.amount
             </if>
             <if test="record.memberId != null">
diff --git a/src/test/java/cc/mrbird/febs/AgentTest.java b/src/test/java/cc/mrbird/febs/AgentTest.java
new file mode 100644
index 0000000..9b1f167
--- /dev/null
+++ b/src/test/java/cc/mrbird/febs/AgentTest.java
@@ -0,0 +1,24 @@
+package cc.mrbird.febs;
+
+import cc.mrbird.febs.rabbit.producter.AgentProducer;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+
+/**
+ * @author wzy
+ * @date 2021-09-25
+ **/
+@Slf4j
+@SpringBootTest
+public class AgentTest {
+
+    @Autowired
+    private AgentProducer agentProducer;
+
+    @Test
+    public void agentTest() {
+        agentProducer.sendOrderReturn(1L);
+    }
+}

--
Gitblit v1.9.1