From 3a99364128a0216834bff767766141f117055566 Mon Sep 17 00:00:00 2001
From: Huahai Yang <huahai.yang@gmail.com>
Date: Wed, 4 Jan 2023 17:49:15 -0800
Subject: [PATCH] add inline.clj

---
 src/datalevin/inline.clj | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 src/datalevin/inline.clj

diff --git a/src/datalevin/inline.clj b/src/datalevin/inline.clj
new file mode 100644
index 00000000..82c9d2e1
--- /dev/null
+++ b/src/datalevin/inline.clj
@@ -0,0 +1,37 @@
+(ns ^:no-doc datalevin.inline
+  (:refer-clojure :exclude [assoc update]))
+
+(defn assoc
+  {:inline
+   (fn
+     ([m k v]
+      `(clojure.lang.RT/assoc ~m ~k ~v))
+     ([m k v & kvs]
+      (assert (even? (count kvs)))
+      `(assoc (assoc ~m ~k ~v) ~@kvs)))}
+  ([map key val] (clojure.lang.RT/assoc map key val))
+  ([map key val & kvs]
+   (let [ret (clojure.lang.RT/assoc map key val)]
+     (if kvs
+       (if (next kvs)
+         (recur ret (first kvs) (second kvs) (nnext kvs))
+         (throw (IllegalArgumentException.
+                  "assoc expects even number of arguments after map/vector, found odd number")))
+       ret))))
+
+(defn update
+  {:inline
+   (fn
+     ([m k f & more]
+      `(let [m# ~m k# ~k]
+         (assoc m# k# (~f (get m# k#) ~@more)))))}
+  ([m k f]
+   (assoc m k (f (get m k))))
+  ([m k f x]
+   (assoc m k (f (get m k) x)))
+  ([m k f x y]
+   (assoc m k (f (get m k) x y)))
+  ([m k f x y z]
+   (assoc m k (f (get m k) x y z)))
+  ([m k f x y z & more]
+   (assoc m k (apply f (get m k) x y z more))))