Skip to content

Commit 7589142

Browse files
Support prometheus metrics for opea microservices (#225)
* add preload env in run.sh Signed-off-by: lvliang-intel <[email protected]> * Support prometheus metrics for all opea microservices Signed-off-by: lvliang-intel <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: lvliang-intel <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 4468ed2 commit 7589142

File tree

3 files changed

+5
-0
lines changed

3 files changed

+5
-0
lines changed

comps/cores/mega/http_service.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from typing import Optional
55

66
from fastapi import FastAPI
7+
from prometheus_fastapi_instrumentator import Instrumentator
78
from uvicorn import Config, Server
89

910
from .base_service import BaseService
@@ -32,6 +33,7 @@ def __init__(
3233
self.uvicorn_kwargs = uvicorn_kwargs or {}
3334
self.cors = cors
3435
self._app = self._create_app()
36+
Instrumentator().instrument(self._app).expose(self._app)
3537

3638
@property
3739
def app(self):

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ httpx
55
opentelemetry-api
66
opentelemetry-exporter-otlp
77
opentelemetry-sdk
8+
prometheus-fastapi-instrumentator
89
pyyaml
910
requests
1011
shortuuid

tests/cores/mega/test_microservice.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ def test_add_route(self):
3131
response = self.client.post("/v1/add", json={"text": "Hello, "})
3232
self.assertEqual(response.status_code, 200)
3333
self.assertEqual(response.json()["text"], "Hello, OPEA Project!")
34+
response = self.client.get("/metrics")
35+
self.assertEqual(response.status_code, 200)
3436

3537

3638
if __name__ == "__main__":

0 commit comments

Comments
 (0)