152 lines
4.6 KiB
Python
152 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
调试时间戳问题
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import datetime
|
||
sys.path.append(os.path.dirname(__file__))
|
||
|
||
from influxdb_client import InfluxDBClient, Point
|
||
from influxdb_client.client.write_api import SYNCHRONOUS
|
||
from config_model import AppConfig
|
||
from pathlib import Path
|
||
|
||
def test_timestamp_writing():
|
||
"""测试不同时间戳的写入"""
|
||
print("🕐 测试时间戳写入")
|
||
|
||
# 加载配置
|
||
config_path = Path("default.json")
|
||
config = AppConfig.load(config_path)
|
||
|
||
influx_config = config.influx
|
||
|
||
client = InfluxDBClient(
|
||
url=influx_config.url,
|
||
token=influx_config.token,
|
||
org=influx_config.org
|
||
)
|
||
|
||
write_api = client.write_api(write_options=SYNCHRONOUS)
|
||
bucket = getattr(influx_config, 'bucket', 'PCM')
|
||
measurement = getattr(influx_config, 'measurement', 'experiment_status')
|
||
|
||
# 测试不同的时间戳格式
|
||
now_local = datetime.datetime.now()
|
||
now_utc = datetime.datetime.now(datetime.timezone.utc)
|
||
|
||
print(f"本地时间: {now_local}")
|
||
print(f"UTC时间: {now_utc}")
|
||
|
||
try:
|
||
# 1. 写入本地时间戳
|
||
print("\n1️⃣ 写入本地时间戳...")
|
||
point1 = Point(measurement) \
|
||
.field("status", "test_local") \
|
||
.time(now_local)
|
||
|
||
write_api.write(bucket=bucket, record=point1)
|
||
print("✅ 本地时间戳写入成功")
|
||
|
||
# 2. 写入UTC时间戳
|
||
print("\n2️⃣ 写入UTC时间戳...")
|
||
point2 = Point(measurement) \
|
||
.field("status", "test_utc") \
|
||
.time(now_utc)
|
||
|
||
write_api.write(bucket=bucket, record=point2)
|
||
print("✅ UTC时间戳写入成功")
|
||
|
||
# 3. 写入当前时间（无时区）
|
||
print("\n3️⃣ 写入当前时间（模拟quick_test_data.py）...")
|
||
point3 = Point(measurement) \
|
||
.field("status", "test_current") \
|
||
.time(datetime.datetime.now())
|
||
|
||
write_api.write(bucket=bucket, record=point3)
|
||
print("✅ 当前时间写入成功")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 写入失败: {e}")
|
||
return
|
||
|
||
client.close()
|
||
|
||
# 等待一下让数据落盘
|
||
import time
|
||
print("\n⏳ 等待2秒让数据落盘...")
|
||
time.sleep(2)
|
||
|
||
# 查询刚写入的数据
|
||
print("\n🔍 查询刚写入的数据...")
|
||
|
||
from influx_service import InfluxService, InfluxConnectionParams
|
||
|
||
params = InfluxConnectionParams(
|
||
url=influx_config.url,
|
||
org=influx_config.org,
|
||
token=influx_config.token
|
||
)
|
||
|
||
service = InfluxService(params)
|
||
|
||
try:
|
||
# 查询最近5分钟的数据
|
||
df = service.query(
|
||
bucket=bucket,
|
||
measurement=measurement,
|
||
fields=['status'],
|
||
filters={},
|
||
time_range="-5m"
|
||
)
|
||
|
||
if df.empty:
|
||
print("❌ 没有找到刚写入的数据")
|
||
|
||
# 尝试更大的时间范围
|
||
print("🔍 尝试查询最近1小时的数据...")
|
||
df_hour = service.query(
|
||
bucket=bucket,
|
||
measurement=measurement,
|
||
fields=['status'],
|
||
filters={},
|
||
time_range="-1h"
|
||
)
|
||
|
||
if df_hour.empty:
|
||
print("❌ 1小时内也没有数据")
|
||
else:
|
||
print(f"✅ 1小时内找到 {len(df_hour)} 条数据")
|
||
# 显示最新的几条
|
||
if '_time' in df_hour.columns and '_value' in df_hour.columns:
|
||
df_sorted = df_hour.sort_values('_time').tail(5)
|
||
for _, row in df_sorted.iterrows():
|
||
time_local = row['_time'].astimezone()
|
||
time_str = time_local.strftime('%H:%M:%S')
|
||
value = row['_value']
|
||
print(f" {time_str}: {value}")
|
||
else:
|
||
print(f"✅ 找到 {len(df)} 条刚写入的数据")
|
||
|
||
# 显示数据
|
||
if '_time' in df.columns and '_value' in df.columns:
|
||
df_sorted = df.sort_values('_time')
|
||
for _, row in df_sorted.iterrows():
|
||
time_local = row['_time'].astimezone()
|
||
time_str = time_local.strftime('%H:%M:%S')
|
||
value = row['_value']
|
||
print(f" {time_str}: {value}")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 查询失败: {e}")
|
||
|
||
def main():
|
||
print("时间戳问题调试工具")
|
||
print("=" * 30)
|
||
test_timestamp_writing()
|
||
|
||
if __name__ == "__main__":
|
||
main()
|