Merge pull request #656 from matrix-org/erikj/get_event

Dedupe requested event list in _get_events
matrix-org · Mar 18, 2016 · 7f79a64 · 7f79a64
2 parents 916227b + 8595ff7
commit 7f79a64
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 15 deletions.
diff --git a/synapse/events/__init__.py b/synapse/events/__init__.py
@@ -168,5 +168,7 @@ def __str__(self):
 
     def __repr__(self):
         return "<FrozenEvent event_id='%s', type='%s', state_key='%s'>" % (
-            self.event_id, self.type, self.get("state_key", None),
+            self.get("event_id", None),
+            self.get("type", None),
+            self.get("state_key", None),
         )
diff --git a/synapse/storage/events.py b/synapse/storage/events.py
@@ -526,6 +526,9 @@ def _get_events(self, event_ids, check_redacted=True,
         if not event_ids:
             defer.returnValue([])
 
+        event_id_list = event_ids
+        event_ids = set(event_ids)
+
         event_map = self._get_events_from_cache(
             event_ids,
             check_redacted=check_redacted,
@@ -535,23 +538,18 @@ def _get_events(self, event_ids, check_redacted=True,
 
         missing_events_ids = [e for e in event_ids if e not in event_map]
 
-        if not missing_events_ids:
-            defer.returnValue([
-                event_map[e_id] for e_id in event_ids
-                if e_id in event_map and event_map[e_id]
-            ])
-
-        missing_events = yield self._enqueue_events(
-            missing_events_ids,
-            check_redacted=check_redacted,
-            get_prev_content=get_prev_content,
-            allow_rejected=allow_rejected,
-        )
+        if missing_events_ids:
+            missing_events = yield self._enqueue_events(
+                missing_events_ids,
+                check_redacted=check_redacted,
+                get_prev_content=get_prev_content,
+                allow_rejected=allow_rejected,
+            )
 
-        event_map.update(missing_events)
+            event_map.update(missing_events)
 
         defer.returnValue([
-            event_map[e_id] for e_id in event_ids
+            event_map[e_id] for e_id in event_id_list
             if e_id in event_map and event_map[e_id]
         ])