返回顶部
首页 > 资讯 > 精选 >Go Http Server框架如何快速实现
  • 641
分享到

Go Http Server框架如何快速实现

2023-07-04 19:07:03 641人浏览 安东尼
摘要

这篇“Go Http Server框架如何快速实现”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“Go Http Server

这篇“Go Http Server框架如何快速实现”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“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

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   TestHttpA1HelloB1

以上就是关于“Go Http Server框架如何快速实现”这篇文章的内容,相信大家都有了一定的了解,希望小编分享的内容对大家有帮助,若想了解更多相关的知识内容,请关注编程网精选频道。

--结束END--

本文标题: Go Http Server框架如何快速实现

本文链接: https://www.lsjlt.com/news/346950.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作