Update cookie to ordered map

buerokratt · Dec 22, 2023 · 0ba30c0 · 0ba30c0
1 parent 1e753c4
commit 0ba30c0
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/main/java/ee/buerokratt/ruuter/domain/steps/ReturnStep.java b/src/main/java/ee/buerokratt/ruuter/domain/steps/ReturnStep.java
@@ -49,16 +49,17 @@ private Map<String, String> formatHeaders(DslInstance di) {
 
     private Map.Entry<String, Object> addDefaultCookies(Map.Entry<String, Object> entry, DslInstance di) {
         if ("Set-Cookie".equals(entry.getKey())) {
-            if (!((HashMap<String, Object>) entry.getValue()).containsKey("Path"))
-                ((HashMap<String, Object>) entry.getValue()).put("Path", "/" );
+            LinkedHashMap<String, Object> cookie = new LinkedHashMap<>((HashMap<String, Object>) entry.getValue());
+            if (!cookie.containsKey("Path"))
+                cookie.put("Path", "/" );
 
-            if (!((HashMap<String, Object>) entry.getValue()).containsKey("HttpOnly"))
-                ((HashMap<String, Object>) entry.getValue()).put("HttpOnly", true);
-
-            if (!((HashMap<String, Object>) entry.getValue()).containsKey("SetSecure"))
-                ((HashMap<String, Object>) entry.getValue()).put("Secure", true);
+            if (!cookie.containsKey("HttpOnly"))
+                cookie.put("HttpOnly", true);
 
+            if (!cookie.containsKey("SetSecure"))
+                cookie.put("Secure", true);
 
+            entry.setValue(cookie);
         }
         return entry;
     }