-
Notifications
You must be signed in to change notification settings - Fork 13
Expand file tree
/
Copy pathtelemetry.py
More file actions
187 lines (166 loc) · 6.54 KB
/
telemetry.py
File metadata and controls
187 lines (166 loc) · 6.54 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
"""Impressions API module."""
import logging
from splitio.api import APIException, headers_from_metadata
from splitio.api.client import HttpClientException
from splitio.models.telemetry import HTTPExceptionsAndLatencies
_LOGGER = logging.getLogger(__name__)
class TelemetryAPI(object): # pylint: disable=too-few-public-methods
"""Class that uses an httpClient to communicate with the Telemetry API."""
def __init__(self, client, sdk_key, sdk_metadata, telemetry_runtime_producer):
"""
Class constructor.
:param client: HTTP Client responsble for issuing calls to the backend.
:type client: HttpClient
:param sdk_key: User sdk_key token.
:type sdk_key: string
"""
self._client = client
self._sdk_key = sdk_key
self._metadata = headers_from_metadata(sdk_metadata)
self._telemetry_runtime_producer = telemetry_runtime_producer
self._client.set_telemetry_data(HTTPExceptionsAndLatencies.TELEMETRY, self._telemetry_runtime_producer)
def record_unique_keys(self, uniques):
"""
Send unique keys to the backend.
:param uniques: Unique Keys
:type json
"""
try:
response = self._client.post(
'telemetry',
'v1/keys/ss',
self._sdk_key,
body=uniques,
extra_headers=self._metadata
)
if not 200 <= response.status_code < 300:
raise APIException(response.body, response.status_code)
except HttpClientException as exc:
_LOGGER.debug(
'Error posting unique keys because an exception was raised by the HTTPClient'
)
_LOGGER.debug('Error: ', exc_info=True)
raise APIException('Unique keys not flushed properly.') from exc
def record_init(self, configs):
"""
Send init config data to the backend.
:param configs: configs
:type json
"""
try:
response = self._client.post(
'telemetry',
'v1/metrics/config',
self._sdk_key,
body=configs,
extra_headers=self._metadata,
)
if not 200 <= response.status_code < 300:
raise APIException(response.body, response.status_code)
except HttpClientException as exc:
_LOGGER.debug(
'Error posting init config because an exception was raised by the HTTPClient'
)
_LOGGER.debug('Error: ', exc_info=True)
def record_stats(self, stats):
"""
Send runtime stats to the backend.
:param stats: stats
:type json
"""
try:
response = self._client.post(
'telemetry',
'v1/metrics/usage',
self._sdk_key,
body=stats,
extra_headers=self._metadata,
)
if not 200 <= response.status_code < 300:
raise APIException(response.body, response.status_code)
except HttpClientException as exc:
_LOGGER.debug(
'Error posting runtime stats because an exception was raised by the HTTPClient'
)
_LOGGER.debug('Error: ', exc_info=True)
raise APIException('Runtime stats not flushed properly.') from exc
class TelemetryAPIAsync(object): # pylint: disable=too-few-public-methods
"""Async Class that uses an httpClient to communicate with the Telemetry API."""
def __init__(self, client, sdk_key, sdk_metadata, telemetry_runtime_producer):
"""
Class constructor.
:param client: HTTP Client responsble for issuing calls to the backend.
:type client: HttpClient
:param sdk_key: User sdk_key token.
:type sdk_key: string
"""
self._client = client
self._sdk_key = sdk_key
self._metadata = headers_from_metadata(sdk_metadata)
self._telemetry_runtime_producer = telemetry_runtime_producer
self._client.set_telemetry_data(HTTPExceptionsAndLatencies.TELEMETRY, self._telemetry_runtime_producer)
async def record_unique_keys(self, uniques):
"""
Send unique keys to the backend.
:param uniques: Unique Keys
:type json
"""
try:
response = await self._client.post(
'telemetry',
'v1/keys/ss',
self._sdk_key,
body=uniques,
extra_headers=self._metadata
)
if not 200 <= response.status_code < 300:
raise APIException(response.body, response.status_code)
except HttpClientException as exc:
_LOGGER.debug(
'Error posting unique keys because an exception was raised by the HTTPClient'
)
_LOGGER.debug('Error: ', exc_info=True)
raise APIException('Unique keys not flushed properly.') from exc
async def record_init(self, configs):
"""
Send init config data to the backend.
:param configs: configs
:type json
"""
try:
response = await self._client.post(
'telemetry',
'v1/metrics/config',
self._sdk_key,
body=configs,
extra_headers=self._metadata,
)
if not 200 <= response.status_code < 300:
raise APIException(response.body, response.status_code)
except HttpClientException as exc:
_LOGGER.debug(
'Error posting init config because an exception was raised by the HTTPClient'
)
_LOGGER.debug('Error: ', exc_info=True)
async def record_stats(self, stats):
"""
Send runtime stats to the backend.
:param stats: stats
:type json
"""
try:
response = await self._client.post(
'telemetry',
'v1/metrics/usage',
self._sdk_key,
body=stats,
extra_headers=self._metadata,
)
if not 200 <= response.status_code < 300:
raise APIException(response.body, response.status_code)
except HttpClientException as exc:
_LOGGER.debug(
'Error posting runtime stats because an exception was raised by the HTTPClient'
)
_LOGGER.debug('Error: ', exc_info=True)
raise APIException('Runtime stats not flushed properly.') from exc