Skip to content

Commit 628cc3f

Browse files
roegimchrza
authored andcommitted
Take into account connect_timeout_ms & request_timeout_ms
Also remove the unnecessary endpointUrl field from custom type aws_commons._lambda_function_arn_1
1 parent 6a1d771 commit 628cc3f

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

aws_s3--0.0.1.sql

+7
Original file line numberDiff line numberDiff line change
@@ -315,14 +315,21 @@ CREATE OR REPLACE FUNCTION aws_lambda._boto3_invoke(IN function_name aws_commons
315315
LANGUAGE plpython3u
316316
AS $$
317317
import boto3
318+
from botocore.client import Config
318319

319320
settings = plpy.execute("""SELECT
320321
coalesce(current_setting('aws_commons.connect_timeout_ms', true), '1000')::INTEGER AS connect_timeout_ms,
321322
coalesce(current_setting('aws_commons.request_timeout_ms', true), '3000')::INTEGER AS request_timeout_ms,
322323
coalesce(current_setting('aws_commons.endpoint_override', true), 'http://localstack:4566') AS endpoint_override""")[0]
323324

325+
connect_timeout = settings['connect_timeout_ms'] / 1000
326+
request_timeout = settings['request_timeout_ms'] / 1000
327+
328+
config = Config(connect_timeout=connect_timeout, read_timeout=request_timeout)
329+
324330
client=boto3.client(
325331
service_name='lambda',
332+
config=config,
326333
region_name=function_name['region'],
327334
endpoint_url=settings['endpoint_override'],
328335
aws_access_key_id='localstack',

0 commit comments

Comments
 (0)