Golang1.7 Http和Tcp使用同一个端口做服务

前端之家收集整理的这篇文章主要介绍了Golang1.7 Http和Tcp使用同一个端口做服务前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
先看一下标准库中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);ok {
        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
    }
}
原文链接:https://www.f2er.com/go/189345.html

猜你在找的Go相关文章