amarillo/amarillo-metrics
Archived
5
0
Fork 0
Code Issues 1 Pull requests Packages Projects Releases Wiki Activity

Setup function

Browse source
This commit is contained in:
Csaba 2023-12-07 15:59:26 +01:00
parent 68a37fd1f4
commit ea385d7382
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
amarillo/plugins/metrics/metrics.py
View file

@ -8,6 +8,8 @@ from fastapi import APIRouter, HTTPException, Depends, Request
from datetime import datetime from datetime import datetime
from prometheus_client.exposition import generate_latest from prometheus_client.exposition import generate_latest
from prometheus_client import Gauge, Counter from prometheus_client import Gauge, Counter
from prometheus_fastapi_instrumentator import Instrumentator
from prometheus_fastapi_instrumentator import metrics as pfi_metrics
from prometheus_fastapi_instrumentator.metrics import Info from prometheus_fastapi_instrumentator.metrics import Info
from fastapi import Depends, HTTPException, FastAPI from fastapi import Depends, HTTPException, FastAPI
from fastapi.security import HTTPBasic, HTTPBasicCredentials from fastapi.security import HTTPBasic, HTTPBasicCredentials
@ -46,4 +48,17 @@ def metrics(credentials: HTTPBasicCredentials = Depends(security)):
) )
# total_requests_metric.labels(endpoint="/amarillo-metrics").inc() # total_requests_metric.labels(endpoint="/amarillo-metrics").inc()
return PlainTextResponse(content=generate_latest()) return PlainTextResponse(content=generate_latest())
#TODO: maybe replace with an @setup decorator? would make it more obvious this is invoked from outside
def setup(app: FastAPI):
app.include_router(router)
instrumentator = Instrumentator().instrument(app)
instrumentator.add(pfi_metrics.default())
instrumentator.add(amarillo_trips_number_total())
instrumentator.instrument(app)