這篇文章跟大家介紹有關Golang的相關知識,聊聊Go Http Server框架的怎麼快速實現的,希望對大家有幫助。
在Go想使用http server,最簡單的方法是使用http/net
err := http.ListenAndServe(":8080", nil)if err != nil { panic(err.Error())}http.HandleFunc("/hello", func(writer http.ResponseWriter, request *http.Request) { writer.Write([]byte("Hello"))})
#定義handle func【相關推薦:Go影片教學】
type HandlerFunc func(ResponseWriter, *Request)
標準庫的http 伺服器實作很簡單,開啟一個端口,註冊一個實作HandlerFunc
的handler
同時標準函式庫也提供了一個完全接管請求的方法
func main() { err := http.ListenAndServe(":8080", &Engine{}) if err != nil { panic(err.Error()) }}type Engine struct {}func (e *Engine) ServeHTTP(w http.ResponseWriter, r *http.Request) { if r.URL.Path == "/hello" { w.Write([]byte("Hello")) }}
定義ServerHTTP
type Handler interface { ServeHTTP(ResponseWriter, *Request)}
如果我們需要寫一個HTTP Server 框架,那麼就需要實作這個方法,同時net /http 的輸入輸出流都不是很方便,我們也需要包裝,再加上一個簡單的Route,不要在ServeHTTP 裡面寫Path。
這裡稍微總結一下
一個實作ServeHTTP 的Engine
一個包裝HTTP 原始輸入輸出流的Context
一個實現路由匹配的Route
Route 這邊為了簡單,使用Map做完全匹配
import ( "net/http")type Engine struct { addr string route map[string]handFunc}type Context struct { w http.ResponseWriter r *http.Request handle handFunc}type handFunc func(ctx *Context) errorfunc NewServer(addr string) *Engine { return &Engine{ addr: addr, route: make(map[string]handFunc), }}func (e *Engine) Run() { err := http.ListenAndServe(e.addr, e) if err != nil { panic(err) }}func (e *Engine) Get(path string, handle handFunc) { e.route[path] = handle}func (e *Engine) handle(writer http.ResponseWriter, request *http.Request, handle handFunc) { ctx := &Context{ w: writer, r: request, handle: handle, } ctx.Next()}func (e *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) { handleF := e.route[req.URL.Path] e.handle(w, req, handleF)}func (c *Context) Next() error { return c.handle(c)}func (c *Context) Write(s string) error { _, err := c.w.Write([]byte(s)) return err}
我們寫一個Test驗證一下我們的Http Server
func TestHttp(t *testing.T) { app := NewServer(":8080") app.Get("/hello", func(ctx *Context) error { return ctx.Write("Hello") }) app.Run()}
這邊我們包裝的Handle 使用了是Return error 模式,相比標準庫只Write 不Return ,避免了不Write 之後忘記Return 導致的錯誤,這通常很難發現。
一個Http Server 還需要一個middleware 功能,這裡的想法就是在Engine 中存放一個handleFunc 的數組,支援在外部註冊,當一個請求打過來時創建一個新Ctx,將Engine 中全域的HandleFunc複製到Ctx 中,再使用c.Next() 實作套娃式呼叫。
package httpimport ( "net/http")type Engine struct { addr string route map[string]handFunc middlewares []handFunc}type Context struct { w http.ResponseWriter r *http.Request index int handlers []handFunc}type handFunc func(ctx *Context) errorfunc NewServer(addr string) *Engine { return &Engine{ addr: addr, route: make(map[string]handFunc), middlewares: make([]handFunc, 0), }}func (e *Engine) Run() { err := http.ListenAndServe(e.addr, e) if err != nil { panic(err) }}func (e *Engine) Use(middleware handFunc) { e.middlewares = append(e.middlewares, middleware)}func (e *Engine) Get(path string, handle handFunc) { e.route[path] = handle}func (e *Engine) handle(writer http.ResponseWriter, request *http.Request, handle handFunc) { handlers := make([]handFunc, 0, len(e.middlewares)+1) handlers = append(handlers, e.middlewares...) handlers = append(handlers, handle) ctx := &Context{ w: writer, r: request, index: -1, handlers: handlers, } ctx.Next()}func (e *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) { handleF := e.route[req.URL.Path] e.handle(w, req, handleF)}func (c *Context) Next() error { c.index++ if c.index < len(c.handlers) { return c.handlers[c.index](c) } return nil}func (c *Context) Write(s string) error { _, err := c.w.Write([]byte(s)) return err}
實作方法很簡單,這裡我們驗證一下是否可以支援前置和後置中間件
func TestHttp(t *testing.T) { app := NewServer(":8080") app.Get("/hello", func(ctx *Context) error { fmt.Println("Hello") return ctx.Write("Hello") }) app.Use(func(ctx *Context) error { fmt.Println("A1") return ctx.Next() }) app.Use(func(ctx *Context) error { err := ctx.Next() fmt.Println("B1") return err }) app.Run()}
輸出:
=== RUN TestHttp A1 Hello B1
總共不過100行程式碼,我們就實作了一個簡易的Http Server ,後續細節可以去看Gin的源碼,主要關注點在Route 方面,前綴樹的實作。
以上是Go Http Server框架怎麼快速實現?一文搞定的詳細內容。更多資訊請關注PHP中文網其他相關文章!