forked from FabianVowie/Lithium
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
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)
|
|
}
|