feat: add user_id index to profiles table for faster lookups

The index on user_id column in profiles table improves query performance for user-specific operations, reducing full table scans during authentication and profile retrieval.
This commit is contained in:
2026-06-14 01:57:17 +00:00
parent 61c6382c39
commit b10a3debd1
8 changed files with 444 additions and 3 deletions
+148
View File
@@ -0,0 +1,148 @@
# retoor <retoor@molodetz.nl>
import json
import requests
from tests.conftest import BASE_URL
from devplacepy.database import (
create_deepsearch_session,
get_table,
refresh_snapshot,
)
from devplacepy.services.jobs import queue
def _json_headers():
return {"Accept": "application/json"}
def _seed_done_session(owner_id="ds-session-owner"):
uid = queue.enqueue(
"deepsearch",
{"query": "the question", "depth": 2, "max_pages": 10},
"user",
owner_id,
"DeepSearch: the question",
)
create_deepsearch_session(
uid, "user", owner_id, "the question", 2, 10, f"ds_{uid.replace('-', '')}"
)
report = {
"query": "the question",
"summary": "A grounded summary.",
"findings": [
{"title": "Finding one", "detail": "Detail.", "confidence": 0.8, "citations": [1]}
],
"gaps": ["An open gap."],
"sources": [{"url": "https://example.com", "title": "Example", "source": "httpx"}],
"score": 70,
"confidence": 0.7,
"source_diversity": 0.5,
"page_count": 3,
"chunk_count": 12,
}
result = {
"query": "the question",
"score": 70,
"confidence": 0.7,
"source_diversity": 0.5,
"page_count": 3,
"chunk_count": 12,
"report": report,
}
get_table("jobs").update(
{"uid": uid, "status": queue.DONE, "result": json.dumps(result)}, ["uid"]
)
get_table("deepsearch_sessions").update(
{
"uid": uid,
"status": "done",
"score": 70,
"confidence": 0.7,
"source_diversity": 0.5,
"page_count": 3,
"chunk_count": 12,
"summary": "A grounded summary.",
},
["uid"],
)
refresh_snapshot()
return uid
def _clear():
refresh_snapshot()
jobs = get_table("jobs")
for row in list(jobs.find(kind="deepsearch")):
jobs.delete(uid=row["uid"])
sessions = get_table("deepsearch_sessions")
for row in list(sessions.find()):
sessions.delete(uid=row["uid"])
def test_session_json_shape(app_server):
try:
uid = _seed_done_session()
r = requests.get(
f"{BASE_URL}/tools/deepsearch/{uid}/session", headers=_json_headers()
)
assert r.status_code == 200, r.text
body = r.json()
assert body["uid"] == uid
assert body["status"] == "done"
assert body["query"] == "the question"
assert body["score"] == 70
assert body["chat_ws_url"] == f"/tools/deepsearch/{uid}/chat"
assert body["export_md_url"] == f"/tools/deepsearch/{uid}/export.md"
assert body["findings"]
assert body["sources"]
assert "viewer_is_admin" in body
assert "viewer_owns" in body
finally:
_clear()
def test_session_html_renders_without_jinja_global_collision(app_server):
try:
uid = _seed_done_session()
r = requests.get(f"{BASE_URL}/tools/deepsearch/{uid}/session")
assert r.status_code == 200, r.text
assert "the question" in r.text
assert "dp-deepsearch-chat" in r.text
finally:
_clear()
def test_session_unknown_uid_404(app_server):
r = requests.get(
f"{BASE_URL}/tools/deepsearch/nope/session", headers=_json_headers()
)
assert r.status_code == 404
def test_export_markdown(app_server):
try:
uid = _seed_done_session()
r = requests.get(f"{BASE_URL}/tools/deepsearch/{uid}/export.md")
assert r.status_code == 200, r.text
assert "DeepSearch report" in r.text
assert "Finding one" in r.text
finally:
_clear()
def test_export_json(app_server):
try:
uid = _seed_done_session()
r = requests.get(f"{BASE_URL}/tools/deepsearch/{uid}/export.json")
assert r.status_code == 200, r.text
body = r.json()
assert body["query"] == "the question"
assert body["findings"]
finally:
_clear()
def test_export_unknown_uid_404(app_server):
r = requests.get(f"{BASE_URL}/tools/deepsearch/nope/export.md")
assert r.status_code == 404
+63
View File
@@ -0,0 +1,63 @@
# retoor <retoor@molodetz.nl>
import requests
from tests.conftest import BASE_URL
from devplacepy.database import get_table, refresh_snapshot
from devplacepy.services.jobs import queue
def _json_headers():
return {"Accept": "application/json"}
def _clear():
refresh_snapshot()
jobs = get_table("jobs")
for row in list(jobs.find(kind="deepsearch")):
jobs.delete(uid=row["uid"])
def test_status_shape_for_pending_job(app_server):
try:
uid = queue.enqueue(
"deepsearch",
{"query": "a research question", "depth": 2, "max_pages": 10},
"user",
"ds-status-owner",
"DeepSearch: a research question",
)
refresh_snapshot()
r = requests.get(
f"{BASE_URL}/tools/deepsearch/{uid}", headers=_json_headers()
)
assert r.status_code == 200, r.text
body = r.json()
assert body["uid"] == uid
assert body["kind"] == "deepsearch"
assert body["status"] == "pending"
assert body["query"] == "a research question"
assert body["ws_url"] == f"/tools/deepsearch/{uid}/ws"
assert body["session_url"] is None
assert body["score"] is None
assert body["page_count"] == 0
finally:
_clear()
def test_status_unknown_uid_404(app_server):
r = requests.get(
f"{BASE_URL}/tools/deepsearch/nope-not-a-job", headers=_json_headers()
)
assert r.status_code == 404
def test_status_rejects_non_deepsearch_kind(app_server):
try:
uid = queue.enqueue("other", {}, "user", "ds-kind-owner", "x")
refresh_snapshot()
r = requests.get(
f"{BASE_URL}/tools/deepsearch/{uid}", headers=_json_headers()
)
assert r.status_code == 404
finally:
get_table("jobs").delete(kind="other")