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

zerver/migrations/0705_stream_subscriber_count_data_migration.py

@@ -0,0 +1,56 @@
+from django.db import connection, migrations, transaction
+from django.db.backends.base.schema import BaseDatabaseSchemaEditor
+from django.db.migrations.state import StateApps
+from psycopg2 import sql
+
+
+def set_stream_subscribe_count(apps: StateApps, schema_editor: BaseDatabaseSchemaEditor) -> None:
+    Realm = apps.get_model("zerver", "realm")
+    Stream = apps.get_model("zerver", "Stream")
+    Subscription = apps.get_model("zerver", "subscription")
+    Recipient = apps.get_model("zerver", "recipient")
+
+    # Here we compute per-stream subscriber_count in the sub-query
+    # which returns a stream_subscribers_table, then Update it
+    # in-place using that computed value. The whole query is then
+    # executed by one batch per realm.
+    query = sql.SQL(
+        """UPDATE {stream_table} AS stream
+            SET subscriber_count = stream_subscribers_table.count
+            FROM (
+                SELECT recipient.type_id AS stream_id,
+                COUNT(subscription.user_profile_id) AS count
+                FROM {subscription_table} AS subscription
+                JOIN {recipient_table} AS recipient
+                    ON subscription.recipient_id = recipient.id
+                WHERE
+                    recipient.type = 2
+                    AND subscription.active = True
+                    AND subscription.is_user_active = True
+                GROUP BY stream_id
+            ) AS stream_subscribers_table
+            WHERE
+                stream.realm_id = %(realm_id)s
+                AND stream.id = stream_subscribers_table.stream_id;
+        """
+    ).format(
+        stream_table=sql.Identifier(Stream._meta.db_table),
+        subscription_table=sql.Identifier(Subscription._meta.db_table),
+        recipient_table=sql.Identifier(Recipient._meta.db_table),
+    )
+
+    for realm in Realm.objects.all().iterator():
+        with connection.cursor() as cursor, transaction.atomic(durable=True):
+            cursor.execute(query, {"realm_id": realm.id})
+
+
+class Migration(migrations.Migration):
+    atomic = False
+
+    dependencies = [
+        ("zerver", "0704_stream_subscriber_count"),
+    ]
+
+    operations = [
+        migrations.RunPython(set_stream_subscribe_count),
+    ]