From 34adad8e80fd4566da4e3c1771489bb6bdaf7a62 Mon Sep 17 00:00:00 2001
From: Mathieu Velten <mathieuv@matrix.org>
Date: Fri, 12 May 2023 15:00:42 +0200
Subject: [PATCH] Add rooms_to_purge table and related methods

---
 synapse/storage/databases/main/roommember.py  | 78 +++++++++++++++++++
 .../main/delta/76/06_rooms_to_purge.sql       | 26 +++++++
 2 files changed, 104 insertions(+)
 create mode 100644 synapse/storage/schema/main/delta/76/06_rooms_to_purge.sql

diff --git a/synapse/storage/databases/main/roommember.py b/synapse/storage/databases/main/roommember.py
index e068f27a10..406ecd263a 100644
--- a/synapse/storage/databases/main/roommember.py
+++ b/synapse/storage/databases/main/roommember.py
@@ -17,6 +17,7 @@ from itertools import chain
 from typing import (
     TYPE_CHECKING,
     AbstractSet,
+    Any,
     Collection,
     Dict,
     FrozenSet,
@@ -1284,6 +1285,83 @@ class RoomMemberWorkerStore(EventsWorkerStore, CacheInvalidationWorkerStore):
         # If any rows still exist it means someone has not forgotten this room yet
         return not rows[0][0]
 
+    async def upsert_room_to_purge(
+        self,
+        room_id: str,
+        delete_id: str,
+        status: str,
+        error: Optional[str] = None,
+        timestamp: Optional[int] = None,
+        shutdown_params: Optional[str] = None,
+        shutdown_response: Optional[str] = None,
+    ) -> None:
+        """Insert or update a room to shutdown/purge.
+
+        Args:
+            room_id: The room ID to shutdown/purge
+            delete_id: The delete ID identifying this action
+            status: Current status of the delete. Cf `DeleteStatus` for possible values
+            error: Error message to return, if any
+            timestamp: Time of the last update. If status is `wait_purge`,
+                then it specifies when to do the purge, with an empty value specifying ASAP
+            shutdown_params: JSON representation of shutdown parameters, cf `ShutdownRoomParams`
+            shutdown_response: JSON representation of shutdown current status, cf `ShutdownRoomResponse`
+        """
+        await self.db_pool.simple_upsert(
+            "rooms_to_purge",
+            {
+                "room_id": room_id,
+                "delete_id": delete_id,
+            },
+            {
+                "room_id": room_id,
+                "delete_id": delete_id,
+                "status": status,
+                "error": error,
+                "timestamp": timestamp,
+                "shutdown_params": shutdown_params,
+                "shutdown_response": shutdown_response,
+            },
+            desc="upsert_room_to_purge",
+        )
+
+    async def delete_room_to_purge(self, room_id: str, delete_id: str) -> None:
+        """Remove a room from the list of rooms to purge.
+
+        Args:
+            room_id: The room ID matching the delete to remove
+            delete_id: The delete ID identifying the delete to remove
+        """
+
+        await self.db_pool.simple_delete(
+            "rooms_to_purge",
+            keyvalues={
+                "room_id": room_id,
+                "delete_id": delete_id,
+            },
+            desc="delete_room_to_purge",
+        )
+
+    async def get_rooms_to_purge(self) -> List[Dict[str, Any]]:
+        """Returns all rooms to shutdown/purge. This includes those that has
+        been interrupted by a stop/restart of synapse, but also scheduled ones
+        like locally forgotten rooms.
+        """
+        return await self.db_pool.simple_select_list(
+            table="rooms_to_purge",
+            keyvalues={},
+            retcols=(
+                "room_id",
+                "delete_id",
+                "timestamp",
+                "status",
+                "error",
+                "shutdown_params",
+                "shutdown_response",
+            ),
+            desc="rooms_to_purge_fetch",
+        )
+
     async def get_rooms_user_has_been_in(self, user_id: str) -> Set[str]:
         """Get all rooms that the user has ever been in.
 
diff --git a/synapse/storage/schema/main/delta/76/06_rooms_to_purge.sql b/synapse/storage/schema/main/delta/76/06_rooms_to_purge.sql
new file mode 100644
index 0000000000..2b5871e675
--- /dev/null
+++ b/synapse/storage/schema/main/delta/76/06_rooms_to_purge.sql
@@ -0,0 +1,26 @@
+/* Copyright 2023 The Matrix.org Foundation C.I.C
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+-- cf upsert_room_to_purge docstring for the meaning of the fields.
+CREATE TABLE IF NOT EXISTS rooms_to_purge(
+    room_id text NOT NULL,
+    delete_id text NOT NULL,
+    "status" text NOT NULL,
+    "error" text,
+    "timestamp" bigint,
+    "shutdown_params" text,
+    "shutdown_response" text,
+    UNIQUE(room_id, delete_id)
+);