#1152 - Refactor EJson internals extracting io.ebean.service.SpiJsonService

src/main/java/io/ebeaninternal/json/ModifyAwareMap.java

@@ -0,0 +1,122 @@
+package io.ebeaninternal.json;
+
+import java.util.Collection;
+import java.util.LinkedHashSet;
+import java.util.Map;
+import java.util.Set;
+
+/**
+ * Map that is wraps an underlying map for the purpose of detecting changes.
+ */
+public class ModifyAwareMap<K, V> implements Map<K, V>, ModifyAwareOwner {
+
+  private static final long serialVersionUID = 1;
+
+  final ModifyAwareOwner owner;
+
+  /**
+   * The underlying map.
+   */
+  final Map<K, V> map;
+
+  public ModifyAwareMap(Map<K, V> underyling) {
+    this.map = underyling;
+    this.owner = new ModifyAwareFlag();
+  }
+
+  public ModifyAwareMap(ModifyAwareOwner owner, Map<K, V> underyling) {
+    this.owner = owner;
+    this.map = underyling;
+  }
+
+  @Override
+  public String toString() {
+    return map.toString();
+  }
+
+  @Override
+  public boolean isMarkedDirty() {
+    return owner.isMarkedDirty();
+  }
+
+  @Override
+  public void markAsModified() {
+    owner.markAsModified();
+  }
+
+  @Override
+  public void resetMarkedDirty() {
+    owner.resetMarkedDirty();
+  }
+
+  @Override
+  public int size() {
+    return map.size();
+  }
+
+  @Override
+  public boolean isEmpty() {
+    return map.isEmpty();
+  }
+
+  @Override
+  public boolean containsKey(Object key) {
+    return map.containsKey(key);
+  }
+
+  @Override
+  public boolean containsValue(Object value) {
+    return map.containsValue(value);
+  }
+
+  @Override
+  public V get(Object key) {
+    return map.get(key);
+  }
+
+  @Override
+  public V put(K key, V value) {
+    markAsModified();
+    return map.put(key, value);
+  }
+
+  @Override
+  public V remove(Object key) {
+    V value = map.remove(key);
+    if (value != null) {
+      markAsModified();
+    }
+    return value;
+  }
+
+  @Override
+  public void putAll(Map<? extends K, ? extends V> m) {
+    markAsModified();
+    map.putAll(m);
+  }
+
+
+  @Override
+  public void clear() {
+    if (!map.isEmpty()) {
+      markAsModified();
+    }
+    map.clear();
+  }
+
+  @Override
+  public Set<K> keySet() {
+    return new ModifyAwareSet<>(this, map.keySet());
+  }
+
+  @Override
+  public Collection<V> values() {
+    return new ModifyAwareSet<>(this, new LinkedHashSet<>(map.values()));
+  }
+
+  @Override
+  public Set<Map.Entry<K, V>> entrySet() {
+    return new ModifyAwareSet<>(this, map.entrySet());
+  }
+
+}