How to store multiple files using GridFs in golang

Vishal answered on June 19, 2023 Popularity 1/10 Helpfulness 1/10

answer How to store multiple files using GridFs in golang

How to store multiple files using GridFs in golang

Comment

package main

import (
	"context"
	"fmt"
	"io"
	"log"
	"mime/multipart"
	"net/http"
	"os"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"go.mongodb.org/mongo-driver/mongo/gridfs"
)

// MongoDB connection details
const (
	uri        = "mongodb://localhost:27017"
	dbName     = "fileStorageDB"
	collection = "fs"
)

// UploadFiles uploads multiple files to GridFS
func UploadFiles(w http.ResponseWriter, r *http.Request) {
	// Parse the multipart form data
	err := r.ParseMultipartForm(32 << 20) // 32 MB maximum file size
	if err != nil {
		http.Error(w, "Failed to parse form data", http.StatusBadRequest)
		return
	}

	// Connect to MongoDB
	client, err := mongo.Connect(context.Background(), options.Client().ApplyURI(uri))
	if err != nil {
		log.Fatal(err)
	}
	db := client.Database(dbName)

	// Create a new GridFSBucket
	bucket, err := gridfs.NewBucket(
		db,
	)
	if err != nil {
		log.Fatal(err)
	}

	// Iterate over each file in the form data
	for _, fileHeaders := range r.MultipartForm.File {
		for _, fileHeader := range fileHeaders {
			// Open the uploaded file
			file, err := fileHeader.Open()
			if err != nil {
				log.Println("Failed to open file:", err)
				continue
			}
			defer file.Close()

			// Create a new GridFS upload stream
				uploadStream, err := bucket.OpenUploadStream(fileHeader.Filename)
				if err != nil {
					log.Println("Failed to create upload stream:", err)
					continue
				}
				defer uploadStream.Close()

				// Copy the file data to the GridFS upload stream
				_, err = io.Copy(uploadStream, file)
				if err != nil {
					log.Println("Failed to upload file:", err)
					continue
				}

				log.Println("File uploaded:", fileHeader.Filename)
			}
		}
	}

	fmt.Fprintf(w, "Files uploaded successfully!")
}

// DownloadFile downloads a file from GridFS
func DownloadFile(w http.ResponseWriter, r *http.Request) {
	// Get the filename from the request path
	filename := r.URL.Query().Get("filename")

	// Connect to MongoDB
	client, err := mongo.Connect(context.Background(), options.Client().ApplyURI(uri))
	if err != nil {
		log.Fatal(err)
	}
	db := client.Database(dbName)

	// Create a new GridFSBucket
	bucket, err := gridfs.NewBucket(
		db,
	)
	if err != nil {
		log.Fatal(err)
	}

	// Open a GridFS download stream for the file
	downloadStream, err := bucket.OpenDownloadStreamByName(filename)
	if err != nil {
		log.Fatal(err)
	}
	defer downloadStream.Close()

	// Set the appropriate content type for the response
	w.Header().Set("Content-Type", "application/octet-stream")

	// Copy the file data to the response writer
	_, err = io.Copy(w, downloadStream)
	if err != nil {
		log.Fatal(err)
	}
}

func main() {
	// Set up HTTP routes
	http.HandleFunc("/upload", UploadFiles)
	http.HandleFunc("/download", DownloadFile)

	// Start the server
	log.Println("Server started on port 8080")
	log.Fatal(http.ListenAndServe(":8080", nil))
}

Popularity 1/10 Helpfulness 1/10 Language go

Source: Grepper

Tags: go store

Link to this answer
Share Copy Link

Contributed on Jun 19 2023

Vishal

0 Answers Avg Quality 2/10

How to store multiple files using GridFs in golang

Contents

More Related Answers

How to store multiple files using GridFs in golang

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.