У меня есть макрос, который берет тело:
(defmacro blah [& body] (dostuffwithbody))
Но я хотел бы добавить дополнительный аргумент ключевого слова ему также, поэтому при вызове это могло быть похожим на любой из них:
(blah :specialthingy 0 body morebody lotsofbody)
(blah body morebody lotsofboy)
Как я могу сделать это? Обратите внимание, что я использую Clojure 1.2, таким образом, я также использую новый дополнительный аргумент ключевого слова destructuring материал. Я наивно пытался сделать это:
(defmacro blah [& {specialthingy :specialthingy} & body])
Но очевидно который не удался хорошо. Как я могу выполнить это или что-то подобное?
Возможно, что-то вроде следующего (также смотрите, как defn
и некоторые другие макросы в clojure. core
определены):
(defmacro blah [& maybe-option-and-body]
(let [has-option (= :specialthingy (first maybe-option-and-body))
option-val (if has-option (second maybe-option-and-body)) ; nil otherwise
body (if has-option (nnext maybe-option-and-body) maybe-option-and-body)]
...))
В качестве альтернативы вы можете быть более изощренными; это может быть целесообразно, если вы думаете, что в какой-то момент захотите иметь несколько возможных вариантов:
(defn foo [& args]
(let [aps (partition-all 2 args)
[opts-and-vals ps] (split-with #(keyword? (first %)) aps)
options (into {} (map vec opts-and-vals))
positionals (reduce into [] ps)]
[options positionals]))
(foo :a 1 :b 2 3 4 5)
; => [{:a 1, :b 2} [3 4 5]]
Michał Marczyk хорошо ответил на ваш вопрос, но если вы готовы принять (foo {} остальные аргументы), то есть карту с необязательными ключевыми словами в качестве первого аргумента (пустую в этом примере), то этот более простой код будет работать отлично:
(defn foo [{:keys [a b]} & rest] (list a b rest))
(foo {} 2 3)
=> (nil nil (2 3))
(foo {:a 1} 2 3)
=> (1 nil (2 3))
То же самое работает для defmacro
. Преимущество этого способа в том, что вам не нужно запоминать специальный синтаксис для необязательных параметров ключевых слов и реализовывать код для обработки этого специального синтаксиса (возможно, другие люди, которые будут вызывать ваш макрос, привыкли указывать пары ключ-значение в карте, а не вне ее). Недостатком, конечно, является то, что вам всегда придется предоставлять карту, даже если вы не хотите предоставлять никаких аргументов в виде ключевых слов.
Небольшое улучшение, когда вы хотите избавиться от этого недостатка - проверка того, предоставили ли вы map в качестве первого элемента списка аргументов:
(defn foo [& rest]
(letfn [(inner-foo [{:keys [a b]} & rest] (list a b rest))]
(if (map? (first rest))
(apply inner-foo rest)
(apply inner-foo {} rest))))
(foo 1 2 3)
=> (nil nil (1 2 3))
(foo {:a 2 :b 3} 4 5 6)
=> (2 3 (4 5 6))