fix(ext): 修改 extension 作用域 (#3)

Author: KomoriDev

nonebot_plugin_prometheus/__init__.py

@@ -12,7 +12,10 @@
 
 from nonebot_plugin_prometheus import api as api
 from nonebot_plugin_prometheus.config import Config
-from nonebot_plugin_prometheus.extension import MessageCounterExtension
+from nonebot_plugin_prometheus.extension import (
+    MessageSendCounter,
+    MessageReceiveCounter,
+)
 
 __plugin_meta__ = PluginMetadata(
     name="Prometheus 监控",
@@ -26,13 +29,10 @@
 
 __all__ = [prometheus_client, Counter, Gauge, Histogram, Summary]
 
-add_global_extension(MessageCounterExtension)
+add_global_extension(MessageSendCounter)
 
 metrics = (
     Command("metrics", help_text="查询指标数据")
     .usage(__plugin_meta__.usage)
-    .build(
-        block=True,
-        use_cmd_start=True,
-    )
+    .build(block=True, use_cmd_start=True, extensions=[MessageReceiveCounter()])
 )

nonebot_plugin_prometheus/extension.py

@@ -8,21 +8,31 @@
 )
 
 
-class MessageCounterExtension(Extension):
+class MessageReceiveCounter(Extension):
     @property
     def priority(self) -> int:
         return 15
 
     @property
     def id(self) -> str:
-        return "MessageCounter"
+        return "MessageReceiveCounter"
 
     async def receive_wrapper(
         self, bot: Bot, event: Event, command: Alconna, receive: UniMessage
     ) -> UniMessage:
         received_messages_counter.labels(bot.self_id, bot.adapter.get_name()).inc()
         return receive
 
+
+class MessageSendCounter(Extension):
+    @property
+    def priority(self) -> int:
+        return 15
+
+    @property
+    def id(self) -> str:
+        return "MessageSendCounter"
+
     async def send_wrapper(
         self, bot: Bot, event: Event, send: Union[str, Message, UniMessage]
     ) -> Union[str, Message, UniMessage]: