Coverage for mindsdb / utilities / otel / metric_handlers / __init__.py: 0%

1from mindsdb.utilities.otel.prepare import metrics, OTEL_SERVICE_NAME 

2 

3_query_request_counter = None 

4 

5 

6def get_query_request_counter(): 

7 """ 

8 Get the query request counter 

9 """ 

10 global _query_request_counter 

11 

12 # Create the counter if it does not exist 

13 if _query_request_counter is None: 

14 meter_name = f"{OTEL_SERVICE_NAME}.query_service_meter" 

15 

16 # Get the meter from the main metrics object 

17 meter = metrics.get_meter(meter_name) 

18 

19 _query_request_counter = meter.create_counter( 

20 name="query_request_count", 

21 description="Counts the number of times the SQL query is called", 

22 unit="1", 

23 ) 

24 

25 return _query_request_counter 

26 

27def increment_otel_query_request_counter(metadata: dict) -> None: 

28 query_request_counter = get_query_request_counter() 

29 query_request_counter.add(1, metadata)