Go 快速起步:创建 WebSocket 服务器(聊天室)

先了解 WebSocket 协议

        和 HTTP 协议一样,WebSocket  协议也建立在 TCP/IP 协议基础上,但不一样的是 HTTP 协议 为单向协议,即只能客户端向服务器请求资源,服务器才能将数据传送给浏览器。 

        而 WebSocket 协议 是一种双向通信协议,在建立连接后,WebSocket 服务器和 Browser/UA 都能主动的向对方发送或接收数据。

        WebSocket 可以认为是与 HTTP 属于同一个层级的兄弟协议,但实际上它又有些依赖 HTTP 协议,因为 WebSocket 的初始连接是通过 HTTP 协议的三次握手实现,完成连接之后再向服务器请求升级成为 websocket 协议;

Go 快速起步:创建 WebSocket 服务器(聊天室)_第1张图片

简单即时聊天室的实现思路

  1. 引入了 github.com/gorilla/websocket 包,先创建 http 服务器,然后升级为 websocket 服务器;
  2. 当用户打开聊天页面,则建立连接,并将用户信息推送给ws服务端,服务端注册该用户到连接池;
  3. 用户点击发送消息时,循环用户连接池拿到用户的连接,主动将消息推送给每个用户;
  4. 这个思路比较简单,只通过 ReadMessage 接口通讯,所以需要为每个消息约定一下消息类型,比如注册、注销、消息);

简单版的 serv.go

package main

// go get -u -v github.com/gorilla/websocket 安装第三方包
import (
	"encoding/json"
	"fmt"
	"github.com/gorilla/websocket"
	"net/http"
)

func main() {
	// http请求,跳转到一个简单的前端页面
	http.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
		http.ServeFile(writer, request, "home.html")
	})
	// http请求,连接成功后申请升级为websocket服务
	http.HandleFunc("/ws", indexHandler)

	err := http.ListenAndServe("0.0.0.0:8888", nil)
	if err != nil {
		return
	}
}

// 消息体
type replyMsg struct {
	Type string `json:"type,omitempty"`
	Uid  string `json:"uid,omitempty"`
	Msg  string `json:"msg,omitempty"`
}

var (
	upgrader = websocket.Upgrader{
		// 允许所有CORS跨域请求
		CheckOrigin: func(r *http.Request) 

你可能感兴趣的:(Go,websocket,网络协议,网络)