Add event/update counter to metrics

This commit is contained in:
Tulir Asokan
2019-06-01 21:10:01 +03:00
parent a279835cf8
commit 145eb8f611
2 changed files with 15 additions and 5 deletions
+8 -3
View File
@@ -52,13 +52,16 @@ UpdateMessage = Union[UpdateShortChatMessage, UpdateShortMessage, UpdateNewChann
UpdateMessageContent = Union[UpdateShortMessage, UpdateShortChatMessage, Message, MessageService]
try:
from prometheus_client import Histogram
from prometheus_client import Histogram, Counter
UPDATE_COUNT = Counter("telegram_update_count", "Number of Telegram updates processed",
["update_type"])
UPDATE_TIME = Histogram("telegram_update", "Time spent processing Telegram updates",
["update_type"])
except ImportError:
Histogram = None
UPDATE_TIME = None
UPDATE_COUNT = None
class AbstractUser(ABC):
session_container = None # type: AlchemySessionContainer
@@ -166,8 +169,10 @@ class AbstractUser(ABC):
await self._update(update)
except Exception:
self.log.exception("Failed to handle Telegram update")
if UPDATE_TIME:
UPDATE_TIME.labels(update_type=type(update).__name__).observe(time.time() - start_time)
if UPDATE_TIME and UPDATE_COUNT:
update_type = type(update).__name__
UPDATE_TIME.labels(update_type=update_type).observe(time.time() - start_time)
UPDATE_COUNT.labels(update_type=update_type).inc()
async def get_dialogs(self, limit: int = None) -> List[Union[Chat, Channel]]:
if self.is_bot: