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

fyxichen · · 2092 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

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

有疑问加站长微信联系(非本文作者)

本文来自:CSDN博客

感谢作者:fyxichen

查看原文:Golang1.7 Http和Tcp使用同一个端口做服务

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2092 次点击  
加入收藏 微博
2 回复  |  直到 1999-12-31 16:00:00
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传