[Checkins] SVN: ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/ notify activity monitors at the end of Connection.close instead of in DB._returnToPool, so that it also happens for subconnections. based on r81414

[David Glick]

Log message for revision 120934:
  notify activity monitors at the end of Connection.close instead of in DB._returnToPool, so that it also happens for subconnections. based on r81414

Changed:
  U   ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/ActivityMonitor.py
  U   ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/Connection.py
  U   ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/DB.py
  U   ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/tests/testConnection.py

-=-
Modified: ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/ActivityMonitor.py
===================================================================
--- ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/ActivityMonitor.py	2011-03-15 03:08:49 UTC (rev 120933)
+++ ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/ActivityMonitor.py	2011-03-15 03:10:43 UTC (rev 120934)
@@ -15,6 +15,7 @@
 
 $Id$"""
 
+import threading
 import time
 
 
@@ -24,14 +25,13 @@
     This simple implementation just keeps a small log in memory
     and iterates over the log when getActivityAnalysis() is called.
 
-    It assumes that log entries are added in chronological sequence,
-    which is only guaranteed because DB.py holds a lock when calling
-    the closedConnection() method.
+    It assumes that log entries are added in chronological sequence.
     """
 
     def __init__(self, history_length=3600):
         self.history_length = history_length  # Number of seconds
         self.log = []                     # [(time, loads, stores)]
+        self.trim_lock = threading.Lock()
 
     def closedConnection(self, conn):
         log = self.log
@@ -41,6 +41,8 @@
         self.trim(now)
 
     def trim(self, now):
+        self.trim_lock.acquire()
+        
         log = self.log
         cutoff = now - self.history_length
         n = 0
@@ -49,6 +51,8 @@
             n = n + 1
         if n:
             del log[:n]
+        
+        self.trim_lock.release()
 
     def setHistoryLength(self, history_length):
         self.history_length = history_length

Modified: ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/Connection.py
===================================================================
--- ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/Connection.py	2011-03-15 03:08:49 UTC (rev 120933)
+++ ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/Connection.py	2011-03-15 03:10:43 UTC (rev 120934)
@@ -322,6 +322,10 @@
                 # get back here.
         else:
             self.opened = None
+            
+        am = self._db._activity_monitor
+        if am is not None:
+            am.closedConnection(self)        
 
     def db(self):
         """Returns a handle to the database this connection belongs to."""

Modified: ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/DB.py
===================================================================
--- ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/DB.py	2011-03-15 03:08:49 UTC (rev 120933)
+++ ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/DB.py	2011-03-15 03:10:43 UTC (rev 120934)
@@ -495,10 +495,6 @@
             assert connection._db is self
             connection.opened = None
 
-            am = self._activity_monitor
-            if am is not None:
-                am.closedConnection(connection)
-
             if connection.before:
                 self.historical_pool.repush(connection, connection.before)
             else:

Modified: ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/tests/testConnection.py
===================================================================
--- ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/tests/testConnection.py	2011-03-15 03:08:49 UTC (rev 120933)
+++ ZODB/branches/davisagli-subconn-activity-monitors/src/ZODB/tests/testConnection.py	2011-03-15 03:10:43 UTC (rev 120934)
@@ -327,6 +327,39 @@
         >>> cn._Connection__onCloseCallbacks
         """
 
+    def test_close_dispatches_to_activity_monitors(self):
+        r"""doctest that connection close updates activity monitors
+        
+        Set up a multi-database:
+        
+            >>> db1 = ZODB.DB(None)
+            >>> db2 = ZODB.DB(None, databases=db1.databases, database_name='2',
+            ...               cache_size=10)
+            >>> conn1 = db1.open()
+            >>> conn2 = conn1.get_connection('2')
+        
+        Add activity monitors to both dbs:
+        
+            >>> from ZODB.ActivityMonitor import ActivityMonitor
+            >>> db1.setActivityMonitor(ActivityMonitor())
+            >>> db2.setActivityMonitor(ActivityMonitor())
+        
+        Commit a transaction that affects both connections:
+        
+            >>> conn1.root()[0] = conn1.root().__class__()
+            >>> conn2.root()[0] = conn2.root().__class__()
+            >>> transaction.commit()
+
+        After closing the primary connection, both monitors should be up to
+        date:
+        
+            >>> conn1.close()
+            >>> len(db1.getActivityMonitor().log)
+            1
+            >>> len(db2.getActivityMonitor().log)
+            1
+        """
+
     def test_db(self):
         r"""doctest of db() method