如何在不创建副本的情况下截断ByteArray以减少内存分配

Kotlin/Java里的ByteArray本身是固定长度的，没法直接修改原数组的长度，所以要避免创建副本的话，得换个思路——不返回新数组，而是传递「原数组+有效长度」的组合，或者用视图类包装。

给你两个实用方案：

方案1：只返回有效长度，复用原数组 #

不用创建新数组，先找到0的位置，返回截断后的有效长度，后续所有操作都基于原数组和这个长度来处理，完全零额外内存分配：

companion object {
    // 获取截断后的有效元素长度，不生成副本
    private fun ByteArray.getTruncatedLength(): Int {
        val zeroIndex = this.indexOf(0.toByte())
        return if (zeroIndex != -1) zeroIndex else this.size
    }

    // 示例：后续处理逻辑
    fun handleByteArray(arr: ByteArray) {
        val validLen = arr.getTruncatedLength()
        // 遍历只到有效长度
        for (i in 0 until validLen) {
            // 处理arr[i]...
        }
        // 或者用ByteBuffer限定范围
        val buffer = ByteBuffer.wrap(arr, 0, validLen)
        // 操作buffer...
    }
}

方案2：用ByteBuffer做视图包装 #

ByteBuffer可以直接包装原数组，指定起始索引和有效长度，本质是原数组的视图，不会创建副本，内存开销可以忽略：

companion object {
    private fun ByteArray.truncateAtZero(): ByteBuffer {
        val zeroIndex = this.indexOf(0.toByte())
        val validLen = if (zeroIndex != -1) zeroIndex else this.size
        return ByteBuffer.wrap(this, 0, validLen)
    }
}

后续直接操作这个ByteBuffer即可，它的读写都会限定在有效范围内，底层还是原数组。

要注意：之所以你的原方法必须创建副本，是因为ByteArray的长度不可变，任何试图"缩短"数组本身的操作都只能通过复制实现。所以要减少内存分配，就得放弃返回新ByteArray的思路，改用上面两种方式。

内容的提问来源于stack exchange，提问作者mach