likes
comments
collection
share

Go语言中常见100问题-#23 判断空切片最佳方法

作者站长头像
站长
· 阅读数 9

前言

案例引入

下面通过一个具体的例子进行说明,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
    }

    // 其他操作
    
    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
    }

    // 其他操作
    return operations
}
方法二

通过切片的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来判断。