go语言通过TCP协议实现聊天室样例

1、服务端:

复制代码
package main

import (
	"fmt"
	"net"
	"sync"
)

type ChatServer struct {
	clients    map[string]net.Conn
	clientsMux sync.Mutex
}

func NewChatServer() *ChatServer {
	return &ChatServer{
		clients: make(map[string]net.Conn),
	}
}

func (s *ChatServer) broadcast(message string, sender string) {
	s.clientsMux.Lock()
	defer s.clientsMux.Unlock()

	for clientAddr, conn := range s.clients {
		if clientAddr != sender {
			fmt.Fprintf(conn, "[%s]: %s\n", sender, message)
		}
	}
}

func (s *ChatServer) handleConnection(conn net.Conn) {
	defer conn.Close()

	clientAddr := conn.RemoteAddr().String()

	fmt.Printf("New connection from %s\n", clientAddr)

	s.clientsMux.Lock()
	s.clients[clientAddr] = conn
	s.clientsMux.Unlock()

	for {
		buffer := make([]byte, 1024)
		n, err := conn.Read(buffer)
		if err != nil {
			fmt.Printf("Error reading from %s: %v\n", clientAddr, err)
			break
		}

		message := string(buffer[:n])
		fmt.Printf("[%s]: %s\n", clientAddr, message)

		s.broadcast(message, clientAddr)
	}
}

func main() {
	server := NewChatServer()

	listener, err := net.Listen("tcp", ":8080")
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer listener.Close()

	fmt.Println("TCP chat server listening on :8080")

	for {
		conn, err := listener.Accept()
		if err != nil {
			fmt.Println("Error accepting connection:", err)
			continue
		}
		go server.handleConnection(conn)
	}
}

2、客户端:

复制代码
package main

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

func readMessages(conn net.Conn) {
	reader := bufio.NewReader(conn)
	for {
		message, err := reader.ReadString('\n')
		if err != nil {
			fmt.Println("Error reading message:", err)
			break
		}
		fmt.Print(message)
	}
}

func main() {
	conn, err := net.Dial("tcp", "127.0.0.1:8080")
	if err != nil {
		fmt.Println("Error connecting to server:", err)
		return
	}
	defer conn.Close()

	go readMessages(conn)

	for {
		scanner := bufio.NewScanner(os.Stdin)
		scanner.Scan()
		message := scanner.Text()

		_, err := conn.Write([]byte(message + "\n"))
		if err != nil {
			fmt.Println("Error sending message:", err)
			break
		}
	}
}

3、实现效果

相关推荐
晨非辰7 分钟前
#C语言——学习攻略:深挖指针路线(四续)——函数指针数组--转移表
c语言·开发语言·经验分享·学习·visual studio
学习编程的gas23 分钟前
C++:STL中list的使用和模拟实现
开发语言·c++
今晚打老虎41 分钟前
c++之基础B(第一课)
开发语言·c++
wa的一声哭了42 分钟前
Python多进程并行multiprocess基础
开发语言·jvm·人工智能·python·机器学习·语言模型·自然语言处理
liujing102329291 小时前
vmware虚拟机中 ubuntu 20.04通过nat设置静态ip(固定ip)
服务器·tcp/ip·ubuntu
枫叶丹41 小时前
【Qt开发】信号与槽(三)-> 自定义信号和槽
开发语言·qt
奔跑吧邓邓子1 小时前
从0到1学PHP(三):PHP 流程控制:掌控程序的走向
开发语言·php·流程控制
R-G-B1 小时前
【31】C# WinForm入门到精通 ——保存文件SaveFileDialog 【属性、方法、事件、实例、源码】
开发语言·c#·c# winform·c#保存文件·savefiledialog
啊阿狸不会拉杆1 小时前
《Java 程序设计》第 11 章 - 泛型与集合
java·开发语言·jvm·数据结构·算法
广东小62 小时前
【昇腾】基于Atlas 200I DK A2开发者套件修改usb0的默认IP重启后被恢复的问题处理_20250730
网络·网络协议·tcp/ip