Go语言中高效接收HTTP二进制数据教程

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
)

// handleBinaryUploadToMemory 处理二进制数据上传，并将其读取到内存
func handleBinaryUploadToMemory(w http.ResponseWriter, req *http.Request) {
    // 确保请求方法是POST
    if req.Method != http.MethodPost {
        http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
        return
    }

    // 从请求体中读取所有数据到字节切片
    buf, err := ioutil.ReadAll(req.Body)
    if err != nil {
        log.Printf("Error reading request body: %v", err)
        http.Error(w, "Failed to read request body", http.StatusInternalServerError)
        return
    }

    // 在这里可以对二进制数据 buf 进行任何操作
    // 例如：打印大小、保存到数据库、进行解压等
    fmt.Printf("Received %d bytes of binary data in memory.\n", len(buf))
    // 假设我们只是简单地将数据大小返回给客户端
    fmt.Fprintf(w, "Successfully received %d bytes of data.", len(buf))

    // 注意：req.Body 会在处理函数返回后由 net/http 自动关闭，
    // 但如果需要提前关闭或确保资源释放，可以显式调用 defer req.Body.Close()
    // 通常在ReadAll之后，Body已经被完全读取，关闭与否影响不大。
}

func main() {
    http.HandleFunc("/upload/memory", handleBinaryUploadToMemory)
    log.Println("Server started on :8080. Listening for /upload/memory...")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

package main

import (
    "fmt"
    "io"
    "io/ioutil"
    "log"
    "net/http"
    "os"
)

// handleBinaryUploadToFile 处理二进制数据上传，并将其流式写入到文件
func handleBinaryUploadToFile(w http.ResponseWriter, req *http.Request) {
    // 确保请求方法是POST
    if req.Method != http.MethodPost {
        http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
        return
    }

    // 确保请求体在函数返回时被关闭，释放资源
    defer req.Body.Close()

    // 创建一个临时文件来存储上传的数据
    // 第一个参数是目录（空字符串表示系统默认临时目录），第二个参数是文件名前缀
    tempFile, err := ioutil.TempFile("", "uploaded_binary_")
    if err != nil {
        log.Printf("Error creating temporary file: %v", err)
        http.Error(w, "Failed to create temporary file", http.StatusInternalServerError)
        return
    }
    // 确保临时文件在函数返回时被关闭
    defer tempFile.Close()
    // 确保临时文件在函数返回时被删除，避免磁盘空间占用
    defer os.Remove(tempFile.Name()) // 生产环境中可能需要根据业务逻辑决定是否删除

    // 使用 io.Copy 将请求体的数据直接复制到临时文件
    bytesWritten, err := io.Copy(tempFile, req.Body)
    if err != nil {
        log.Printf("Error writing to temporary file: %v", err)
        http.Error(w, "Failed to write data to file", http.StatusInternalServerError)
        return
    }

    fmt.Printf("Received %d bytes of binary data, saved to temporary file: %s\n", bytesWritten, tempFile.Name())
    fmt.Fprintf(w, "Successfully received %d bytes of data, saved to %s.", bytesWritten, tempFile.Name())

    // 在这里可以对 tempFile.Name() 指示的文件进行后续处理，例如移动到指定目录、解压、分析等。
}

func main() {
    http.HandleFunc("/upload/file", handleBinaryUploadToFile)
    log.Println("Server started on :8080. Listening for /upload/file...")
    log.Fatal(http.ListenAndServe(":8080", nil))
}