go语言多房间聊天室websocket（二）

最新推荐文章于 2024-09-25 20:17:13 发布
最新推荐文章于 2024-09-25 20:17:13 发布
阅读量1.8k 收藏 14
点赞数 6
文章标签： go
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/swan_tang/article/details/103538543
版权

这一篇将继续实现go语言的多房间聊天室，第一篇实现了基本的功能，今天就继续完善，实现各种功能。
广播功能：就是游戏里的全服通告，
信息过滤：对言论信息处理，类似于不能发黄赌毒信息，
禁言功能：此处警告并禁言5分钟;
踢出群聊功能：三次不合法信息后，提醒以后并踢出群聊;

功能代码：

由于内容较多，分成两个文件来写了;
hub.go文件:

package main

import (
	"github.com/gorilla/websocket"
	"log"
	"net/http"
	"strings"
	"time"
)

const (
	writeWait = 10 * time.Second

	pongWait = 60 * time.Second

	pingPeriod = (pongWait * 9) / 10

	maxMessageSize = 512
)

var( upgrader = websocket.Upgrader{
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}
)

type connection struct {
	ws            *websocket.Conn
	send          chan []byte
	numberv       int
	forbiddenword bool
	timelog       int64
}


func (m message) readPump() {
	c := m.conn

	defer func() {
		h.unregister <- m
		c.ws.Close()
	}()

	c.ws.SetReadLimit(maxMessageSize)
	c.ws.SetReadDeadline(time.Now().Add(pongWait))
	c.ws.SetPongHandler(func(string) error { c.ws.SetReadDeadline(time.Now().Add(pongWait)); return nil })

	for {
		_, msg, err := c.ws.ReadMessage()
		if err != nil {
			if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
				log.Printf("error: %v", err)
			}
			//log.Printf("error: %v", err)
			break
		}
		go m.Kickout(msg)

	}
}


// 信息处理，不合法言论 禁言警告，超过3次，踢出群聊；
func (m message) Kickout(msg []byte) {
	c := m.conn
	// 判断是否有禁言时间,并超过5分钟禁言时间,没有超过进入禁言提醒
	nowT:=int64(time.Now().Unix())
	if nowT-c.timelog<300{
		h.warnmsg <- m
	}
	// 不合法信息3次，判断是否有不合法信息，没有进行信息发布
	if c.numberv < 3 {
		// 信息过滤，设置含有此字符串内字符为不合法信息，
		basestr := "死亡崩薨"
		teststr := string(msg[:])
		for _, ev := range teststr {
			// 判断字符串中是否含有某个字符，true/false
			reslut := strings.Contains(basestr, string(ev))
			if reslut == true {
				c.numberv += 1
				c.forbiddenword = true // 禁言为真
				// 记录禁言开始时间,禁言时间内任何信息不能发送
				c.timelog = int64(time.Now().Unix())
				h.warnings <- m
				break
			}
		}
		// 不禁言，消息合法 可以发送
		if c.forbiddenword != true {
			// 设置广播消息, 所有房间内都可以收到信息;给广播消息开头加一个特定字符串为标识，当然也有其他方法;
			// 此例 设置以开头0为标识, 之后去掉0 ;
			if msg[0] == 48 {
				head := string("所有玩家请注意:")
				data := head + string(msg[1:])
				m := message{[]byte(data), m.roomid, c}
				h.broadcastss <- m
			} else if msg[0] != 48 { //不是0，就是普通消息
				m := message{msg, m.roomid, c}
				h.broadcast <- m
			}
		}

		// 不合法信息超过三次，踢出群
	} else {
		h.kickoutroom <- m
		log.Println("要被踢出群聊了...")
		c.ws.Close()  // 此处关闭了踢出的连接,也可以不关闭做其他操作,
	}

}

func (c *connection) write(mt int, payload []byte) error {
	c.ws.SetWriteDeadline(time.Now().Add(writeWait))
	return c.ws.WriteMessage(mt, payload)
}


func (s *message) writePump() {
	c := s.conn

	ticker := time.NewTicker(pingPeriod)

	defer func() {
		ticker.Stop()
		c.ws.Close()
	}()

	for {
		select {
		case message, ok := <-c.send:
			if !ok {
				c.write(websocket.CloseMessage, []byte{})
				return
			}
			if err := c.write(websocket.TextMessage, message); err != nil {
				return
			}
		case <-ticker.C:
			if err := c.write(websocket.PingMessage, []byte{}); err != nil {
				return
			}
		}
	}
}

func serverWs(w http.ResponseWriter, r *http.Request) {
	r.ParseForm()
	roomid := r.Form["roomid"][0]

	ws, err := upgrader.Upgrade(w, r, nil)

	if err != nil {
		log.Println(err)
		return
	}

	c := &connection{send: make(chan []byte, 256), ws: ws}
	m := message{nil, roomid, c}

	h.register <- m

	go m.writePump()
	go m.readPump()
}

server.go文件:

package main

import (
	"fmt"
	"log"
	"net/http"
)


type message struct {
	data []byte
	roomid string
	conn *connection
}


type hub struct {
	rooms map[string]map[*connection]bool
	broadcast   chan  message
	broadcastss chan  message
	warnings    chan  message
	register    chan  message
	unregister  chan  message
	kickoutroom chan  message
	warnmsg    chan  message

}

var h = hub{
	broadcast:  make(chan message),
	broadcastss:make(chan message),
	warnings:   make(chan message),
	warnmsg:   make(chan message),
	register:   make(chan message),
	unregister: make(chan message),
	kickoutroom: make(chan message),
	rooms:       make(map[string]map[*connection]bool),
}


func (h *hub) run() {

	for {
		select {
		case m := <-h.register:    //传输链接
			conns:=h.rooms[m.roomid]
			if conns == nil {  // 链接保存到相应的房间
			    conns = make(map[*connection]bool)
				h.rooms[m.roomid] = conns
				fmt.Println("在线人数:==",len(conns))
				fmt.Println("rooms:==",h.rooms)
			}
			h.rooms[m.roomid][m.conn] = true
			fmt.Println("在线人数:==",len(conns))
			fmt.Println("rooms:==",h.rooms)

			for con := range conns {
				delmsg := "系统消息：欢迎新伙伴加入" + m.roomid + "聊天室！！！"
				data := []byte(delmsg)
				select {
				case con.send <- data:
				}
			}

		case m := <-h.unregister:  //断开链接
			conns:=h.rooms[m.roomid]
			if conns != nil {
				if _, ok := conns[m.conn]; ok {
					delete(conns, m.conn) //删除链接
					close(m.conn.send)
					for con := range conns {
						delmsg := "系统消息：有小伙伴离开了" + m.roomid + "聊天室！欢送！！！"
						data := []byte(delmsg)
						select {
						case con.send <- data:
							}
						if len(conns) == 0 { // 链接都断开，删除房间
							delete(h.rooms, m.roomid)
						}
					}
				}
			}

		case m := <-h.kickoutroom:   //3次不合法信息后，被踢出群聊
			conns:=h.rooms[m.roomid]
			notice:="由于您多次发送不合法信息,已被踢出群聊！！！"
			select {
			case m.conn.send <- []byte(notice):
			}
			if conns != nil {
				if _, ok := conns[m.conn]; ok {
					delete(conns, m.conn)
					close(m.conn.send)
					if len(conns) == 0 {
						delete(h.rooms, m.roomid)
					}
				}
			}


		case m := <-h.warnings:  //不合法信息警告
			conns:=h.rooms[m.roomid]
			if conns != nil {
				if _, ok := conns[m.conn]; ok {
					notice := "警告:您发布不合法信息，将禁言5分钟，三次后将被踢出群聊！！！"
					//starttime:=
					select {
					case m.conn.send <- []byte(notice):
					}
				}
			}

		case m := <-h.warnmsg:  //禁言中提示
			conns:=h.rooms[m.roomid]
			if conns != nil {
				if _, ok := conns[m.conn]; ok {
					notice := "您还在禁言中,暂时不能发送信息！！！"
					select {
					case m.conn.send <- []byte(notice):
					}
				}
			}


		case m := <-h.broadcast:  //传输群信息/房间信息
			conns := h.rooms[m.roomid]
			for con := range conns {
				if con==m.conn{  //自己发送的信息，不用再发给自己
					continue
				}
				select {
				case con.send <- m.data:
				default:
					close(con.send)
					delete(conns, con)
					if len(conns) == 0 {
						delete(h.rooms, m.roomid)
					}
				}
			}


		case m := <-h.broadcastss:  //传输全员广播信息
			for _,conns := range h.rooms {
				for con:=range conns{
					if con==m.conn{ //自己发送的信息，不用再发给自己
						continue
					}
					select {
					case con.send <- m.data:
					default:
						close(con.send)
						delete(conns, con)
						if len(conns) == 0 {
							delete(h.rooms, m.roomid)
						}
					}
				}

			}



		}
	}
}

// 要运行同包下的几个文件，都要执行
//go run server.go hub.go
func main() {

	go h.run()

	http.HandleFunc("/",serverWs )

	err := http.ListenAndServe(":8899", nil)

	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}
结果:

ws://127.0.0.1:8899/?roomid=tang
ws://127.0.0.2:8899/?roomid=tang
ws://127.0.0.3:8899/?roomid=wang
ws://127.0.0.4:8899/?roomid=wang
ws://127.0.0.5:8899/?roomid=wang
记录的打印信息,把请求连接加入多房间聊天室rooms,此处创建了两个房间,
在这里插入图片描述
以下是发送信息的显示结果:
发现有一点没有处理好,就是发的消息有的情况不需要让发信息人看到(显示),
只需要看到提醒信息…有兴趣的小伙伴可以进一步改进,
对于 ***github.com/gorilla/websocket***这个库,我了解使用的还不太多, 只是用它实现了基本功能, 大家也可以深入研究一下,
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

我有辣条 跟我走
关注
使用Go和WebSockets构建实时聊天服务器
FlyWine的博客
04-10 826
使用Go和WebSockets构建实时聊天服务器 源代码连接 Go Chat 现在web应用变得越来越复杂，前端开发人员的工资也是水涨船高。现在的web程序有些是可以实时更新的，用户无需主动调用服务器或刷新浏览器。在早期，开发人员需要依靠ajax请求来创建非常接近实时的应用程序，现在他们能够利用websocket的强大功能来创建完全实时的应用程序。 关于什么是websocket查看这篇文章we...
Go-webSocket实现广播
Lin_Bolun的博客
10-11 1832
WebSocket 协议本质上是一个基于 TCP 的协议。 为了建立一个 WebSocket 连接，客户端浏览器首先要向服务器发起一个 HTTP 请求，这个请求和通常的 HTTP 请求不同，包含了一些附加头信息，其中附加头信息"Upgrade: WebSocket"表明这是一个申请协议升级的 HTTP 请求，服务器端解析这些附加的头信息然后产生应答信息返回给客户端，客户端和服务器端的 WebSocket 连接就建立起来了，双方就可以通过这个连接通道自由的传递信息，并且这个连接会持续存在直到客户端或者服务器端
golang websocket 简易聊天室
free0day的博客
09-18 312
先说点无关的,最近忙于工作没有更新博客,今天休息顺便把golang websocket研究了一下,挺好玩的,写了一个聊天室,分享给大家. websocket包 : code.google.com/p/go.net/websocket 文档 : http://go.pkgdoc.org/code.google.com/p/go.net/websocket 首先安装websocket包 g...
【Go】-Websocket的使用
最新发布
knoci的博客
09-25 1269
可以理解为实现这样一种功能：服务器端可以即时地将数据的更新或变化反应到客户端，例如消息推送等功能都是通过这种技术实现的。但是在Web中，由于浏览器的限制，实现即时通讯需要借助一些方法。这种限制出现的主要原因是，一般的Web通信都是浏览器先发送请求到服务器，服务器再进行响应完成数据的现实更新。轮询、长轮询(comet)、长连接(SSE)、WebSocket。它们大体可以分为两类，一种是在HTTP基础上实现的，包括短轮询、长轮询(comet)、长连接(SSE)；
Go 快速起步：创建 WebSocket 服务器（聊天室）
Phplayers的博客
06-30 1209
和 HTTP 协议一样，WebSocket 协议 也是建立在 TCP/IP协议 基础上的协议 ，但不一样的是 HTTP 协议 为单向协议，即浏览器只能向服务器请求资源，服务器才能将数据传送给浏览器。 WebSocket 协议 是一种双向通信协议，在建立连接后，WebSocket服务器和 Browser/UA 都能主动的向对方发送或接收数据！...
GO websocket 实现简易聊天室
qq_52209929的博客
10-18 1239
GO websocket 实现简易聊天室
桫哥-GOlang基础-Go语言实战：多人聊天室（下）
06-12
在本课程"桫哥-GOlang基础-Go语言实战：多人聊天室（下）"中，我们将深入探讨如何使用Go语言构建一个支持多人实时交流的聊天室应用。这个项目不仅涵盖了Go语言的基础语法，还涉及到了网络编程、并发处理以及框架封装...
[Go WebSocket] 单房间的聊天室
Hull Qin
09-19 4353
如果你想实现WebSocket，基于压测数据，强烈建议你用Go！本文手把手教你实现一个支持多人聊天的聊天室服务，很适合用于新手入门！每一行代码都有详细注释。
Go websocket 聊天室demo以及k8s 部署
dz45693的专栏
01-09 2859
本来打算练习go websocket 做一个示例，结果在网上找了一个聊天室的示例【Go websocket 聊天室的详细实现和详细分析_上官二狗的博客-CSDN博客_go websocket 聊天室】，感觉不错就拿来用一下。 介绍 首先需要有一个客户端 client 的 manager ，manager 里应该保存所有的client 信息 所以在我们的程序里定义了 ClientManager 这个结构体 用 clients 这个 map 结构来保存所有的连接信息 遍历 clients 通过使用 bro.
Laravel + connmix 开发分布式 WebSocket 聊天室
撸代码的乡下人
05-06 1118
connmix 是一个基于 go + lua 开发面向消息编程的分布式长连接引擎，可用于互联网、即时通讯、APP开发、网络游戏、硬件通讯、智能家居、物联网等领域的开发，支持 java,php,go,nodejs 等各种语言的客户端。
使用Go和WebSocket构建实时聊天应用
sybh的博客
06-06 147
用户注册和登录创建和加入聊天室在聊天室中发送和接收消息后端: Go (Gorilla WebSocket)前端: Vue.js数据库: PostgreSQL。
Go-golang实现的长连接服务适合基于房间的聊天推送场景
08-13
golang实现的长连接服务，适合基于房间的聊天推送场景
html5+go+websocket不到150行代码，实现一个在线实时聊天的功能
qq_39962403的博客
03-06 1403
在了解什么是websocket之前，我们下说一说http，因为HTTP我们太熟了。我们知道，HTTP是一种基于应用层的网络协议，往往都是一个请求，一个相应。websocket呢，也是一种基于应用层的网络协议，但是它不仅可以实现请求-相应这种模式，还可以实现主动推送，即你不请求，我也可以给你发消息通知。它实现了浏览器与服务器之间的双工通信。浏览器和服务器只需要完成一次握手，两者就可以创建一个持续的链接。
websocket+golang聊天室
diaoliu7156的博客
12-19 381
原文地址: http://www.niu12.com/article/3 websocket+golang聊天室main.go和index.html放在同一目录下main.gopackage mainimport ( "encoding/json" "fmt" "golang.org/x/net/websocket" "net/http" "time")ty...
字节为何选择go语言作为开发语言，gin框架实现一个webSocket聊天功能的案例
m0_46168848的博客
09-01 801
我们可以理解进程是计算机资源分配的最小单元，也是CPU分配资源的最小单位，具有独立的内存。go天生支持高并发，符合字节跳动这种流媒体的业务场景，而JAVA的优势是生态比较强大，从web应用到大数据、云计算，JAVA都有成熟的解决方案。go诞生于Google公司的工程师，是出自c语言之父之手，底层是拿c开发的，所以可想而知go具备高性能和天生支持高并发的特性。，像docker，k8s，底层都是拿go语言实现的。所以java有它的优势，go也有它的优势，没有谁比谁更好这一说，只有合适。
go websocket 单聊 推送(群聊)实现
Xly_blog
04-29 1135
直接上代码了 在main方法里进行端口监听 在server包实现具体逻辑，首先定义3个结构体 分别是所有客户端管理者(申明了有两个map 这两个map分别是绑定的是所有客户端连接，和单个客户端连接),后面两个分别是单个客户端和上传数据类型 接下来就是具体的实现逻辑 这只是一个demo示例，其中只开了一个read不断读取的协程，没有开启write协程，让他们通过管道来通...
golang聊天室的搭建(实现room,群聊,指定room推送)加锁版本
土豆科技的博客
07-07 2871
package main import ( "log" "net/http" "github.com/gorilla/websocket" "sync" "fmt" "strconv" "encoding/json" ) //定义我们的消息对象 type Message struct { Room_id int
[Go WebSocket] 多房间的聊天室（四）黑天鹅事件
Hull Qin
09-23 4066
多房间的聊天室的第四篇。上次我们开发了「多房间聊天室」的自动清理无人房间的逻辑，避免了内存无限增长。但是上次的实现并不完美，无法保证100%正确，今天我们做个优化。
WebSocket并搭建简易聊天室
玄霄
07-05 3752
WebSocketWebSocketWebSocket特点代码基本框架客户端服务端代码示例发送与接收客户端实现服务端实现长连接在线聊天室设计思路客户端服务端代码存在的问题用户身份解决方法用户姓名的显示代码并发优化解决方法 WebSocket WebSocket，是一个基于TCP的通讯协议，适用于网站应用。 HTTP：短连接的TCP协议 WebSocket: 长连接的TCP协议 常见的应用场景:基于Web的聊天室、在线客服 WebSocket特点 单一TCP长连接，采用全双工通信模式（HTTP是半双工，
Go语言实现WebSocket多人聊天室
在实际应用中，可能需要更复杂的逻辑，比如管理多个并发的WebSocket连接，或者实现聊天室功能。为了实现聊天室，你可以维护一个连接池，每个连接代表一个用户，当接收到新的消息时，遍历连接池并发送消息到所有活跃...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值