Golang 中函數做爲值與類型

微信公衆號「後端進階」，專一後端技術分享：Java、Golang、WEB框架、分佈式中間件、服務治理等等。
老司機傾囊相授，帶你一路進階，來不及解釋了快上車！

在 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 被調用以後的執行結果。