135 lines
3.8 KiB
Python
135 lines
3.8 KiB
Python
"""
|
||
InfluxDB 客户端模块
|
||
"""
|
||
from PyQt6.QtCore import QObject, pyqtSignal as Signal, QTimer, pyqtSlot as Slot
|
||
from influxdb_client import InfluxDBClient as InfluxClientLib
|
||
from typing import Dict, List, Optional
|
||
import logging
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class InfluxDBClient(QObject):
|
||
"""
|
||
InfluxDB 客户端，用于获取实时数据
|
||
"""
|
||
dataReceived = Signal(dict) # 数据接收信号，参数为 {field_name: value}
|
||
errorOccurred = Signal(str) # 错误信号
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self._client: Optional[InfluxClientLib] = None
|
||
self._url: str = ""
|
||
self._token: str = ""
|
||
self._org: str = ""
|
||
self._bucket: str = ""
|
||
self._query: str = ""
|
||
self._timer: Optional[QTimer] = None
|
||
self._is_connected: bool = False
|
||
|
||
@Slot(str, str, str, str)
|
||
def connect(self, url: str, token: str, org: str, bucket: str):
|
||
"""连接到 InfluxDB"""
|
||
try:
|
||
self._url = url
|
||
self._token = token
|
||
self._org = org
|
||
self._bucket = bucket
|
||
|
||
if self._client:
|
||
self._client.close()
|
||
|
||
self._client = InfluxClientLib(
|
||
url=url,
|
||
token=token,
|
||
org=org
|
||
)
|
||
|
||
# 测试连接
|
||
query_api = self._client.query_api()
|
||
test_query = f'from(bucket: "{bucket}") |> range(start: -1m) |> limit(n: 1)'
|
||
query_api.query(test_query)
|
||
|
||
self._is_connected = True
|
||
logger.info(f"Connected to InfluxDB: {url}")
|
||
|
||
except Exception as e:
|
||
self._is_connected = False
|
||
error_msg = f"连接失败: {str(e)}"
|
||
logger.error(error_msg)
|
||
self.errorOccurred.emit(error_msg)
|
||
|
||
@Slot(str)
|
||
def setQuery(self, query: str):
|
||
"""设置查询语句"""
|
||
self._query = query
|
||
|
||
@Slot(int)
|
||
def startQuery(self, interval_ms: int = 1000):
|
||
"""开始定时查询，interval_ms 为毫秒"""
|
||
if not self._is_connected:
|
||
self.errorOccurred.emit("未连接到 InfluxDB")
|
||
return
|
||
|
||
if self._timer:
|
||
self._timer.stop()
|
||
|
||
self._timer = QTimer(self)
|
||
self._timer.timeout.connect(self._executeQuery)
|
||
self._timer.start(interval_ms)
|
||
# 立即执行一次
|
||
self._executeQuery()
|
||
|
||
@Slot()
|
||
def stopQuery(self):
|
||
"""停止查询"""
|
||
if self._timer:
|
||
self._timer.stop()
|
||
|
||
def _executeQuery(self):
|
||
"""执行查询"""
|
||
if not self._is_connected or not self._query:
|
||
return
|
||
|
||
try:
|
||
query_api = self._client.query_api()
|
||
result = query_api.query(self._query)
|
||
|
||
# 解析查询结果
|
||
data = {}
|
||
for table in result:
|
||
for record in table.records:
|
||
field = record.get_field()
|
||
value = record.get_value()
|
||
# 如果同一个字段有多个值，取最后一个
|
||
data[field] = value
|
||
|
||
if data:
|
||
self.dataReceived.emit(data)
|
||
|
||
except Exception as e:
|
||
error_msg = f"查询失败: {str(e)}"
|
||
logger.error(error_msg)
|
||
self.errorOccurred.emit(error_msg)
|
||
|
||
@Slot()
|
||
def disconnect(self):
|
||
"""断开连接"""
|
||
if self._timer:
|
||
self._timer.stop()
|
||
|
||
if self._client:
|
||
self._client.close()
|
||
self._client = None
|
||
|
||
self._is_connected = False
|
||
|
||
def __del__(self):
|
||
self.disconnect()
|
||
|
||
|
||
|
||
|
||
|
||
|