-
Notifications
You must be signed in to change notification settings - Fork 7
/
exporter.py
48 lines (32 loc) · 1.19 KB
/
exporter.py
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
from typing import Optional
from fastapi import FastAPI
from fast_tools.base import RouteTrie
from fast_tools.exporter import PrometheusMiddleware, get_metrics, init_registry
app: "FastAPI" = FastAPI()
route_trie: "RouteTrie" = RouteTrie()
app.add_middleware(PrometheusMiddleware, route_trie=route_trie, block_url_set={"/metrics"})
app.add_route("/metrics", get_metrics)
@app.on_event("startup")
async def startup_event() -> None:
init_registry()
route_trie.insert_by_app(app)
@app.get("/")
async def root() -> dict:
return {"Hello": "World"}
@app.get("/api/users/{user_id}/items/{item_id}")
async def read_user_item(user_id: int, item_id: str, q: Optional[str] = None, short: bool = False) -> dict:
"""
copy from:https://fastapi.tiangolo.com/tutorial/query-params/#multiple-path-and-query-parameters
"""
item = {"item_id": item_id, "owner_id": user_id}
if q:
item.update({"q": q})
if not short:
item.update({"description": "This is an amazing item that has a long description"})
return item
@app.get("/api/users/login")
async def user_login() -> str:
return "ok"
if __name__ == "__main__":
import uvicorn # type: ignore
uvicorn.run(app)