fix: anchor partition aging to server date for v0.4.2

Author: O2eg

pyproject.toml

@@ -4,14 +4,14 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pg-workload"
-version = "0.4.1"
+version = "0.4.2"
 description = "PostgreSQL workload emulator for diagnostic and observability testing"
 readme = "README.md"
 requires-python = ">=3.10"
 license = "MIT"
 license-files = ["LICENSE", "THIRD_PARTY_NOTICES.md"]
 authors = [
-  { name = "O2eg", email = "oleg.ispu@yandex.ru" }
+  { name = "O2eg", email = "oleg.ispu@gmail.com" }
 ]
 keywords = ["postgresql", "workload", "emulation", "diagnostics", "observability"]
 dependencies = [

src/pg_workload/__init__.py

@@ -1,6 +1,6 @@
 """PostgreSQL workload generation and scheduling tools."""
 
-__version__ = "0.4.1"
+__version__ = "0.4.2"
 
 from pg_workload.assets import bundled_profiles_root, initialize_project
 from pg_workload.cli import build_parser

src/pg_workload/bundled/profiles/partition_aging/generator.py

@@ -4,7 +4,6 @@
 import math
 import os
 import subprocess
-from datetime import date, timedelta
 
 DAYS_BACK = 30
 DAYS_AHEAD = 2
@@ -23,20 +22,25 @@ def main() -> None:
 
     row_count = scaled(200_000, args.scale, 2_000)
 
-    today = date.today()
-    partitions = []
-    for offset in range(-DAYS_BACK, DAYS_AHEAD + 1):
-        day = today + timedelta(days=offset)
-        nxt = day + timedelta(days=1)
-        partitions.append(
-            f"CREATE TABLE partition_aging.events_{day:%Y_%m_%d} "
-            "PARTITION OF partition_aging.events "
-            f"FOR VALUES FROM ('{day.isoformat()}') TO ('{nxt.isoformat()}');"
-        )
-
-    sql = (
-        "\n".join(partitions)
-        + f"""
+    partition_count = DAYS_BACK + DAYS_AHEAD + 1
+    sql = f"""
+        DO $partition_setup$
+        DECLARE
+            partition_day date;
+        BEGIN
+            FOR day_offset IN -{DAYS_BACK}..{DAYS_AHEAD} LOOP
+                partition_day := current_date + day_offset;
+                EXECUTE format(
+                    'CREATE TABLE partition_aging.%I '
+                    'PARTITION OF partition_aging.events '
+                    'FOR VALUES FROM (%L) TO (%L)',
+                    'events_' || to_char(partition_day, 'YYYY_MM_DD'),
+                    partition_day,
+                    partition_day + 1
+                );
+            END LOOP;
+        END
+        $partition_setup$;
 
         SELECT setseed(0.55083122);
 
@@ -49,15 +53,14 @@ def main() -> None:
             'event ' || g || ' session=' || floor(power(random(), 2) * 100000)::integer
         FROM generate_series(1, {row_count}) AS g;
     """
-    )
     subprocess.run(
         [os.environ.get("PG_WORKLOAD_PSQL", "psql"), "-X", "-q", "-v", "ON_ERROR_STOP=1"],
         input=sql,
         text=True,
         check=True,
     )
 
-    print(f"Generated partition_aging: partitions={len(partitions)}, events={row_count}")
+    print(f"Generated partition_aging: partitions={partition_count}, events={row_count}")
 
 
 if __name__ == "__main__":

tests/test_workload.py

@@ -48,6 +48,11 @@ def test_all_profile_manifests_are_valid(self):
             self.assertEqual(errors, [])
             imdb_generator = Path(tmpdir) / "data" / "imdb" / "generator.py"
             self.assertIn("g::bigint * 104729", imdb_generator.read_text(encoding="utf-8"))
+            partition_generator = (Path(tmpdir) / "data" / "partition_aging" / "generator.py").read_text(
+                encoding="utf-8"
+            )
+            self.assertIn("partition_day := current_date + day_offset", partition_generator)
+            self.assertNotIn("date.today()", partition_generator)
 
     def test_default_pg_major_is_18(self):
         parser = self.workload.build_parser()