• 技术文章 >后端开发 >Golang

    Go Http Server框架怎么快速实现?一文搞定

    藏色散人藏色散人2022-12-14 20:43:18转载1786

    本篇文章给大家介绍有关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。

    这里稍微总结一下

    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中文网其它相关文章!

    声明:本文转载于:learnku,如有侵犯,请联系admin@php.cn删除
    专题推荐:Golang
    上一篇:聊聊Go中的循环依赖检测工具(go-cyclic) 下一篇:自己动手写 PHP MVC 框架(40节精讲/巨细/新人进阶必看)

    相关文章推荐

    • 一文浅析Golang中的切片(Slice)• 一文深析MongoDB存储引擎(附原理图)• golang grpc消息过大?教你快速解决!• 分享一套Go编码规范!欢迎收藏!• 聊聊Go中的循环依赖检测工具(go-cyclic)
    1/1

    PHP中文网