clojure中的函数定义宏

时间:2013-12-26 20:40:25

标签: clojure clojurescript

我正在尝试制作一个可以改变它的宏:

(defn-check my-checked-function
  check-function
  [a A
   b B]
  (do-something a b))

成:

(defn my-checked-function [a b]
  {:pre [(= (check-function a) A) 
         (= (check-function b) B)]}
  (do-something a b))

我是clojure的新手,有人可以告诉我该怎么做吗?

1 个答案:

答案 0 :(得分:7)

(defmacro defn-check [name check-fn args & tail]
  (let [docstring (if (string? (first tail)) (first tail))
        tail (if docstring (next tail) tail)
        params (take-nth 2 args)
        checks (take-nth 2 (next args))
        cf (gensym "cf__")]
    `(let [~cf ~check-fn]
       (defn ~name
         ~@(if docstring [docstring])
          ~(vec params)
         {:pre ~(mapv (fn [p c]
                        `(= (~cf ~p) ~c))
                      params
                      checks)}
         ~@tail))))

这允许您根据需要指定文档字符串,但不允许指定属性映射(请参阅(doc defn);添加对此的支持将非常简单。)

示例:

user> (defn-check funky-add odd? [x true y false] "foo" (+ x y))
#'user/funky-add
user> (doc funky-add)
-------------------------
user/funky-add
([x y])
  foo
nil
user> (funky-add 1 2)
3
user> (funky-add 2 1)
AssertionError Assert failed: (clojure.core/= (cf__2268 x) true)  user/eval2269/funky-add--2270 (form-init1446120099766722611.clj:1)