mirror of
https://github.com/gomods/athens
synced 2026-02-03 11:00:32 +00:00
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package minio
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/gomods/athens/pkg/config"
|
|
"github.com/gomods/athens/pkg/errors"
|
|
"github.com/gomods/athens/pkg/storage"
|
|
minio "github.com/minio/minio-go"
|
|
)
|
|
|
|
type storageImpl struct {
|
|
minioClient *minio.Client
|
|
minioCore *minio.Core
|
|
bucketName string
|
|
}
|
|
|
|
func (s *storageImpl) versionLocation(module, version string) string {
|
|
return fmt.Sprintf("%s/%s", module, version)
|
|
}
|
|
|
|
// NewStorage returns a connected Minio or DigitalOcean Spaces storage
|
|
// that implements storage.Backend
|
|
func NewStorage(conf *config.MinioConfig, timeout time.Duration) (storage.Backend, error) {
|
|
const op errors.Op = "minio.NewStorage"
|
|
endpoint := conf.Endpoint
|
|
accessKeyID := conf.Key
|
|
secretAccessKey := conf.Secret
|
|
bucketName := conf.Bucket
|
|
region := conf.Region
|
|
useSSL := conf.EnableSSL
|
|
minioCore, err := minio.NewCore(endpoint, accessKeyID, secretAccessKey, useSSL)
|
|
minioClient, err := minio.New(endpoint, accessKeyID, secretAccessKey, useSSL)
|
|
if err != nil {
|
|
return nil, errors.E(op, err)
|
|
}
|
|
|
|
err = minioClient.MakeBucket(bucketName, region)
|
|
if err != nil {
|
|
// Check to see if we already own this bucket
|
|
exists, _ := minioClient.BucketExists(bucketName)
|
|
if !exists {
|
|
// MakeBucket Error takes priority
|
|
return nil, errors.E(op, err)
|
|
}
|
|
}
|
|
return &storageImpl{minioClient, minioCore, bucketName}, nil
|
|
}
|