多房间聊天室 websocket(一)

由于之前工作需要做聊天系统, 给游戏服务做客服系统,就了解了go语言的websocket 。
之前没有做过这方面的,就在网上找go语言实现的多房间聊天室的文章博客什么的, 发现资料很少啊,就找到了一篇讲java实现多房间聊天室的,就借鉴了一下实现思路,还有github上第三方写的websocket库——github.com/gorilla/websocket 基本实现了功能。

这篇就先实现基本功能,也方便初学者理解,注释会写清楚一些,下一篇会结合go语言的channl与goroutine实现更复杂的功能;

服务器功能代码

此处提供的服务器功能代码,之前说过,客户端负责向服务器发送信息,服务器可以控制怎么处理这个信息。客户端的功能还是找 “websocket在线测试工具”实现的。

package main

import (
	"fmt"
	"github.com/gorilla/websocket"
	"log"
	"net/http"
	"strconv"
)

// 实现多房间聊天室思路
//创建map以roomName为key,访问者请求连接为value;value可以是map,也可以是切片数组;
//因为房间内有多个请求连接,为记录用户和请求,此处以用户id为key,请求为value;
// {roomid:{uid:conn}} 
var rooms = make(map[string]map[int]*websocket.Conn)

var (
	// 服务器应用程序从HTTP请求处理程序调用Upgrader.Upgrade方法以获取* Conn;
	upgraderdd = websocket.Upgrader {
		// 读取存储空间大小
		ReadBufferSize:1024,
		// 写入存储空间大小
		WriteBufferSize:1024,
		// 允许跨域
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
	}
)

// 连接websocket进入房间,首先在请求里指定要进入的房间
//  (也可以先进入个人房间,再转入其他房间都可以的,就看怎么处理连接对象conn,在这里先指定房间了)
// 所以在请求里需要带两个参数,房间名roomid和用户id
func wsEcho(w http.ResponseWriter, r *http.Request) {
	r.ParseForm()
	roomid:=r.Form["roomid"][0] //从请求里获取房价名roomid
	uidd:=r.Form["uid"][0]      // 从请求里获取用户id
	uid,_:=strconv.Atoi(uidd)
	fmt.Println("roomid:===",roomid)
	fmt.Println("uid:==",uid)
	// conn就是建立连接后的连接对象 
	conn,_ := upgraderdd.Upgrade(w, r, nil)

	defer conn.Close()

	func(conn *websocket.Conn){
		if rooms[roomid] == nil{
			rooms[roomid] = map[int]*websocket.Conn{uid: conn}
		}else {
			rooms[roomid][uid] = conn
		}
	}(conn)
	fmt.Println("rooms:==",rooms)
	
	for {
		// 某个链接发来数据data
		msgTy, data,_:= conn.ReadMessage()
		rooms[roomid][uid]=conn  //把房间名和用户id及连接对象conn保存到map里

		//把数据返给当前房间里(除自己)所有的链接
		for k,v :=range rooms[roomid]{
			if k==uid{
				continue
			}
			err := v.WriteMessage(msgTy,data)
			if err!=nil{
				log.Println("error:==",err)
			}
			log.Printf("Write msg to client: recved: %s \n",data)
		}
	}

}

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

结果

代码设置里有允许跨域请求,所以允许不同ip访问,但是端口要是“8888” ,所以测试用来三个ip来访问,我为了省事就只使用了一个房间,也可以多开几个连接,换不同的房间名就可以了。。。。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
如上三个图所示,三个请求是 :
ws://127.0.0.1:8888/?roomid=tang&uid=01
ws://127.0.0.2:8888/?roomid=tang&uid=02
ws://127.0.0.3:8888/?roomid=tang&uid=03
房间名为tang的房间里有三个连接,用户为01、02、03;
首先01发送“古筝计划”,02和03就收到此信息,
02发送“ 掩体计划”,01和03就收到此信息,
也可以同时有多个连接多个房间,只有一个房间内的用户才可以收到信息;
ws://127.0.0.1:8888/?roomid=tang&uid=01
ws://127.0.0.1:8888/?roomid=tang&uid=02
ws://127.0.0.1:8888/?roomid=zhao&uid=03
ws://127.0.0.1:8888/?roomid=zhao&uid=04
ws://127.0.0.1:8888/?roomid=zhao&uid=05

相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页