-
-
Notifications
You must be signed in to change notification settings - Fork 87
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #350 from Forest-Troll/master
Added prometheus integration
- Loading branch information
Showing
9 changed files
with
159 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,86 @@ | ||
"""Prometheus metrics collectors.""" | ||
import logging | ||
from typing import Callable | ||
|
||
import sqlalchemy | ||
from prometheus_client import REGISTRY, Gauge, make_asgi_app | ||
from sqlalchemy.ext.asyncio import AsyncSession | ||
from sqlalchemy.orm import contains_eager | ||
|
||
from spoolman.database import models | ||
|
||
registry = REGISTRY | ||
|
||
PREFIX = "spoolman" | ||
|
||
SPOOL_PRICE = Gauge("%s_spool_price" % PREFIX, "Total Spool price", ["spool_id", "filament_id"]) | ||
SPOOL_USED_WEIGHT = Gauge("%s_spool_weight_used" % PREFIX, "Spool Used Weight", ["spool_id", "filament_id"]) | ||
FILAMENT_INFO = Gauge( | ||
"%s_filament_info" % PREFIX, | ||
"Filament information", | ||
["filament_id", "vendor", "name", "material", "color"], | ||
) | ||
FILAMENT_DENSITY = Gauge("%s_filament_density" % PREFIX, "Density of filament", ["filament_id"]) | ||
FILAMENT_DIAMETER = Gauge("%s_filament_diameter" % PREFIX, "Diameter of filament", ["filament_id"]) | ||
FILAMENT_WEIGHT = Gauge("%s_filament_weight" % PREFIX, "Net weight of filament", ["filament_id"]) | ||
|
||
logger = logging.getLogger(__name__) | ||
|
||
|
||
def make_metrics_app() -> Callable: | ||
"""Start ASGI prometheus app with global registry.""" | ||
logger.info("Start metrics app") | ||
return make_asgi_app(registry=registry) | ||
|
||
|
||
metrics_app = make_asgi_app() | ||
|
||
|
||
async def spool_metrics(db: AsyncSession) -> None: | ||
"""Get metrics by Spools from DB and write to prometheus. | ||
Args: | ||
db: async db session | ||
""" | ||
stmt = sqlalchemy.select(models.Spool).where( | ||
sqlalchemy.or_( | ||
models.Spool.archived.is_(False), | ||
models.Spool.archived.is_(None), | ||
), | ||
) | ||
rows = await db.execute(stmt) | ||
result = list(rows.unique().scalars().all()) | ||
for row in result: | ||
if row.price is not None: | ||
SPOOL_PRICE.labels(str(row.id), str(row.filament_id)).set(row.price) | ||
SPOOL_USED_WEIGHT.labels(str(row.id), str(row.filament_id)).set(row.used_weight) | ||
|
||
|
||
async def filament_metrics(db: AsyncSession) -> None: | ||
"""Get metrics and info by Filaments from DB and write to prometheus. | ||
Args: | ||
db: async db session | ||
""" | ||
stmt = ( | ||
sqlalchemy.select(models.Filament) | ||
.options(contains_eager(models.Filament.vendor)) | ||
.join(models.Filament.vendor, isouter=True) | ||
) | ||
rows = await db.execute(stmt) | ||
result = list(rows.unique().scalars().all()) | ||
for row in result: | ||
vendor_name = "-" | ||
if row.vendor is not None: | ||
vendor_name = row.vendor.name | ||
FILAMENT_INFO.labels( | ||
str(row.id), | ||
vendor_name, | ||
row.name, | ||
row.material, | ||
row.color_hex, | ||
).set(1) | ||
FILAMENT_DENSITY.labels(str(row.id)).set(row.density) | ||
FILAMENT_DIAMETER.labels(str(row.id)).set(row.diameter) | ||
if row.weight is not None: | ||
FILAMENT_WEIGHT.labels(str(row.id)).set(row.weight) |