先看一下标准库中http server的实现
type tcpKeepAliveListener struct {
*net.TCPListener
}
func (ln tcpKeepAliveListener) Accept() (c net.Conn, err error) {
tc, err := ln.AcceptTCP()
if err != nil {
return
}
tc.SetKeepAlive(true)
tc.SetKeepAlivePeriod(3 * time.Minute)
return tc, nil
}
这里可以看出:tc.SetKeepAlivePeriod(3 * time.Minute) 设置了超时时间为3分钟,这个先到这里下面会用到
func main() {
http.HandleFunc("/", route)
http.ListenAndServe(":1789", nil)
}
func route(w http.ResponseWriter, r *http.Request) {
log.Printf("Addr->%s\tURI->%s\n", r.RemoteAddr, r.URL.Path)
defer r.Body.Close()
switch r.URL.Path {
case "/":
w.Write([]byte("welcome to work-stacks"))
case "/gettcp":
gettcp(w, r)
default:
http.NotFound(w, r)
}
}
func gettcp(w http.ResponseWriter, r *http.Request) {
//这里返回的*bufio.ReadWriter 没有处理, 生产环境注意要情况bufferd
conn, _, err := w.(http.Hijacker).Hijack()
if err != nil {
log.Printf("获取Hijacks失败:%s\n", err.Error())
return
}
if tcp,ok := conn.(*net.TCPConn)
tcp.SetKeepAlivePeriod(60 * time.Minute)
}
//然后就可以做自己要做的操作了.
conn.Close()
}
func main() {
req, err := http.NewRequest("GET", "http://127.0.0.1:1789/gettcp", nil)
if err != nil {
log.Println("创建请求失败:%s\n", err.Error())
return
}
conn, _ := net.Dial("tcp", "127.0.0.1:1789")
defer conn.Close()
hclient := httputil.NewClientConn(conn, nil)
err = hclient.Write(req)
if err != nil {
log.Printf("发送请求失败:%s\n", err)
return
}
}
