sig_core
/
srpmproc
mirror of https://github.com/rocky-linux/srpmproc.git
8
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

enhancement: make it possible to set custom s3 endpoints

This commit is contained in:
Mustafa Gezen 2021-09-04 01:57:29 +02:00
parent f6c9dba678
commit 825228e627
Signed by: mustafa
GPG Key ID: DCDF010D946438C1
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/blob/s3/s3.go
View File

@ -23,9 +23,11 @@ package s3
import (
"bytes"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3"
"github.com/aws/aws-sdk-go/service/s3/s3manager"
"github.com/spf13/viper"
"io/ioutil"
"log"
)
@ -36,7 +38,25 @@ type S3 struct {
}
func New(name string) *S3 {
sess := session.Must(session.NewSession())
awsCfg := &aws.Config{}
if accessKey := viper.GetString("s3-access-key"); accessKey != "" {
awsCfg.Credentials = credentials.NewStaticCredentials(accessKey, viper.GetString("s3-secret-key"), "")
}
if endpoint := viper.GetString("s3-endpoint"); endpoint != "" {
awsCfg.Endpoint = aws.String(endpoint)
}
if region := viper.GetString("s3-region"); region != "" {
awsCfg.Region = aws.String(region)
}
if disableSsl := viper.GetBool("s3-disable-ssl"); disableSsl {
awsCfg.DisableSSL = aws.Bool(true)
}
sess := session.Must(session.NewSession(awsCfg))
uploader := s3manager.NewUploader(sess)
return &S3{