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

先了解 WebSocket 协议

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

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

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

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

  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.
  • 3
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值