fix(aws): update close if aws auth provider is nil (#3607)

Signed-off-by: Samantha Coyle <sam@diagrid.io>
Co-authored-by: Yaron Schneider <schneider.yaron@live.com>
This commit is contained in:
Sam 2024-11-20 11:24:49 -06:00 committed by GitHub
parent f3bd794b12
commit e2b27d3538
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
10 changed files with 40 additions and 10 deletions

View File

@ -122,5 +122,8 @@ func (d *DynamoDB) GetComponentMetadata() (metadataInfo metadata.MetadataMap) {
}
func (d *DynamoDB) Close() error {
if d.authProvider != nil {
return d.authProvider.Close()
}
return nil
}

View File

@ -266,7 +266,10 @@ func (a *AWSKinesis) Close() error {
close(a.closeCh)
}
a.wg.Wait()
if a.authProvider != nil {
return a.authProvider.Close()
}
return nil
}
func (a *AWSKinesis) ensureConsumer(ctx context.Context, streamARN *string) (*string, error) {

View File

@ -153,7 +153,10 @@ func (s *AWSS3) Init(ctx context.Context, metadata bindings.Metadata) error {
}
func (s *AWSS3) Close() error {
if s.authProvider != nil {
return s.authProvider.Close()
}
return nil
}
func (s *AWSS3) Operations() []bindings.OperationKind {

View File

@ -176,5 +176,8 @@ func (a *AWSSES) GetComponentMetadata() (metadataInfo contribMetadata.MetadataMa
}
func (a *AWSSES) Close() error {
if a.authProvider != nil {
return a.authProvider.Close()
}
return nil
}

View File

@ -128,5 +128,8 @@ func (a *AWSSNS) GetComponentMetadata() (metadataInfo metadata.MetadataMap) {
}
func (a *AWSSNS) Close() error {
if a.authProvider != nil {
return a.authProvider.Close()
}
return nil
}

View File

@ -173,7 +173,10 @@ func (a *AWSSQS) Close() error {
close(a.closeCh)
}
a.wg.Wait()
if a.authProvider != nil {
return a.authProvider.Close()
}
return nil
}
func (a *AWSSQS) parseSQSMetadata(meta bindings.Metadata) (*sqsMetadata, error) {

View File

@ -875,7 +875,10 @@ func (s *snsSqs) Close() error {
s.subscriptionManager.Close()
}
if s.authProvider != nil {
return s.authProvider.Close()
}
return nil
}
func (s *snsSqs) Features() []pubsub.Feature {

View File

@ -182,5 +182,8 @@ func (s *ssmSecretStore) GetComponentMetadata() (metadataInfo metadata.MetadataM
}
func (s *ssmSecretStore) Close() error {
if s.authProvider != nil {
return s.authProvider.Close()
}
return nil
}

View File

@ -170,5 +170,8 @@ func (s *smSecretStore) GetComponentMetadata() (metadataInfo metadata.MetadataMa
}
func (s *smSecretStore) Close() error {
if s.authProvider != nil {
return s.authProvider.Close()
}
return nil
}

View File

@ -275,7 +275,10 @@ func (d *StateStore) GetComponentMetadata() (metadataInfo metadata.MetadataMap)
}
func (d *StateStore) Close() error {
if d.authProvider != nil {
return d.authProvider.Close()
}
return nil
}
func (d *StateStore) getDynamoDBMetadata(meta state.Metadata) (*dynamoDBMetadata, error) {