luciano/vmware-go-kcl-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Automatically resolve default KinesisEndpoint

Browse source 
This commit fixes #5 by returning `aws.EndpointNotFoundError` from the
endpoint resolver when no `KinesisEndpoint` is defined, which will
resolve the default AWS endpoint. This is the same process used by the
DynamoDB checkpointer to resolve the default endpoint.

Signed-off-by: Caleb Stewart <caleb.stewart94@gmail.com>
This commit is contained in:
Caleb Stewart 2022-10-13 13:37:51 -04:00
parent c862165130
commit e2a45c53c3
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
clientlibrary/worker/worker.go
View file

@ -160,11 +160,15 @@ func (w *Worker) initialize() error {
log.Infof("Creating Kinesis client") log.Infof("Creating Kinesis client")
resolver := aws.EndpointResolverWithOptionsFunc(func(service, region string, options ...interface{}) (aws.Endpoint, error) { resolver := aws.EndpointResolverWithOptionsFunc(func(service, region string, options ...interface{}) (aws.Endpoint, error) {
return aws.Endpoint{ if service == kinesis.ServiceID && len(w.kclConfig.KinesisEndpoint) > 0 {
PartitionID: "aws", return aws.Endpoint{
URL: w.kclConfig.KinesisEndpoint, PartitionID: "aws",
SigningRegion: w.regionName, URL: w.kclConfig.KinesisEndpoint,
}, nil SigningRegion: w.regionName,
}, nil
}
// returning EndpointNotFoundError will allow the service to fallback to it's default resolution
return aws.Endpoint{}, &aws.EndpointNotFoundError{}
}) })
cfg, err := awsConfig.LoadDefaultConfig( cfg, err := awsConfig.LoadDefaultConfig(