Go语言：使用mgo将文件高效流式存储至MongoDB GridFS

func uploadfilePageHandler(w http.ResponseWriter, req *http.Request) {
  // 1. 捕获 multipart 表单中的文件信息
  file, handler, err := req.FormFile("filename")
  if err != nil {
    // 错误处理
    http.Error(w, "无法获取上传文件", http.StatusInternalServerError)
    fmt.Println("获取文件失败:", err)
    return
  }
  defer file.Close() // 确保文件句柄被关闭

  // 2. 将整个文件内容读入内存 - 这是问题的根源
  data, err := ioutil.ReadAll(file)
  if err != nil {
    // 错误处理
    http.Error(w, "无法读取文件内容", http.StatusInternalServerError)
    fmt.Println("读取文件内容失败:", err)
    return
  }

  // 3. 指定 MongoDB 数据库和 GridFS 实例
  my_db := mongo_session.DB("... database name...")
  gridFS := my_db.GridFS("fs")

  // 4. 在 GridFS 中创建文件
  unique_filename := handler.Filename // 或生成一个唯一文件名
  my_file, err := gridFS.Create(unique_filename)
  if err != nil {
    // 错误处理
    http.Error(w, "无法在GridFS中创建文件", http.StatusInternalServerError)
    fmt.Println("创建GridFS文件失败:", err)
    return
  }
  defer my_file.Close() // 确保 GridFS 文件句柄被关闭

  // 5. 将内存中的数据写入 GridFS
  n, err := my_file.Write(data)
  if err != nil {
    // 错误处理
    http.Error(w, "无法将数据写入GridFS", http.StatusInternalServerError)
    fmt.Println("写入GridFS失败:", err)
    return
  }

  fmt.Printf("%d bytes written to the Mongodb instance\n", n)
  // ... 其他业务逻辑，如重定向等
}

import (
    "fmt"
    "io" // 导入 io 包
    "net/http"
    // "github.com/globalsign/mgo" // 如果使用旧版mgo
    // "github.com/mongodb/mongo-go-driver/mongo" // 如果使用新版官方驱动，GridFS API会有所不同
)

// 假设 mongo_session 已经是一个有效的 *mgo.Session
// var mongo_session *mgo.Session

func uploadfilePageHandler(w http.ResponseWriter, req *http.Request) {
  // 1. 捕获 multipart 表单中的文件信息
  file, handler, err := req.FormFile("filename")
  if err != nil {
    http.Error(w, "无法获取上传文件", http.StatusInternalServerError)
    fmt.Println("获取文件失败:", err)
    return
  }
  defer file.Close() // 确保上传文件句柄被关闭

  // 2. 指定 MongoDB 数据库和 GridFS 实例
  my_db := mongo_session.DB("... database name...") // 替换为你的数据库名
  gridFS := my_db.GridFS("fs") // 默认的 GridFS 集合前缀是 "fs"

  // 3. 在 GridFS 中创建文件，获取 io.WriteCloser 接口
  unique_filename := handler.Filename // 可以根据需要生成唯一文件名
  my_file, err := gridFS.Create(unique_filename)
  if err != nil {
    http.Error(w, "无法在GridFS中创建文件", http.StatusInternalServerError)
    fmt.Println("创建GridFS文件失败:", err)
    return
  }
  defer my_file.Close() // 确保 GridFS 文件句柄被关闭，这会触发最终的写入和元数据保存

  // 4. 使用 io.Copy 直接将文件内容从上传流写入 GridFS
  // file (req.FormFile 返回) 是 io.Reader
  // my_file (gridFS.Create 返回) 是 io.Writer
  n, err := io.Copy(my_file, file)
  if err != nil {
    http.Error(w, "无法将数据流式写入GridFS", http.StatusInternalServerError)
    fmt.Println("流式写入GridFS失败:", err)
    return
  }

  fmt.Printf("%d bytes written to the Mongodb instance using streaming\n", n)
  // ... 其他业务逻辑，如返回成功信息或重定向
}