Commit 0d6acef7 authored by Julien Muchembled's avatar Julien Muchembled

Optimize pickles of rotated buckets in WorkflowHistoryList

This often divides the size by ~3,
and also memory usage if it is loaded again.
parent 7a1763a9
......@@ -27,9 +27,11 @@
#
##############################################################################
from six.moves import map
from App.special_dtml import HTMLFile
from Acquisition import aq_inner
from AccessControl.requestmethod import postonly
from DateTime import DateTime
from Products.DCWorkflow.DCWorkflow import DCWorkflowDefinition
from . import Permissions
from .ConflictFree import DoublyLinkList
......@@ -235,3 +237,44 @@ class WorkflowHistoryList(DoublyLinkList):
else:
self.__class__ = WorkflowHistoryList
super(WorkflowHistoryList, self).__setstate__(state)
def __getstate__(self):
bucket = self._next
if getattr(bucket, '_p_changed', None) is not None and \
self._tail_count < bucket._tail_count:
# This bucket won't be modified anymore so optimize this last change,
# which is even more useful that there are often very few items added
# within a transaction and many identical strings are rarely deduplicated.
self._log = dedupStrings(self._log)
return super(WorkflowHistoryList, self).__getstate__()
def dedupStrings(obj):
_sequence_type_list = frozenset, list, set, tuple
_by_id = {} # usually useless for workflow history lists
# BBB: With Python 3, a single dict will be enough and it will even be
# possible to dedup hashable tuples/frozensets (workflow history
# lists usually don't contain these 2 types).
_bytes = {}
_str = {}
def dedup(obj):
id_ = id(obj)
try:
return _by_id[id_]
except KeyError:
t = getattr(obj, '__class__', None)
if t is DateTime:
obj._tz = dedup(obj._tz)
elif t is bytes:
obj = _bytes.setdefault(obj, obj)
elif t is unicode:
obj = _str.setdefault(obj, obj)
elif t in _sequence_type_list:
obj = t(map(dedup, obj))
elif t is dict:
obj = {dedup(k): dedup(v) for k, v in obj.iteritems()}
else:
return obj
_by_id[id_] = obj
return obj
return dedup(obj)
......@@ -33,6 +33,7 @@ from Products.ERP5Type.ConflictFree import DoublyLinkList
from Products.ERP5Type.Workflow import WorkflowHistoryList
from Products.ERP5Type.patches.WorkflowTool import \
WorkflowHistoryList as LegacyWorkflowHistoryList
from Products.ERP5Type.tests.ERP5TypeTestCase import ERP5TypeTestCase
orig_maybe_rotate = DoublyLinkList._maybe_rotate.__func__
......@@ -191,3 +192,32 @@ class TestWorkflowHistoryList(TestCase):
self.assertEqual(6, ddl._tail_count)
self.assertEqual(3, ddl._prev._tail_count)
self.assertEqual(0, ddl._prev._prev._tail_count)
class TestDedup(ERP5TypeTestCase):
def getBusinessTemplateList(self):
return ('erp5_base',)
def test(self):
self.login()
deduped = []
def dedupStrings(obj):
new = orig_dedupStrings(obj)
self.assertEqual(new, obj)
deduped.append(len(new))
return new
from Products.ERP5Type import Workflow
orig_dedupStrings = Workflow.dedupStrings
try:
Workflow.dedupStrings = dedupStrings
ob = self.portal.person_module.newContent()
whl = ob.workflow_history["edit_workflow"]
while whl._prev is None:
ob.edit()
self.commit()
whl._p_deactivate()
finally:
Workflow.dedupStrings = orig_dedupStrings
self.assertEqual(deduped, [24])
self.assertEqual(len(list(whl)), 25)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment