29 lines
907 B
Python
29 lines
907 B
Python
|
from aioinflux import InfluxDBClient
|
||
|
|
|
||
|
|
from pluralkit.bot import logger
|
||
|
|
|
||
|
|
client = None
|
||
|
|
async def connect():
|
||
|
|
global client
|
||
|
|
client = InfluxDBClient(host="influx", db="pluralkit")
|
||
|
|
await client.create_database(db="pluralkit")
|
||
|
|
|
||
|
|
async def report_db_query(query_name, time, success):
|
||
|
|
await client.write({
|
||
|
|
"measurement": "database_query",
|
||
|
|
"tags": {"query": query_name},
|
||
|
|
"fields": {"response_time": time, "success": int(success)}
|
||
|
|
})
|
||
|
|
|
||
|
|
async def report_command(command_name, execution_time, response_time):
|
||
|
|
await client.write({
|
||
|
|
"measurement": "command",
|
||
|
|
"tags": {"command": command_name},
|
||
|
|
"fields": {"execution_time": execution_time, "response_time": response_time}
|
||
|
|
})
|
||
|
|
|
||
|
|
async def report_webhook(time, success):
|
||
|
|
await client.write({
|
||
|
|
"measurement": "webhook",
|
||
|
|
"fields": {"response_time": time, "success": int(success)}
|
||
|
|
})
|