golang中 type func() 用法分析

冷不防 2023-01-02 03:29 145阅读 0赞

在看golang 的http服务部分代码时，被golang 中的 type func()写法难住了，一时没看懂代码。后来查资料后，有了一点理解。  
在golang中可以通过这样简单实现一个http服务

package main
    
    import "net/http"
    
    func mHttp() {
        http.HandleFunc("/", h)
        http.ListenAndServe("0.0.0.0:8888",nil)
    }
    func h(w http.ResponseWriter, r *http.Request) {
    
    }

`http.HandleFunc()`是一个注册函数，传一个string类型的路由，和一个函数，函数的参数为**(http.ResponseWriter, \*http.Request)**。跟踪进入函数，在golang 源码net/http/server.go文件中

func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
        DefaultServeMux.HandleFunc(pattern, handler)
    }

在**HandleFunc**调用了`DefaultServeMux.HandleFunc(pattern, handler)`  
至于这些函数是干啥的先不做探讨，这不是本文的重点。  
再次跟进函数

func (mux *ServeMux) HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
        if handler == nil {
            panic("http: nil handler")
        }
        mux.Handle(pattern, HandlerFunc(handler))
    }

在**mux.Handle(pattern, HandlerFunc(handler))** 的第二个参数`HandlerFunc(handler)`是什么鬼。  
跟进看一下

type HandlerFunc func(ResponseWriter, *Request)
    
    func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
        f(w, r)
    }

原来**HandlerFunc** 是用 **type** 定义的函数，而函数的类型就是最开始传入的类型`func(ResponseWriter, *Request)`  
**ServeHTTP**是**HandlerFunc**的一个方法（注意一下，golang中方法和函数不是一回事）。并且**HandlerFunc**实现了 **Handler**接口  
Handler接口定义:

type Handler interface {
        ServeHTTP(ResponseWriter, *Request)
    }

回到**HandleFunc**方法中，`mux.Handle(pattern, HandlerFunc(handler))`的第二个参数是把传入的函数 **handler** 强转成 **HandlerFunc**类型，这样**handler**就实现了Handler接口。  
到这我们明白`HandlerFunc(handler)` 是把普通函数强转成type定义的函数。  
现在写一个简单的demo验证一下：

package main
    
    import "fmt"
    
    func main() {
       one(2, callback)
    }
    
    //需要传递函数
    func callback(i int) {
       fmt.Println("i am callBack")
       fmt.Println(i)
    }
    
    //main中调用的函数
    func one(i int, f func(int)) {
       two(i, fun(f))
    }
    
    //one()中调用的函数
    func two(i int, c Call) {
       c.call(i)
    }
    
    //定义的type函数
    type fun func(int)
    
    //fun实现的Call接口的call()函数
    func (f fun) call(i int) {
       f(i)
    }
    
    //接口
    type Call interface {
       call(int)
    }

先看一下程序的运行结果:

![26c285bdfc644fb2046519e2344816e5.png][]

我们在**main()**函数中调用了**one()**函数，并传入了**callback()**函数，最终调用了我们传入的**callback()**函数。

理一下思路：

使用**type**定义函数 **func(int)**  
定义 **Call** 接口，Call中有一个函数 **call(int)**  
在**main()**中调用**one(2, callback)**，在**one()**中调用**two()**，传入**two()**函数前，对callback函数实现了类型转换，从普通函数转换成type定义的函数。  
在 **two()** 中调用传入的 **c** 因为 **c** 实现了 **Call** 接口，所以可以调用 **call()** 函数，最终调用了我们传入的 **callback()** 函数。

[26c285bdfc644fb2046519e2344816e5.png]: /images/20221120/994b5a5962fb4315a8d818231bef24c6.png