Golang websocket client读取数据

Golang既可以写websocket的server端也可以写websocket的client端，前者网上的资料很多后者甚少，今天遇到写client的需求，在此做个总结。

• 测试地址：火币网
• websocket包：golang.org/x/net/websocket

1. 建立连接。连接成功建立后，client和server均可以随时往数据通道里写数据同时也可以从中读取数据。

var wsurl = "wss://api.huobi.pro/ws"
var origin = "http://api.huobi.pro/"
ws, err := websocket.Dial(wsurl, "", origin)
if err != nil {
    panic(err)
}

2. 写数据。在通道已建立的前提下，写数据操作通过一行代码即可完成：

func sendMessage(data []bytes) {
	ws.Write(msg)
}

3.1 读数据。最简单的方法是调用func (ws *Conn) Read(msg []byte) (n int, err error)方法，定义一个用来接收数据的[]byte数组当作参数传入，但是由于不知道server发来的数据长度，所以一般是定义一个足够大的字节[]byte数组，这样读取一来浪费内存二来处理起来麻烦，不建议使用；

func readMessage(ws *websocket.Conn) {
	data := make([]byte, 1024*10)
	_, err := ws.Read(data)
	if err != nil {
		log.Println(err)
	}
}

3.2 读数据。对于都是以json方式传输的数据，websocket包提供了将每条message读取到一个interface{}中的方法，等同于json.Unmarshal。

func readJsonMessage(ws *websocket.Conn) {
	var data interfact{} // data的类型为接收的JSON类型struct
	err := websocket.Message.Receive(ws, data)
	if err != nil {
		log.Println(err)
	}
}

3.3 读数据。3.2是将接收到的数据直接unmarshal到struct里了，而我的需求比这个要麻烦一点：server发来的数据[]byte数组是压缩过的，所以接收到数据后第一步应该解压缩然后才能unmarshal，所以不能再用3.2的方式，参照3.2的源码，实现方式如下。

func readOriginMessage(ws *websocket.Conn) {
	again:
		fr, err := ws.NewFrameReader()
		if err != nil {
			log.Printf("new frame reader err %v", err)
			return
		}
		frame, err := ws.HandleFrame(fr)
		if err != nil {
			log.Printf("handle frame err %v", err)
			return
		}
		if frame == nil {
			goto again
		}
		
		bytes, err := ioutil.ReadAll(frame)
		if err != nil {
			log.Printf("read frame data err %v", err)
		}
		unzipData, err := utils.UnzipByte(bytes)
		if err != nil {
			log.Printf("unzip data err %v", err)
		}

		var message map[string]interface{}
		e := json.Unmarshal(unzipData, &message)
		if e != nil {
			log.Printf("unmarshal err %v", e)
		}
		
		log.Printf("message content= %+v", message)
}