访问 Clojure.zip 按路径模式划分的树



>假设我有一棵树,我想在那里访问 - 这应该包括修改访问的项目的可能性 - 所有与路径匹配的项目

(def visit-path [:b :all :x :all])

我使用 :all 作为通配符来匹配所有子节点。在以下示例树中,

(def my-tree
  {:a "a"
   :b
   {:b-1
    {:x
     {:b-1-1 "b11"
      :b-1-2 "b12"}}
    :b-2
    {:x
     {:b-2-1 "b21"}}}})

那将是项目

  • [:B-1-1 "B11"]
  • [:B-1-2 "B12"]
  • [:B-2-1 "B21"]

有没有一种优雅的方法可以使用 clojure 核心来做到这一点?


仅供参考,我确实通过创建自己的模式访问者来解决这个问题

(defn visit-zipper-pattern
  [loc pattern f]

但是,尽管此函数通常可用,但它非常复杂,结合了堆栈消耗递归和尾调用递归。因此,当调用该方法时,例如

(visit-zipper-pattern (map-zipper my-tree) visit-path
  (fn [[k v]] [k (str "mod-" v)]))

使用 https://stackoverflow.com/a/15020649/709537 中的map-zipper,它将树转换为

{:a "a"
 :b {:b-1
     {:x
      {:b-1-1 "mod-b11"
       :b-1-2 "mod-b12"}}
     :b-2
     {:x
      {:b-2-1 "mod-b21"}}}}

以下内容将起作用 - 请注意,1( 在处理:all键时,它可能会分配不需要的对象,2( 您需要决定如何处理边缘情况,例如非映射叶上的:all

(defn traverse [[k & rest-ks :as pattern] f tree]
  (if (empty? pattern)
    (f tree)
    (if (= k :all)
      (reduce #(assoc %1 %2 (traverse rest-ks f (get tree %2)))
              tree (keys tree))
      (cond-> tree (contains? tree k)
              (assoc k (traverse rest-ks f (get tree k)))))))

对于更有效的解决方案,最好按照上述建议使用 https://github.com/nathanmarz/specter。

最新更新