30 lines
722 B
Python
30 lines
722 B
Python
"""PostgreSQL execution operations used by the direct DWH adapter."""
|
|
|
|
from sqlalchemy import Engine
|
|
|
|
from tht.execute import (
|
|
ExecResult,
|
|
PlanSummary,
|
|
require_positive_int,
|
|
run_controlled,
|
|
)
|
|
from tht.execute import (
|
|
explain as _explain,
|
|
)
|
|
|
|
DEFAULT_TIMEOUT_MS = 30_000
|
|
|
|
|
|
def run_query(engine: Engine, sql: str, *, limit: int, timeout_ms: int = DEFAULT_TIMEOUT_MS) -> ExecResult:
|
|
limit = require_positive_int(limit, name="limit")
|
|
return run_controlled(
|
|
engine,
|
|
sql,
|
|
limit=limit,
|
|
timeout_ms=timeout_ms,
|
|
)
|
|
|
|
|
|
def explain(engine: Engine, sql: str, *, timeout_ms: int = DEFAULT_TIMEOUT_MS) -> PlanSummary:
|
|
return _explain(engine, sql, timeout_ms=timeout_ms)
|