提交warp

Author: MentosL

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

@@ -2,6 +2,8 @@
 
 import org.springframework.messaging.MessageHeaders;
 
+import java.util.Collections;
+import java.util.HashMap;
 import java.util.Map;
 import java.util.UUID;
 
@@ -11,31 +13,39 @@
  */
 public class MessageHeaderWarp extends MessageHeaders {
 
+    private final Map<String, Object> headers;
+    private final UUID id;
+    private final Long timestamp;
 
     public MessageHeaderWarp(MessageHeaders messageHeaders) {
-        super(messageHeaders);
-        this.put(ID,messageHeaders.get(ID));
-        this.put(TIMESTAMP,messageHeaders.get(TIMESTAMP));
+        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 Object put(String key, Object value){
-        if (value == null){
-            return null;
-        }
-        super.getRawHeaders().put(key,value);
-        return value;
-    }
 
-    public void remove(String key){
-        super.getRawHeaders().remove(key);
-    }
 
+
+
+    public Map<String, Object> getHeaders() {
+        return Collections.unmodifiableMap(headers);
+    }
 }