package main import ( "bufio" "fmt" "log" "net" ) func main() { listener, err := net.Listen("tcp", "localhost:8000") if err != nil { log.Fatal(err) } go broadcaster() for { conn, err := listener.Accept() if err != nil { log.Print(err) continue } go handleConn(conn) } } type client chan<- string // 对外发送消息的通道 var ( entering = make(chan client) leaving = make(chan client) messages = make(chan string) // 所有连接的客户端 ) 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) } } } func handleConn(conn net.Conn) { ch := make(chan string) // 对外发送客户消息的通道 go clientWriter(conn, ch) who := conn.RemoteAddr().String() ch <- "欢迎 " + who messages <- who + " 上线" entering <- ch input := bufio.NewScanner(conn) for input.Scan() { messages <- who + ": " + input.Text() } // 注意:忽略 input.Err() 中可能的错误 leaving <- ch messages <- who + " 下线" conn.Close() } func clientWriter(conn net.Conn, ch <-chan string) { for msg := range ch { fmt.Fprintln(conn, msg) // 注意:忽略网络层面的错误 } }代码中 main 函数里面写的代码非常简单,其实服务器要做的事情总结一下无非就是获得 listener 对象,然后不停的获取链接上来的 conn 对象,最后把这些对象丢给处理链接函数去进行处理。
type client chan<- string // 对外发送消息的通道 var ( entering = make(chan client) leaving = make(chan client) messages = make(chan string) // 所有连接的客户端 ) 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) } } }在 main 函数里面使用 goroutine 开启了一个 broadcaster 函数来负责广播所有用户发送的消息。
func handleConn(conn net.Conn) { ch := make(chan string) // 对外发送客户消息的通道 go clientWriter(conn, ch) who := conn.RemoteAddr().String() ch <- "欢迎 " + who messages <- who + " 上线" entering <- ch input := bufio.NewScanner(conn) for input.Scan() { messages <- who + ": " + input.Text() } // 注意:忽略 input.Err() 中可能的错误 leaving <- ch messages <- who + " 下线" conn.Close() } func clientWriter(conn net.Conn, ch <-chan string) { for msg := range ch { fmt.Fprintln(conn, msg) // 注意:忽略网络层面的错误 } }handleConn 函数会为每个过来处理的 conn 都创建一个新的 channel,开启一个新的 goroutine 去把发送给这个 channel 的消息写进 conn。
// netcat 是一个简单的TCP服务器读/写客户端 package main import ( "io" "log" "net" "os" ) func main() { conn, err := net.Dial("tcp", "localhost:8000") if err != nil { log.Fatal(err) } done := make(chan struct{}) go func() { io.Copy(os.Stdout, conn) // 注意:忽略错误 log.Println("done") done <- struct{}{} // 向主Goroutine发出信号 }() mustCopy(conn, os.Stdin) conn.Close() <-done // 等待后台goroutine完成 } func mustCopy(dst io.Writer, src io.Reader) { if _, err := io.Copy(dst, src); err != nil { log.Fatal(err) } }当有 n 个客户 session 在连接的时候,程序并发运行着
2n+2
个相互通信的 goroutine,它不需要隐式的加锁操作。clients map 限制在广播器这一个 goroutine 中被访问,所以不会并发访问它。唯一被多个 goroutine 共享的变量是通道以及 net.Conn 的实例,它们又都是并发安全的。go build
命令编译服务端和客户端,并运行生成的可执行文件。
本文链接:http://task.lmcjl.com/news/14396.html