tcp协议写的一个广播消息

package main

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

var buffServer = make([]byte, 1024)
var buffClient = make([]byte, 1024)
var clients = make(map[string]net.Conn)
var messages = make(chan string, 10)

func TcpServer(port string) {
    port = ":" + port
    tcpAddr, err := net.ResolveTCPAddr("tcp4", port)
    if err != nil {
        log.Printf("error: %s\n", err)
        return
    }
    l, err := net.ListenTCP("tcp", tcpAddr)
    defer l.Close()
    if err != nil {
        log.Printf("error: %s\n", err)
        return
    }
    log.Printf("开启了服务端:%s\n", "正在监听客户端的请求")
    go BoradCast(clients, messages)
    for {
        conn, err := l.Accept()
        if err != nil {
            log.Printf("error: %s\n", err)
        }
        clients[conn.RemoteAddr().String()] = conn
        go HandleServer(conn, messages)
    }
}
func HandleServer(conn net.Conn, messages chan string) {
    for {
        _, err := conn.Read(buffServer)
        if err != nil {
            log.Printf("error: %s\n", err)
            conn.Close()
            return
        }
        messages <- string(buffServer)
    }
}
func BoradCast(clients map[string]net.Conn, messages chan string) {

    for {
        msg := <-messages
        for index, client := range clients {
            _, err := client.Write([]byte(msg))
            if err != nil {
                log.Printf("error: %s\n", err)
                delete(clients, index)
            }

        }
    }

}
func TcpClient(serverAddr string) {
    tcpAddr, err := net.ResolveTCPAddr("tcp4", serverAddr)
    if err != nil {
        log.Printf("error: %s\n", err)
        return
    }
    conn, err := net.DialTCP("tcp", nil, tcpAddr)
    defer conn.Close()
    if err != nil {
        log.Printf("error: %s\n", err)
        return
    }
    log.Printf("客户端:%s开始连接\n", conn.LocalAddr().String())
    go ClientSend(conn)
    for {
        _, err := conn.Read(buffClient)
        if err != nil {
            log.Printf("error: %s\n", err)
            return
        }
        log.Printf("收到消息:%s\n", string(buffClient))
    }
}
func ClientSend(conn net.Conn) {

    var input string
    username := conn.LocalAddr().String()
    for {

        fmt.Scanln(&input)
        if input == "/quit" {
            log.Printf("Info: %s\n", "ByeBye")
            conn.Close()
            os.Exit(0)
        }
        _, err := conn.Write([]byte(username + ":" + input))
        if err != nil {
            log.Printf("error: %s\n", err)
            return
        }

    }

}
//开启TcpServer: go run main.go server 端口号(8080)
//开启TcpClient: go run main.go client 本地ip地址:端口号(8080)
func main() {
    if len(os.Args) != 3 {
        log.Println("command error")
        os.Exit(0)
    }
    if os.Args[1] == "server" {
        TcpServer(os.Args[2])
    }
    if os.Args[1] == "client" {
        TcpClient(os.Args[2])
    }
}