Go语言实现并发聊天室实战：通道与多 goroutine 应用

package main

import (
	"bufio"
	"fmt"
	"log"
	"net"
)

// 定义单向通道类型，仅允许向外部发送字符串
type client chan <- string 

var (
	entering = make(chan client) // 用户登录信号通道
	leaving = make(chan client)  // 用户退出信号通道
	messages = make(chan string) // 公共消息广播通道
)

func main() {
	listener, err := net.Listen("tcp", "localhost:8000")
	if err != nil {
		log.Fatal("network is broken", err)
	}
	// 启动广播器 goroutine
	go broadcaster()
	for {
		conn, err := listener.Accept()
		if err != nil {
			log.Print(err)
			continue
		}
		// 每个新连接启动单独的处理器
		go handleConn(conn)
	}
}

// broadcaster 负责将所有消息广播给在线用户
func broadcaster() {
	clients := make(map[client]bool) // 存储活跃客户端通道及其在线状态
	for {
		select {
		case msg := <-messages:
			// 遍历并发送消息给所有已注册的客户端
			for cli := range clients {
				cli <- msg
			}
		case cli := <-entering:
			// 注册用户
			clients[cli] = true
		case cli := <-leaving:
			// 注销用户并关闭通道
			delete(clients, cli)
			close(cli)
		}
	}
}

// handleConn 处理单个客户端的连接交互
func handleConn(conn net.Conn) {
	ch := make(chan string) // 创建该用户的专用输出通道
	go clientWriter(conn, ch) // 启动写辅助 goroutine
	
	who := conn.RemoteAddr().String()
	ch <- "You are " + who     // 通知用户自己的身份
	entering <- ch             // 注册进入广播系统
	messages <- who + " has arrived" // 广播欢迎消息

	input := bufio.NewScanner(conn) // 扫描输入流
	for input.Scan() {
		// 将用户输入的内容广播出去
		messages <- who + ":" + input.Text()
	}
	
	// 处理断连或输入结束的情况
	leaving <- ch
	messages <- who + " has left"
	conn.Close()
}

// clientWriter 将从 ch 收到的消息写入网络连接
func clientWriter(conn net.Conn, ch <- chan string) {
	for msg := range ch {
		fmt.Fprintln(conn, msg)
	}
}

package main

import (
	"io"
	"log"
	"net"
	"os"
)

func main() {
	conn, err := net.Dial("tcp", "localhost:8000")
	if err != nil {
		log.Fatal("Connection refused!", err)
	}
	defer conn.Close()

	// 启动后台 goroutine，将服务器发来的内容复制到标准输出 (终端)
	go mesCopy(os.Stdout, conn)
	
	// 在前台/主流程中，将标准输入复制到服务器连接
	mesCopy(conn, os.Stdin)
}

// mesCopy 将 res 中的数据拷贝到 des
func mesCopy(des io.Writer, res io.Reader) {
	if _, err := io.Copy(des, res); err != nil {
		log.Print("Transfer error!")
	}
}