mirror of
https://github.com/gomods/athens
synced 2026-02-03 11:00:32 +00:00
e5ce3f7c6e
* Set a default value for ATHENS_MONGO_CONNECTION_STRING env var - ATHENS_MONGO_CONNECTION_STRING will default to mongodb://127.0.0.1:27017 if not set * Fix Get function call * Update user of env.MongoConnectionString
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package actions
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/gomods/athens/pkg/config/env"
|
|
"github.com/gomods/athens/pkg/storage"
|
|
"github.com/gomods/athens/pkg/storage/fs"
|
|
"github.com/gomods/athens/pkg/storage/gcp"
|
|
"github.com/gomods/athens/pkg/storage/mem"
|
|
"github.com/gomods/athens/pkg/storage/minio"
|
|
"github.com/gomods/athens/pkg/storage/mongo"
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
// GetStorage returns storage backend based on env configuration
|
|
func GetStorage() (storage.Backend, error) {
|
|
storageType := env.StorageTypeWithDefault("memory")
|
|
var storageRoot string
|
|
var err error
|
|
|
|
switch storageType {
|
|
case "memory":
|
|
return mem.NewStorage()
|
|
case "mongo":
|
|
connectionString := env.MongoConnectionString()
|
|
certPath := env.MongoCertPath()
|
|
return mongo.NewStorageWithCert(connectionString, certPath)
|
|
case "disk":
|
|
storageRoot, err = env.DiskRoot()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
s, err := fs.NewStorage(storageRoot, afero.NewOsFs())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not create new storage from os fs (%s)", err)
|
|
}
|
|
return s, nil
|
|
case "minio":
|
|
endpoint, err := env.MinioEndpoint()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
accessKeyID, err := env.MinioAccessKeyID()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
secretAccessKey, err := env.MinioSecretAccessKey()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bucketName := env.MinioBucketNameWithDefault("gomods")
|
|
useSSL := true
|
|
if useSSLVar := env.MinioSSLWithDefault("yes"); strings.ToLower(useSSLVar) == "no" {
|
|
useSSL = false
|
|
}
|
|
return minio.NewStorage(endpoint, accessKeyID, secretAccessKey, bucketName, useSSL)
|
|
case "gcp":
|
|
return gcp.New(context.Background())
|
|
default:
|
|
return nil, fmt.Errorf("storage type %s is unknown", storageType)
|
|
}
|
|
}
|