Go语言如何实现支持多类型切片的通用分块函数？

你的核心问题在于原代码使用[]any导致类型丢失、无法直接适配具体类型切片，且硬编码处理块大小的逻辑极不灵活。Go 1.18+ 支持的泛型是解决这类通用切片操作的最优方案，以下是符合你需求的实现：

通用泛型实现 #

这个版本支持传入任意数量的块大小：先依次使用你指定的每个大小，剩余元素重复使用最后一个指定大小分块，不足时取剩余所有元素。

func SliceToChunks[T any](arr []T, sizes ...int) [][]T {
    if len(sizes) == 0 {
        return [][]T{arr} // 无指定块大小时，返回完整切片作为单个块
    }

    var chunks [][]T
    currentIdx := 0
    totalLen := len(arr)
    sizeCount := len(sizes)

    // 先按传入的块大小依次分块
    for i := 0; i < sizeCount; i++ {
        if currentIdx >= totalLen {
            break
        }
        size := sizes[i]
        nextIdx := currentIdx + size
        if nextIdx > totalLen {
            nextIdx = totalLen
        }
        chunks = append(chunks, arr[currentIdx:nextIdx])
        currentIdx = nextIdx
    }

    // 剩余元素用最后一个指定大小重复分块
    if currentIdx < totalLen {
        repeatSize := sizes[sizeCount-1]
        for currentIdx < totalLen {
            nextIdx := currentIdx + repeatSize
            if nextIdx > totalLen {
                nextIdx = totalLen
            }
            chunks = append(chunks, arr[currentIdx:nextIdx])
            currentIdx = nextIdx
        }
    }

    return chunks
}

适配你的示例场景 #

用你的测试代码验证：

type TypeA int
type TypeB string

func main() {
    // 测试sliceA：50个元素，按5、15分块
    sliceA := make([]TypeA, 50)
    for i := range sliceA {
        sliceA[i] = TypeA(i+1)
    }
    chunksA := SliceToChunks(sliceA, 5, 15)
    // 输出结果：4个块，长度分别为5、15、15、15

    // 测试sliceB：50个元素，按20、25分块
    sliceB := make([]TypeB, 50)
    for i := range sliceB {
        sliceB[i] = TypeB(fmt.Sprintf("item%d", i+1))
    }
    chunksB := SliceToChunks(sliceB, 20, 25)
    // 输出结果：3个块，长度分别为20、25、5
}

为什么原方案不可行 #

1. 类型安全问题：[]TypeA和[]any是不同类型，无法直接转换，强行断言会导致代码冗余且不安全；
2. 逻辑僵化：硬编码处理n1到n6的方式无法适配任意长度的切片和灵活的块大小需求；
3. 初始化错误：make([][]any, len(arr))会创建一个长度等于原切片元素数的切片，完全不符合分块的预期。

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