From a7032abb2e64f79be5823b770230cb223cc22ff1 Mon Sep 17 00:00:00 2001
From: Erik Johnston <erik@matrix.org>
Date: Mon, 5 Sep 2016 15:07:23 +0100
Subject: [PATCH] Correctly handle reindexing state groups that already have an
 edge

---
 synapse/storage/state.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/synapse/storage/state.py b/synapse/storage/state.py
index 589a4fec6e..af3ddd962d 100644
--- a/synapse/storage/state.py
+++ b/synapse/storage/state.py
@@ -673,6 +673,17 @@ class StateStore(SQLBaseStore):
                 if not row or not state_group:
                     return True, count
 
+                txn.execute(
+                    "SELECT state_group FROM state_group_edges"
+                    " WHERE state_group = ?",
+                    (state_group,)
+                )
+
+                # If we reach a point where we've already started inserting
+                # edges we should stop.
+                if txn.fetchall():
+                    return True, count
+
                 txn.execute(
                     "SELECT coalesce(max(id), 0) FROM state_groups"
                     " WHERE id < ? AND room_id = ?",
@@ -709,6 +720,14 @@ class StateStore(SQLBaseStore):
                             if prev_state.get(key, None) != value
                         }
 
+                        self._simple_delete_txn(
+                            txn,
+                            table="state_group_edges",
+                            keyvalues={
+                                "state_group": state_group,
+                            }
+                        )
+
                         self._simple_insert_txn(
                             txn,
                             table="state_group_edges",