详解Golang开启http服务的三种方式

    前言

    都说go标准库实用,Api设计简洁。这次就用go 标准库中的net/http包实现一个简洁的http web服务器,包括三种版本。

    v1最简单版

    直接使用http.HandleFunc(partern,function(http.ResponseWriter,*http.Request){})

    HandleFunc接受两个参数,第一个为路由地址,第二个为处理方法。

    
    //v1
    func main() {
      http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
         w.Write([]byte("httpserver v1"))
        })
        http.HandleFunc("/bye", sayBye)
        log.Println("Starting v1 server ...")
        log.Fatal(http.ListenAndServe(":1210", nil))
     }
    
    func sayBye(w http.ResponseWriter, r *http.Request) {
       w.Write([]byte("bye bye ,this is v1 httpServer"))
    }

    v2自定义Handler

    查看标准库源码,v1版本实际上是调用了handle方法,传入的HandlerFunc实现了Handler的ServeHTTP方法,实际上是ServeHTTP在做http请求处理。

    HandleFunc调用.png

    HandleFunc实现Handler.png

    Handler接口定义.png

    由此我们可以自定义自己的Handler,v2版本代码如下:

    
    // v2
    func main() {
      mux := http.NewServeMux()
      mux.Handle("/", &myHandler{})
      mux.HandleFunc("/bye", sayBye)
    
      log.Println("Starting v2 httpserver")
      log.Fatal(http.ListenAndServe(":1210", mux))
    }
    type myHandler struct{}
    
    func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
       w.Write([]byte("this is version 2"))
    }
    func sayBye(w http.ResponseWriter, r *http.Request) {
       w.Write([]byte("bye bye ,this is v2 httpServer"))
    }

    v3自定义server配置

    前面对Handler开了一次刀,下面我们看看http.ListenAndServe()中有些什么秘密。

    ListenAndServe.png

    原来这里可以自定义http服务器配置,都在Server这个结构体中,这个对象能配置监听地址端口,配置读写超时时间,配置handler,配置请求头最大字节数…,所有稍微改造一下v2的程序得到v3版:

    
    // v3
    func main() {
      mux := http.NewServeMux()
      mux.Handle("/", &myHandler{})
      mux.HandleFunc("/bye", sayBye)
    
      server := &http.Server{
        Addr:     ":1210",
        WriteTimeout: time.Second * 3,      //设置3秒的写超时
        Handler:   mux,
      }
      log.Println("Starting v3 httpserver")
      log.Fatal(server.ListenAndServe())
    }
    
    type myHandler struct{}
    
    func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
      w.Write([]byte("this is version 3"))
    }
    
    func sayBye(w http.ResponseWriter, r *http.Request) {
        // 睡眠4秒 上面配置了3秒写超时,所以访问 “/bye“路由会出现没有响应的现象
      time.Sleep(4 * time.Second)       
      w.Write([]byte("bye bye ,this is v3 httpServer"))
    }

    拓展一下(如何平滑关闭http服务)

    在go1.8中新增了一个新特性,利用Shutdown(ctx context.Context) 优雅地关闭http服务。

    文档中描述:

    Shutdown 将无中断的关闭正在活跃的连接,然后平滑的停止服务。处理流程如下:

    • 首先关闭所有的监听;
    • 然后关闭所有的空闲连接;
    • 然后无限期等待连接处理完毕转为空闲,并关闭;
    • 如果提供了 带有超时的Context,将在服务关闭前返回 Context的超时错误;

    利用这个特性改造一下v3版本的程序,实现一个关闭http的提示

    
    // 主动关闭服务器
    var server *http.Server
    func main() {
      
     // 一个通知退出的chan
    quit := make(chan os.Signal)
    signal.Notify(quit, os.Interrupt)
    
    mux := http.NewServeMux()
    mux.Handle("/", &myHandler{})
    mux.HandleFunc("/bye", sayBye)
    
    server = &http.Server{
      Addr:     ":1210",
      WriteTimeout: time.Second * 4,
      Handler:   mux,
    }
    
    go func() {
      // 接收退出信号
      <-quit
      if err := server.Close(); err != nil {
        log.Fatal("Close server:", err)
      }
    }()
    
    log.Println("Starting v3 httpserver")
    err := server.ListenAndServe()
    if err != nil {
      // 正常退出
      if err == http.ErrServerClosed {
        log.Fatal("Server closed under request")
      } else {
        log.Fatal("Server closed unexpected", err)
      }
    }
     log.Fatal("Server exited")
    
    }
    
    type myHandler struct{}
    
    func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
     w.Write([]byte("this is version 3"))
    }
    
     // 关闭http
    func sayBye(w http.ResponseWriter, r *http.Request) {
       w.Write([]byte("bye bye ,shutdown the server"))   // 没有输出
       err := server.Shutdown(nil)
       if err != nil {
        log.([]byte("shutdown the server err"))
       }
     }

    尝试访问 http://localhost:1210/bye 在控制台会得到以下提示结果,平滑关闭http服务成功:

    成功平滑关闭.png

    到此这篇关于详解Golang开启http服务的三种方式的文章就介绍到这了,更多相关Golang开启http服务内容请搜索lingkb以前的文章或继续浏览下面的相关文章希望大家以后多多支持lingkb!