Go语言使用mgo驱动高效存储文件至MongoDB GridFS：流式上传实践

func uploadfilePageHandler(w http.ResponseWriter, req *http.Request) {
  file, handler, err := req.FormFile("filename")
  if err != nil {
    // ... 错误处理
    return
  }
  defer file.Close() // 确保关闭文件句柄

  // 将整个文件内容读取到内存中
  data, err := ioutil.ReadAll(file)
  if err != nil {
    // ... 错误处理
    return
  }

  // ... 获取MongoDB会话和数据库
  my_db := mongo_session.DB("... database name...")

  // 创建GridFS文件
  my_file, err := my_db.GridFS("fs").Create(handler.Filename) // 使用原始文件名或生成唯一文件名
  if err != nil {
    // ... 错误处理
    return
  }
  defer my_file.Close() // 确保关闭GridFS文件

  // 将内存中的数据写入GridFS
  n, err := my_file.Write(data)
  if err != nil {
    // ... 错误处理
    return
  }

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

package main

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

    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/gridfs"
)

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

func init() {
    // 实际应用中需要替换为你的MongoDB连接字符串
    session, err := mgo.Dial("mongodb://localhost:27017")
    if err != nil {
        log.Fatalf("Failed to connect to MongoDB: %v", err)
    }
    session.SetMode(mgo.Monotonic, true)
    mongo_session = session
    log.Println("MongoDB session initialized.")
}

func uploadfilePageHandler(w http.ResponseWriter, req *http.Request) {
    // 1. 从HTTP请求中获取上传文件
    file, handler, err := req.FormFile("filename")
    if err != nil {
        http.Error(w, fmt.Sprintf("Failed to get file from form: %v", err), http.StatusBadRequest)
        return
    }
    defer file.Close() // 确保关闭上传文件的文件句柄

    // 2. 指定MongoDB数据库
    my_db := mongo_session.DB("your_database_name") // 替换为你的数据库名

    // 3. 创建GridFS文件对象
    // 可以使用上传文件的原始文件名，或者生成一个唯一的名称
    unique_filename := handler.Filename // 或者 uuid.New().String() + "_" + handler.Filename
    my_file, err := my_db.GridFS("fs").Create(unique_filename)
    if err != nil {
        http.Error(w, fmt.Sprintf("Failed to create GridFS file: %v", err), http.StatusInternalServerError)
        return
    }
    defer my_file.Close() // 确保关闭GridFS文件，这会触发文件的最终写入和元数据保存

    // 4. 使用io.Copy进行流式传输
    // 将上传文件（io.Reader）直接复制到GridFS文件（io.Writer）
    n, err := io.Copy(my_file, file)
    if err != nil {
        http.Error(w, fmt.Sprintf("Failed to write file to GridFS: %v", err), http.StatusInternalServerError)
        return
    }

    // 5. 写入成功日志或响应
    log.Printf("Successfully uploaded %d bytes to GridFS as %s\n", n, unique_filename)
    w.WriteHeader(http.StatusOK)
    w.Write([]byte(fmt.Sprintf("File '%s' uploaded successfully, %d bytes.", unique_filename, n)))

    // ... 其他业务逻辑，例如重定向或返回JSON响应
}

func main() {
    http.HandleFunc("/upload", uploadfilePageHandler)
    log.Println("Server started on :8080, upload endpoint: /upload")
    log.Fatal(http.ListenAndServe(":8080", nil))
}