mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-25 07:36:27 +00:00
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
from typing import TYPE_CHECKING, List
|
|
|
|
from gram_core.base_service import BaseService
|
|
from gram_core.dependence.influxdb import InfluxDatabase
|
|
|
|
if TYPE_CHECKING:
|
|
from influxdb_client.client.flux_table import FluxTable
|
|
from influxdb_client.client.influxdb_client_async import InfluxDBClientAsync
|
|
from influxdb_client import Point
|
|
|
|
|
|
class ActionLogRepository(BaseService.Component):
|
|
|
|
def __init__(self, influxdb: InfluxDatabase):
|
|
self.influxdb = influxdb
|
|
self.client = influxdb.client
|
|
self.bucket = "hkrpg"
|
|
|
|
async def add(self, p: List["Point"]) -> bool:
|
|
async with self.client() as client:
|
|
client: "InfluxDBClientAsync"
|
|
return await client.write_api().write(self.bucket, record=p)
|
|
|
|
async def count_uptime_period(self, uid: int) -> "FluxTable":
|
|
async with self.client() as client:
|
|
client: "InfluxDBClientAsync"
|
|
query = (
|
|
'import "date"'
|
|
'import "timezone"'
|
|
'option location = timezone.location(name: "Asia/Shanghai")'
|
|
'from(bucket: "{}")'
|
|
"|> range(start: -180d)"
|
|
'|> filter(fn: (r) => r["_measurement"] == "action_log")'
|
|
'|> filter(fn: (r) => r["_field"] == "status")'
|
|
'|> filter(fn: (r) => r["_value"] == 1)'
|
|
'|> filter(fn: (r) => r["uid"] == "{}")'
|
|
"|> aggregateWindow(every: 1h, fn: count)"
|
|
).format(self.bucket, uid)
|
|
query += '|> map(fn: (r) => ({r with hour: date.hour(t: r._time)}))|> yield(name: "hourly_count")'
|
|
tables = await client.query_api().query(query)
|
|
for table in tables:
|
|
return table
|
|
|
|
async def get_data(self, uid: int, day: int = 30) -> "FluxTable":
|
|
async with self.client() as client:
|
|
client: "InfluxDBClientAsync"
|
|
query = (
|
|
'from(bucket: "{}")'
|
|
"|> range(start: -{}d)"
|
|
'|> filter(fn: (r) => r["_measurement"] == "action_log")'
|
|
'|> filter(fn: (r) => r["uid"] == "{}")'
|
|
'|> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'
|
|
).format(self.bucket, day, uid)
|
|
tables = await client.query_api().query(query)
|
|
for table in tables:
|
|
return table
|