golang之websocket 源码分析

下载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
}

在此处根据header.length 接受完整个frame的数据.

而HandleFrame中主要是判断frame的payloadtype.

第二点. 这个是golang的struct 的坑了.

对于小写的成员. 其他包是无法访问的. 所以json的Marshal和Unmarshal 为空. 然而又不会报错或者异常. 害我纠结了半天.

解决方法有两个:

1. 成员变量首字母大写.

2. 实现json.Marshaler接口

对于第二种, 可以参考这篇文章. http://blog.csdn.net/varding/article/details/38560681