首页 > 代码库 > golang之websocket 源码分析
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.2func (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
golang之websocket 源码分析