本文共 2609 字，大约阅读时间需要 8 分钟。

由于之前工作需要做聊天系统, 给游戏服务做客服系统，就了解了go语言的websocket 。

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

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

服务器功能代码

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

package mainimport (	"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和用户idfunc 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

转载地址：http://wnqxi.baihongyu.com/