詳解Golang中函數作為值與型別

2020-11-13 15:00:54

下面由教學欄目給大家介紹Golang中函數作為值與型別,希望對需要的朋友有所幫助!

簡介

在 Go 語言中,我們可以把函數作為一種變數,用 type 去定義它,那麼這個函數型別就可以作為值傳遞,甚至可以實現方法,這一特性是在太靈活了,有時候我們甚至可以利用這一特性進行型別轉換。作為值傳遞的條件是型別具有相同的引數以及相同的返回值。

函數的型別轉換

Go 語言的型別轉換基本格式如下:

type_name(expression)

舉個例子:

package main        import "fmt"        type CalculateType func(int, int) 
// 宣告了一個函數型別        
// 該函數型別實現了一個方法    
func (c *CalculateType) Serve() {      
fmt.Println("我是一個函數型別")    }        
// 加法函數    
func add(a, b int) {      
fmt.Println(a + b)    }        
// 乘法函數    
func mul(a, b int) {      
fmt.Println(a * b)    }        
func main() {      a := CalculateType(add) 
// 將add函數強制轉換成CalculateType型別      
b := CalculateType(mul) 
// 將mul函數強制轉換成CalculateType型別      
a(2, 3)      b(2, 3)      a.Serve()      b.Serve()    }        
// 5    
// 6    
// 我是一個函數型別    
// 我是一個函數型別

如上,宣告了一個 CalculateType 函數型別,並實現 Serve() 方法,並將擁有相同引數的 add 和 mul 強制轉換成 CalculateType 函數型別,同時這兩個函數都擁有了 CalculateType 函數型別的 Serve() 方法。

函數作引數傳遞

package main        import "fmt"        
type CalculateType func(a, b int) int 
// 宣告了一個函數型別        
// 加法函數    func add(a, b int) int {      return a + b    }        
// 乘法函數    func mul(a, b int) int {      return a * b    }        
func Calculate(a, b int, f CalculateType) int {      return f(a, b)    }        
func main() {      a, b := 2, 3      fmt.Println(Calculate(a, b, add))      
fmt.Println(Calculate(a, b, mul))    }    
// 5    
// 6

如上例子,Calculate 的 f 引數型別為 CalculateType,add 和 mul 函數具有和 CalculateType 函數型別相同的引數和返回值,因此可以將 add 和 mul 函數作為引數傳入 Calculate 函數中。

net/http 包原始碼例子

// HandleFunc registers the handler function for the given pattern    
// in the DefaultServeMux.    
// The documentation for ServeMux explains how patterns are matched.    func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {      DefaultServeMux.HandleFunc(pattern, handler)    }
// HandleFunc registers the handler function for the given pattern.    func (mux *ServeMux) HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {      mux.Handle(pattern, HandlerFunc(handler))    }
type HandlerFunc func(ResponseWriter, *Request)        
// ServeHTTP calls f(w, r).    
func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {      f(w, r)    }

剛開始看到這段原始碼的時候,真的有點懵逼了,這段原始碼的目的是為了將我們的 Handler 強制實現 ServeHTTP() 方法,如下例子:

func sayHi(w http.ResponseWriter, r *http.Request) 
{      
io.WriteString(w, "hi")    
}        
func main() {      
http.HandlerFunc("/", sayHi)      
http.ListenAndserve(":8080", nil)    
}

因為 HandlerFunc 是一個函數型別,而 sayHi 函數擁有和 HandlerFunc 函數型別一樣的引數值,因此可以將 sayHi 強制轉換成 HandlerFunc,因此 sayHi 也擁有了 ServeHTTP() 方法,也就實現了 Handler 介面,同時,HandlerFunc 的 ServeHTTP 方法執行了它自己本身,也就是 sayHi 函數,這也就可以看出來了,sayHi 就是 Handler 被呼叫之後的執行結果。

以上就是詳解Golang中函數作為值與型別的詳細內容,更多請關注TW511.COM其它相關文章!