RAMQueue.py 4.77 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

class RAMQueue(Queue):
  """
    A simple RAM based queue
  """
  def __init__(self):
    Queue.__init__(self)
39
    self.queue_dict = {}
Jean-Paul Smets's avatar
Jean-Paul Smets committed
40 41
    self.last_uid = 0
    
42 43 44 45 46 47
  def getQueue(self, activity_tool):
    path = activity_tool.getPhysicalPath()
    if not self.queue_dict.has_key(path):
      self.queue_dict[path] = []
    return self.queue_dict[path]
    
48
  def finishQueueMessage(self, activity_tool, m):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
49 50 51 52
    if m.is_registered:
      # XXX - Some lock is required on this section
      self.last_uid = self.last_uid + 1
      m.uid = self.last_uid
53
      self.getQueue(activity_tool).append(m)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
54

55 56
  def finishDeleteMessage(self, activity_tool, m):
    i = 0
57 58
    queue = self.getQueue(activity_tool)
    for my_message in queue:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
59
      if my_message.uid == m.uid:
60
        del queue[i]
61 62
        return
      i = i + 1
63

Jean-Paul Smets's avatar
Jean-Paul Smets committed
64
  def dequeueMessage(self, activity_tool, processing_node):
Jean-Paul Smets's avatar
fixed  
Jean-Paul Smets committed
65 66 67
    for m in self.getQueue(activity_tool):
      if not m.validate(self, activity_tool):
        self.deleteMessage(activity_tool, m) # Trash messages which are not validated (no error handling)
68
        get_transaction().commit() # Start a new transaction
Jean-Paul Smets's avatar
fixed  
Jean-Paul Smets committed
69 70 71 72
        return 0    # Keep on ticking
      activity_tool.invoke(m)
      if m.is_executed:
        self.deleteMessage(activity_tool, m) # Trash messages which are not validated (no error handling)
73
        get_transaction().commit() # Start a new transaction
Jean-Paul Smets's avatar
fixed  
Jean-Paul Smets committed
74 75 76
        return 0    # Keep on ticking         
      else:
        # Start a new transaction and keep on to next message
77 78 79
        get_transaction().commit()
    return 1 # Go to sleep

Jean-Paul Smets's avatar
Jean-Paul Smets committed
80

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

Sebastien Robin's avatar
Sebastien Robin committed
88
  def flush(self, activity_tool, object_path, invoke=0, method_id=None, **kw):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
89 90
    # Parse each message in registered
    for m in activity_tool.getRegisteredMessageList(self):
91
      if object_path == m.object_path and (method_id is None or method_id == m.method_id):
92 93 94 95 96 97 98 99 100
        if not m.validate(self, activity_tool):
          activity_tool.unregisterMessage(self, m) # Trash messages which are not validated (no error handling)
        else:            
          if invoke:
            activity_tool.invoke(m)
            if m.is_executed:
              activity_tool.unregisterMessage(self, m)
          else:              
            activity_tool.unregisterMessage(self, m)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
101
    # Parse each message in queue
102 103
    for m in self.getQueue(activity_tool):
      if object_path == m.object_path and (method_id is None or method_id == m.method_id):
104
        if not m.validate(self, activity_tool):
105
          self.deleteMessage(activity_tool, m) # Trash messages which are not validated (no error handling)
106 107 108 109
        else:            
          if invoke:
            activity_tool.invoke(m)
            if m.is_executed:
110
              self.deleteMessage(activity_tool, m) # Only delete if no error happens
111
          else:              
112
            self.deleteMessage(activity_tool, m)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
113

Jean-Paul Smets's avatar
Jean-Paul Smets committed
114
  def getMessageList(self, activity_tool, processing_node=None):
115
    new_queue = []
116
    for m in self.getQueue(activity_tool):
117 118 119 120
      m.processing_node = 1
      m.priority = 0
      new_queue.append(m)
    return new_queue
Jean-Paul Smets's avatar
Jean-Paul Smets committed
121 122

registerActivity(RAMQueue)