20 lines
383 B
Go
20 lines
383 B
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func NewStorage(cfg *viper.Viper) (Storage, error) {
|
|
storage_type := cfg.GetString("storage.db.type")
|
|
|
|
switch storage_type {
|
|
case "mongodb":
|
|
return NewMongoDBStorageFromConfig(cfg)
|
|
case "mock":
|
|
return NewMockStorage(), nil
|
|
default:
|
|
return nil, fmt.Errorf("storage type %v is not supported", storage_type)
|
|
}
|
|
} |