chore(memory): ability to provide storage

picosh · Oct 16, 2024 · c39198b · c39198b
1 parent 01fa9e4
commit c39198b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/storage/memory.go b/storage/memory.go
@@ -18,9 +18,9 @@ type StorageMemory struct {
 var _ ObjectStorage = &StorageMemory{}
 var _ ObjectStorage = (*StorageMemory)(nil)
 
-func NewStorageMemory() (*StorageMemory, error) {
+func NewStorageMemory(st map[string]map[string]string) (*StorageMemory, error) {
 	return &StorageMemory{
-		storage: map[string]map[string]string{},
+		storage: st,
 	}, nil
 }
 

diff --git a/util.go b/util.go
@@ -19,7 +19,7 @@ func EnvDriverDetector(logger *slog.Logger) (storage.ObjectStorage, error) {
 	logger.Info("driver detected", "driver", driver)
 
 	if driver == "memory" {
-		return storage.NewStorageMemory()
+		return storage.NewStorageMemory(map[string]map[string]string{})
 	} else if driver == "minio" {
 		url := GetEnv("MINIO_URL", "")
 		user := GetEnv("MINIO_ROOT_USER", "")