提交warp

Author: MentosL

arex-instrumentation/mq/arex-integration-mqtt/src/main/java/io/arex/inst/mqtt/warp/MessageHeaderWarp.java

@@ -0,0 +1,51 @@
+package io.arex.inst.mqtt.warp;
+
+import org.springframework.messaging.MessageHeaders;
+
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
+
+/**
+ * @author : MentosL
+ * @date : 2023/5/9 23:12
+ */
+public class MessageHeaderWarp extends MessageHeaders {
+
+    private final Map<String, Object> headers;
+    private final UUID id;
+    private final Long timestamp;
+
+    public MessageHeaderWarp(MessageHeaders messageHeaders) {
+        super(null);
+        this.headers = new HashMap<>();
+        for (Map.Entry<String, Object> entry : messageHeaders.entrySet()) {
+            this.headers.put(entry.getKey(), entry.getValue());
+        }
+        this.id = messageHeaders.getId();
+        this.timestamp = messageHeaders.getTimestamp();
+    }
+
+    public MessageHeaderWarp(Map<String, Object> headers) {
+        super(headers);
+        this.headers = headers;
+        this.id = getIdGenerator().generateId();
+        this.timestamp = System.currentTimeMillis();
+    }
+
+    public MessageHeaderWarp(Map<String, Object> headers, UUID id, Long timestamp) {
+        super(headers, id, timestamp);
+        this.headers = headers;
+        this.id = id;
+        this.timestamp = timestamp;
+    }
+
+
+
+
+
+    public Map<String, Object> getHeaders() {
+        return Collections.unmodifiableMap(headers);
+    }
+}