#29: add catalog to last_update table (#396)

Author: Kraysent

app/data/repositories/layer2/repository.py

@@ -25,10 +25,13 @@ def __init__(self, storage: postgres.PgStorage, logger: structlog.stdlib.BoundLo
         self._storage = storage
 
     def get_last_update_time(self) -> datetime.datetime:
-        return self._storage.query_one("SELECT dt FROM layer2.last_update")["dt"]
+        return self._storage.query_one("SELECT dt FROM layer2.last_update WHERE catalog = %s", params=["all"])["dt"]
 
     def update_last_update_time(self, dt: datetime.datetime):
-        self._storage.exec("UPDATE layer2.last_update SET dt = %s", params=[dt])
+        self._storage.exec(
+            "UPDATE layer2.last_update SET dt = %s WHERE catalog = %s",
+            params=[dt, "all"],
+        )
 
     def get_orphaned_pgcs(self, catalogs: list[model.RawCatalog]) -> dict[str, list[int]]:
         result: dict[str, list[int]] = {}

postgres/migrations/V024__last_update_catalog.sql

@@ -0,0 +1,8 @@
+/* pgmigrate-encoding: utf-8 */
+ALTER TABLE layer2.last_update ADD COLUMN catalog text;
+
+UPDATE layer2.last_update SET catalog = 'all';
+
+ALTER TABLE layer2.last_update ALTER COLUMN catalog SET NOT NULL;
+
+ALTER TABLE layer2.last_update ADD PRIMARY KEY (catalog);

tests/lib/postgres.py

@@ -137,7 +137,7 @@ def clear(self):
         ):
             self.storage.exec(f"TRUNCATE {table['table_schema']}.{table['table_name']} CASCADE")
 
-        self.storage.exec("INSERT INTO layer2.last_update VALUES (to_timestamp(0))")
+        self.storage.exec("INSERT INTO layer2.last_update (dt, catalog) VALUES (to_timestamp(0), 'all')")
 
     def get_storage(self) -> postgres.PgStorage:
         return self.storage