首页 > 编程语言 > go语言求任意类型切片的长度操作
2021
05-25

go语言求任意类型切片的长度操作

最近用go写程序时遇到一个问题——求任意类型切片的长度。

作为一个初学者,刚刚学了接口和切片,知道了每个类型都实现了一个空接口interface{},那么如果接口类型作为函数的参数,那它应该是可以接收任意类型的实参的

带着这样的想法就写出了下面的代码:

func size(ins []interface{}) int {
        return len(ins)
}

然后调用

a := []int{1, 2, 3, 4}
fmt.Println(size(a))

但编译的时候报了以下错误:

cannot use a (type []int) as type []interface {} in argument to size

从报错的信息来看,是go语言不支持将任意类型的切片转换为接口切片所导致的,为了确定是go语言本身不支持所导致以及探究不支持的原因,于是在网上查阅了一些资料,最权威的应该是来自于go的官方文档

这上边的解释是说,由于非接口类型的切片与接口类型的切片在内存中的空间布局不一样,如果要做这样的隐式转换,将会比较耗时,因此go不支持此种转换。

如果确实需要用到传接口切片的情况,则需要由程序员自己来提前做转换,在传参的时候确保实参是接口切片类型,这样才能通过编译,这也是官方推荐的做法

代码如下所示:

// 获得一个int类型的切片
var dataSlice []int = foo()
// 创建接口类型的切片
var interfaceSlice []interface{} = make([]interface{}, len(dataSlice))
// 依次转换每个元素
for i, d := range dataSlice {
 interfaceSlice[i] = d
}
// 调用上面的size方法
size(interfaceSlice)

如果按照上面的写法来传参,那么求切片的长度就显得太费劲了,还不如直接调用 len(dataSlice) 完事。

事情发展到这里,有点不甘心,于是继续查资料,发现go语言的反射机制可以解决这个问题。

首先将上面的size函数的参数改为接口类型(a interface{}),然后在里面用reflect.TypeOf(a).Kind()来判断实参的类型,如果是切片类型,则用reflect.ValueOf()来获得该切片,最后返回切片的长度

代码如下所示:

func Size(a interface{}) int {
        if reflect.TypeOf(a).Kind() != reflect.Slice {
                return -1
        }
        ins := reflect.ValueOf(a)
        return ins.Len()
}

补充:Go语言中切片的长度与容量的变化

在学习go语言的切片信息时,发现切片的容量变化非常让人摸不着头脑,为了更记忆深刻就写下了这篇,如有错误之处,请大家指正

一、当前切片的长度与容量相等情况:

package main
import (
    "fmt"
)
func main() {
	numbers := []int{0,1,2}  
	printSlice(numbers)
	//通过append给numbers增加信息,如果当前切片的长度与容量相等,增加信息的长度小于等于原来的长度,
	那么切片的长度变为相加之和,容量变为原来的2倍(图片一的第二行结果)
	numbers = append(numbers, 10,5,6) 
	printSlice(numbers)
	
	//通过append给numbers增加信息,如果当前切片A的长度与容量相等,增加信息B的长度大于切片A原来的长度,
	那么切片的长度变为相加之和,容量变为:B长度+A长度+(B长度-A长度)%2(图片一的第三行结果)
	numbers = append(numbers, 12,13,15,16,17,18,19,20,21,22,23) 
	printSlice(numbers)
}
func printSlice(x []int){
   fmt.Printf("len=%d cap=%d slice=%v\n",len(x),cap(x),x)
}

图片一:

在这里插入图片描述

二、如果当前切片的长度小于容量情况:

1、增加信息的长度与当前长度和小于等于容量

func main() {
   /* 创建切片 */
   numbers := []int{0,1,2}  
   printSlice(numbers)
   numbers = append(numbers, 10,5) 
   printSlice(numbers)
   //通过append给numbers增加信息,如果当前切片的长度小于容量,增加信息的长度与当前长度和小于等于容量,
   那么numbers的长度变为相加之和,容量不变(图片二的第三行结果)
   numbers = append(numbers, 11)  
   printSlice(numbers)
}

图片二:

在这里插入图片描述

2、增加信息B的长度与当前A的长度大于A容量并且小于A容量的2倍

func main() {
   /* 创建切片 */
   numbers := []int{0,1,2}  
   printSlice(numbers)
   numbers = append(numbers, 10,5) 
   printSlice(numbers)
   //通过append给numbers增加信息,如果当前切片A的长度小于容量,增加信息B的长度与当前A的长度大于A容量并且小于A容量的2倍,
   那么numbers的长度变为相加之和,容量变为:A容量*2(图片三的第三行结果)
   numbers = append(numbers, 11,12)  
   printSlice(numbers)
}

图片三:

在这里插入图片描述

3、增加信息B的长度与当前A的长度大于A容量的2倍

func main() {
	/* 创建切片 */
	numbers := []int{0,1,2}  
	printSlice(numbers)
	numbers = append(numbers, 10,5) 
	printSlice(numbers)
	//通过append给numbers增加信息,如果当前切片A的长度小于容量,增加信息B的长度与当前A的长度大于A容量的2倍,
	那么numbers的长度变为相加之和C。容量变为:B长度+A长度+(B长度-A长度)%2(图片四的第三行结果)
	numbers = append(numbers, 11,12,13,15,16,17,18,19,20)  
	printSlice(numbers)
}

图片四:

在这里插入图片描述

以上为个人经验,希望能给大家一个参考,也希望大家多多支持自学编程网。如有错误或未考虑完全的地方,望不吝赐教。

编程技巧