使用net/http包搭建服务器
func SayHello(w http.ResponseWriter, req *http.Request) {
w.Write([]byte("Hello"))
}
func main() {
http.HandleFunc("/", SayHello)
http.ListenAndServe(":3000", nil)
// TODO
}
问题是如何正常关闭这个服务器?因为没有界面,服务器关闭后还有后续工作,所以不考虑结束进程的方式(ctrl+c)。
增加一个close的路由地址,
http.HandleFunc("/close", closeServer)
然后自己写方法
package main
import (
"net"
"net/http"
"time"
)
type HTTP struct {
listener net.Listener
}
func (h *HTTP) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello"))
}
func main() {
l, _ := net.Listen("tcp", ":3000")
h := new(HTTP)
h.listener = l
go func() {
http.Serve(h.listener, h)
}()
time.Sleep(time.Second * 10)
h.listener.Close()
select{}
}
这里有一种方案供你参考。