图稿库需求变更

sd-mq/src/main/java/com/sd/mq/entity/ArtworkMessage.java

@@ -3,13 +3,9 @@ package com.sd.mq.entity;
 import lombok.Getter;
 import lombok.Setter;
 
-import java.io.Serializable;
-
 @Getter
 @Setter
-public class ArtworkMessage implements Serializable {
-
-    private static final long serialVersionUID = 1L;
+public class ArtworkMessage extends BaseMessage {
 
     /**
      * 保存路径

sd-mq/src/main/java/com/sd/mq/entity/BaseMessage.java

@@ -0,0 +1,16 @@
+package com.sd.mq.entity;
+
+import lombok.Getter;
+import lombok.Setter;
+
+import java.io.Serializable;
+
+@Getter
+@Setter
+public class BaseMessage implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    private Long messageId;
+
+}

sd-mq/src/main/java/com/sd/mq/entity/TempArtworkMessage.java

@@ -3,13 +3,9 @@ package com.sd.mq.entity;
 import lombok.Getter;
 import lombok.Setter;
 
-import java.io.Serializable;
-
 @Getter
 @Setter
-public class TempArtworkMessage implements Serializable {
-
-    private static final long serialVersionUID = 1L;
+public class TempArtworkMessage extends BaseMessage {
 
     /**
      * 图片url地址

sd-mq/src/main/java/com/sd/mq/util/RabbitMqUtil.java

@@ -1,6 +1,8 @@
 package com.sd.mq.util;
 
+import cn.hutool.core.util.IdUtil;
 import cn.hutool.extra.spring.SpringUtil;
+import com.sd.mq.entity.BaseMessage;
 import org.springframework.amqp.rabbit.core.RabbitTemplate;
 
 public class RabbitMqUtil {
@@ -10,8 +12,11 @@ public class RabbitMqUtil {
     /**
      * 发送消息
      */
-    public static void send(String exchange, String routingKey, Object object) {
-        rabbitTemplate.convertAndSend(exchange, routingKey, object);
+    public static void send(String exchange, String routingKey, BaseMessage message) {
+        if (message.getMessageId() == null) {
+            message.setMessageId(IdUtil.getSnowflake().nextId());
+        }
+        rabbitTemplate.convertAndSend(exchange, routingKey, message);
     }
 
 }