Clojure：使用不带头的流。它是否正确？

Question

我想在某处放置一个惰性序列以根据需要提供数据。我知道我必须避免抓住序列的头部。我想出了以下解决方案，我错过了什么吗？

(defn headless [s] 
  (let [a (atom s)] 
    (fn 
      ([]  (let [s @a r (first s)] 
             (swap! a rest) r)) 
      ([n] (let [s @a rs (take n s)] 
             (swap! a #(drop n %)) rs))))) 

使用示例，这个简单的生成器只给出自然数。

(def nums (headless (iterate inc 0)))

(nums 5)
; (0 1 2 3 4)

(nums)
;5

更新：“测试”应该使用 dorun，而不是 doall。查看 lgrapenthin 的解决方案

一个（不太现实的）测试

(doall (map #(nums %) (repeat 20)))

在所有 4 个内核使用 5 分钟后崩溃并出现异常（OutOfMemoryError Java 堆空间）

Answer 1

您的代码有效。

这个表格：

(doall (map #(nums %) (repeat 20)))

将生成无限数量的 (nums 20) 并且永远不会返回。您可以改用dorun 来删除生成的(nums 20)，而不是将它们保存在内存中。但是，它不会返回，因为(repeat 20) 会生成一个无限的惰性序列。

headless 更易读的版本

(defn headless [s] 
  (let [a (atom s)] 
    (fn 
      ([]  (let [s @a]
             (swap! a rest)
             (first s))
      ([n] (let [s @a]
             (swap! a (partial drop n))
             (take n s)))))))