dqj29136 2017-10-17 07:36
浏览 178
已采纳

使用GoLang mgo将PDF保存在MongoDb中

I read multiple blogs about saving the file using mgo, but unable to find a solution for below specific need, help shoutout!

Below inserts object in MongoDb:

var dbSchoolPojo dbSchool
i := bson.NewObjectId()
dbSchoolPojo.ID = i
coll := db.C("school")
coll.Insert(&dbSchoolPojo)

Below able to get hold of file:

file, handler, err := r.FormFile("pdf") //Able to get file from r *http.Request

Now, before inserting object, I need to set above file like:

dbSchoolPojo.pdf = file.Bytes(); //Of course Bytes() is invalid method

My structs object:

type dbSchool struct {
    ID  bson.ObjectId  `json:"id" bson:"_id,omitempty"`
    ...
    Pdf  []byte  `json:"pdf" bson:"pdf"`
    ...
}

In layman's term, the question is: How do I insert file (received from HTML form) in mongoDb via GoLang struct using mgo driver?

Thanks for reading! :)


Update:

PDF is stored in MongoDB like below:

Binary('EWHKDSH876KJHlkjdswsddsfsd3232432njnjkn2dljDSFSDFIJSFD333...')

Below code works without error, but doesn't serve PDF file:

func DownloadPdf(w http.ResponseWriter, r *http.Request, db mongoDB) {
    var school dbSchool
    coll := db.C("schools")

    incomingId = "59e6404e2f68182a74610f19"; //This mongo DB _id is received from GET URL request

    err := coll.Find(bson.M{"_id": bson.ObjectIdHex(incomingId)}).
        Select(bson.M{"pdf": 1}).One(&school)
    if err != nil {
        serve404(w, r, db)
        return
    }

    buffer := school.Pdf
    w.Header().Set("Content-Disposition", "attachment; filename=abc.pdf")
    w.Header().Set("Content-Type", "application/pdf")
    w.Header().Set("Content-Length", strconv.Itoa(len(buffer)))

    if _, err := w.Write(buffer); err != nil {
        log.Println("unable to serve image.") //This line is not executed
    }
}

JQuery code to request content:

      $(".downloadPdfFile").click(function() {
            var domain = document.location.origin;
            window.open(domain+'/brochure/59e6404e2f68182a74610f19', '_blank');
         });
  • 写回答

1条回答 默认 最新

  • douling8087 2017-10-17 08:10
    关注

    The file returned by Request.FormFile() is of type multipart.File which is:

    type File interface {
        io.Reader
        io.ReaderAt
        io.Seeker
        io.Closer
    }
    

    It implements io.Reader, so you can simply read its content e.g. with ioutil.ReadAll():

    data, err := io.ReadAll(file)
    // Handle error
    

    Then:

    dbSchoolPojo.Pdfdata = data
    

    But storing big files as part of documents is not optimal / efficient. Instead take a look at the MongoDB GridFS that is also supported by mgo: Database.GridFS().

    Here's how you can store a file in MongoDB GridFS (example taken from GridFS.Create()):

    func check(err error) {
        if err != nil {
            panic(err.String())
        }
    }
    file, err := db.GridFS("fs").Create("myfile.txt")
    check(err)
    n, err := file.Write([]byte("Hello world!"))
    check(err)
    err = file.Close()
    check(err)
    fmt.Printf("%d bytes written
    ", n)
    

    Using GridFS, you can also save the file without having to read all its content into memory first, if you "stream" the file content into the mgo.GridFile, as it implements io.Writer. Call io.Copy():

    // ...
    file, handler, err := r.FormFile("pdfFile")
    
    // ...
    mongoFile, err := db.GridFS("fs").Create("somepdf.pdf")
    check(err)
    
    // Now stream from HTTP request into the mongo file:
    written, err := io.Copy(mongoFile, file)
    // Handle err
    
    err = mongoFile.Close()
    check(err)
    

    Edit: update answering your update

    When you serve the PDF, you query the document in a wrong way:

    err := coll.Find(bson.M{"_id": bson.ObjectIdHex(incomingId)}).
        Select(bson.M{"pdf": 1}).One(&school)
    

    You select the pdf field to retrieve, but your documents in MongoDB have no such field:

    type dbSchool struct {
        ID  bson.ObjectId  `json:"id" bson:"_id,omitempty"`
        ...
        Pdfdata  []byte  `json:"pdf"`
        ...
    }
    

    This type definition will result having a pdfdata field, but not pdf. Add the proper mgo struct tag:

    type dbSchool struct {
        ID  bson.ObjectId  `json:"id" bson:"_id,omitempty"`
        ...
        Pdfdata  []byte  `json:"pdf" bson:"pdf"`
        ...
    }
    

    Or change the selected field from pdf to pdfdata.

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 数学建模——参会安排怎么做
  • ¥15 电脑键盘实现触摸功能
  • ¥25 matlab无法将表达式转换为双数组怎么解决?
  • ¥15 单片机汇编语言相关程序
  • ¥20 家用射频美容仪技术规格
  • ¥15 大家帮我看看为什么错了
  • ¥15 unity互动琴弦抖动效果
  • ¥15 做了个的二极管反向饱和电流测量电路,但是测试达不到效果
  • ¥15 nginx无证书访问https失败
  • ¥15 树莓派启动AP热点传入数据