cast datomic.db.Db results from peer/transact to LocalDb. before returning (#1)

Author: theronic

src/compute/datomic_client_memdb/core.clj

@@ -7,6 +7,9 @@
 
 (defonce clients (atom {}))
 
+(defn update-vals [m ks f]
+  (reduce #(update-in % [%2] f) m ks))
+
 (defn- throw-unsupported
   [data]
   (throw (ex-info "Unsupported operation." data)))
@@ -61,7 +64,8 @@
     (LocalDb. (peer/db conn) db-name))
 
   (transact [_ arg-map]
-    @(peer/transact conn (:tx-data arg-map)))
+    (-> @(peer/transact conn (:tx-data arg-map))
+        (update-vals #{:db-before :db-after} #(LocalDb. % db-name))))
 
   (tx-range [_ arg-map]
     (peer/tx-range (peer/log conn) (:start arg-map) (:end arg-map)))
@@ -116,4 +120,4 @@
     (let [new-client (map->Client {:db-lookup      (atom {})
                                    :client-arg-map arg-map})]
       (swap! clients assoc arg-map new-client)
-      new-client)))
+      new-client)))