Go语言websocket (一)

之前一段时间工作需要使用websocket写一个聊天系统,是给游戏服务做的客服系统; 就学习了一下go语言中的websocket…

我对socket的理解

socket是网络连接的底层实现，服务器和客户端建立连接进行信息交互。
客户端的功能：向服务器发送连接请求（就像A给B打电话，等对方接听），给服务器发送信息（A说话）;
服务器的功能：是一直监听发来请求（B的工作就是一直等着来电话，接电话），连接后，客户端发来什么信息，服务器就返回什么信息（A说什么，B就给A回复什么，不会主动说什么话），之后客户端的连接就结束了，服务器还是一直等待请求;
据了解比较早的服务器也是只有静态页面，客户端请求什么，服务器就返回什么页面;

net 包

net包go语言自带的网络连接相关包, 其中有IP、TCP连接socket、UDP连接socket 等，其他没有具体了解;

golang.org/x/net/websocket

这个包是go语言内写的websocket， socket是一个短连接，客户端发送完信息就结束连接了，不符合我们的实际应用，于是就有了对socket的处理升级为长连接，也就是全双工通信的理念。 这个websocket包就算go语言自带的长连接，不过我没有具体研究使用，看到很多网友说这个包功能性不完善，建议使用另一个第三方库，在github上写的一个websocket包，我使用的是这个包。

github.com/gorilla/websocket

这个包使用的比较多，在这里我就不多说了，有需要可以去github上去看看：地址
server 启动以后会注册两个 Handler。
websocketHandler 用于提供浏览器端发送 Upgrade 请求并升级为 WebSocket 连接。
pushHandler 用于提供外部推送端发送推送数据的请求。
浏览器首先连接 websocketHandler （默认地址为 ws://ip:port/ws）升级请求为 WebSocket 连接，当连接建立之后需要发送注册信息进行注册。
推送端发送数据的请求到 pushHandler（默认地址为 ws://ip:port/push），
Upgrader *websocket.Upgrader，这是 gorilla/websocket 包的对象，它用来升级 HTTP 请求。

golang.org/x/net/websocket测试

使用此包写了一个websocket ,只写了一个服务端功能, 可以找 “websocket在线测试工具” 当作客户端 模拟发送请求;

package main

import (
	"fmt"
	"golang.org/x/net/websocket"
	"log"
	"net/http"
)
func Echo(ws *websocket.Conn) {
	// websocket.Conn  用来作为客户端和服务器端交互的通道
	var err error
	// 只是用来记录接收请求的次数
    var i int
	for {
		var reply string
		// 建立连接后 接收来自客户端的信息reply
		if err = websocket.Message.Receive(ws, &reply); err != nil {
			fmt.Println("Error! Can't receive message...")
			break
		}
		fmt.Println("Received from client: " + reply)
        i++
        // 把收到的信息进行处理,也可以做信息过滤,也可以返回固定的信息
		msg := "Received:  " + reply
		fmt.Println("Sending to client: " + msg)
        fmt.Println(i)
		// 把信息返回发送给客户端 
		if err = websocket.Message.Send(ws, msg); err != nil {
			fmt.Println("Error! Can't send message...")
			break
		}
	}
}

func main() {
	http.Handle("/", websocket.Handler(Echo))


	if err := http.ListenAndServe(":8888", nil); err != nil {
		log.Fatal("ListenAndServe:", err)
	}   // 访问服务器的地址,ip没有限制,端口是8888,
	   //  ws://127.0.0.1:8888 
}

模拟结果:
连接成功后模拟发送信息, 客户端显示的页面…

记录服务端里的打印信息:

github.com/gorilla/websocket测试

也是只写了服务器的功能…

package main

import (
	"log"
	"net/http"
	"github.com/gorilla/websocket"
)
//Conn类型表示WebSocket连接。服务器应用程序从HTTP请求处理程序调用Upgrader.Upgrade方法以获取* Conn：
// var upgrader = websocket.Upgrader{}
var (
	upgrader = websocket.Upgrader {
		// 读取存储空间大小
		ReadBufferSize:1024,
		// 写入存储空间大小
		WriteBufferSize:1024,
		// 允许跨域
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
	}
)

func wsHandler(w http.ResponseWriter, r *http.Request) {
	//   完成握手 升级为 WebSocket长连接，使用conn发送和接收消息。
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("upgrade:", err)
		return
	}
	defer conn.Close()

	//调用连接的WriteMessage和ReadMessage方法以一片字节发送和接收消息。实现如何回显消息：
	//p是一个[]字节，messageType是一个值为websocket.BinaryMessage或websocket.TextMessage的int。
	for {
		messageType, msg, err := conn.ReadMessage()
		if err != nil {
			log.Println("Reading error...",err)
			return
		}
		log.Printf("Read from client msg:%s \n", msg)


		if err := conn.WriteMessage(messageType, msg); err != nil {
		//if err := conn.WriteMessage(1, []byte("今天。。。"));err != nil {
			log.Println("Writeing error...",err)
			return
		}
		log.Printf("Write msg to client: recved: %s \n",msg )
	}
}

func main() {
	http.HandleFunc("/",wsHandler)      // ws://127.0.0.1:8888/rom
	// 监听 地址 端口
	err := http.ListenAndServe(":8888", nil)
	if err != nil {
		log.Fatal("ListenAndServe", err.Error())
	}
}

测试结果:
模拟客户端发送信息…

记录日志打印结果…