ThinkChat2.0新版上线,更智能更精彩,支持会话、画图、阅读、搜索等,送10W Token,即刻开启你的AI之旅 广告
## 5.5. 函數值 在Go中,函數被看作第一類值(first-class values):函數像其他值一樣,擁有類型,可以被賦值給其他變量,傳遞給函數,從函數返迴。對函數值(function value)的調用類似函數調用。例子如下: ```Go func square(n int) int { return n * n } func negative(n int) int { return -n } func product(m, n int) int { return m * n } f := square fmt.Println(f(3)) // "9" f = negative fmt.Println(f(3)) // "-3" fmt.Printf("%T\n", f) // "func(int) int" f = product // compile error: can't assign func(int, int) int to func(int) int ``` 函數類型的零值是nil。調用值爲nil的函數值會引起panic錯誤: ```Go var f func(int) int f(3) // 此處f的值爲nil,會引起panic錯誤 ``` 函數值可以與nil比較: ```Go var f func(int) int if f != nil { f(3) } ``` 但是函數值之間是不可比較的,也不能用函數值作爲map的key。 函數值使得我們不僅僅可以通過數據來參數化函數,亦可通過行爲。標準庫中包含許多這樣的例子。下面的代碼展示了如何使用這個技巧。string.Map對字符串中的每個字符調用add1函數,併將每個add1函數的返迴值組成一個新的字符串返迴給調用者。 ```Go func add1(r rune) rune { return r + 1 } fmt.Println(strings.Map(add1, "HAL-9000")) // "IBM.:111" fmt.Println(strings.Map(add1, "VMS")) // "WNT" fmt.Println(strings.Map(add1, "Admix")) // "Benjy" ``` 5.2節的findLinks函數使用了輔助函數visit,遍歷和操作了HTML頁面的所有結點。使用函數值,我們可以將遍歷結點的邏輯和操作結點的邏輯分離,使得我們可以複用遍歷的邏輯,從而對結點進行不同的操作。 ```Go gopl.io/ch5/outline2 // forEachNode針對每個結點x,都會調用pre(x)和post(x)。 // pre和post都是可選的。 // 遍歷孩子結點之前,pre被調用 // 遍歷孩子結點之後,post被調用 func forEachNode(n *html.Node, pre, post func(n *html.Node)) { if pre != nil { pre(n) } for c := n.FirstChild; c != nil; c = c.NextSibling { forEachNode(c, pre, post) } if post != nil { post(n) } } ``` 該函數接收2個函數作爲參數,分别在結點的孩子被訪問前和訪問後調用。這樣的設計給調用者更大的靈活性。舉個例子,現在我們有startElemen和endElement兩個函數用於輸出HTML元素的開始標籤和結束標籤`<b>...</b>`: ```Go var depth int func startElement(n *html.Node) { if n.Type == html.ElementNode { fmt.Printf("%*s<%s>\n", depth*2, "", n.Data) depth++ } } func endElement(n *html.Node) { if n.Type == html.ElementNode { depth-- fmt.Printf("%*s</%s>\n", depth*2, "", n.Data) } } ``` 上面的代碼利用fmt.Printf的一個小技巧控製輸出的縮進。`%*s`中的`*`會在字符串之前填充一些空格。在例子中,每次輸出會先填充`depth*2`數量的空格,再輸出"",最後再輸出HTML標籤。 如果我們像下面這樣調用forEachNode: ```Go forEachNode(doc, startElement, endElement) ``` 與之前的outline程序相比,我們得到了更加詳細的頁面結構: ``` $ go build gopl.io/ch5/outline2 $ ./outline2 http://gopl.io <html> <head> <meta> </meta> <title> </title> <style> </style> </head> <body> <table> <tbody> <tr> <td> <a> <img> </img> ... ``` **練習 5.7:** 完善startElement和endElement函數,使其成爲通用的HTML輸出器。要求:輸出註釋結點,文本結點以及每個元素的屬性(< a href='...'>)。使用簡略格式輸出沒有孩子結點的元素(卽用`<img/>`代替`<img></img>`)。編寫測試,驗證程序輸出的格式正確。(詳見11章) **練習 5.8:** 脩改pre和post函數,使其返迴布爾類型的返迴值。返迴false時,中止forEachNoded的遍歷。使用脩改後的代碼編寫ElementByID函數,根據用戶輸入的id査找第一個擁有該id元素的HTML元素,査找成功後,停止遍歷。 ```Go func ElementByID(doc *html.Node, id string) *html.Node ``` **練習 5.9:** 編寫函數expand,將s中的"foo"替換爲f("foo")的返迴值。 ```Go func expand(s string, f func(string) string) string ```