goLang简单聊天室

最新推荐文章于 2024-07-19 04:03:36 发布
最新推荐文章于 2024-07-19 04:03:36 发布
阅读量986 收藏 1
点赞数 1
分类专栏： goLang 文章标签： golang go go聊天室 聊天室
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/a568283992/article/details/117199968
版权

Go语言 TCP 聊天室 编码解码 Redis
关键词由CSDN通过智能技术生成

有一段时间没写博客了，现在因为工作需要学习了go，做一个小Demo，简单的聊天室。

  1. 首先服务端开启监听8889端口，监听每一个客户端。
  2. 客户端需要先登录，登录成功后，服务端会发送给每一个客户端一个消息，XXX用户登陆成功，目前程序还没写完全，后续会不断更新~~
  • message.go:

首先需要一个消息传送的protocol，因为是基于TCP协议，传输过程中会有粘包和拆包问题，因此定义一个协议来保证数据完整性。

package message

const (
	LoginMesType    = "LoginMes"
	RegisterMesType = "RegisterMesType"
	MesType         = "MesType"
)

type Message struct {
	Type string `json:"type"` // 消息类型
	Data string `json:"data"` // 消息内容
}

type LoginMes struct {
	UserId   int    `json:"userId"`   // 用户id
	UserPwd  string `json:"userPwd"`  // 用户密码
	UserName string `json:"userName"` // 用户名
}
  • coding.go:

这个包是用来编码和解码的作用，当发送数据时需要分为2段编码，然后转为字节进行发送。接收数据时，通过反序列化直接就可以获得数据。

package coding

import (
	"bufio"
	"bytes"
	"encoding/binary"
	"encoding/json"
	"fmt"
)

func Encode(i interface{}) (b []byte, err error) {
	//将message进行序列化
	data, err := json.Marshal(i)
	if err != nil {
		fmt.Println("json Marshal error")
		return
	}
	//这个时候data就是要发送的消息
	var pkgLen uint32
	var pkg = new(bytes.Buffer)
	pkgLen = uint32(len(data))
	// 写入头
	err = binary.Write(pkg, binary.LittleEndian, pkgLen)
	// 写入体
	err = binary.Write(pkg, binary.LittleEndian, data)
	b = pkg.Bytes()
	return
}

func Decode(reader *bufio.Reader) (b []byte, err error) {
	// 读取消息
	preBytes, err := reader.Peek(4)
	var length uint32
	err = binary.Read(bytes.NewBuffer(preBytes), binary.LittleEndian, &length)
	if err != nil {
		fmt.Println("读取头4字节失败", err)
		return
	}
	if uint32(reader.Buffered()) < length+4 {
		return []byte{}, fmt.Errorf("读取到%d，应为%d", reader.Buffered(), length+4)
	}
	fmt.Println("body长度为", length)
	p := make([]byte, length+4)
	_, err = reader.Read(p)
	if err != nil {
		fmt.Println("读取头数据体失败")
		return
	}
	b = p[4:]
	return
}
  • server.go

服务器主要代码，因为是个简单demo，没有考虑很多。

package main

import (
	"bufio"
	"coding"
	"encoding/json"
	"fmt"
	"message"
	"net"
	"redisCli"
	"strconv"
)

var r *redisCli.RedisCli

var online map[net.Conn]message.LoginMes

func init() {
	r, _ = redisCli.NewRedisCli("tcp", "127.0.0.1:6379")
	online = make(map[net.Conn]message.LoginMes, 10)
}

func handleLogin(b []byte) (loginMes message.LoginMes) {
	_ = json.Unmarshal(b, &loginMes)
	//使用redis操作是否存在该账号
	str, err := r.GetString(strconv.Itoa(loginMes.UserId) + ":" + loginMes.UserPwd)
	if err != nil {
		return
	}
	//if str == "" {
	//	fmt.Println("登录失败")
	//	return
	//}
	loginMes.UserName = str
	return
}

func handleRegister(b []byte) {

}

func process(conn net.Conn) {
	// 读取客户端发送的信息
	reader := bufio.NewReader(conn)
	mes := message.Message{}
	b, err := coding.Decode(reader)
	if err != nil {
		fmt.Println("decode失败")
		return
	}
	err = json.Unmarshal(b, &mes)
	if err != nil {
		fmt.Println("反序列化失败")
		return
	}
	dataBytes := []byte(mes.Data)
	switch mes.Type {
	case message.LoginMesType: // 登录消息
		loginMes := handleLogin(dataBytes)
		online[conn] = loginMes
		mes := message.Message{
			Type: message.MesType,
			Data: loginMes.UserName + "加入群聊啦",
		}
		b, err := coding.Encode(mes)
		if err != nil {
			fmt.Println("序列化失败")
			return
		}
		for c, _ := range online {
			_, _ = c.Write(b)
		}
	case message.RegisterMesType: // 注册消息
		handleRegister(dataBytes)
	}
}

func main() {
	fmt.Println("服务器从8889端口监听")
	listen, err := net.Listen("tcp", "0.0.0.0:8889")
	if err != nil {
		fmt.Println("net listen error")
		return
	}
	//监听成功等待
	for {
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("net client error")
			return
		}
		//连接成功就保持通讯
		go process(conn)
	}
}
  • client.go:

客户端代码

package main

import (
	"bufio"
	"coding"
	"context"
	"fmt"
	"message"
	"sync"
)

var (
	userId  int
	userPwd string
)

var wg sync.WaitGroup

func main() {
	// 接收用户的选择
	var key int
	// 判断是否还继续显示菜单
	var loop = true
	for loop {
		fmt.Println("----------欢迎登录多人聊天系统----------")
		fmt.Println("\t\t\t 1 登录聊天系统")
		fmt.Println("\t\t\t 2 注册用户")
		fmt.Println("\t\t\t 3 退出系统")
		fmt.Println("\t\t\t 请选择(1-3):")
		_, _ = fmt.Scanf("%d\n", &key)
		switch key {
		case 1:
			fmt.Println("登录聊天系统")
			loop = false
		case 2:
			fmt.Println("注册用户")
		case 3:
			fmt.Println("退出系统")
			loop = false
		default:
			fmt.Println("输入有误 请重新输入")
		}
	}

	if key == 1 {
		// 说明用户要登录
		fmt.Println("请输入用户id")
		_, _ = fmt.Scanf("%d\n", &userId)
		fmt.Println("请输入用户密码")
		_, _ = fmt.Scanf("%s\n", &userPwd)
		// 先把登录的函数写到另一个文件中
		conn, err := login(userId, userPwd)
		if err != nil {
			fmt.Println("登录失败了")
		} else {
			fmt.Println("登录成功了")
			// 这里需要监听服务器发来的信息
			ctx, cancel := context.WithCancel(context.Background())
			wg.Add(2)
			// 开启接收端进程
			go func(ctx context.Context, cancel context.CancelFunc) {
				fmt.Println("开启接收进程")
				defer wg.Done()
				defer cancel()
				for {
					select {
					case <-ctx.Done():
						fmt.Println("退出接收进程")
						return
					default:
						b, err := coding.Decode(bufio.NewReader(conn))
						if err != nil {
							fmt.Println("退出接收进程, err:", err)
							return
						}
						str := string(b)
						fmt.Println("收到消息：", str)
						if "exit" == str {
							return
						}
					}
				}
			}(ctx, cancel)
			// 开启发送端进程
			go func(ctx context.Context) {
				defer wg.Done()
				fmt.Println("开启写入进程")
				for {
					select {
					case <-ctx.Done():
						fmt.Println("退出写入进程")
						return
					default:
						var str string
						_, _ = fmt.Scanf("%s\n", &str)
						mes := message.Message{
							Type: message.MesType,
							Data: str,
						}
						b, err := coding.Encode(mes)
						if err != nil {
							fmt.Println("序列化失败")
							continue
						}
						_, _ = conn.Write(b)
					}
				}
			}(ctx)
			wg.Wait()
		}
	} else if key == 2 {
		fmt.Println("进行用户注册")
	}

}



// 写一个函数，完成登录
func login(userId int, userPwd string) (conn net.Conn, err error) {
	// 开始定协议
	conn, err = net.Dial("tcp", "localhost:8889")
	if err != nil {
		fmt.Println("net Dial error")
		return
	}
	// 准备conn发送消息
	var mes message.Message
	mes.Type = message.LoginMesType
	// 创建一个LoginMes 结构体
	var loginMes message.LoginMes
	loginMes.UserId = userId
	loginMes.UserPwd = userPwd
	b, err := json.Marshal(loginMes)
	if err != nil {
		fmt.Println("json Marshal error")
		return
	}
	mes.Data = string(b)
	data, err := coding.Encode(mes)
	if err != nil {
		fmt.Println("data Marshal error")
		return
	}
	_, _ = conn.Write(data)
	return
}

效果：

  • 启动服务器

  • 加入第一个客户端，第一个客户端显示如下

  • 加入第二个客户端，第一个客户端显示如下

SineyCoder
关注
  • 1
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
专栏目录
golang实现简单聊天室
hechunbao987的博客
04-20 685
server端 package main import ( "fmt" "net" "strconv" "strings" ) type user struct { conn net.Conn name string } //定义用户结构体 var online map[string]*user //该map储存在线用户 var msg chan string //定义全局的消息channel,接受到的消息会储存在这里,然后通过manager读取消息,做出相应处理 func man
golang通过socket实现简单多人聊天室
热门推荐
qq_15267283的博客
02-06 1万+
聊天室包括以下功能: 聊天 超时检查断开连接 查看用户列表 改名 退出 server.go package main import ( "fmt" "net" "strings" "time" ) func main() { // 监听聊天室端口 port := "8888" listener, err := net.Listen("tcp", ":"+port) if err != nil { fmt.Println(err) return } fmt.Printf("
golang 聊天室
行者游学
07-25 1887
使用golang实现简单的websocket聊天室
golang|练手项目|在线聊天室（极简版）
最新发布
wysjr的博客
07-19 587
现在是凌晨三点，从11点30开始写，刚开始想着30分钟完成，12点睡觉，没想到居然拖到现在。写文章的中间总是参杂着许多细碎的事情，不是去洗衣服，就是吃东西，看微信，抖音，这些与写文章无关的事物。唉，做事总是如此，设立目标，却不贯彻执行，面对有点难度的事情，潜意识反应总是逃避，并顺带找一个看似正当的借口麻痹自己（如：好像有点饿，先吃点东西再看看怎么解决吧），以减轻内心的谴责。但回归理性又何尝不明白，逃避并没有解决问题，不断的逃避，问题就会不断累积，总会有一天会陷入无处可逃的境地。2.遇到困难直面解决！
golang实现聊天室
ChangUper的博客
08-25 814
用golang实现聊天室用户上线以及群发消息功能
golang 简易聊天
weixin_34186128的博客
04-11 120
client.go ------------------------------ package main import ( "net" "fmt" "bufio" "echo/codec" "time" ) var quitSemaphore chan bool func main() { var tcpAd...
Golang 搭建 Web 聊天室
06-07
【Golang 搭建 Web 聊天室】是一个关于使用 Go 语言构建实时通信应用的教程，重点是利用 WebSocket 协议实现一个在线聊天室。WebSocket 是一种在 Web 端支持双向通信的协议，它克服了 HTTP 的单向传输限制，允许...
golang基于websocket实现的简易聊天室程序
09-21
### Golang 基于 WebSocket 实现的简易聊天室程序详解 #### 一、引言 随着互联网技术的发展，实时通信的需求越来越广泛。WebSocket 作为一种在单个 TCP 连接上进行全双工通信的协议，能够实现实时的数据交换。本篇...
一百行Golang代码实现简单并发聊天室
09-18
在本文中，我们将深入探讨如何使用Golang编写一个简单的并发聊天室。这个聊天室的实现主要依赖于Golang的并发特性，尤其是goroutines和channels。首先，我们要了解两个关键概念：`onlinemap`全局map和`Message`全局...
golang完成聊天室功能
weixin_48445640的博客
05-02 1647
go语言易于实现高并发，简易聊天室如下： package main import ( "fmt" "net" ) /* 主go程：创建监听socket，for循环accept获取到客户端的conn，开启goroutine HandlerConnect HandlerConnect:创建用户结构体，存入onlineMap，发送用户登录广播 Manager:监听全局的channel message，将读到的消息广播给onlineMap中的所有用户 WriteMsgToClient:读取每个用户自带cha
GolangChat:聊天系统
03-11
高朗聊天 聊天系统 1.0版本已实现实现单撩实现群撩实现用户上线的全网通知实现用户昵称实现聊天日志的存储和查看 2.0版本等待
Golang实现并发聊天室
01-20
目录结构前言以及项目简介门槛项目演示截图项目流程图各个功能模块详细说明全局变量以及关键函数广播用户上线广播用户消息展示在线用户名修改用户名用户退出超时处理总结源码 前言以及项目简介 Golang是一门极为出色的语言，近些年也越发火热。每一种语言火起来都有它的道理，Golang就是以其独树一帜的并发处理和高性能博得了众多程序员的青睐，不少的C/C++、Java、PHP开发者都已经或逐渐转型扑向Go语言的怀抱。 从当初刚刚接触Go到现在大概有25天了，一直在看某马的培训视频，也确实学到了不少东西。这个并发聊天室就是他们GO语言与区块链就业班的阶段性学习项目。该项目处于整个课程中的第二阶段——并发编
golang简易版聊天室
Lesroad
12-01 471
功能需求： 创建一个聊天室，实现群聊和单聊的功能，直接输入为群聊，@某人后输入为单聊 效果图： 群聊： 单聊： 服务端： package main import ( "fmt" "net" "strings" ) type info struct { conn net.Conn name string } var...
golang实现聊天室（一）
csdn_kou的博客
01-27 484
一个简单的tcp通讯 客户端发一下，服务端收一下打印 客户端 func main() { var conn, err = net.DialTimeout("tcp", "127.0.0.1:8888", 30*time.Second) if err != nil { return } conn.Write([]byte("string not ")) conn.Close() ...
golang聊天室
baixiao9898的博客
03-05 137
golang对并发的支持非常好，就是因为他有一个goroutine机制和channel机制，实现并发场景轻松高效。 下面我们实现一个聊天室程序，代码不多，但是功能很强大 //代码转载自：http://blog.csdn.net/ygrx/article/details/11773151 package main import ( "fmt" "net" "os...
Golang实现完整聊天室（内附源码）
paterl的博客
08-12 1692
一篇带你掌握IM聊天原理
[Golang]简单的聊天室实现
Sunface撩技术
03-06 3628
最近写了一个chrome的websocket extension, server端用的是Java的Netty框架, 后来发现golang 实现websocket 非常简单，高效， 下面是例子， 简单实现了用户登录，广播，相当于聊天室！[plain] view plaincopyprint?package main    import (      "code.google.com/p/go.n
golang websocekt 聊天室流程
02-03
golang websocekt 聊天室流程一般包括以下几个步骤： 1. 建立服务端：首先需要编写一个基于golang的websocekt服务端，负责接收客户端连接，并处理客户端发送的消息。可以使用gorilla/websocket库来实现websocekt...
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值