1 files changed, 50 insertions, 0 deletions
diff --git a/product/database/transactions/IdentityMapProxy.cs b/product/database/transactions/IdentityMapProxy.cs
new file mode 100644
index 0000000..a251d10
--- /dev/null
+++ b/product/database/transactions/IdentityMapProxy.cs
@@ -0,0 +1,50 @@
+using System;
+using System.Collections.Generic;
+using gorilla.commons.utility;
+
+namespace momoney.database.transactions
+{
+    public class IdentityMapProxy<Key, Value> : IIdentityMap<Key, Value> where Value : Identifiable<Guid>
+    {
+        readonly IIdentityMap<Key, Value> real_map;
+        readonly IChangeTracker<Value> change_tracker;
+
+        public IdentityMapProxy(IChangeTracker<Value> change_tracker, IIdentityMap<Key, Value> real_map)
+        {
+            this.change_tracker = change_tracker;
+            this.real_map = real_map;
+        }
+
+        public IEnumerable<Value> all()
+        {
+            return real_map.all();
+        }
+
+        public void add(Key key, Value value)
+        {
+            change_tracker.register(value);
+            real_map.add(key, value);
+        }
+
+        public void update_the_item_for(Key key, Value new_value)
+        {
+            real_map.update_the_item_for(key, new_value);
+        }
+
+        public bool contains_an_item_for(Key key)
+        {
+            return real_map.contains_an_item_for(key);
+        }
+
+        public Value item_that_belongs_to(Key key)
+        {
+            return real_map.item_that_belongs_to(key);
+        }
+
+        public void evict(Key key)
+        {
+            change_tracker.delete(real_map.item_that_belongs_to(key));
+            real_map.evict(key);
+        }
+    }
+}
+\ No newline at end of file