Go语言中通过HTTP接收二进制数据实践指南

package main

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

// handleUploadInMem 处理将二进制数据读入内存的请求
func handleUploadInMem(w http.ResponseWriter, req *http.Request) {
    // 确保请求体在使用完毕后关闭
    defer req.Body.Close()

    // 限制请求体大小，防止恶意上传导致内存溢出
    // 例如，限制为10MB
    req.Body = http.MaxBytesReader(w, req.Body, 10*1024*1024) 

    // 读取请求体中的所有数据到字节切片
    buf, err := ioutil.ReadAll(req.Body)
    if err != nil {
        http.Error(w, fmt.Sprintf("无法读取请求体: %v", err), http.StatusInternalServerError)
        log.Printf("读取请求体错误: %v", err)
        return
    }

    // 此时，buf 包含了完整的二进制数据。
    // 你可以在这里对 buf 进行处理，例如保存到数据库、进行解压等。
    fmt.Printf("成功接收到 %d 字节的二进制数据 (内存处理方式)\n", len(buf))
    // 示例：打印数据的前100字节（如果数据足够长）
    if len(buf) > 0 {
        fmt.Printf("数据预览: %s...\n", string(buf[:min(len(buf), 100)]))
    }

    w.WriteHeader(http.StatusOK)
    w.Write([]byte("二进制数据已成功接收并读入内存！"))
}

func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}

func main() {
    http.HandleFunc("/upload/inmem", handleUploadInMem)
    log.Println("服务器正在监听 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

package main

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

// handleUploadToFile 处理将二进制数据流式写入文件的请求
func handleUploadToFile(w http.ResponseWriter, req *http.Request) {
    // 确保请求体在使用完毕后关闭
    defer req.Body.Close()

    // 限制请求体大小，防止恶意上传导致磁盘空间耗尽
    // 例如，限制为100MB
    req.Body = http.MaxBytesReader(w, req.Body, 100*1024*1024) 

    // 创建一个临时文件来存储上传的数据
    // 第一个参数是目录（空字符串表示系统默认临时目录），第二个参数是文件名前缀
    tempFile, err := ioutil.TempFile("", "uploaded_binary_")
    if err != nil {
        http.Error(w, fmt.Sprintf("无法创建临时文件: %v", err), http.StatusInternalServerError)
        log.Printf("创建临时文件错误: %v", err)
        return
    }
    // 确保临时文件在使用完毕后关闭
    defer tempFile.Close()
    // 通常，在处理完成后，你还会希望删除这个临时文件
    defer os.Remove(tempFile.Name()) // 生产环境中，可能需要将文件移动到最终位置

    // 将请求体的数据流式复制到临时文件
    bytesWritten, err := io.Copy(tempFile, req.Body)
    if err != nil {
        http.Error(w, fmt.Sprintf("无法写入文件: %v", err), http.StatusInternalServerError)
        log.Printf("写入文件错误: %v", err)
        return
    }

    // 此时，二进制数据已成功写入 tempFile.Name() 指定的临时文件
    fmt.Printf("成功接收到 %d 字节的二进制数据并写入文件: %s\n", bytesWritten, tempFile.Name())

    w.WriteHeader(http.StatusOK)
    w.Write([]byte(fmt.Sprintf("二进制数据已成功接收并写入文件：%s", tempFile.Name())))
}

func main() {
    http.HandleFunc("/upload/inmem", handleUploadInMem) // 保持内存处理示例
    http.HandleFunc("/upload/tofile", handleUploadToFile)
    log.Println("服务器正在监听 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}