使用Go实现一个百行聊天服务器

2023-12-16 22:44

本文主要是介绍使用Go实现一个百行聊天服务器,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

前段时间, redis作者不是整了个c语言版本的聊天服务器嘛, 地址, 代码量拢共不过百行.

于是, 心血来潮下, 我也整了个Go语言版本. 简单来说就是实现了一个聊天室的功能. 将所有注释空行都去掉, 刚好100行实现. 废话不多说, 先上代码:

package mainimport ("fmt""log""net""strings""sync"
)type Server struct { // 服务端内容clients map[string]*Clientlock    sync.Mutex
}func (s *Server) delClient(client *Client) { // 客户端关闭s.lock.Lock()defer s.lock.Unlock()delete(s.clients, client.name)
}func (s *Server) addClient(client *Client) { // 客户端关闭s.lock.Lock()defer s.lock.Unlock()s.clients[client.name] = client
}func (s *Server) sendMsgToOtherClient(msg string, client *Client) { // 将消息发送给其他所有客户端s.lock.Lock()defer s.lock.Unlock()// 将消息转发给其他客户端for _, c := range s.clients {if c != client {c.msgCh <- "msg-> " + client.name + ": " + msg + "\n"}}
}type Client struct { // 定义客户端conn   net.Connname   string      // 当前客户端的名称msgCh  chan string // 发送消息的管道server *Server
}func (c *Client) receive() { // 接收消息for msg := range c.msgCh {_, _ = c.conn.Write([]byte(msg))}
}func (c *Client) close() {c.server.delClient(c)close(c.msgCh)_ = c.conn.Close()
}func (c *Client) handle() { // 开始处理连接_, _ = c.conn.Write([]byte(fmt.Sprintf("hello %s!\n", c.name))) // 发送欢迎信息c.server.sendMsgToOtherClient("join", c)                        // 通知大家, 有人加入了聊天室defer c.close()for {buf := make([]byte, 2048)n, err := c.conn.Read(buf) // 接收客户端发送的消息if err != nil {log.Printf("receive client data error: %s", err.Error())return}msg := strings.TrimSpace(string(buf[:n]))if len(msg) == 0 {continue}if msg == "quit" {c.server.sendMsgToOtherClient("quit", c) // 通知大家, 有人退出了聊天室return}c.server.sendMsgToOtherClient(msg, c)}
}func main() {// 监听端口listener, err := net.Listen("tcp", ":8080")if err != nil {panic(err)}server := &Server{clients: make(map[string]*Client),lock:    sync.Mutex{},}nameIndex := 1for {conn, err := listener.Accept() // 建立连接if err != nil {log.Println(err)continue}client := &Client{conn:   conn,msgCh:  make(chan string, 100),name:   fmt.Sprintf("user%d", nameIndex),server: server,}nameIndex++server.addClient(client)go client.handle()go client.receive()log.Printf("new client: %s\n", conn.RemoteAddr())}
}

可以直接telnet作为客户端连接, 实现的功能简单来说就是一个大的聊天室, 用户发的消息会同步发给所有用户. (因代码篇幅和复杂度原因, 就不对代码做详细说明了)

看下聊天室效果:

image-20231216220013277

当然了, 还是有很多极限情况没有处理, 很多异常情况没有判断. 比如:

  • 客户端数量没有控制
  • 缺少客户端心跳
  • 缺少错误处理
  • 缺少安全性检查
  • 客户端发送消息的长度限制
  • 等等…

但总得来说, 作为一个玩具还是可以的, 而且也达到百行的要求咯.(这也是我改了几版才改到百行的…)

这篇关于使用Go实现一个百行聊天服务器的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/502153

相关文章

解决Maven项目idea找不到本地仓库jar包问题以及使用mvn install:install-file

《解决Maven项目idea找不到本地仓库jar包问题以及使用mvninstall:install-file》:本文主要介绍解决Maven项目idea找不到本地仓库jar包问题以及使用mvnin... 目录Maven项目idea找不到本地仓库jar包以及使用mvn install:install-file基

Python使用getopt处理命令行参数示例解析(最佳实践)

《Python使用getopt处理命令行参数示例解析(最佳实践)》getopt模块是Python标准库中一个简单但强大的命令行参数处理工具,它特别适合那些需要快速实现基本命令行参数解析的场景,或者需要... 目录为什么需要处理命令行参数?getopt模块基础实际应用示例与其他参数处理方式的比较常见问http

python实现svg图片转换为png和gif

《python实现svg图片转换为png和gif》这篇文章主要为大家详细介绍了python如何实现将svg图片格式转换为png和gif,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录python实现svg图片转换为png和gifpython实现图片格式之间的相互转换延展:基于Py

Python利用ElementTree实现快速解析XML文件

《Python利用ElementTree实现快速解析XML文件》ElementTree是Python标准库的一部分,而且是Python标准库中用于解析和操作XML数据的模块,下面小编就来和大家详细讲讲... 目录一、XML文件解析到底有多重要二、ElementTree快速入门1. 加载XML的两种方式2.

C 语言中enum枚举的定义和使用小结

《C语言中enum枚举的定义和使用小结》在C语言里,enum(枚举)是一种用户自定义的数据类型,它能够让你创建一组具名的整数常量,下面我会从定义、使用、特性等方面详细介绍enum,感兴趣的朋友一起看... 目录1、引言2、基本定义3、定义枚举变量4、自定义枚举常量的值5、枚举与switch语句结合使用6、枚

Java的栈与队列实现代码解析

《Java的栈与队列实现代码解析》栈是常见的线性数据结构,栈的特点是以先进后出的形式,后进先出,先进后出,分为栈底和栈顶,栈应用于内存的分配,表达式求值,存储临时的数据和方法的调用等,本文给大家介绍J... 目录栈的概念(Stack)栈的实现代码队列(Queue)模拟实现队列(双链表实现)循环队列(循环数组

使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)

《使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)》PPT是一种高效的信息展示工具,广泛应用于教育、商务和设计等多个领域,PPT文档中常常包含丰富的图片内容,这些图片不仅提升了... 目录一、引言二、环境与工具三、python 提取PPT背景图片3.1 提取幻灯片背景图片3.2 提取

C++如何通过Qt反射机制实现数据类序列化

《C++如何通过Qt反射机制实现数据类序列化》在C++工程中经常需要使用数据类,并对数据类进行存储、打印、调试等操作,所以本文就来聊聊C++如何通过Qt反射机制实现数据类序列化吧... 目录设计预期设计思路代码实现使用方法在 C++ 工程中经常需要使用数据类,并对数据类进行存储、打印、调试等操作。由于数据类

Python实现图片分割的多种方法总结

《Python实现图片分割的多种方法总结》图片分割是图像处理中的一个重要任务,它的目标是将图像划分为多个区域或者对象,本文为大家整理了一些常用的分割方法,大家可以根据需求自行选择... 目录1. 基于传统图像处理的分割方法(1) 使用固定阈值分割图片(2) 自适应阈值分割(3) 使用图像边缘检测分割(4)

Android实现在线预览office文档的示例详解

《Android实现在线预览office文档的示例详解》在移动端展示在线Office文档(如Word、Excel、PPT)是一项常见需求,这篇文章为大家重点介绍了两种方案的实现方法,希望对大家有一定的... 目录一、项目概述二、相关技术知识三、实现思路3.1 方案一:WebView + Office Onl