Go를 사용하여 대규모 동시 요청을 효율적으로 처리

package main

import (
    "context"
    "log"
    "net/http"
    "sync"
)

const maxConcurrency = 10

var wg sync.WaitGroup

func main() {
    http.HandleFunc("/upload", uploadHandler)
    http.ListenAndServe(":8080", nil)
}

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    wg.Add(1)
    defer wg.Done()

    // 创建一个 goroutine 来处理文件上传
    go func() {
        defer r.Body.Close()
        if err := handleFileUpload(r.Body); err != nil {
            log.Println(err)
        }
    }()
    
    // 限制并发上传的数量
    ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
    defer cancel()
    select {
    case <-ctx.Done():
        http.Error(w, "Too many concurrent uploads", http.StatusTooManyRequests)
        return
    default:
    }
}

func handleFileUpload(r io.Reader) error {
    // 省略实际的文件上传处理
    return nil
}