pankajastro commented on code in PR #28850:
URL: https://github.com/apache/airflow/pull/28850#discussion_r1112816670
##########
airflow/providers/amazon/aws/hooks/base_aws.py:
##########
@@ -843,3 +848,82 @@ def _parse_s3_config(config_file_name: str, config_format:
str | None = "boto",
config_format=config_format,
profile=profile,
)
+
+
+class AwsBaseHookAsync(AwsBaseHook):
+ """
+ Interacts with AWS using aiobotocore asynchronously.
+
+ .. note::
+ AwsBaseHookAsync uses aiobotocore to create asynchronous S3 hooks.
Hence, AwsBaseHookAsync
+ only supports the authentication mechanism that aiobotocore supports.
Currently, AwsBaseHookAsync supports
+ only AWS STS client method ``assume_role`` provided in the Airflow
connection extra args via aiobotocore.
+
+ :param aws_conn_id: The Airflow connection used for AWS credentials.
+ If this is None or empty then the default boto3 behaviour is used. If
+ running Airflow in a distributed manner and aws_conn_id is None or
+ empty, then default boto3 configuration would be used (and must be
+ maintained on each worker node).
+ :param verify: Whether or not to verify SSL certificates.
+ :param region_name: AWS region_name. If not specified then the default
boto3 behaviour is used.
+ :param client_type: boto3.client client_type. Eg 's3', 'emr' etc
+ :param resource_type: boto3.resource resource_type. Eg 'dynamodb' etc
+ :param config: Configuration for botocore client.
+ .. seealso::
+ `AWS API
<https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html>`_
+ """
+
+ async def get_client_async(self) -> AioBaseClient:
+ """Create an Async Client object to communicate with AWS services."""
+
+ connection_object = await
sync_to_async(self.get_connection)(self.aws_conn_id)
+
Review Comment:
yeah, fixed the behaviour
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscr...@airflow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
