reduce-kv 在 subvec 上失败

Question

reduce-kv 在以元素索引作为“键”参数传递的向量上按预期工作。但是，它在与子向量一起使用时失败，因为 clojure.lang.APersistentVector$SubVector 没有实现 IKVReduce。

(reduce-kv + 0 (link: 1 2 3))
9

(reduce-kv + 0 (subvec (link: 1 2 3) 1))
IllegalArgumentException No implementation of method: :kv-reduce of protocol: #'clojure.core.protocols/IKVReduce found for class: clojure.lang.APersistentVector$SubVector clojure.core/-cache-protocol-fn (core_deftype.clj:583)

一种解决方案是将子向量复制到一个向量中

(reduce-kv + 0 (into (link: ) (subvec (link: 1 2 3) 1)))
6

但是，vec不会在这里工作。从Clojure 1.7开始，vec将返回一个subvec而不是复制。

(reduce-kv + 0 (vec (subvec (link: 1 2 3) 1)))
IllegalArgumentException No implementation of method: :kv-reduce of protocol: #'clojure.core.protocols/IKVReduce found for class: clojure.lang.APersistentVector$SubVector clojure.core/-cache-protocol-fn (core_deftype.clj:583)

Clojure用户期望子向量支持所有正常的向量操作，而这个异常很令人困惑。问题的原因是子向量返回了一个 clojure.lang.APersistentVector$SubVector，它没有实现 clojure.lang.IKVReduce。PersistentVector继承自 APersistentVector 并实现了 IKVReduce，但 SubVector 没有得到任何 IKVReduce 支持。这很可能是疏忽。

附上了两个补丁。第一个补丁通过在 core.clj 中扩展 IKVReduce 协议来修复了这个问题，这对于Clojure用户来说作为解决方案很方便。第二个补丁，由Alex Miller建议，直接在 APersistentVector$SubVector 中添加了 IKVReduce 接口的 Java 实现。应该审查的是第二个补丁。两个补丁中包含相同的测试。

Answer 1

评论者：[email protected]

下面是我的当前解决方案

`
(extend-type clojure.lang.APersistentVector$SubVector
clojure.core.protocols/IKVReduce
(kv-reduce [subv f init]

(transduce (map-indexed vector)
           (fn ([ret] ret) ([ret [k v]] (f ret k v)))
           init
           subv)))

`

在我的测试中，通常将subvec复制到常规向量中速度更快，但我更喜欢transduce修复的样式。如果在APersistentVector.java中添加原生Java实现可能会更快。如果Clojure/core团队想要补丁，我愿意做这项工作。

Answer 2

_评论者：[email protected]_

使用更多交互改善性能的修订解决方案。与常规向量reduce-kv的速度相当。


(when-not (satisfies?   clojure.core.protocols/IKVReduce (subvec [1] 0))
  (extend-type clojure.lang.APersistentVector$SubVector
    clojure.core.protocols/IKVReduce
    (kv-reduce
      [subv f init]
      (let [cnt (.count subv)]
        (loop [k 0 ret init]
          (if (< k cnt)
            (let [val (.nth subv k)
                ret (f ret k val)]
              (if (reduced? ret)
                @ret
                (recur (inc k) ret)))
            ret))))))

Answer 3

评论者：[email protected]

支持SubVector的IKVReduce

Answer 4

评论者：[email protected]

补丁还包括在常规向量和子向量化reduce-kv测试。

Answer 5

评论者：alexmiller

为什么不在APersistentVector$SubVector中直接实现IKVReduce？

Answer 6

评论者：[email protected]

Java 实现的 SubVector 上的 IKVReduce

Answer 7

评论者：[email protected]

是的，这很有道理。我犹豫了一下，因为我不想重构层次结构，将 SubVector 成为一个 PersistentVector 的子类，但这并不是修复这个错误的必要条件。CLJ-2065-Support-IKVReduce-on-SubVector.patch 只是 Java 方面，加上相同的测试。

Answer 8

评论者：[email protected]

更新补丁，以内联 nth() 调用，避免不必要的边界检查。

Answer 9

参考资料： https://clojure.atlassian.net/browse/CLJ-2065（由 [email protected] 报告）

Answer 10

JIRA 工单有点过时了。它只说这个错误会影响 1.9，但我可以确认它在 1.10.3 中仍然存在。

Comments

是的，通常我们假设这只是一种“在哪里注意到的”，除非由于回归或其他原因有进一步的说明，所以没什么大不了的。