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
return self._start_zig()
166
return self._start_python()
167
168
def _start_zig(self) -> bool:
169
binary = self.project_root / "zig-out" / "bin" / "prefect-server"
170
if not binary.exists():
···
175
db_path = self.project_root / "prefect.db"
176
if db_path.exists():
177
db_path.unlink()
178
179
env = os.environ.copy()
180
env["PREFECT_SERVER_LOGGING_LEVEL"] = "WARNING"
···
165
return self._start_zig()
166
return self._start_python()
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
+
184
def _start_zig(self) -> bool:
185
binary = self.project_root / "zig-out" / "bin" / "prefect-server"
186
if not binary.exists():
···
191
db_path = self.project_root / "prefect.db"
192
if db_path.exists():
193
db_path.unlink()
194
+
elif self.db_backend == "postgres":
195
+
# reset postgres schema for clean state
196
+
self._reset_postgres()
197
198
env = os.environ.copy()
199
env["PREFECT_SERVER_LOGGING_LEVEL"] = "WARNING"