status-im
/
timbre
mirror of https://github.com/status-im/timbre.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Make defnp support multi-arity functions (@maurolopes)

This commit is contained in:
Mauro Lopes 2014-08-07 02:34:25 -03:00 committed by Peter Taoussanis
parent ee49e44a46
commit 6b6395db10
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/taoensso/timbre/profiling.clj
View File

@ -218,12 +218,17 @@
(defmacro defnp "Like `defn` but wraps body in `p` macro."
{:arglists '([name ?doc-string ?attr-map [params] ?prepost-map body])}
[name & sigs]
(let [[name [params & sigs]] (encore/name-with-attrs name sigs)
prepost-map (when (and (map? (first sigs)) (next sigs)) (first sigs))
body (if prepost-map (next sigs) sigs)]
`(defn ~name ~params ~(or prepost-map {})
(pspy ~(clojure.core/name name)
~@body))))
(let [[name sigs] (encore/name-with-attrs name sigs)
[sigs func->str] (if (vector? (first sigs)) ;only one arity
[(list sigs) (fn [name _] (clojure.core/name name))]
[sigs (fn [name params] (str (clojure.core/name name) \_ (count params)))])
new-sigs (map (fn [[params & others]]
(let [[prepost-map & body] (if (and (map? (first others)) (next others)) ;has prepost map
others
(cons {} others))]
`(~params ~prepost-map (pspy ~(func->str name params) ~@body))))
sigs)]
`(defn ~name ~@new-sigs)))
(comment (defnp foo "Docstring "[x] "boo" (* x x))
(macroexpand '(defnp foo "Docstring" [x] "boo" (* x x)))