mirror of
https://github.com/gomods/athens
synced 2026-02-03 07:30:32 +00:00
900cb4f7c1
Upgrades the AWS SDK to v2. AWS S3 bucket urls will now error if they are not prefixed with a schema (example: https://).
77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
package actions
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/gomods/athens/pkg/config"
|
|
"github.com/gomods/athens/pkg/errors"
|
|
"github.com/gomods/athens/pkg/storage"
|
|
"github.com/gomods/athens/pkg/storage/azureblob"
|
|
"github.com/gomods/athens/pkg/storage/external"
|
|
"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/s3"
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
// GetStorage returns storage backend based on env configuration.
|
|
func GetStorage(storageType string, storageConfig *config.Storage, timeout time.Duration, client *http.Client) (storage.Backend, error) {
|
|
const op errors.Op = "actions.GetStorage"
|
|
switch storageType {
|
|
case "memory":
|
|
return mem.NewStorage()
|
|
case "mongo":
|
|
if storageConfig.Mongo == nil {
|
|
return nil, errors.E(op, "Invalid Mongo Storage Configuration")
|
|
}
|
|
return mongo.NewStorage(storageConfig.Mongo, timeout)
|
|
case "disk":
|
|
if storageConfig.Disk == nil {
|
|
return nil, errors.E(op, "Invalid Disk Storage Configuration")
|
|
}
|
|
rootLocation := storageConfig.Disk.RootPath
|
|
s, err := fs.NewStorage(rootLocation, afero.NewOsFs())
|
|
if err != nil {
|
|
errStr := fmt.Sprintf("could not create new storage from os fs (%s)", err)
|
|
return nil, errors.E(op, errStr)
|
|
}
|
|
return s, nil
|
|
case "minio":
|
|
if storageConfig.Minio == nil {
|
|
return nil, errors.E(op, "Invalid Minio Storage Configuration")
|
|
}
|
|
return minio.NewStorage(storageConfig.Minio, timeout)
|
|
case "gcp":
|
|
if storageConfig.GCP == nil {
|
|
return nil, errors.E(op, "Invalid GCP Storage Configuration")
|
|
}
|
|
return gcp.New(context.Background(), storageConfig.GCP, timeout)
|
|
case "s3":
|
|
if storageConfig.S3 == nil {
|
|
return nil, errors.E(op, "Invalid S3 Storage Configuration")
|
|
}
|
|
return s3.New(storageConfig.S3, timeout, func(config *aws.Config) {
|
|
config.HTTPClient = client
|
|
})
|
|
case "azureblob":
|
|
if storageConfig.AzureBlob == nil {
|
|
return nil, errors.E(op, "Invalid AzureBlob Storage Configuration")
|
|
}
|
|
return azureblob.New(storageConfig.AzureBlob, timeout)
|
|
case "external":
|
|
if storageConfig.External == nil {
|
|
return nil, errors.E(op, "Invalid External Storage Configuration")
|
|
}
|
|
return external.NewClient(storageConfig.External.URL, client), nil
|
|
default:
|
|
return nil, fmt.Errorf("storage type %s is unknown", storageType)
|
|
}
|
|
}
|