golang：聊天服務器

《GO程序設計語言》設計中案例，僅作爲筆記進行收藏。簡單的聊天服務器，使用telnet 發送信息。

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 <- "You are " + who
	messages <- who + " has arrived"
	entering <- ch

	input := bufio.NewScanner(conn)
	for input.Scan() {
		// 注意，忽略 input.Err() 中可能的錯誤
		messages <- who + ": " + input.Text()
	}

	leaving <- ch
	messages <- who + " has left"
	conn.Close()
}

func clientWriter(conn net.Conn, ch <-chan string) {
	for msg := range ch {
		// 注意，忽略網絡層面的錯誤
		fmt.Fprintln(conn, msg)
	}
}