Purge the room entry too

roomserver/storage/postgres/rooms_table.go

@@ -82,6 +82,9 @@ const bulkSelectRoomIDsSQL = "" +
 const bulkSelectRoomNIDsSQL = "" +
 	"SELECT room_nid FROM roomserver_rooms WHERE room_id = ANY($1)"
 
+const purgeRoomSQL = "" +
+	"DELETE FROM roomserver_rooms WHERE room_nid = $1"
+
 type roomStatements struct {
 	insertRoomNIDStmt                  *sql.Stmt
 	selectRoomNIDStmt                  *sql.Stmt
@@ -93,6 +96,7 @@ type roomStatements struct {
 	selectRoomIDsStmt                  *sql.Stmt
 	bulkSelectRoomIDsStmt              *sql.Stmt
 	bulkSelectRoomNIDsStmt             *sql.Stmt
+	purgeRoomStmt                      *sql.Stmt
 }
 
 func CreateRoomsTable(db *sql.DB) error {
@@ -114,6 +118,7 @@ func PrepareRoomsTable(db *sql.DB) (tables.Rooms, error) {
 		{&s.selectRoomIDsStmt, selectRoomIDsSQL},
 		{&s.bulkSelectRoomIDsStmt, bulkSelectRoomIDsSQL},
 		{&s.bulkSelectRoomNIDsStmt, bulkSelectRoomNIDsSQL},
+		{&s.purgeRoomStmt, purgeRoomSQL},
 	}.Prepare(db)
 }
 
@@ -288,6 +293,13 @@ func (s *roomStatements) BulkSelectRoomNIDs(ctx context.Context, txn *sql.Tx, ro
 	return roomNIDs, nil
 }
 
+func (s *roomStatements) PurgeRoom(
+	ctx context.Context, txn *sql.Tx, roomNID types.RoomNID,
+) error {
+	_, err := sqlutil.TxStmt(txn, s.purgeRoomStmt).ExecContext(ctx, roomNID)
+	return err
+}
+
 func roomNIDsAsArray(roomNIDs []types.RoomNID) pq.Int64Array {
 	nids := make([]int64, len(roomNIDs))
 	for i := range roomNIDs {

roomserver/storage/shared/storage.go

@@ -1390,6 +1390,9 @@ func (d *Database) PurgeRoom(ctx context.Context, roomID string) error {
 		if err := d.EventsTable.PurgeEvents(ctx, txn, roomNID); err != nil {
 			return fmt.Errorf("failed to purge events: %w", err)
 		}
+		if err := d.RoomsTable.PurgeRoom(ctx, txn, roomNID); err != nil {
+			return fmt.Errorf("failed to purge room: %w", err)
+		}
 		return nil
 	})
 }

roomserver/storage/sqlite3/rooms_table.go

@@ -309,3 +309,9 @@ func (s *roomStatements) BulkSelectRoomNIDs(ctx context.Context, txn *sql.Tx, ro
 	}
 	return roomNIDs, nil
 }
+
+func (s *roomStatements) PurgeRoom(
+	ctx context.Context, txn *sql.Tx, roomNID types.RoomNID,
+) error {
+	return fmt.Errorf("not implemented on SQLite")
+}

roomserver/storage/tables/interface.go

@@ -82,6 +82,7 @@ type Rooms interface {
 	SelectRoomIDsWithEvents(ctx context.Context, txn *sql.Tx) ([]string, error)
 	BulkSelectRoomIDs(ctx context.Context, txn *sql.Tx, roomNIDs []types.RoomNID) ([]string, error)
 	BulkSelectRoomNIDs(ctx context.Context, txn *sql.Tx, roomIDs []string) ([]types.RoomNID, error)
+	PurgeRoom(ctx context.Context, txn *sql.Tx, roomNID types.RoomNID) error
 }
 
 type StateSnapshot interface {