omnibill/fsys
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

add option to use an existing s3 client.

Browse source
This commit is contained in:
Shane C. 2024-11-22 13:31:14 -05:00
parent ff784c6585
commit 2edba5d03f
Signed by: Shane C.
GPG key ID: E46B5FEA35B22FF9
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
storage.go
View file

@ -42,6 +42,9 @@ type Config struct {
S3AccessKey string
S3BucketName string
// Override default s3 client.
MinioClient *minio.Client
}
// New creates a new storage interface
@ -54,26 +57,29 @@ func New(config Config) (*Storage, error) {
switch config.Type {
case "minio", "s3":
s3Client, err := minio.New(config.S3Endpoint, &minio.Options{
Creds: credentials.NewStaticV4(config.S3AccessID, config.S3AccessKey, ""),
Secure: config.S3Secure,
})
if err != nil {
return nil, err
if config.MinioClient != nil {
newStorage.s3Client = config.MinioClient
} else {
s3Client, err := minio.New(config.S3Endpoint, &minio.Options{
Creds: credentials.NewStaticV4(config.S3AccessID, config.S3AccessKey, ""),
Secure: config.S3Secure,
})
if err != nil {
return nil, err
}
newStorage.s3Client = s3Client
}
bucketExists, err := s3Client.BucketExists(newStorage.ctx, config.S3BucketName)
bucketExists, err := newStorage.s3Client.BucketExists(newStorage.ctx, config.S3BucketName)
if err != nil {
return nil, err
}
if !bucketExists {
if err := s3Client.MakeBucket(newStorage.ctx, config.S3BucketName, minio.MakeBucketOptions{}); err != nil {
if err := newStorage.s3Client.MakeBucket(newStorage.ctx, config.S3BucketName, minio.MakeBucketOptions{}); err != nil {
return nil, err
}
}
newStorage.s3Client = s3Client
case "local":
if statInfo, err := os.Stat(config.Path); err != nil && errors.Is(err, fs.ErrNotExist) {
if err := os.MkdirAll(config.Path, 0740); err != nil {