golang內置函數len的小技巧

Posted on by

len是很常用的內置函數，可以測量字符串、slice、array、channel以及map的長度/元素個數。

不過你真的瞭解len嗎？也許還有一些你不知道的小知識。

我們來看一道GO101的題目，這題也被GO語言愛好者周刊轉載：

package main

import "fmt"

func main() {
    var x *struct {
        s [][32]byte
    }
    
    fmt.Println(len(x.s[99]))
}

題目問你這段代碼的運行結果，選項有編譯錯誤、panic、32和0。

我們分析一下，別看x的聲明定義一大長串，實際上就是定義瞭一個有個[][32]byte的結構體，然後x是這個結構體的指針。

然後我們沒有初始化x，所以x是一個值為nil的指針。看到這裡你也許以及有答案瞭，對nil指針解引用訪問它的成員s，那不就是panic嘛。即使引用x的成員合法，我們的s也沒有初始化，訪問沒有初始化的slice也會panic。

然而這麼想你就錯瞭，代碼的實際運行結果是32！

為什麼呢？我們看看len的幫助文檔：

For some arguments, such as a string literal or a simple array expression, the result can be a constant. See the Go language specification’s “Length and capacity” section for details.

這句話很重要，對於結果是數組的表達式，len可能會是一個編譯期常量，而且數組類型的長度在編譯期是可知的，所以熟悉c++的朋友大概會立刻想到這樣的常量是不需要進行實際求值的，簡單類型推導即可獲得。不過口說無憑，我們看看spec裡的描述：

The expression len(s) is constant if s is a string constant. The expressions len(s) and cap(s) are constants if the type of s is an array or pointer to an array and the expression s does not contain channel receives or (non-constant) function calls; in this case s is not evaluated. Otherwise, invocations of len and cap are not constant and s is evaluated.

如果表達式是字符串常量那麼len(s)也是常量。如果表達式s的類型是array或者array的指針，且表達式不是channel的接收操作或是函數調用，那麼len(s)是常量，且表達式s不會被求值；否則len和cap會對s進行求值，其計算結果也不是一個常量。

其實說的很清楚瞭，但還有三點需要說明。

第一個是視為常量的表達式裡為什麼不能含有chan的接收操作和函數調用？

這個答案很簡單，因為這兩個操作都是使用這明確希望發生“副作用”的。特別是從chan裡接收數據，還會導致goroutine阻塞，而我們的常量len表達式不會進行求值，這些你期望會發生的副作用便不會產生，會引發一些隱蔽的bug。

第二個是我們註意到瞭函數調用前用non-constant修飾瞭，這是什麼意思？

按字面意思，一部分函數調用其實是可以在編譯期完成計算被當成常量處理的，而另一些不可以。

在進一步深入之前我們先要看看golang裡哪些東西是常量/常量表達式。

  • 首先是各種字面量以及對字面量的類型轉換產生的值瞭，無需多說。
  • 一部分內置函數：len、cap、imag、real、complex，它們在參數是常量的時候本身也是常量表達式。
  • unsafe.Sizeof，因為類型的大小也是編譯期就能確定的，所以它是常量表達式也很好理解。
  • 所有的常量之間的運算（加減乘除位運算等，除瞭非常量表達式函數的調用）都是常量表達式。

從上面的描述裡可以看出兩點，內置函數和unsafe.Sizeof的調用我們可以看成是constant function calls，所有常量表達式除瞭浮點數和復數表達式都可以在編譯期完成計算。而其他函數比如用戶自定義函數的調用，雖然仍然有可能在編譯期被求值優化，但本身不屬於常量表達式。所以語言標準會加以區分。比如下面這個：

func add(x, y int) int {
    return x + y
}

func main() {
    fmt.Println(add(512, 513)) // 1025
}

如果我們看生成的匯編，會發現求值已經完成，不需要調用add：

MOVQ    $1025, (SP)
PCDATA  $1, $0
CALL    runtime.convT64(SB)
MOVQ    8(SP), AX
XORPS   X0, X0
MOVUPS  X0, “”..autotmp_16+64(SP)
LEAQ    type.int(SB), CX
MOVQ    CX, “”..autotmp_16+64(SP)
MOVQ    AX, “”..autotmp_16+72(SP)
NOP
MOVQ    os.Stdout(SB), AX
LEAQ    go.itab.*os.File,io.Writer(SB), CX
MOVQ    CX, (SP)
MOVQ    AX, 8(SP)
LEAQ    “”..autotmp_16+64(SP), AX
MOVQ    AX, 16(SP)
MOVQ    $1, 24(SP)
MOVQ    $1, 32(SP)
NOP
CALL    fmt.Fprintln(SB)
MOVQ    80(SP), BP
ADDQ    $88, SP
RET

很明顯的，1025已經在編譯期求值瞭，然而add的調用不是常量表達式，所以下面的代碼會報錯：

const number = add(512, 513) // error!!!

// example.go:11:7: const initializer add(512, 513) is not a constant

spec給出的實例是調用的內置函數，內置函數也隻有在參數是常量的情況下被調用才算做常量表達式：

const (
 c4 = len([10]float64{imag(2i)})  // imag(2i) is a constant and no function call is issued
 c5 = len([10]float64{imag(z)})   // invalid: imag(z) is a (non-constant) function call
)
var z complex128

所以len的表達式裡如果用瞭non-constant的函數調用，那麼就len本身不能算是常量表達式瞭。

這就有瞭最後一個疑問，題目中的x不是常量，為什麼len的結果是常量呢？

標準隻說表達式裡不能有chan的接收和非常量表達式的函數調用，沒說其他的不可以。你也可以這麼理解，表達式都有結果值，任何值除瞭無類型常量（這裡顯然不是）都是要有一個確定的類型的，隻要這個類型是數組或者數組的指針，那len就能獲得數組的長度，而這一切不需要s一定是常量表達式，編譯器可以簡單推導出表達式的值的類型。不能包含non-constant function calls和chan接收是我在第一點裡解釋的，杜絕所有可能的副作用發生從而保證即使不對表達式求值程序也是正確的，不包含這兩個操作的表達式既可以是常量的也可以不是，所以這裡我們能用x.s[99]作為len的參數。

說瞭這麼多，隻要len的參數類型為array或者array的指針並且符合要求，就不會進行求值，而題目裡的表達式正好滿足這點，所以雖然我們看起來是會導致panic的代碼，但是本身並未進行實際求值，因此程序可以正常運行。另外cap也遵循同樣的規則。

最後，還有個小測驗，檢驗一下自己的學習吧：

// 以下哪些語句是正確的，哪些是錯誤的
var slice [][]*[10]int

const (
    a = len(slice[10000000000000][4]) // 1
    b = len(slice[1]) // 2
    c = len(slice) // 3
    d = len([1]int{1024}) // 4
    e = len([1]int{add(512, 512)}) // 5
    g = len([unsafe.Sizeof(slice)]int{}) // 6
    g = len([unsafe.Sizeof(slice)]int{int(unsafe.Sizeof(slice))}) // 7
)

參考
https://golang.org/ref/spec#Length_and_capacity

到此這篇關於golang內置函數len的小技巧的文章就介紹到這瞭,更多相關golang內置函數len內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：