加入队列

src/main/java/com/kiisoo/ic/common/AbstractMsgDisruptor.java

@@ -0,0 +1,95 @@
+package com.kiisoo.ic.common;
+
+import com.lmax.disruptor.BlockingWaitStrategy;
+import com.lmax.disruptor.RingBuffer;
+import com.lmax.disruptor.dsl.Disruptor;
+import com.lmax.disruptor.dsl.ProducerType;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+
+import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ThreadFactory;
+
+/**
+ *  消息Disruptor
+ * Created by hc on 2019/10/22
+ */
+@Setter
+@Getter
+@Slf4j
+public abstract class AbstractMsgDisruptor {
+
+    /**
+     *
+     */
+    Disruptor<MsgEvent> disruptor;
+
+    /**
+     * 消息生产者
+     */
+    MsgEventProducer producer;
+
+    /**
+     * 初始化
+     */
+    @PostConstruct
+    public void init(){
+
+        String name = getName();
+
+        ThreadFactory producerFactory = Executors.defaultThreadFactory();
+
+        // 创建缓冲池
+        MsgEventFactory eventFactory = new MsgEventFactory();
+
+        disruptor = new Disruptor<>(eventFactory, getRingBufferSize(), producerFactory,
+                ProducerType.SINGLE, new BlockingWaitStrategy());
+
+        RingBuffer<MsgEvent> ringBuffer = disruptor.getRingBuffer();
+
+        disruptor.handleEventsWithWorkerPool(getMsgEventHandle());
+
+        disruptor.start();
+
+        producer = new MsgEventProducer(ringBuffer);
+    }
+
+    /**
+     * 结束关闭
+     */
+    @PreDestroy
+    public void close(){
+        disruptor.shutdown();
+    }
+
+    /**
+     * 消息发送
+     * @param dataMsg
+     */
+    public void send(DisruptorDataI dataMsg){
+        producer.onData(dataMsg);
+    }
+
+    /**
+     * 环形队列槽数量
+     * 必须是2的N次方
+     * @return 数量
+     */
+    public abstract int getRingBufferSize();
+
+    /**
+     * 处理消息对象
+     * 多个Woker同时消费(消费者不重复消费消息)
+     * @return 处理woker
+     */
+    public abstract AbstractMsgEventHandler[] getMsgEventHandle();
+
+    /**
+     * 主题名称
+     */
+    public abstract String getName();
+
+}

src/main/java/com/kiisoo/ic/common/AbstractMsgEventHandler.java

@@ -0,0 +1,22 @@
+package com.kiisoo.ic.common;
+
+import com.lmax.disruptor.WorkHandler;
+
+/**
+ *  数据槽工厂
+ * Created by hc on 2019/10/22
+ */
+public abstract class AbstractMsgEventHandler implements WorkHandler<MsgEvent> {
+
+    @Override
+    public void onEvent(MsgEvent msgEvent) throws Exception {
+        DisruptorDataI dataMsg = msgEvent.getData();
+        handler(dataMsg);
+    }
+
+    /**
+     * 消息对象处理
+     * @param dataMsg 消息
+     */
+    public abstract void handler(DisruptorDataI dataMsg);
+}

src/main/java/com/kiisoo/ic/common/DisruptorCallackI.java

@@ -0,0 +1,9 @@
+package com.kiisoo.ic.common;
+
+/**
+ * 消费者回调接口
+ */
+public interface DisruptorCallackI {
+
+    void handel(DisruptorDataI dataMsg) throws Exception;
+}

src/main/java/com/kiisoo/ic/common/DisruptorConsume.java

@@ -0,0 +1,26 @@
+package com.kiisoo.ic.common;
+
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * 消费者
+ */
+@Setter
+@Slf4j
+public class DisruptorConsume extends AbstractMsgEventHandler {
+
+    /** 回调业务 */
+    private DisruptorCallackI disruptorCallack;
+
+    @Override
+    public void handler(DisruptorDataI dataMsg) {
+        try {
+            disruptorCallack.handel(dataMsg);
+        }catch (Exception e){
+            log.error("DisruptorConsume error",e);
+        }
+
+    }
+
+}

src/main/java/com/kiisoo/ic/common/DisruptorDataI.java

@@ -0,0 +1,8 @@
+package com.kiisoo.ic.common;
+
+/**
+ * 数据
+ */
+public interface DisruptorDataI {
+
+}

src/main/java/com/kiisoo/ic/common/MsgEvent.java

@@ -0,0 +1,19 @@
+package com.kiisoo.ic.common;
+
+import lombok.Getter;
+import lombok.Setter;
+
+/**
+ *  数据槽
+ * Created by hc on 2019/10/22
+ */
+@Getter
+@Setter
+public class MsgEvent {
+
+    /**
+     * 接口收到的一次一批消息
+     */
+    private DisruptorDataI data;
+
+}

src/main/java/com/kiisoo/ic/common/MsgEventFactory.java

@@ -0,0 +1,15 @@
+package com.kiisoo.ic.common;
+
+import com.lmax.disruptor.EventFactory;
+
+/**
+ *  数据槽工厂
+ * Created by hc on 2019/10/22
+ */
+public class MsgEventFactory implements EventFactory<MsgEvent> {
+
+    @Override
+    public MsgEvent newInstance() {
+        return new MsgEvent();
+    }
+}

src/main/java/com/kiisoo/ic/common/MsgEventProducer.java

@@ -0,0 +1,26 @@
+package com.kiisoo.ic.common;
+
+import com.lmax.disruptor.RingBuffer;
+
+/**
+ *  消息生产者
+ * Created by hc on 2019/10/22
+ */
+public class MsgEventProducer {
+
+    private final RingBuffer<MsgEvent> ringBuffer;
+
+    public MsgEventProducer(RingBuffer<MsgEvent> ringBuffer){
+        this.ringBuffer = ringBuffer;
+    }
+
+    public void onData(DisruptorDataI dataMsg){
+        long sequence = ringBuffer.next();
+        try{
+            MsgEvent event = ringBuffer.get(sequence);
+            event.setData(dataMsg);
+        }finally{
+            ringBuffer.publish(sequence);
+        }
+    }
+}

src/main/java/com/kiisoo/ic/synchronous/controller/SynchronousController.java

@@ -5,6 +5,8 @@ import com.google.gson.JsonObject;
 import com.kiisoo.ic.customer.CustomerService;
 import com.kiisoo.ic.store.entity.PoiStore;
 import com.kiisoo.ic.synchronous.entity.TurnBackDTO;
+import com.kiisoo.ic.synchronous.entity.TurnBackVO;
+import com.kiisoo.ic.synchronous.service.BackReceiveService;
 import com.kiisoo.ic.utils.DataImportUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -32,6 +34,8 @@ public class SynchronousController {
         this.customerService = customerService;
     }
 
+    @Autowired
+    private BackReceiveService backReceiveService;
     /**
      * 导入店铺接口
      * @return
@@ -75,16 +79,16 @@ public class SynchronousController {
         return DataImportUtil.succ(1);
     }
 
-//    /**
+    /**
-//     * 好友添加回调通知
+     * 好友添加回调通知
-//     * @return 是否成功
+     * @return 是否成功
-//     */
+     */
-//    @RequestMapping(value = "/ics/customer",method = RequestMethod.POST,consumes = "application/json")
+    @RequestMapping(value = "/ics/queue/customer",method = RequestMethod.POST,consumes = "application/json")
-//    @ResponseBody
+    @ResponseBody
-//    public String turnQueueBack(@RequestBody List<TurnBackDTO> turnBackDTOS){
+    public String turnQueueBack(@RequestBody List<TurnBackDTO> turnBackDTOS){
-//        String str = JSON.toJSONString(turnBackDTOS);
+        TurnBackVO turnBackVO = new TurnBackVO();
-//        System.out.println(str);
+        turnBackVO.setTurnBackDTOS(turnBackDTOS);
-//        turnBackDTOS.forEach(customerService::turnBack);
+        backReceiveService.send(turnBackVO);
-//        return DataImportUtil.succ(1);
+        return DataImportUtil.succ(1);
-//    }
+    }
 }

src/main/java/com/kiisoo/ic/synchronous/entity/TurnBackVO.java

@@ -0,0 +1,13 @@
+package com.kiisoo.ic.synchronous.entity;
+
+import com.kiisoo.ic.common.DisruptorDataI;
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+public class TurnBackVO implements DisruptorDataI {
+
+    List<TurnBackDTO> turnBackDTOS;
+
+}

src/main/java/com/kiisoo/ic/synchronous/service/BackDisposeService.java

@@ -0,0 +1,22 @@
+package com.kiisoo.ic.synchronous.service;
+
+import com.alibaba.fastjson.JSON;
+import com.kiisoo.ic.common.DisruptorCallackI;
+import com.kiisoo.ic.common.DisruptorDataI;
+import com.kiisoo.ic.customer.CustomerService;
+import com.kiisoo.ic.synchronous.entity.TurnBackVO;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+@Service
+public class BackDisposeService implements DisruptorCallackI {
+
+    @Autowired
+    private CustomerService customerService;
+
+    @Override
+    public void handel(DisruptorDataI dataMsg) throws Exception {
+        TurnBackVO channelStateOrderDTO = (TurnBackVO)dataMsg;
+        channelStateOrderDTO.getTurnBackDTOS().forEach(customerService::turnBack);
+    }
+}

src/main/java/com/kiisoo/ic/synchronous/service/BackReceiveService.java

@@ -0,0 +1,33 @@
+package com.kiisoo.ic.synchronous.service;
+
+import com.kiisoo.ic.common.*;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+@Service
+public class BackReceiveService extends AbstractMsgDisruptor{
+
+    @Autowired
+    private BackDisposeService backDisposeService;
+
+    @Override
+    public int getRingBufferSize() {
+        return 1024;
+    }
+
+    @Override
+    public AbstractMsgEventHandler[] getMsgEventHandle() {
+        AbstractMsgEventHandler[] list = new AbstractMsgEventHandler[10];
+        for(int i = 0; i < 10; i++){
+            DisruptorConsume disruptorConsume = new DisruptorConsume();
+            disruptorConsume.setDisruptorCallack(backDisposeService);
+            list[i] =  disruptorConsume;
+        }
+        return list;
+    }
+
+    @Override
+    public String getName() {
+        return null;
+    }
+}

src/main/resources/application-dev.yml

@@ -8,7 +8,7 @@ mybatis:
 
 spring:
   datasource:
-    url: jdbc:mysql://192.168.0.215:3306/ic?characterEncoding=utf8&allowMultiQueries=true&autoReconnect=true
+    url: jdbc:mysql://192.168.0.215:3306/bsd?characterEncoding=utf8&allowMultiQueries=true&autoReconnect=true
     username: p2p
     password: p2p
     driver-class-name: com.mysql.jdbc.Driver