在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【相關推薦:】
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 方面,字首樹的實現。
php入門到就業線上直播課:
以上就是Go Http Server框架怎麼快速實現?一文搞定的詳細內容,更多請關注TW511.COM其它相關文章!