java - 使用 Java 流将数组拆分为子数组 - 有状态映射器

Question

我有一个包含 K*N 个元素的 double[]。我想将其拆分为 N 个大小为 K 的 long[] 数组的流(列表/数组/...)。我已经找到了一个解决方案，但它看起来很脏并且需要一个有状态的映射器(我认为它们的意思是无国籍):

private class DataToLong implements DoubleFunction<long[]> {
    int d = 0;
    long[] buf = new long[K];
    @Override
    public long[] apply(double value) {
        buf[d] = BitTools.toSortableLong(value);
        d++;
        long[] ret = null;
        if (d >= K) {
            ret = buf;
            buf = new long[K];
            d = 0;
        }
        return ret;
    }
}

public void load(double[] data, int K) {
    Arrays.stream(data).mapToObj(new DataToLong())
                            .filter((x)-> x != null).forEach((buf)->{
        //here we do something with 'buf'
    });
}

上面的代码看似可行，但实际上比非流版本的代码要长，而且违反了Mapper的无状态要求。有没有更好的方法来实现同样的事情？

Answer 1

IntStream.range(0, n).mapToObj(
   i -> {
     long[] arr = new long[k];
     for (int j = 0; j < k; j++) {
       arr[j] = BitTools.toSortableLong(data[k * i + j]);
     }
     return arr;
   });