类型断言和反射

在 Go 语言中，当函数的输入参数类型为 any（在 Go 1.18 及以后版本可用，之前版本为 interface{}）时，若要判断这个 any 类型的值是否为切片或者数组，可以使用类型断言和反射两种方法，下面分别进行详细介绍：

方法一：使用类型断言

类型断言是一种检查接口值底层具体类型的方式，通过类型断言可以尝试将 any 类型的值转换为切片或数组类型，若转换成功则说明该值为对应的类型。

package main

import (
    "fmt"
)

func checkIfSliceOrArray(input any) {
    // 尝试将 input 断言为切片类型
    if _, ok := input.([]any); ok {
        fmt.Println("输入是切片")
        return
    }
    // 尝试将 input 断言为数组类型（这里以长度为 3 的 int 数组为例）
    if _, ok := input.([3]int); ok {
        fmt.Println("输入是数组")
        return
    }
    fmt.Println("输入既不是切片也不是数组")
}

func main() {
    slice := []int{1, 2, 3}
    array := [3]int{4, 5, 6}
    num := 10

    checkIfSliceOrArray(slice)
    checkIfSliceOrArray(array)
    checkIfSliceOrArray(num)
}

代码解释：

• 在 checkIfSliceOrArray 函数中，首先使用类型断言 input.([]any) 尝试将 input 转换为切片类型，如果转换成功（ok 为 true），则说明输入是切片。
• 接着使用类型断言 input.([3]int) 尝试将 input 转换为长度为 3 的 int 数组类型，如果转换成功，则说明输入是数组。
• 如果以上两种类型断言都失败，则说明输入既不是切片也不是数组。

方法二：使用反射

反射是 Go 语言提供的一种在运行时检查和操作对象的机制，通过反射可以获取对象的类型信息，从而判断其是否为切片或数组。

package main

import (
    "fmt"
    "reflect"
)

func checkIfSliceOrArrayWithReflect(input any) {
    // 获取 input 的反射类型
    inputType := reflect.TypeOf(input)
    if inputType != nil {
        switch inputType.Kind() {
        case reflect.Slice:
            fmt.Println("输入是切片")
        case reflect.Array:
            fmt.Println("输入是数组")
        default:
            fmt.Println("输入既不是切片也不是数组")
        }
    } else {
        fmt.Println("输入是 nil")
    }
}

func main() {
    slice := []int{1, 2, 3}
    array := [3]int{4, 5, 6}
    num := 10

    checkIfSliceOrArrayWithReflect(slice)
    checkIfSliceOrArrayWithReflect(array)
    checkIfSliceOrArrayWithReflect(num)
}

代码解释：

• 在 checkIfSliceOrArrayWithReflect 函数中，使用 reflect.TypeOf(input) 获取 input 的反射类型。
• 通过 inputType.Kind() 获取该类型的具体种类，然后使用 switch 语句判断其是否为 reflect.Slice 或 reflect.Array，如果是则分别输出相应信息，否则说明输入既不是切片也不是数组。
• 如果 inputType 为 nil，则说明输入是 nil。

两种方法的比较

• 类型断言：代码简洁，性能较高，但只能针对特定的类型进行判断，如果需要判断多种不同类型的切片或数组，需要编写多个类型断言语句。
• 反射：更加灵活，可以在运行时动态判断任意类型的切片或数组，但反射操作会带来一定的性能开销，因为它涉及到运行时的类型检查和操作。在性能要求较高的场景下，应谨慎使用反射。