有什么方法可以控制春季批次的剩余项目吗

Question

如果我有这样的项目

-> [1,2,3,4,5,6,7,8,9,10]

如果ChunkSize=3，春季批次ItemReader读取类似的项目

[1,2,3]
[4,5,6]
[7,8,9]
[10]

但是，我想

[1,2,3]
[4,5,6]
[7,8,9,10]

有办法做到这一点吗？

Answer 1

for (i in 1..12) {   // added for testing
val list = IntArray(i) { it + 1 }.toList()
val result = list
.chunked(3)
.groupBy { it.count() }
.run {
values.flatten().run {
when (keys.size) {
1    -> this
else -> dropLast(2).plusElement(takeLast(2).flatten())
}
}
}
println(result)
}

测试输出：

[[1]]
[[1, 2]]
[[1, 2, 3]]
[[1, 2, 3, 4]]
[[1, 2, 3, 4, 5]]
[[1, 2, 3], [4, 5, 6]]
[[1, 2, 3], [4, 5, 6, 7]]
[[1, 2, 3], [4, 5, 6, 7, 8]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9, 10]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9, 10, 11]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]]

Answer 2

您可以通过检查结果中最后一项的大小来进行调整，如果它小于chunkSize，则将其附加到结果的倒数第二项：

fun main() {
val l = listOf(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
val chunkSize = 3
val c = l.chunked(chunkSize)
.let {
println(it)  // basic chunked result
if (it.size >= 2 && it.last().size < chunkSize)
it.dropLast(2) + listOf(it.takeLast(2).flatten())
else
it
}
println(c)
}

listOf(1, 2, 3, 4, 5, 6, 7, 8, 9, 10):的输出

[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9, 10]]

输出完全正确的结束块，带有listOf(1, 2, 3, 4, 5, 6, 7, 8, 9):

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

这里有一个扩展函数，它使用泛型来实现这一点，因此对于任何类型的Iterables，分块+合并：

fun <T> Iterable<T>.chunkedMergeSmall(chunkSize: Int) =
chunked(chunkSize)
.let {
println(it)
if (it.size >= 2 && it.last().size < chunkSize)
it.dropLast(2) + listOf(it.takeLast(2).flatten())
else
it
}
// println(list.chunkedMergeSmall(3))