概述

今天我们会结合之前几节课的知识来综合实战一下, 实现一个聊天室.

服务端实现

运行的时候我们可以开启一个服务端和 N 个客户端, 来实现聊天室.

代码:

package main

import (
	"fmt"
	"log"
	"net"
	"os"
	"strings"
	"time"
)

// ================常量================
const (

	// 日志路径
	log_path = "chat_room/"
)


// ===============全局变量==============

// 文件日志
var logFile *os.File

// 日志类
var logger *log.Logger

// 客户端连接, key: ip端口, value: 连接对象
var onlineConns = make(map[string]net.Conn)

// 消息队列, 缓冲区
var message_quene = make(chan string, 1024)

// 消息, 处理程序退出
var quitchan = make(chan bool)

// 消息协程
func comsume_msg() 
	for 
		select 
		// 取出消息
		case msg := <- message_quene:
			process_msg(msg)
		// 处理退出
		case <- quitchan:
			break
		
	


// 消息解析协程
func process_msg(message string) 

	// 字符串切割
	contents := strings.Split(message, "#")
	if len(contents) > 1 

		// 取出地址
		address := contents[0]

		// 取出消息
		message := contents[1]

		// 删除首尾
		address = strings.Trim(address, " ")

		// 在线连接
		conn, ok := onlineConns[address]

		if ok 
			_, err := conn.Write([]byte(message))
			fmt.Println("发送消息:", message, "目的地:", address)
			if err != nil 
				fmt.Println("在线连接发送失败")
			
		
	 else 


		// 查看list
		contents = strings.Split(message, "&")
		if contents[1] == "list" 

			var str = ""

			// 向每个客户端发送信息
			for i := range onlineConns 

				//
				str += "||||" + i
			

			// 在线连接
			conn, ok := onlineConns[contents[0]]

			if ok 
				_, err := conn.Write([]byte(str))
				if err != nil 
					fmt.Println("在线发送失败:", err)
				
			
			fmt.Println("发送消息:", str, "目的地:", conn.RemoteAddr())
			logger.Println("发送消息:", str, "目的地:", conn.RemoteAddr())
		
	



// 接收消息
func receive_info(conn net.Conn) 

	// 缓冲
	buffer := make([]byte, 1024)

	// 循环读取
	for 

		// 读取数据
		nums, err := conn.Read(buffer)
		if err != nil 
			break
		

		if nums != 0 

			// 获取地址
			address := conn.RemoteAddr()

			// 获取消息
			message := string(buffer[:nums])

			// 调试输出
			fmt.Println("收到消息:", message, "来自:", address)
			logger.Println("收到消息:", message, "来自:", address)

			// 处理客户端退出
			if message == "exit" 

				// 调试输出
				fmt.Println("客户端:", conn.RemoteAddr(), "正在退出...")
				logger.Println("客户端:", conn.RemoteAddr(), "正在退出...")

				// 退出
				client_exit(conn)
			 else
				// 消息队列存储消息
				message_quene <- message
			
		
	


// 处理退出
func client_exit(conn net.Conn) 

	// 获取地址
	address := fmt.Sprint(conn.RemoteAddr())

	// 客户端退出时, 从map中移除地址
	delete(onlineConns, address)

	// 关闭连接
	conn.Close()

	// 输出当前列表
	fmt.Println("客户端列表:\\n--------------------")
	logger.Println("客户端列表:\\n--------------------")


	for i := range onlineConns 
		fmt.Println(i)
	


// 错误处理
func error_check(err error) 
	if err != nil 
		fmt.Println("Error:", err)
		os.Exit(1)
	


func main() 

	fmt.Println("服务端正在启动...")

	// 打开日志文件
	name := fmt.Sprintf("%d_%02d_%02d_%02d_%02d_%02d_server.log",
		time.Now().Year(),
		time.Now().Month(),
		time.Now().Day(),
		time.Now().Hour(),
	    time.Now().Minute(),
		time.Now().Second(),
	)

	log_file, err := os.OpenFile((log_path + name), os.O_RDWR | os.O_CREATE, 0)
	if err != nil 
		fmt.Println("日志文件打开失败:", err.Error())
		os.Exit(-1)
	

	// 关闭
	defer log_file.Close()

	// 创建一个日志对象
	logger = log.New(log_file, "\\r\\n", log.Ldate | log.Ltime | log.Llongfile)
	logger.Println("写入日志, 服务器正在启动...")

	// 创建TCP服务端
	listen_socket, err := net.Listen("tcp", "127.0.0.1:8888")
	error_check(err)

	// 关闭
	defer listen_socket.Close()

	fmt.Println("服务端启动完毕, 等待连接...")

	// 协程
	go comsume_msg()

	// 启动连接
	for 

		// 连接新客户端
		conn, err := listen_socket.Accept()
		error_check(err)

		fmt.Println("服务端连接到客服端, 客户端地址:", conn.RemoteAddr())
		logger.Println("服务端连接到客服端, 客户端地址:", conn.RemoteAddr())


		// 获取地址
		address := fmt.Sprint(conn.RemoteAddr())

		// 添加到全局变量
		onlineConns[address] = conn

		// 遍历每一个连接
		fmt.Println("客户端列表:\\n--------------------")
		logger.Println("客户端列表:\\n---------------------------------------------------")
		for i := range onlineConns 
			fmt.Println(i)
			logger.Println(i)
		

		// 发送消息
		go receive_info(conn)
	

客户端实现

package main

import (
	"bufio"
	"fmt"
	"net"
	"os"
)

// 发送消息
func send_msg(conn net.Conn) 

	// 循环发送
	for 

		// 读取键盘输入
		reader := bufio.NewReader(os.Stdin)

		// 读取一行
		data, _, _ := reader.ReadLine()

		// 发送输入的字符串
		_, err := conn.Write(data)
		fmt.Println("发送消息:", string(data))
		if err != nil 
			conn.Close()
			fmt.Println("Error:", err, "客户端关闭")
			os.Exit(0)
		

		// 收到exit, 关闭客户端
		if string(data) == "exit" 
			fmt.Println("客户端关闭")
			os.Exit(0)
		
	


func main() 

	fmt.Println("客服端正在建立连接...")

	// 建立网络连接
	conn, err := net.Dial("tcp", "127.0.0.1:8888")
	if err != nil 
		fmt.Println("网络连接错误")
		os.Exit(1)
	
	fmt.Println("客户端成功连接到服务端, 服务端地址:", conn.RemoteAddr())

	// 发送消息
	go send_msg(conn)

	// 接收消息
	buffer := make([]byte, 1024)

	// 循环接收
	for 
		// 读取消息
		nums, err := conn.Read(buffer)
		if err != nil 
			fmt.Println("读取消息出错, 退出客户端")
			os.Exit(0)
		
		fmt.Println("收到消息:", string(buffer[:nums]))
	

输出结果:
server:

服务端正在启动...
服务端启动完毕, 等待连接...
服务端连接到客服端, 客户端地址: 127.0.0.1:63776
客户端列表:
--------------------
127.0.0.1:63776
服务端连接到客服端, 客户端地址: 127.0.0.1:63777
客户端列表:
--------------------
127.0.0.1:63776
127.0.0.1:63777
收到消息: 127.0.0.1:63776&list 来自: 127.0.0.1:63776
发送消息: ||||127.0.0.1:63776||||127.0.0.1:63777 目的地: 127.0.0.1:63776
收到消息: 127.0.0.1:63777#hi, client2 来自: 127.0.0.1:63776
发送消息: hi, client2 目的地: 127.0.0.1:63777
收到消息: 127.0.0.1:63777&list 来自: 127.0.0.1:63777
发送消息: ||||127.0.0.1:63776||||127.0.0.1:63777 目的地: 127.0.0.1:63777
收到消息: 127.0.0.1:63776#hi, client1 来自: 127.0.0.1:63777
发送消息: hi, client1 目的地: 127.0.0.1:63776
收到消息: exit 来自: 127.0.0.1:63776
客户端: 127.0.0.1:63776 正在退出...
客户端列表:
--------------------
127.0.0.1:63777
收到消息: exit 来自: 127.0.0.1:63777
客户端: 127.0.0.1:63777 正在退出...
客户端列表:
--------------------

client1:

客服端正在建立连接...
客户端成功连接到服务端, 服务端地址: 127.0.0.1:8888
127.0.0.1:63776&list
发送消息: 127.0.0.1:63776&list
收到消息: ||||127.0.0.1:63776||||127.0.0.1:63777
127.0.0.1:63777#hi, client2
发送消息: 127.0.0.1:63777#hi, client2
收到消息: hi, client1
exit
发送消息: exit
客户端关闭

client2:

客服端正在建立连接...
客户端成功连接到服务端, 服务端地址: 127.0.0.1:8888
收到消息: hi, client2
127.0.0.1:63777&list
发送消息: 127.0.0.1:63777&list
收到消息: ||||127.0.0.1:63776||||127.0.0.1:63777
127.0.0.1:63776#hi, client1
发送消息: 127.0.0.1:63776#hi, client1
exit
发送消息: exit
客户端关闭

日志

2021/08/27 01:08:04 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:209: 写入日志, 服务器正在启动...

2021/08/27 01:08:07 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:231: 服务端连接到客服端, 客户端地址: 127.0.0.1:63776

2021/08/27 01:08:07 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:242: 客户端列表:
---------------------------------------------------

2021/08/27 01:08:07 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:245: 127.0.0.1:63776

2021/08/27 01:08:09 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:231: 服务端连接到客服端, 客户端地址: 127.0.0.1:63777

2021/08/27 01:08:09 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:242: 客户端列表:
---------------------------------------------------

2021/08/27 01:08:09 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:245: 127.0.0.1:63776

2021/08/27 01:08:09 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:245: 127.0.0.1:63777

2021/08/27 01:08:20 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:135: 收到消息: 127.0.0.1:63776&list 来自: 127.0.0.1:63776

2021/08/27 01:08:20 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:104: 发送消息: ||||127.0.0.1:63776||||127.0.0.1:63777 目的地: 127.0.0.1:63776

2021/08/27 01:08:31 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:135: 收到消息: 127.0.0.1:63777#hi, client2 来自: 127.0.0.1:63776

2021/08/27 01:08:42 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:135: 收到消息: 127.0.0.1:63777&list 来自: 127.0.0.1:63777

2021/08/27 01:08:42 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:104: 发送消息: ||||127.0.0.1:63776||||127.0.0.1:63777 目的地: 127.0.0.1:63777

2021/08/27 01:08:53 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:135: 收到消息: 127.0.0.1:63776#hi, client1 来自: 127.0.0.1:63777

2021/08/27 01:09:04 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:135: 收到消息: exit 来自: 127.0.0.1:63776

2021/08/27 01:09:04 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:142: 客户端: 127.0.0.1:63776 正在退出...

2021/08/27 01:09:04 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:168: 客户端列表:
--------------------

2021/08/27 01:09:11 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:135: 收到消息: exit 来自: 127.0.0.1:63777

2021/08/27 01:09:11 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:142: 客户端: 127.0.0.1:63777 正在退出...

2021/08/27 01:09:11 C:/Users/Windows/Desktop/project2/chat_room/server2/server.go:168: 客户端列表:
--------------------