详解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服务内容请搜索小牛知识库以前的文章或继续浏览下面的相关文章希望大家以后多多支持小牛知识库！