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

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

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

go 复制代码
package main

import (
	"fmt"
	"log"
	"net"
	"strings"
	"sync"
)

type Server struct { // 服务端内容
	clients map[string]*Client
	lock    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.Conn
	name   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 := 1
	for {
		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作为客户端连接, 实现的功能简单来说就是一个大的聊天室, 用户发的消息会同步发给所有用户. (因代码篇幅和复杂度原因, 就不对代码做详细说明了)

看下聊天室效果:

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

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

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

相关推荐
头发那是一根不剩了1 小时前
Spring Boot 多数据源切换:AbstractRoutingDataSource
数据库·spring boot·后端
网硕互联的小客服1 小时前
服务器经常出现蓝屏是什么原因导致的?如何排查和修复?
运维·服务器·stm32·单片机·网络安全
witton1 小时前
Go语言网络游戏服务器模块化编程
服务器·开发语言·游戏·golang·origin·模块化·耦合
成都极云科技1 小时前
成都算力租赁新趋势:H20 八卡服务器如何重塑 AI 产业格局?
大数据·服务器·人工智能·云计算·gpu算力
喜欢吃豆1 小时前
从零构建MCP服务器:FastMCP实战指南
运维·服务器·人工智能·python·大模型·mcp
刘阿宾1 小时前
【华为昇腾|CUDA】服务器A6000显卡部署LLM实战记录
服务器·华为·语言模型·gpu算力·kylin
叹一曲当时只道是寻常1 小时前
Softhub软件下载站实战开发(十六):仪表盘前端设计与实现
前端·golang
海外空间恒创科技1 小时前
恒创科技:香港站群服务器做seo站群优化效果如何
运维·服务器·科技
搬码临时工1 小时前
内网服务器怎么设置公网远程访问? windows桌面连接和Linux自带SSH外网异地跨网用完整步骤教程
运维·服务器·ssh
草履虫建模1 小时前
Redis:高性能内存数据库与缓存利器
java·数据库·spring boot·redis·分布式·mysql·缓存