Go语言必知必会100问题-23 判断空切片最佳方法

前言

在Go语言必知必会100问题-22 空切片与nil切片有区别吗？中分析了空切片和nil切片是有区别的，那如何正确的判断一个切片是否为空切片呢？

案例引入

下面通过一个具体的例子进行说明，getOperations 函数返回一个float32类型的切片，在 handle 函数中会检查返回的切片中是否有元素，如果含有元素，调用 handle函数。

func handleOperations(id string) {
    operations := getOperations(id)
    if operations != nil {
        handle(operations)
    }
}
 
func getOperations(id string) []float32 {
    operations := make([]float32, 0)
 
    if id == "" {
        return operations
    }
 
    // Add elements to operations
 
    return operations
}

在函数 handleOperations 通过比较operations是否等于nil决定是否调用hanle函数。这样写存在问题，因为getOperations返回的切片永远不为nil, 即使返回的切片中不含任何元素，它也不为nil， 所以 operations != nil永远为true.

解决方法

如何处理呢？ 有两种解决方法。

方法1

解决方法一：对getOperations函数进行修改，如果id为空，直接返回nil，而不是operations, 修改后的代码如下。这种修改方法有局限性，有时候我们是无法修改被调用方代码的，例如我们在调用第三方库的时候。

func getOperations(id string) []float32 {
    operations := make([]float32, 0)
 
    if id == "" {
        return nil
    }
 
    // Add elements to operations
 
    return operations
}

方法2

通过切片的length是否为0来判断。无论是空切片还是nil切片，它们的长度都为0.

func handleOperations(id string) {
    operations := getOperations(id)
    if len(operations) != 0 {
        handle(operations)
    }
}

思考总结

综上，我们并不是总能够修改调用方逻辑，所以通过判断切片的长度是否为0判别切片中是否存在元素是最佳方法。与此同时，正如Go wiki中陈述的，在我们设计接口的时候，应该避免区分空切片和nil切片，对它们做区分会导致细微的程序错误。在返回切片时，如果返回nil切片或空切片，要确保不会产生语义上和技术上的差异，两者表达的应该是同一个意思。这项原则同样适用于map类型，通过检查map的长度为0判断是否为空，而不是通过nil来判断。