mirror of
https://github.com/gomods/athens
synced 2026-02-03 11:00:32 +00:00
afd01813e9
* Default to in-memory storage for the proxy This makes it easier for someone to try out or just run the proxy by executing the binary without further environment setup. * Remove obsolete comment
76 lines
2.0 KiB
Go
76 lines
2.0 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/gomods/athens/pkg/storage/rdbms"
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
// GetStorage returns storage backend based on env configuration
|
|
func GetStorage() (storage.BackendConnector, error) {
|
|
storageType := env.StorageTypeWithDefault("memory")
|
|
var storageRoot string
|
|
var err error
|
|
|
|
switch storageType {
|
|
case "memory":
|
|
return mem.NewStorage()
|
|
case "mongo":
|
|
storageRoot, err = env.MongoURI()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return mongo.NewStorage(storageRoot), nil
|
|
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 storage.NoOpBackendConnector(s), nil
|
|
case "postgres", "sqlite", "cockroach", "mysql":
|
|
storageRoot, err = env.RdbmsName()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return rdbms.NewRDBMSStorage(storageRoot), 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
|
|
}
|
|
s, err := minio.NewStorage(endpoint, accessKeyID, secretAccessKey, bucketName, useSSL)
|
|
return storage.NoOpBackendConnector(s), err
|
|
case "gcp":
|
|
return gcp.New(context.Background())
|
|
default:
|
|
return nil, fmt.Errorf("storage type %s is unknown", storageType)
|
|
}
|
|
}
|