zzstoatzz.io
prefect server in zig
make benchmark tests stateless by resetting postgres schema
reset postgres schema before each benchmark run to ensure clean state,
matching the sqlite behavior of deleting the db file.
🤖 Generated with [Claude Code](https://claude.com/claude-code)
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
+1
-1
loq.toml
+1
-1
loq.toml
+19
scripts/benchmark
+19
scripts/benchmark
···
165
165
return self._start_zig()
166
166
return self._start_python()
167
167
168
+
def _reset_postgres(self) -> None:
169
+
"""Reset postgres schema for clean test state."""
170
+
try:
171
+
subprocess.run(
172
+
[
173
+
"docker", "compose", "exec", "-T", "postgres",
174
+
"psql", "-U", "prefect", "-c",
175
+
"DROP SCHEMA public CASCADE; CREATE SCHEMA public; GRANT ALL ON SCHEMA public TO prefect;",
176
+
],
177
+
cwd=self.project_root,
178
+
capture_output=True,
179
+
check=True,
180
+
)
181
+
except subprocess.CalledProcessError:
182
+
pass # postgres might not be running yet, that's ok
183
+
168
184
def _start_zig(self) -> bool:
169
185
binary = self.project_root / "zig-out" / "bin" / "prefect-server"
170
186
if not binary.exists():
···
175
191
db_path = self.project_root / "prefect.db"
176
192
if db_path.exists():
177
193
db_path.unlink()
194
+
elif self.db_backend == "postgres":
195
+
# reset postgres schema for clean state
196
+
self._reset_postgres()
178
197
179
198
env = os.environ.copy()
180
199
env["PREFECT_SERVER_LOGGING_LEVEL"] = "WARNING"