snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

pgroonga/migrations/0002_html_escape_subject.py

@@ -0,0 +1,28 @@
+from django.db import connection, migrations
+from django.db.backends.base.schema import BaseDatabaseSchemaEditor
+from django.db.migrations.state import StateApps
+from psycopg2.sql import SQL
+
+from zerver.lib.migrate import do_batch_update
+
+
+def rebuild_pgroonga_index(apps: StateApps, schema_editor: BaseDatabaseSchemaEditor) -> None:
+    with connection.cursor() as cursor:
+        do_batch_update(
+            cursor,
+            "zerver_message",
+            [SQL("search_pgroonga = escape_html(subject) || ' ' || rendered_content")],
+            batch_size=10000,
+        )
+
+
+class Migration(migrations.Migration):
+    atomic = False
+
+    dependencies = [
+        ("pgroonga", "0001_enable"),
+    ]
+
+    operations = [
+        migrations.RunPython(rebuild_pgroonga_index, reverse_code=migrations.RunPython.noop),
+    ]