golang之websocket 源码分析

浏览数：31 / 时间：2015年06月09日

下载go的websocket包.

1. 通过google官方的方法, 需要hg来同步代码. 由于墙的原因, 还需要设置代理. 比较麻烦

2. http://gopm.io/ 通过该网站下载, 这是golang中国提供的解决方法 http://www.golangtc.com/download/package

websocket的实现还是比较简单的. 一共就4个文件. client.go hybi.go server.go websocket.go

示例代码网上到处都是, 就不贴了.

有一篇自己实现了一遍websocket的协议的文章 http://www.cnblogs.com/yjf512/archive/2013/02/18/2915171.html

对于理解协议还是比较好的. 当然这个代码仅限学习.

通过源码来确定两件事:

1. websocket 在接收时会自动组合一个完整的frame抛上来. 即send和recv是一一对应的.

2. 发送json数据为什么是空的(后来发现和websocket没啥关系0.0)

第一点. 是我很少用web框架. 孤陋寡闻了. 后来问了一下java的同事, 现在的框架基本都做了自动拆解包.

golang的websocket支持两种codec. Message和json形式.

Message 就是直接发送字符流.

json 即在内部自动调用json的解析器.

c++中发送一般都控制得很精准. json会浪费很多冗余数据. 虽然比起xml来说更轻量. 但是扩展性和通用性的好处是显而易见的.

这部分代码在websocket.go中.

var JSON = Codec{jsonMarshal, jsonUnmarshal}

var Message = Codec{marshal, unmarshal}

receive的处理如下:

// Receive receives single frame from ws, unmarshaled by cd.Unmarshal and stores in v.
func (cd Codec) Receive(ws *Conn, v interface{}) (err error) {
    .... // 省略
again:
    frame, err := ws.frameReaderFactory.NewFrameReader()
    if err != nil {
        return err
    }
    frame, err = ws.frameHandler.HandleFrame(frame)
    if err != nil {
        return err
    }
    if frame == nil {
        goto again
    }
    payloadType := frame.PayloadType()
    data, err := ioutil.ReadAll(frame)
    if err != nil {
        return err
    }
    return cd.Unmarshal(data, payloadType, v)
}

frame的处理代码在hybi.go中.

// NewFrameReader reads a frame header from the connection, and creates new reader for the frame.
// See Section 5.2 Base Framing protocol for detail.
// http://tools.ietf.org/html/draft-ietf-hybi-thewebsocketprotocol-17#section-5.2
func (buf hybiFrameReaderFactory) NewFrameReader() (frame frameReader, err error) {
    hybiFrame := new(hybiFrameReader)
    // ... 解析websocket的协议头.
    hybiFrame.reader = io.LimitReader(buf.Reader, hybiFrame.header.Length)
    hybiFrame.header.data = bytes.NewBuffer(header)
    hybiFrame.length = len(header) + int(hybiFrame.header.Length)
    return
}