Go语言网络编程：构建自定义二进制协议通信

package main

import (
    "encoding/binary"
    "fmt"
    "net"
)

// packet 定义了我们自定义的二进制协议包结构
type packet struct {
    Type int32     // 包类型
    Id   int32     // 包ID
    Data [100]byte // 载荷数据，固定100字节
}

// ... (packet struct definition) ...

func main() {
    // 在端口2000上设置TCP监听器
    l, err := net.Listen("tcp", ":2000")
    if err != nil {
        // 生产环境中应使用更优雅的错误处理，如log.Fatal
        panic(err.String())
    }
    defer l.Close() // 确保监听器在程序退出时关闭

    fmt.Println("Server listening on :2000")

    for {
        // 接受客户端连接
        conn, err := l.Accept()
        if err != nil {
            panic(err.String())
        }
        // 为每个连接启动一个goroutine处理，实现并发
        go handleClient(conn)
    }
}

// handleClient 处理单个客户端连接的逻辑
func handleClient(conn net.Conn) {
    defer conn.Close() // 确保连接在函数退出时关闭

    fmt.Printf("Client connected from %s\n", conn.RemoteAddr())

    // 等待接收客户端发送的包
    var msg packet
    // 使用binary.Read从连接中读取数据到msg结构体，采用大端字节序
    err := binary.Read(conn, binary.BigEndian, &msg)
    if err != nil {
        fmt.Printf("Error reading packet: %v\n", err)
        return
    }
    // 打印接收到的数据，注意Data是固定大小数组，需要截断或转换为字符串
    fmt.Printf("Received from %s: %s\n", conn.RemoteAddr(), string(msg.Data[:]))

    // 准备响应包
    response := packet{Type: 1, Id: 1}
    // 将字符串复制到Data字段中，注意Data是字节数组
    copy(response.Data[:], "Hello, client from server!")
    // 使用binary.Write将响应包写入连接，采用大端字节序
    err = binary.Write(conn, binary.BigEndian, &response)
    if err != nil {
        fmt.Printf("Error writing response: %v\n", err)
        return
    }
    fmt.Printf("Sent response to %s\n", conn.RemoteAddr())
}

// ... (packet struct definition) ...

func main() {
    // 连接到本地服务器的2000端口
    conn, err := net.Dial("tcp", "localhost:2000")
    if err != nil {
        panic(err.String())
    }
    defer conn.Close() // 确保连接在程序退出时关闭

    fmt.Println("Connected to server.")

    // 准备要发送的包
    msg := packet{Type: 0, Id: 0}
    copy(msg.Data[:], "Hello, server from client!")
    // 使用binary.Write将包写入连接，采用大端字节序
    err = binary.Write(conn, binary.BigEndian, &msg)
    if err != nil {
        panic(err.String())
    }
    fmt.Println("Sent packet to server.")

    // 接收服务器的响应
    var response packet
    // 使用binary.Read从连接中读取数据到response结构体，采用大端字节序
    err = binary.Read(conn, binary.BigEndian, &response)
    if err != nil {
        panic(err.String())
    }
    fmt.Printf("Received response from server: %s\n", string(response.Data[:]))
}