关于clojure：自定义map函数在惰性场景中表现异常

Custom map function behaves oddly in lazy scenarios

为了好玩，我决定编写自己的map版本，但终于学会了如何正确使用lazy-seq并使用它来使地图变得懒惰：

1
2
3
4
5

(defn my-map [f [head & tail]]
(lazy-seq
(if head
(cons (f head) (my-map f tail))
tail)))

它可以工作，但是当我针对 map 测试它的惰性行为时，我注意到了一些不同的东西。我正在使用在处理元素时打印的辅助映射函数：

1
2
3
4
5
6
7

(defn print-map [description-str f coll mapping-f]
(mapping-f
(fn [x]
(do
(print (str description-str":" x))
(f x)))
coll))

当我使用标准 map 函数时，元素一次处理一个，在函数之间交替：

1
2
3
4
5
6

(defn -main []
(let [m map
coll (into '() (range 10 0 -1))
coll2 (print-map"A" identity coll m)
coll3 (print-map"B" identity coll2 m)]
(println (doall coll3))))

打印：

1	A:1 B:1 A:2 B:2 A:3 B:3 A:4 B:4 A:5 B:5 A:6 B:6 A:7 B:7 A:8 B:8 A:9 B:9 A:10 B:10 (1 2 3 4 5 6 7 8 9 10)

请注意两个函数如何处理每个数字，然后才能看到其余元素。

但是当我将-main中的m改为my-map时，处理顺序略有变化：

1	A:1 A:2 B:1 A:3 B:2 A:4 B:3 A:5 B:4 A:6 B:5 A:7 B:6 A:8 B:7 A:9 B:8 A:10 B:9 B:10 (1 2 3 4 5 6 7 8 9 10)

现在第一个函数运行两次开始，第二个函数最后连续运行两次，因此映射不再"同步"。

导致这种情况发生的 my-map 有什么问题？

您在 my-map 中所做的破坏将在您的惰性序列上调用 next。

你可以通过不破坏来避免这种情况：

1
2
3
4
5
6
7
8
9

(defn my-map [f [x :as xs]]
#_(next xs) ;; uncomment to observere similar"broken" behaviour
(lazy-seq
(if x
(cons (f x) (my-map f (rest xs)))
(rest xs))))

;; You can find out what destructing does with this call:
(destructure '[[x & r :as xs] numbers])

next 不像 rest 那样懒惰。