内容简介:都说go标准库实用,Api设计简洁。这次就用go 标准库中的net/http包实现一个简洁的http web服务器,包括三种版本。直接使用http.HandleFunc(partern,function(http.ResponseWriter,*http.Request){})
前言
都说 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
以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网
猜你喜欢:本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。