Skip to content
Snippets Groups Projects
Commit 01016f7a authored by Karel van Klink's avatar Karel van Klink :smiley_cat: Committed by Neda Moeini
Browse files

re-use TCP session in Kentik client

parent 0e94e554
No related branches found
No related tags found
1 merge request!235Add a Kentik service to GSO
Pipeline #88115 passed
...@@ -33,18 +33,19 @@ class KentikClient: ...@@ -33,18 +33,19 @@ class KentikClient:
def __init__(self) -> None: def __init__(self) -> None:
"""Instantiate a new Kentik Client.""" """Instantiate a new Kentik Client."""
self.config = load_oss_params().KENTIK self.config = load_oss_params().KENTIK
self.headers = { self.session = requests.Session()
self.session.headers.update({
"X-CH-Auth-Email": self.config.user_email, "X-CH-Auth-Email": self.config.user_email,
"X-CH-Auth-API-Token": self.config.api_key, "X-CH-Auth-API-Token": self.config.api_key,
"Content-Type": "application/json", "Content-Type": "application/json",
} })
def _send_request( def _send_request(
self, method: Literal["GET", "POST", "PUT", "DELETE"], endpoint: str, data: dict[str, Any] | None = None self, method: Literal["GET", "POST", "PUT", "DELETE"], endpoint: str, data: dict[str, Any] | None = None
) -> Response: ) -> Response:
url = self.config.api_base + endpoint url = self.config.api_base + endpoint
logger.debug("Kentik - Sending request", extra={"method": method, "endpoint": url, "form_data": data}) logger.debug("Kentik - Sending request", extra={"method": method, "endpoint": url, "form_data": data})
result = requests.request(method, url, json=data, headers=self.headers) result = self.session.request(method, url, json=data)
logger.debug("Kentik - Received response", extra=result.__dict__) logger.debug("Kentik - Received response", extra=result.__dict__)
return result return result
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment