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

Posted on by

最近用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)
}

圖片四：

在這裡插入圖片描述

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：