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

来源：互联网发布：柳州市一职校淘宝地址编辑：程序博客网时间：2024/05/16 19:19

先看一下标准库中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    }}

1 0