最近打算用golang做项目,看了一下golang的资料,顺便做一下笔记,以供参考,复习.
由于对golang的熟悉度还不是很高,写的有点生疏了,也希望能对大家有帮助吧.
示例源码:
packagemain
import(
"log"
"net/http"
)
func helloWorld(whttp.ResponseWriter,r*http.Request){
w.Write([]byte("hello,world!"))
}
func main(){
http.HandleFunc("/",sayHelloName)
err:=http.ListenAndServe(":8080",128); background-color:inherit">nil)
iferr!=nil{
log.Fatal("ListenAndServe:",err)
}
http.HandleFunc() 解读
打开源码包 server.go跳到1775行,看到函数
func HandleFunc(pattern string, handlerfunc(ResponseWriter,192); background-color:inherit"> *Request)) {DefaultServeMux.HandleFunc(pattern,192); background-color:inherit"> handler)
}
DefaultServeMux的结构如下:
typeServeMuxstructmu sync.RWMutex
m map[string]muxEntry
hostsbool//whetheranypatternscontainhostnames
muxEntryexplicitbool
h Handler
pattern string
}
DefaultServeMux是一个全局变量在server.go的 1595 行创建的
NewServeMux() *ServeMux {return &ServeMux{m:make(string]muxEntry)} }
var DefaultServeMux = NewServeMux()
接着跟进 DefaultServeMux.HandleFunc(pattern, handler)
(mux *ServeMux)mux.Handle(pattern,192); background-color:inherit"> HandlerFunc(handler))
}
在这个函数里面把前面传进来的helloWorld 函数转换成 HandlerFunc
而 HandlerFunc的类型如下:
HandlerFunc *Request)
ServeHTTPcallsf(w,0); background-color:inherit">r).
(fHandlerFunc) ServeHTTP(w ResponseWriter,192); background-color:inherit"> r *Request)f(w,192); background-color:inherit"> r)
HandlerFunc类型实现了方法 ServeHTTP
再跟进Handle
函数 Handle(pattern Handler)mux.mu.Lock() // 因为map是全局的,涉及到全局调用,因此加锁
defer mux.mu.Unlock()
if pattern ==""panic("http:invalidpattern" + pattern)}
nilnilhandler")
mux.m[pattern].explicitmultipleregistrationsformux.m[pattern] muxEntry{explicit:true,192); background-color:inherit"> h: handler,192); background-color:inherit"> pattern: pattern}// 这里注册路由err := http.ListenAndServe(nil)解读
跟进
函数ListenAndServe,内容如下 ListenAndServe(addrerrorserver := &Server{Addr: addr,192); background-color:inherit"> Handler: handler} server.ListenAndServe()
而参数 Handler 是一个接口,内容如下HandlerinterfaceServeHTTP(ResponseWriter,monospace; background-color:inherit">}
看到 server := &Server{Addr: addr,192); background-color:inherit"> Handler: handler} 这句
因为我们在调用ListenAndServe时,第二个参数是nil,所以这里的handle是nil
继续跟进ListenAndServe内容如下
(srvServer) ListenAndServe()addr srv.Addr
addr":http"
ln,192); background-color:inherit"> err net.Listen("tcp",192); background-color:inherit"> addr) // 在这里调用了tcp监听了 != err
srv.Serve(tcpKeepAliveListener{ln.(*net.TCPListener)})}
跟进
Serve,可以看到在该函数里面调用了rw,192)"> e := l.Accept(),在接受连接的函数并且每一个连接go c.serve() 创建了一个携程
跟进 c.serve() 可以看到在函数里面对请求调用了 ServeHTTP
serverHandler{c.server}.ServeHTTP(w,192); background-color:inherit"> w.req) 跟进去内容如下
(shserverHandler)ServeHTTP(rw ResponseWriter,192); background-color:inherit"> req*Request{handler:= sh.srv.Handler
=={ // 在在这里判断用哪个 handler了= DefaultServeMux
}
req.RequestURI "*"&&.Method "OPTIONS" globalOptionsHandler{}handler.ServeHTTP(rw)
}
很显然看出了,如果在
调用err:=http.ListenAndServe(":8080",nil)时,第二个参数是我们自定义的Handler的话则调用我们的,否则的话就默认调用 DefaultServeMux
由此可以看出,如果我们要自定义Handler的话只要实现ServeHTTP(ResponseWriter,192); background-color:inherit"> *Request)方法即可
例子如下:
package main
import (
"fmt"
"log"
"net/http"
)
MyHandler (thisMyHandler) http.ResponseWriter,192); background-color:inherit"> *http.Request)fmt.Println("myHanlder")"myHanlder"))
main()err http.ListenAndServe( &MyHandler{})"ListenAndServe: err)
}
