Golang并发网络连接技巧与实战解析

listener, err := net.Listen("tcp", ":8080")
if err != nil {
    log.Fatal(err)
}
for {
    conn, err := listener.Accept()
    if err != nil {
        log.Print(err)
        continue
    }
    go handleConn(conn)
}

var sem = make(chan struct{}, 100) // 最多100个并发处理
func handleConn(conn net.Conn) {
sem <- struct{}{} // 获取信号量
defer func() { <-sem }() // 释放信号量
defer conn.Close()
// 处理逻辑
io.Copy(io.Discard, conn)
}

func handleConn(conn net.Conn) {
    ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
    defer cancel()
    defer conn.Close()
for {
    select {
    case <-ctx.Done():
        log.Println("connection timed out or canceled")
        return
    default:
        conn.SetReadDeadline(time.Now().Add(5 * time.Second))
        buf := make([]byte, 1024)
        n, err := conn.Read(buf)
        if err != nil {
            return
        }
        // 处理数据
        conn.Write(buf[:n])
    }
}
}

var bufferPool = sync.Pool{
    New: func() interface{} {
        return make([]byte, 1024)
    },
}
func handleConn(conn net.Conn) {
buf := bufferPool.Get().([]byte)
defer bufferPool.Put(buf)
// 使用buf进行读写
}