yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2026-01-06 04:26:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Support for Endpoint URL in Datalake (#9293)

Browse Source
This commit is contained in:
Mayur Singal 2022-12-15 13:00:32 +05:30 committed by GitHub
parent 9ce4fa8a84
commit 8fe3b09d21
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ingestion/src/metadata/clients/aws_client.py
View File

@ -73,6 +73,10 @@ class AWSClient:
if self.config is not None:
logger.info(f"Getting AWS client for service [{service_name}]")
session = self._get_session()
if self.config.endPointURL is not None:
return session.client(
service_name=service_name, endpoint_url=self.config.endPointURL
)
return session.client(service_name=service_name)
logger.info(f"Getting AWS default client for service [{service_name}]")
@ -81,6 +85,10 @@ class AWSClient:
def get_resource(self, service_name: str) -> Any:
session = self._get_session()
if self.config.endPointURL is not None:
return session.resource(
service_name=service_name, endpoint_url=self.config.endPointURL
)
return session.resource(service_name=service_name)
def get_dynamo_client(self) -> DynamoClient: