RAMDict.py 5.25 KB
Newer Older
Jean-Paul Smets's avatar
Jean-Paul Smets committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
##############################################################################
#
# Copyright (c) 2002 Nexedi SARL and Contributors. All Rights Reserved.
#                    Jean-Paul Smets-Solanes <jp@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

from Products.CMFActivity.ActivityTool import registerActivity
from Queue import Queue
31
from Products.CMFActivity.ActiveObject import DISTRIBUTABLE_STATE, INVOKE_ERROR_STATE, VALIDATE_ERROR_STATE
Jean-Paul Smets's avatar
Jean-Paul Smets committed
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47

from zLOG import LOG

class RAMDict(Queue):
  """
    A simple RAM based queue. It is not compatible with transactions which
    means methods can be called before an object even exists or before
    it is modified. This also means there is no garantee on any kind of sequenciality.

    Dictionnary is global.
  """

  def __init__(self):
    Queue.__init__(self)
    self.dict = {}

48
  def finishQueueMessage(self, activity_tool, m):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
49 50
    if m.is_registered:
      self.dict[(m.object_path, m.method_id)] = m
Jean-Paul Smets's avatar
Jean-Paul Smets committed
51

52 53 54 55 56
  def finishDeleteMessage(self, activity_tool, message):
    for key, m in self.dict.items():
      if m.object_path == message.object_path and m.method_id == message.method_id:
          del self.dict[(m.object_path, m.method_id)]

Jean-Paul Smets's avatar
Jean-Paul Smets committed
57 58
  def registerActivityBuffer(self, activity_buffer):
    class_name = self.__class__.__name__
Jean-Paul Smets's avatar
Jean-Paul Smets committed
59 60
    setattr(activity_buffer, '_%s_message_list' % class_name, [])  
    setattr(activity_buffer, '_%s_uid_dict' % class_name, {})  
Jean-Paul Smets's avatar
Jean-Paul Smets committed
61 62 63 64 65 66 67 68 69 70 71
            
  def isMessageRegistered(self, activity_buffer, activity_tool, m):
    class_name = self.__class__.__name__
    return getattr(activity_buffer, '_%s_uid_dict' % class_name).has_key((m.object_path, m.method_id))
                                   
  def registerMessage(self, activity_buffer, activity_tool, m):
    class_name = self.__class__.__name__
    getattr(activity_buffer, '_%s_message_list' % class_name).append(m)
    getattr(activity_buffer, '_%s_uid_dict' % class_name)[(m.object_path, m.method_id)] = 1
    m.is_registered = 1
          
Jean-Paul Smets's avatar
Jean-Paul Smets committed
72
  def dequeueMessage(self, activity_tool, processing_node):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
73 74 75 76 77 78 79 80 81
    if len(self.dict.keys()) is 0:
      return 1  # Go to sleep
    for key, m in self.dict.items():
      if m.validate(self, activity_tool):
        activity_tool.invoke(m)
        del self.dict[key]
        return 0
    return 1

82
  def hasActivity(self, activity_tool, object, **kw):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
83 84 85 86 87 88
    object_path = object.getPhysicalPath()
    for m in self.dict.values():
      if m.object_path == object_path:
        return 1
    return 0

Jean-Paul Smets's avatar
Jean-Paul Smets committed
89 90 91 92 93 94 95
  def flush(self, activity_tool, object_path, invoke=0, method_id=None, **kw):    
    path = '/'.join(object_path)
    # LOG('Flush', 0, str((path, invoke, method_id)))
    method_dict = {}
    # Parse each message in registered
    for m in activity_tool.getRegisteredMessageList(self):
      if object_path == m.object_path and (method_id is None or method_id == m.method_id):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
96
        activity_tool.unregisterMessage(self, m)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
97 98 99 100 101 102 103 104 105 106 107 108 109 110
        if not method_dict.has_key(method_id):
          if invoke:
            # First Validate
            if m.validate(self, activity_tool):
              activity_tool.invoke(m) # Try to invoke the message - what happens if invoke calls flushActivity ??
              if not m.is_executed:                                                 # Make sure message could be invoked
                # The message no longer exists
                raise ActivityFlushError, (
                    'Could not evaluate %s on %s' % (method_id , path))
            else:
              # The message no longer exists
              raise ActivityFlushError, (
                  'The document %s does not exist' % path)               
    # Parse each message in RAM dict
Jean-Paul Smets's avatar
Jean-Paul Smets committed
111
    for key, m in self.dict.items():
112
      if not m.is_deleted:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
113
        if object_path == m.object_path and (method_id is None or method_id == m.method_id):
114 115
          LOG('CMFActivity RAMDict: ', 0, 'flushing object %s' % '/'.join(m.object_path))
          if invoke: activity_tool.invoke(m)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
116
          self.deleteMessage(activity_tool, m)
117 118 119
        else:
          pass
          #LOG('CMFActivity RAMDict: ', 0, 'not flushing object %s' % '/'.join(m.object_path))
Jean-Paul Smets's avatar
Jean-Paul Smets committed
120

Jean-Paul Smets's avatar
Jean-Paul Smets committed
121
  def getMessageList(self, activity_tool, processing_node=None):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
122 123 124
    return self.dict.values()

registerActivity(RAMDict)