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 被调用以后的执行结果。