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

2023-12-17 06:45

本文主要是介绍使用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/503464

相关文章

vue使用docxtemplater导出word

《vue使用docxtemplater导出word》docxtemplater是一种邮件合并工具,以编程方式使用并处理条件、循环,并且可以扩展以插入任何内容,下面我们来看看如何使用docxtempl... 目录docxtemplatervue使用docxtemplater导出word安装常用语法 封装导出方

SpringBoot3实现Gzip压缩优化的技术指南

《SpringBoot3实现Gzip压缩优化的技术指南》随着Web应用的用户量和数据量增加,网络带宽和页面加载速度逐渐成为瓶颈,为了减少数据传输量,提高用户体验,我们可以使用Gzip压缩HTTP响应,... 目录1、简述2、配置2.1 添加依赖2.2 配置 Gzip 压缩3、服务端应用4、前端应用4.1 N

Linux换行符的使用方法详解

《Linux换行符的使用方法详解》本文介绍了Linux中常用的换行符LF及其在文件中的表示,展示了如何使用sed命令替换换行符,并列举了与换行符处理相关的Linux命令,通过代码讲解的非常详细,需要的... 目录简介检测文件中的换行符使用 cat -A 查看换行符使用 od -c 检查字符换行符格式转换将

Go标准库常见错误分析和解决办法

《Go标准库常见错误分析和解决办法》Go语言的标准库为开发者提供了丰富且高效的工具,涵盖了从网络编程到文件操作等各个方面,然而,标准库虽好,使用不当却可能适得其反,正所谓工欲善其事,必先利其器,本文将... 目录1. 使用了错误的time.Duration2. time.After导致的内存泄漏3. jsO

SpringBoot实现数据库读写分离的3种方法小结

《SpringBoot实现数据库读写分离的3种方法小结》为了提高系统的读写性能和可用性,读写分离是一种经典的数据库架构模式,在SpringBoot应用中,有多种方式可以实现数据库读写分离,本文将介绍三... 目录一、数据库读写分离概述二、方案一:基于AbstractRoutingDataSource实现动态

Python FastAPI+Celery+RabbitMQ实现分布式图片水印处理系统

《PythonFastAPI+Celery+RabbitMQ实现分布式图片水印处理系统》这篇文章主要为大家详细介绍了PythonFastAPI如何结合Celery以及RabbitMQ实现简单的分布式... 实现思路FastAPI 服务器Celery 任务队列RabbitMQ 作为消息代理定时任务处理完整

使用Jackson进行JSON生成与解析的新手指南

《使用Jackson进行JSON生成与解析的新手指南》这篇文章主要为大家详细介绍了如何使用Jackson进行JSON生成与解析处理,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录1. 核心依赖2. 基础用法2.1 对象转 jsON(序列化)2.2 JSON 转对象(反序列化)3.

Java枚举类实现Key-Value映射的多种实现方式

《Java枚举类实现Key-Value映射的多种实现方式》在Java开发中,枚举(Enum)是一种特殊的类,本文将详细介绍Java枚举类实现key-value映射的多种方式,有需要的小伙伴可以根据需要... 目录前言一、基础实现方式1.1 为枚举添加属性和构造方法二、http://www.cppcns.co

使用Python实现快速搭建本地HTTP服务器

《使用Python实现快速搭建本地HTTP服务器》:本文主要介绍如何使用Python快速搭建本地HTTP服务器,轻松实现一键HTTP文件共享,同时结合二维码技术,让访问更简单,感兴趣的小伙伴可以了... 目录1. 概述2. 快速搭建 HTTP 文件共享服务2.1 核心思路2.2 代码实现2.3 代码解读3.

Elasticsearch 在 Java 中的使用教程

《Elasticsearch在Java中的使用教程》Elasticsearch是一个分布式搜索和分析引擎,基于ApacheLucene构建,能够实现实时数据的存储、搜索、和分析,它广泛应用于全文... 目录1. Elasticsearch 简介2. 环境准备2.1 安装 Elasticsearch2.2 J