快捷導(dǎo)航

go語言求任意類型切片的長度操作

更新時間：2021年04月26日 11:22:11 作者：choumin

這篇文章主要介紹了go語言求任意類型切片的長度操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

最近用go寫程序時遇到一個問題——求任意類型切片的長度。

作為一個初學(xué)者，剛剛學(xué)了接口和切片，知道了每個類型都實現(xiàn)了一個空接口interface{}，那么如果接口類型作為函數(shù)的參數(shù)，那它應(yīng)該是可以接收任意類型的實參的

帶著這樣的想法就寫出了下面的代碼：

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

然后調(diào)用

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

但編譯的時候報了以下錯誤：

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

從報錯的信息來看，是go語言不支持將任意類型的切片轉(zhuǎn)換為接口切片所導(dǎo)致的，為了確定是go語言本身不支持所導(dǎo)致以及探究不支持的原因，于是在網(wǎng)上查閱了一些資料，最權(quán)威的應(yīng)該是來自于go的官方文檔

這上邊的解釋是說，由于非接口類型的切片與接口類型的切片在內(nèi)存中的空間布局不一樣，如果要做這樣的隱式轉(zhuǎn)換，將會比較耗時，因此go不支持此種轉(zhuǎn)換。

如果確實需要用到傳接口切片的情況，則需要由程序員自己來提前做轉(zhuǎn)換，在傳參的時候確保實參是接口切片類型，這樣才能通過編譯，這也是官方推薦的做法

代碼如下所示：

// 獲得一個int類型的切片
var dataSlice []int = foo()
// 創(chuàng)建接口類型的切片
var interfaceSlice []interface{} = make([]interface{}, len(dataSlice))
// 依次轉(zhuǎn)換每個元素
for i, d := range dataSlice {
 interfaceSlice[i] = d
}
// 調(diào)用上面的size方法
size(interfaceSlice)

如果按照上面的寫法來傳參，那么求切片的長度就顯得太費勁了，還不如直接調(diào)用 len(dataSlice) 完事。

事情發(fā)展到這里，有點不甘心，于是繼續(xù)查資料，發(fā)現(xiàn)go語言的反射機制可以解決這個問題。

首先將上面的size函數(shù)的參數(shù)改為接口類型(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語言中切片的長度與容量的變化

在學(xué)習(xí)go語言的切片信息時，發(fā)現(xiàn)切片的容量變化非常讓人摸不著頭腦，為了更記憶深刻就寫下了這篇，如有錯誤之處，請大家指正

一、當(dāng)前切片的長度與容量相等情況：

package main
import (
    "fmt"
)
func main() {
	numbers := []int{0,1,2}  
	printSlice(numbers)
	//通過append給numbers增加信息，如果當(dāng)前切片的長度與容量相等，增加信息的長度小于等于原來的長度，
	那么切片的長度變?yōu)橄嗉又停萘孔優(yōu)樵瓉淼?倍(圖片一的第二行結(jié)果)
	numbers = append(numbers, 10,5,6) 
	printSlice(numbers)
	
	//通過append給numbers增加信息，如果當(dāng)前切片A的長度與容量相等，增加信息B的長度大于切片A原來的長度，
	那么切片的長度變?yōu)橄嗉又?，容量變?yōu)椋築長度+A長度+(B長度-A長度)%2（圖片一的第三行結(jié)果）
	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)
}

圖片一：

在這里插入圖片描述

二、如果當(dāng)前切片的長度小于容量情況：

1、增加信息的長度與當(dāng)前長度和小于等于容量

func main() {
   /* 創(chuàng)建切片 */
   numbers := []int{0,1,2}  
   printSlice(numbers)
   numbers = append(numbers, 10,5) 
   printSlice(numbers)
   //通過append給numbers增加信息，如果當(dāng)前切片的長度小于容量，增加信息的長度與當(dāng)前長度和小于等于容量，
   那么numbers的長度變?yōu)橄嗉又?，容量不?圖片二的第三行結(jié)果)
   numbers = append(numbers, 11)  
   printSlice(numbers)
}

圖片二：

在這里插入圖片描述

2、增加信息B的長度與當(dāng)前A的長度大于A容量并且小于A容量的2倍

func main() {
   /* 創(chuàng)建切片 */
   numbers := []int{0,1,2}  
   printSlice(numbers)
   numbers = append(numbers, 10,5) 
   printSlice(numbers)
   //通過append給numbers增加信息，如果當(dāng)前切片A的長度小于容量，增加信息B的長度與當(dāng)前A的長度大于A容量并且小于A容量的2倍，
   那么numbers的長度變?yōu)橄嗉又?，容量變?yōu)椋篈容量*2(圖片三的第三行結(jié)果)
   numbers = append(numbers, 11,12)  
   printSlice(numbers)
}

圖片三：

在這里插入圖片描述

3、增加信息B的長度與當(dāng)前A的長度大于A容量的2倍

func main() {
	/* 創(chuàng)建切片 */
	numbers := []int{0,1,2}  
	printSlice(numbers)
	numbers = append(numbers, 10,5) 
	printSlice(numbers)
	//通過append給numbers增加信息，如果當(dāng)前切片A的長度小于容量，增加信息B的長度與當(dāng)前A的長度大于A容量的2倍，
	那么numbers的長度變?yōu)橄嗉又虲。容量變?yōu)椋築長度+A長度+(B長度-A長度)%2（圖片四的第三行結(jié)果）
	numbers = append(numbers, 11,12,13,15,16,17,18,19,20)  
	printSlice(numbers)
}

圖片四：

在這里插入圖片描述