Skip to content

Commit faf0a91

Browse files
fix: avoid invalid labels
1 parent d48a925 commit faf0a91

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

datapower_net.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -323,6 +323,7 @@ def invoke_api(self, api):
323323
""" invoke api_tests endpoints """
324324
http_call = getattr(requests, api['method'])
325325
try:
326+
invoke_labels = self.labels
326327
result = http_call(
327328
"https://{}:{}{}".format(self.ip, self.apiPort, api['path']),
328329
headers=api.get('headers', None),
@@ -333,31 +334,31 @@ def invoke_api(self, api):
333334
elapsed_time = result.elapsed.microseconds
334335
size = len(result.text)
335336
status = result.status_code
337+
336338
self.trawler.set_gauge(
337339
'datapower',
338340
"invoke_api_{}_size".format(api['name']),
339341
size,
340-
pod_name=self.name, labels=self.labels)
342+
pod_name=self.name, labels=invoke_labels)
341343
self.trawler.set_gauge(
342344
'datapower',
343345
"invoke_api_{}_time".format(api['name']),
344346
elapsed_time,
345-
pod_name=self.name, labels=self.labels)
346-
status_labels = self.labels
347-
status_labels['code'] = status
348-
self.trawler.set_gauge(
347+
pod_name=self.name, labels=invoke_labels)
348+
349+
status_labels = {'code':status}
350+
self.trawler.inc_counter(
349351
'datapower',
350352
"invoke_api_{}_status_total".format(api['name']),
351353
1,
352-
pod_name=self.name, labels=status_labels)
354+
pod_name=self.name, labels={**status_labels, **self.labels})
353355
except requests.RequestException:
354-
status_labels = self.labels
355-
status_labels['code'] = '000'
356-
self.trawler.set_gauge(
356+
status_labels = {'code': '000'}
357+
self.trawler.inc_counter(
357358
'datapower',
358359
"invoke_api_{}_status_total".format(api['name']),
359360
0,
360-
pod_name=self.name, labels=status_labels)
361+
pod_name=self.name, labels={**status_labels, **self.labels})
361362

362363
if __name__ == "__main__":
363364
net = DataPowerNet()

0 commit comments

Comments
 (0)