Merge pull request #13 from calebstewart/fix/issue-5-empty-kinesisendpoint

Automatically resolve default KinesisEndpoint
This commit is contained in:
vmwjc 2023-04-04 14:40:26 -07:00 committed by GitHub
commit 16c5c53a30
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

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) {
if service == kinesis.ServiceID && len(w.kclConfig.KinesisEndpoint) > 0 {
return aws.Endpoint{ return aws.Endpoint{
PartitionID: "aws", PartitionID: "aws",
URL: w.kclConfig.KinesisEndpoint, URL: w.kclConfig.KinesisEndpoint,
SigningRegion: w.regionName, SigningRegion: w.regionName,
}, nil }, 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(