Micro-service for file storage and processing written in Go
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

39 lines
1.1 KiB

package storage
import (
"os"
"path/filepath"
"github.com/spf13/afero"
)
type IStorageProvider interface {
storeRaw(bucketName string, objectName string, data []byte)
storeExisting(bucketName string, objectName string, existingFilePath string)
}
type FileSystemStorageProvider struct {
fileSystem afero.Fs
basePath string
}
func (sp FileSystemStorageProvider) storeRaw(bucketName string, objectName string, data []byte) {
directoryPath := filepath.Join(sp.basePath, bucketName)
sp.fileSystem.MkdirAll(directoryPath, os.ModePerm)
filePath := filepath.Join(directoryPath, objectName)
afero.WriteFile(sp.fileSystem, filePath, data, os.ModePerm)
}
func (sp FileSystemStorageProvider) storeExisting(bucketName string, objectName string, existingFilePath string) {
directoryPath := filepath.Join(sp.basePath, bucketName)
sp.fileSystem.MkdirAll(directoryPath, os.ModePerm)
filePath := filepath.Join(directoryPath, objectName)
bytesRead, _ := afero.ReadFile(sp.fileSystem, existingFilePath)
afero.WriteFile(sp.fileSystem, filePath, bytesRead, os.ModePerm)
}