Limit the max number of grouped objects.

git-svn-id: https://svn.erp5.org/repos/public/erp5/trunk@3856 20353a03-c40f-0410-a6d1-a30d3c3de9de

product/CMFActivity/Activity/SQLDict.py

@@ -32,10 +32,12 @@ from Products.CMFActivity.ActivityTool import registerActivity
 from Queue import VALID, INVALID_ORDER, INVALID_PATH, EXCEPTION, MAX_PROCESSING_TIME, VALIDATION_ERROR_DELAY, SECONDS_IN_DAY
 from RAMDict import RAMDict
 from Products.CMFActivity.ActiveObject import DISTRIBUTABLE_STATE, INVOKE_ERROR_STATE, VALIDATE_ERROR_STATE
+from ZODB.POSException import ConflictError
 
 from zLOG import LOG
 
 MAX_PRIORITY = 5
+MAX_GROUPED_OBJECTS = 300
 
 priority_weight = \
   [1] * 64 + \
@@ -191,7 +193,21 @@ class SQLDict(RAMDict):
         if self.validateMessage(activity_tool, m, uid_list, line.priority, next_processing_date):
           group_method_id = m.activity_kw.get('group_method_id')
           if group_method_id is not None:
+            # Count the number of objects to prevent too many objects.
+            if m.hasExpandMethod():
+              try:
+                count = len(m.getObjectList())
+              except ConflictError:
+                raise
+              except:
+                # Here, simply ignore an exception. The same exception should be handled later.
+                count = 0
+            else:
+              count = 1
+
             group_method = activity_tool.restrictedTraverse(group_method_id)
+            
+            if count < MAX_GROUPED_OBJECTS:
               # Retrieve objects which have the same group method.
               result = activity_tool.SQLDict_readMessage(processing_node=processing_node, priority=priority,
                                                         to_date=now_date, group_method_id=group_method_id)
@@ -207,9 +223,21 @@ class SQLDict(RAMDict):
                 get_transaction().commit() # Release locks before starting a potentially long calculation
                 m = self.loadMessage(line.message, uid = line.uid)
                 if self.validateMessage(activity_tool, m, uid_list, line.priority, next_processing_date):
+                  if m.hasExpandMethod():
+                    try:
+                      count += len(m.getObjectList())
+                    except ConflictError:
+                      raise
+                    except:
+                      # Here, simply ignore an exception. The same exception should be handled later.
+                      pass
+                  else:
+                    count += 1
                   message_list.append(m)
                   uid_list_list.append(uid_list)
                   priority_list.append(line.priority)
+                  if count >= MAX_GROUPED_OBJECTS:
+                    break
                 
           get_transaction().commit() # Release locks before starting a potentially long calculation
           # Try to invoke

product/CMFActivity/ActivityTool.py

@@ -85,20 +85,23 @@ class Message:
     self.kw = kw
     self.is_executed = 0
     self.user_name = str(_getAuthenticatedUser(self))
+    self.object_list = None
     # Store REQUEST Info ?
 
   def getObject(self, activity_tool):
     return activity_tool.unrestrictedTraverse(self.object_path)
     
   def getObjectList(self, activity_tool):
+    if self.object_list is not None:
       try:
         expand_method_id = self.activity_kw['expand_method_id']
-    except KeyError:
-      return [self.getObject()]
-      
         obj = self.getObject(activity_tool)
         # FIXME: how to pass parameters?
-    return getattr(obj, expand_method_id)()
+        self.object_list = getattr(obj, expand_method_id)()
+      except KeyError:
+        self.object_list = [self.getObject(activity_tool)]
+      
+    return self.object_list
       
   def hasExpandMethod(self):
     return self.activity_kw.has_key('expand_method_id')