clojure - 如何在 Clojure 中的嵌套映射中选择键？

Question

假设我有一张这样的 map ( m ):

(def m {:a 1 :b 2 :c {:d 3 :e 4} :e { ... } ....})

我想创建一个仅包含 :a 的新 map , :b和:d来自m ，即结果应该是:

{:a 1 :b 2 :d 3}

我知道我可以使用select-keys轻松获取:a和:b :

(select-keys m [:a :b])

但是有什么好方法可以同时获得 :d ？我正在寻找这样的东西:

(select-keys* m [:a :b [:c :d]])

Clojure 中是否存在这样的函数或者推荐的方法是什么？

Answer 1

在纯 Clojure 中我会这样做:

(defn select-keys* [m paths]
  (into {} (map (fn [p]
                  [(last p) (get-in m p)]))
        paths))

(select-keys* m [[:a] [:b] [:c :d]]) ;;=> {:a 1, :b 2, :d 3}

我更喜欢保持路径的类型为常规，因此所有路径的键序列。在 clojure.spec 中，这将读作

(s/def ::nested-map (s/map-of keyword? 
                              (s/or :num number? :map ::nested-map)))
(s/def ::path (s/coll-of keyword?))
(s/fdef select-keys*
        :args (s/cat :m ::nested-map 
                     :paths (s/coll-of ::path)))