mirror of
https://github.com/marcredhat/kql
synced 2026-06-08 13:23:58 +00:00
marc
27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Minimal PowerQuery smoke test against SDL."""
|
|
import sys, json, time
|
|
from pathlib import Path
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
from harness.sdl_client import power_query, power_query_long_running
|
|
|
|
NOW_MS = int(time.time() * 1000)
|
|
START = NOW_MS - 30 * 24 * 3600 * 1000 # 30d back
|
|
END = NOW_MS
|
|
|
|
q = "dataset='kql-proof' | group n = count() by event_type"
|
|
print(f"Query: {q}")
|
|
print(f"Window: {START} .. {END}")
|
|
t0 = time.time()
|
|
r = power_query(q, START, END)
|
|
print(f"Initial response in {time.time()-t0:.2f}s:")
|
|
print(json.dumps({k: (v if k != 'values' else f'<{len(v)} rows>') for k, v in r.items()},
|
|
indent=2, default=str))
|
|
if r.get("continuationToken") or r.get("token"):
|
|
print("\nPolling for completion ...")
|
|
r = power_query_long_running(q, START, END, max_wait_sec=30)
|
|
print(json.dumps({k: (v if k != 'values' else f'<{len(v)} rows>') for k, v in r.items()},
|
|
indent=2, default=str))
|
|
print("\nColumns:", r.get("columns"))
|
|
print("First 20 values:", r.get("values", [])[:20])
|