golang之websocket 源码分析

liukun2009

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