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:
def __init__(self) -> None:
"""Instantiate a new Kentik Client."""
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-API-Token": self.config.api_key,
"Content-Type": "application/json",
}
})
def _send_request(
self, method: Literal["GET", "POST", "PUT", "DELETE"], endpoint: str, data: dict[str, Any] | None = None
) -> Response:
url = self.config.api_base + endpoint
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__)
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