mirror of
https://github.com/zulip/zulip.git
synced 2025-11-02 13:03:29 +00:00
Upgrade event flag collapsing to support all flag changes.
(imported from commit dadf3852f2ca7aaa85fb97b3d265c990b4e5a38a)
This commit is contained in:
@@ -147,13 +147,10 @@ def compute_full_event_type(event):
|
||||
if event["type"] == "update_message_flags":
|
||||
if event["all"]:
|
||||
# Put the "all" case in its own category
|
||||
return "%s/%s/all" % (event["flag"], event["operation"])
|
||||
return "%s/%s" % (event["flag"], event["operation"])
|
||||
return "all_flags/%s/%s" % (event["flag"], event["operation"])
|
||||
return "flags/%s/%s" % (event["operation"], event["flag"])
|
||||
return event["type"]
|
||||
|
||||
# Virtual events are a mechanism for storing pointer changes and other
|
||||
# easily collapsed event types efficiently.
|
||||
VIRTUAL_EVENT_TYPES = ["pointer", "read/add", "restart"]
|
||||
class EventQueue(object):
|
||||
def __init__(self, id):
|
||||
self.queue = deque()
|
||||
@@ -179,7 +176,8 @@ class EventQueue(object):
|
||||
event['id'] = self.next_event_id
|
||||
self.next_event_id += 1
|
||||
full_event_type = compute_full_event_type(event)
|
||||
if full_event_type in VIRTUAL_EVENT_TYPES:
|
||||
if (full_event_type in ["pointer", "restart"] or
|
||||
full_event_type.startswith("flags/")):
|
||||
if full_event_type not in self.virtual_events:
|
||||
self.virtual_events[full_event_type] = copy.deepcopy(event)
|
||||
return
|
||||
@@ -190,10 +188,10 @@ class EventQueue(object):
|
||||
virtual_event["timestamp"] = event["timestamp"]
|
||||
if full_event_type == "pointer":
|
||||
virtual_event["pointer"] = event["pointer"]
|
||||
elif full_event_type == "read/add":
|
||||
virtual_event["messages"] += event["messages"]
|
||||
elif full_event_type == "restart":
|
||||
virtual_event["server_generation"] = event["server_generation"]
|
||||
elif full_event_type.startswith("flags/"):
|
||||
virtual_event["messages"] += event["messages"]
|
||||
else:
|
||||
self.queue.append(event)
|
||||
|
||||
|
||||
@@ -2542,7 +2542,7 @@ class EventQueueTest(TestCase):
|
||||
"timestamp": "22"},
|
||||
])
|
||||
|
||||
def test_flag_collapsing(self):
|
||||
def test_flag_add_collapsing(self):
|
||||
queue = EventQueue("1")
|
||||
queue.push({"type": "update_message_flags",
|
||||
"flag": "read",
|
||||
@@ -2565,6 +2565,29 @@ class EventQueueTest(TestCase):
|
||||
"messages": [1,2,3,4,5,6],
|
||||
"timestamp": "1"}])
|
||||
|
||||
def test_flag_remove_collapsing(self):
|
||||
queue = EventQueue("1")
|
||||
queue.push({"type": "update_message_flags",
|
||||
"flag": "collapsed",
|
||||
"operation": "remove",
|
||||
"all": False,
|
||||
"messages": [1, 2, 3, 4],
|
||||
"timestamp": "1"})
|
||||
queue.push({"type": "update_message_flags",
|
||||
"flag": "collapsed",
|
||||
"all": False,
|
||||
"operation": "remove",
|
||||
"messages": [5, 6],
|
||||
"timestamp": "1"})
|
||||
self.assertEqual(queue.contents(),
|
||||
[{'id': 1,
|
||||
'type': 'update_message_flags',
|
||||
"all": False,
|
||||
"flag": "collapsed",
|
||||
"operation": "remove",
|
||||
"messages": [1,2,3,4,5,6],
|
||||
"timestamp": "1"}])
|
||||
|
||||
class GetProfileTest(AuthedTestCase):
|
||||
|
||||
def common_update_pointer(self, email, pointer):
|
||||
|
||||
Reference in New Issue
Block a user