Changeset 56801 in vbox for trunk/src/VBox/ValidationKit/testmanager

Timestamp:

Jul 3, 2015 11:10:57 PM (10 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

101451

Message:

testmanager/db.py: Added insertList methods for optimized multi row insertion. Added debugDisableExplain and debugIsExplainEnabled methods for avoid deadlocks in the schulding queue regenration code.

File:

• trunk/src/VBox/ValidationKit/testmanager/core/db.py (modified) (4 diffs)

trunk/src/VBox/ValidationKit/testmanager/core/db.py

@@ -119 +119 @@
         """ See TMDatabaseConnection.callProc()"""
         return self._oDb.callProcInternal(self._oCursor, sProcedure, aoArgs, utils.getCallerName());
+
+    def insertList(self, sInsertSql, aoList, fnEntryFmt):
+        """ See TMDatabaseConnection.insertList. """
+        return self._oDb.insertListInternal(self._oCursor, sInsertSql, aoList, fnEntryFmt, utils.getCallerName());
 
     def fetchOne(self):
@@ -384 +388 @@
         return oRc;
 
+    def insertListInternal(self, oCursor, sInsertSql, aoList, fnEntryFmt, sCallerName):
+        """
+        Optimizes the insertion of a list of values.
+        """
+        oRc = None;
+        asValues = [];
+        for aoEntry in aoList:
+            asValues.append(fnEntryFmt(aoEntry));
+            if len(asValues) > 256:
+                oRc = self.executeInternal(oCursor, sInsertSql + 'VALUES' + ', '.join(asValues), None, sCallerName);
+                asValues = [];
+        if len(asValues) > 0:
+            oRc = self.executeInternal(oCursor, sInsertSql + 'VALUES' + ', '.join(asValues), None, sCallerName);
+        return oRc
+
     def _fetchOne(self, oCursor):
         """Wrapper around Psycopg2.cursor.fetchone."""
@@ -425 +444 @@
         """
         return self.callProcInternal(self._oCursor, sProcedure, aoArgs, utils.getCallerName());
+
+    def insertList(self, sInsertSql, aoList, fnEntryFmt):
+        """
+        Optimizes the insertion of a list of values.
+        """
+        return self.insertListInternal(self._oCursor, sInsertSql, aoList, fnEntryFmt, utils.getCallerName());
 
     def fetchOne(self):
@@ -629 +654 @@
         return True;
 
+    def debugDisableExplain(self):
+        """ Disables explain. """
+        self._oExplainCursor = None;
+        self._oExplainConn   = None
+        return True;
+
+    def debugIsExplainEnabled(self):
+        """ Check if explaining of SQL statements is enabled. """
+        return self._oExplainConn is not None;
+