package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: add missing file from previous commit 386b6f43

Browse Source
This commit is contained in:
Kashiwa
2025-01-09 13:05:33 +00:00
parent 386b6f431f
commit e4d46ce66c
1 changed files with 3 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
dynamodb/dynamodb.go
View File

@@ -44,7 +44,9 @@ func New(config Config) *Storage {
} }
// Create db // Create db
sess := awsdynamodb.NewFromConfig(awscfg) sess := awsdynamodb.NewFromConfig(awscfg, func(o *awsdynamodb.Options) {
o.BaseEndpoint = aws.String(cfg.Endpoint)
})
timeoutCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second) timeoutCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel() defer cancel()
@@ -232,23 +234,10 @@ func (s *Storage) requestContext() (context.Context, context.CancelFunc) {
} }
func returnAWSConfig(cfg Config) (aws.Config, error) { func returnAWSConfig(cfg Config) (aws.Config, error) {
endpoint := aws.EndpointResolverWithOptionsFunc(func(service, region string, options ...interface{}) (aws.Endpoint, error) {
if cfg.Endpoint != "" {
return aws.Endpoint{
PartitionID: "aws",
URL: cfg.Endpoint,
SigningRegion: cfg.Region,
HostnameImmutable: true,
}, nil
}
return aws.Endpoint{}, &aws.EndpointNotFoundError{}
})
if cfg.Credentials != (Credentials{}) { if cfg.Credentials != (Credentials{}) {
credentials := credentials.NewStaticCredentialsProvider(cfg.Credentials.AccessKey, cfg.Credentials.SecretAccessKey, "") credentials := credentials.NewStaticCredentialsProvider(cfg.Credentials.AccessKey, cfg.Credentials.SecretAccessKey, "")
return awsconfig.LoadDefaultConfig(context.TODO(), return awsconfig.LoadDefaultConfig(context.TODO(),
awsconfig.WithRegion(cfg.Region), awsconfig.WithRegion(cfg.Region),
awsconfig.WithEndpointResolverWithOptions(endpoint),
awsconfig.WithCredentialsProvider(credentials), awsconfig.WithCredentialsProvider(credentials),
awsconfig.WithRetryer(func() aws.Retryer { awsconfig.WithRetryer(func() aws.Retryer {
return retry.AddWithMaxAttempts(retry.NewStandard(), cfg.MaxAttempts) return retry.AddWithMaxAttempts(retry.NewStandard(), cfg.MaxAttempts)
@@ -258,7 +247,6 @@ func returnAWSConfig(cfg Config) (aws.Config, error) {
return awsconfig.LoadDefaultConfig(context.TODO(), return awsconfig.LoadDefaultConfig(context.TODO(),
awsconfig.WithRegion(cfg.Region), awsconfig.WithRegion(cfg.Region),
awsconfig.WithEndpointResolverWithOptions(endpoint),
awsconfig.WithRetryer(func() aws.Retryer { awsconfig.WithRetryer(func() aws.Retryer {
return retry.AddWithMaxAttempts(retry.NewStandard(), cfg.MaxAttempts) return retry.AddWithMaxAttempts(retry.NewStandard(), cfg.MaxAttempts)
}), }),