Go 语言中数组与切片作为函数参数的正确姿势

本文旨在深入探讨 go 语言中数组（array）和切片（slice）作为函数参数时的类型差异及处理方法。我们将详细解释为何直接传递数组给接受切片参数的函数会导致类型不匹配错误，并提供两种有效的解决方案：通过切片表达式将数组转换为切片传递，或调整函数签名以直接接受特定大小的数组。文章还将分析两种方法的优缺点，并提供最佳实践建议，帮助开发者避免常见陷阱。

理解 Go 语言中的数组与切片

在 Go 语言中，数组（Array）和切片（Slice）是两种不同的数据结构，尽管它们都用于存储同类型元素的集合，但在底层实现和行为上存在显著差异。

数组 (Array)：数组是具有固定长度的同类型元素的序列。数组的长度是其类型的一部分。例如，[3]string 和 [4]string 是两种完全不同的数组类型。当数组作为函数参数传递时，会进行值拷贝。
切片 (Slice)：切片是对底层数组的一个连续段的引用。它包含三个组件：指向底层数组的指针、长度（当前可见元素的数量）和容量（从切片起点到底层数组末尾的元素数量）。切片的长度是动态的，可以增长。切片作为函数参数传递时，传递的是切片头（包含指针、长度、容量）的值拷贝，但由于切片头中的指针指向同一个底层数组，因此在函数内部对切片元素的修改会影响到外部的底层数组。

由于数组的长度是其类型的一部分，[N]T 和 []T 在 Go 编译器看来是完全不同的类型。[N]T 表示一个长度为 N 的数组，而 []T 表示一个切片，其长度可以在运行时确定。

常见的类型不匹配问题

当尝试将一个数组直接传递给一个期望切片作为参数的函数时，Go 编译器会报告类型不匹配错误。考虑以下示例代码：

package main

import "fmt"

type name struct {
    X string
}

func main() {
    var a [3]name // 声明一个长度为3的name类型数组
    a[0] = name{"Abbed"}
    a[1] = name{"Ahmad"}
    a[2] = name{"Ghassan"}

    // 尝试将数组 'a' 传递给接受切片参数的函数
    nameReader(a) 
} 

// nameReader 函数期望一个 []name 类型的切片
func nameReader(array []name) {
    for i := 0; i < len(array); i++ {
        fmt.Println(array[i].X)
    }
}

运行上述代码会得到如下错误：

.\structtest.go:15: cannot use a (type [3]name) as type []name in function argument

这个错误明确指出，类型为 [3]name 的数组 a 不能被用作类型为 []name 的函数参数。

解决方案

针对上述类型不匹配问题，Go 语言提供了两种主要的解决方案。

方案一：通过切片表达式将数组转换为切片传递（推荐）

这是 Go 语言中最常用且推荐的方法。你可以使用切片表达式 array[:] 将整个数组转换为一个切片，然后将这个切片传递给函数。

package main

import "fmt"

type name struct {
    X string
}

func main() {
    var a [3]name
    a[0] = name{"Abbed"}
    a[1] = name{"Ahmad"}
    a[2] = name{"Ghassan"}

    // 使用切片表达式将数组 'a' 转换为切片后传递
    nameReader(a[:]) 
} 

func nameReader(array []name) {
    for i := 0; i < len(array); i++ {
        fmt.Println(array[i].X)
    }
}

优点：

灵活性： 接受切片参数的函数可以处理任意长度的切片，这使得函数更加通用。
Go 语言惯用法： 切片是 Go 语言处理动态集合的推荐方式。
性能： 传递切片头是高效的，它不会复制整个底层数组。

注意事项：

尽管传递的是切片头的值拷贝，但切片头中的指针指向的是原始数组的内存地址。因此，如果在 nameReader 函数内部修改了 array 中的元素，这些修改会反映到 main 函数中的原始数组 a。

方案二：修改函数签名以接受特定大小的数组

如果你明确知道函数只应处理特定长度的数组，并且希望函数接收数组的完整副本，你可以修改函数签名以直接接受数组类型。

package main

import "fmt"

type name struct {
    X string
}

func main() {
    var a [3]name
    a[0] = name{"Abbed"}
    a[1] = name{"Ahmad"}
    a[2] = name{"Ghassan"}

    // 直接将数组 'a' 传递给接受数组参数的函数
    nameReader(a) 
} 

// nameReader 函数现在期望一个 [3]name 类型的数组
func nameReader(array [3]name) {
    for i := 0; i < len(array); i++ {
        fmt.Println(array[i].X)
    }
}