VirtualBox

source: vbox/trunk/src/VBox/ValidationKit/testmanager/core/db.py@ 70662

Last change on this file since 70662 was 70662, checked in by vboxsync, 7 years ago

ValidationKit/db.py: pylint fix

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 25.0 KB
Line 
1# -*- coding: utf-8 -*-
2# $Id: db.py 70662 2018-01-21 16:28:23Z vboxsync $
3
4"""
5Test Manager - Database Interface.
6"""
7
8__copyright__ = \
9"""
10Copyright (C) 2012-2017 Oracle Corporation
11
12This file is part of VirtualBox Open Source Edition (OSE), as
13available from http://www.virtualbox.org. This file is free software;
14you can redistribute it and/or modify it under the terms of the GNU
15General Public License (GPL) as published by the Free Software
16Foundation, in version 2 as it comes in the "COPYING" file of the
17VirtualBox OSE distribution. VirtualBox OSE is distributed in the
18hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
19
20The contents of this file may alternatively be used under the terms
21of the Common Development and Distribution License Version 1.0
22(CDDL) only, as it comes in the "COPYING.CDDL" file of the
23VirtualBox OSE distribution, in which case the provisions of the
24CDDL are applicable instead of those of the GPL.
25
26You may elect to license modified versions of this file under the
27terms and conditions of either the GPL or the CDDL or both.
28"""
29__version__ = "$Revision: 70662 $"
30
31
32# Standard python imports.
33import datetime;
34import os;
35import sys;
36import psycopg2; # pylint: disable=import-error
37import psycopg2.extensions; # pylint: disable=import-error
38
39# Validation Kit imports.
40from common import utils, webutils;
41from testmanager import config;
42
43# Fix psycho unicode handling in psycopg2 with python 2.x.
44if sys.version_info[0] < 3:
45 psycopg2.extensions.register_type(psycopg2.extensions.UNICODE);
46 psycopg2.extensions.register_type(psycopg2.extensions.UNICODEARRAY);
47else:
48 unicode = str; # pylint: disable=redefined-builtin,invalid-name
49
50
51
52def isDbTimestampInfinity(tsValue):
53 """
54 Checks if tsValue is an infinity timestamp.
55 """
56 ## @todo improve this test...
57 return tsValue.year >= 9999;
58
59def isDbTimestamp(oValue):
60 """
61 Checks if oValue is a DB timestamp object.
62 """
63 if isinstance(oValue, datetime.datetime):
64 return True;
65 if utils.isString(oValue):
66 ## @todo detect strings as well.
67 return False;
68 return getattr(oValue, 'pydatetime', None) != None;
69
70def dbTimestampToDatetime(oValue):
71 """
72 Converts a database timestamp to a datetime instance.
73 """
74 if isinstance(oValue, datetime.datetime):
75 return oValue;
76 if utils.isString(oValue):
77 raise Exception('TODO');
78 return oValue.pydatetime();
79
80def dbTimestampToZuluDatetime(oValue):
81 """
82 Converts a database timestamp to a zulu datetime instance.
83 """
84 tsValue = dbTimestampToDatetime(oValue);
85
86 class UTC(datetime.tzinfo):
87 """UTC TZ Info Class"""
88 def utcoffset(self, _):
89 return datetime.timedelta(0);
90 def tzname(self, _):
91 return "UTC";
92 def dst(self, _):
93 return datetime.timedelta(0);
94 if tsValue.tzinfo is not None:
95 tsValue = tsValue.astimezone(UTC());
96 else:
97 tsValue = tsValue.replace(tzinfo=UTC());
98 return tsValue;
99
100def dbTimestampPythonNow():
101 """
102 Gets the current python timestamp in a database compatible way.
103 """
104 return dbTimestampToZuluDatetime(datetime.datetime.utcnow());
105
106def dbTimestampMinusOneTick(oValue):
107 """
108 Returns a new timestamp that's one tick before the given one.
109 """
110 oValue = dbTimestampToZuluDatetime(oValue);
111 return oValue - datetime.timedelta(microseconds = 1);
112
113def isDbInterval(oValue):
114 """
115 Checks if oValue is a DB interval object.
116 """
117 if isinstance(oValue, datetime.timedelta):
118 return True;
119 return False;
120
121
122class TMDatabaseIntegrityException(Exception):
123 """
124 Herolds a database integrity error up the callstack.
125
126 Do NOT use directly, only thru TMDatabaseConnection.integrityException.
127 Otherwise, we won't be able to log the issue.
128 """
129 pass;
130
131
132class TMDatabaseCursor(object):
133 """ Cursor wrapper class. """
134
135 def __init__(self, oDb, oCursor):
136 self._oDb = oDb;
137 self._oCursor = oCursor;
138
139 def execute(self, sOperation, aoArgs = None):
140 """ See TMDatabaseConnection.execute()"""
141 return self._oDb.executeInternal(self._oCursor, sOperation, aoArgs, utils.getCallerName());
142
143 def callProc(self, sProcedure, aoArgs = None):
144 """ See TMDatabaseConnection.callProc()"""
145 return self._oDb.callProcInternal(self._oCursor, sProcedure, aoArgs, utils.getCallerName());
146
147 def insertList(self, sInsertSql, aoList, fnEntryFmt):
148 """ See TMDatabaseConnection.insertList. """
149 return self._oDb.insertListInternal(self._oCursor, sInsertSql, aoList, fnEntryFmt, utils.getCallerName());
150
151 def fetchOne(self):
152 """Wrapper around Psycopg2.cursor.fetchone."""
153 return self._oCursor.fetchone();
154
155 def fetchMany(self, cRows = None):
156 """Wrapper around Psycopg2.cursor.fetchmany."""
157 return self._oCursor.fetchmany(cRows if cRows is not None else self._oCursor.arraysize);
158
159 def fetchAll(self):
160 """Wrapper around Psycopg2.cursor.fetchall."""
161 return self._oCursor.fetchall();
162
163 def getRowCount(self):
164 """Wrapper around Psycopg2.cursor.rowcount."""
165 return self._oCursor.rowcount;
166
167 def formatBindArgs(self, sStatement, aoArgs):
168 """Wrapper around Psycopg2.cursor.mogrify."""
169 oRet = self._oCursor.mogrify(sStatement, aoArgs);
170 if sys.version_info[0] >= 3 and not isinstance(oRet, str):
171 oRet = oRet.decode('utf-8');
172 return oRet;
173
174 def copyExpert(self, sSqlCopyStmt, oFile, cbBuf = 8192):
175 """ See TMDatabaseConnection.copyExpert()"""
176 return self._oCursor.copy_expert(sSqlCopyStmt, oFile, cbBuf);
177
178 @staticmethod
179 def isTsInfinity(tsValue):
180 """ Checks if tsValue is an infinity timestamp. """
181 return isDbTimestampInfinity(tsValue);
182
183
184class TMDatabaseConnection(object):
185 """
186 Test Manager Database Access class.
187
188 This class contains no logic, just raw access abstraction and utilities,
189 as well as some debug help and some statistics.
190 """
191
192 def __init__(self, fnDPrint = None, oSrvGlue = None):
193 """
194 Database connection wrapper.
195 The fnDPrint is for debug logging of all database activity.
196
197 Raises an exception on failure.
198 """
199
200 sAppName = '%s-%s' % (os.getpid(), os.path.basename(sys.argv[0]),)
201 if len(sAppName) >= 64:
202 sAppName = sAppName[:64];
203 os.environ['PGAPPNAME'] = sAppName;
204
205 dArgs = \
206 { \
207 'database': config.g_ksDatabaseName,
208 'user': config.g_ksDatabaseUser,
209 'password': config.g_ksDatabasePassword,
210 # 'application_name': sAppName, - Darn stale debian! :/
211 };
212 if config.g_ksDatabaseAddress is not None:
213 dArgs['host'] = config.g_ksDatabaseAddress;
214 if config.g_ksDatabasePort is not None:
215 dArgs['port'] = config.g_ksDatabasePort;
216 self._oConn = psycopg2.connect(**dArgs); # pylint: disable=W0142
217 self._oConn.set_client_encoding('UTF-8');
218 self._oCursor = self._oConn.cursor();
219 self._oExplainConn = None;
220 self._oExplainCursor = None;
221 if config.g_kfWebUiSqlTraceExplain and config.g_kfWebUiSqlTrace:
222 self._oExplainConn = psycopg2.connect(**dArgs); # pylint: disable=W0142
223 self._oExplainConn.set_client_encoding('UTF-8');
224 self._oExplainCursor = self._oExplainConn.cursor();
225 self._fTransaction = False;
226 self._tsCurrent = None;
227 self._tsCurrentMinusOne = None;
228
229 assert self.isAutoCommitting() is False;
230
231 # Debug and introspection.
232 self._fnDPrint = fnDPrint;
233 self._aoTraceBack = [];
234
235 # Exception class handles.
236 self.oXcptError = psycopg2.Error;
237
238 if oSrvGlue is not None:
239 oSrvGlue.registerDebugInfoCallback(self.debugInfoCallback);
240
241 # Object caches (used by database logic classes).
242 self.ddCaches = dict();
243
244 def isAutoCommitting(self):
245 """ Work around missing autocommit attribute in older versions."""
246 return getattr(self._oConn, 'autocommit', False);
247
248 def close(self):
249 """
250 Closes the connection and renders all cursors useless.
251 """
252 if self._oCursor is not None:
253 self._oCursor.close();
254 self._oCursor = None;
255
256 if self._oConn is not None:
257 self._oConn.close();
258 self._oConn = None;
259
260 if self._oExplainCursor is not None:
261 self._oExplainCursor.close();
262 self._oExplainCursor = None;
263
264 if self._oExplainConn is not None:
265 self._oExplainConn.close();
266 self._oExplainConn = None;
267
268
269 def _startedTransaction(self):
270 """
271 Called to work the _fTransaction and related variables when starting
272 a transaction.
273 """
274 self._fTransaction = True;
275 self._tsCurrent = None;
276 self._tsCurrentMinusOne = None;
277 return None;
278
279 def _endedTransaction(self):
280 """
281 Called to work the _fTransaction and related variables when ending
282 a transaction.
283 """
284 self._fTransaction = False;
285 self._tsCurrent = None;
286 self._tsCurrentMinusOne = None;
287 return None;
288
289 def begin(self):
290 """
291 Currently just for marking where a transaction starts in the code.
292 """
293 assert self._oConn is not None;
294 assert self.isAutoCommitting() is False;
295 self._aoTraceBack.append([utils.timestampNano(), 'START TRANSACTION', 0, 0, utils.getCallerName(), None]);
296 self._startedTransaction();
297 return True;
298
299 def commit(self, sCallerName = None):
300 """ Wrapper around Psycopg2.connection.commit."""
301 assert self._fTransaction is True;
302
303 nsStart = utils.timestampNano();
304 oRc = self._oConn.commit();
305 cNsElapsed = utils.timestampNano() - nsStart;
306
307 if sCallerName is None:
308 sCallerName = utils.getCallerName();
309 self._aoTraceBack.append([nsStart, 'COMMIT', cNsElapsed, 0, sCallerName, None]);
310 self._endedTransaction();
311 return oRc;
312
313 def maybeCommit(self, fCommit):
314 """
315 Commits if fCommit is True.
316 Returns True if committed, False if not.
317 """
318 if fCommit is True:
319 self.commit(utils.getCallerName());
320 return True;
321 return False;
322
323 def rollback(self):
324 """ Wrapper around Psycopg2.connection.rollback."""
325 nsStart = utils.timestampNano();
326 oRc = self._oConn.rollback();
327 cNsElapsed = utils.timestampNano() - nsStart;
328
329 self._aoTraceBack.append([nsStart, 'ROLLBACK', cNsElapsed, 0, utils.getCallerName(), None]);
330 self._endedTransaction();
331 return oRc;
332
333 #
334 # Internal cursor workers.
335 #
336
337 def executeInternal(self, oCursor, sOperation, aoArgs, sCallerName):
338 """
339 Execute a query or command.
340
341 Mostly a wrapper around the psycopg2 cursor method with the same name,
342 but collect data for traceback.
343 """
344 if aoArgs is not None:
345 sBound = oCursor.mogrify(unicode(sOperation), aoArgs);
346 elif sOperation.find('%') < 0:
347 sBound = oCursor.mogrify(unicode(sOperation), list());
348 else:
349 sBound = unicode(sOperation);
350
351 if sys.version_info[0] >= 3 and not isinstance(sBound, str):
352 sBound = sBound.decode('utf-8'); # pylint: disable=redefined-variable-type
353
354 aasExplain = None;
355 if self._oExplainCursor is not None and not sBound.startswith('DROP'):
356 try:
357 if config.g_kfWebUiSqlTraceExplainTiming:
358 self._oExplainCursor.execute('EXPLAIN (ANALYZE, BUFFERS, COSTS, VERBOSE, TIMING) ' + sBound);
359 else:
360 self._oExplainCursor.execute('EXPLAIN (ANALYZE, BUFFERS, COSTS, VERBOSE) ' + sBound);
361 except Exception as oXcpt:
362 aasExplain = [ ['Explain exception: '], [str(oXcpt)]];
363 try: self._oExplainConn.rollback();
364 except: pass;
365 else:
366 aasExplain = self._oExplainCursor.fetchall();
367
368 nsStart = utils.timestampNano();
369 try:
370 oRc = oCursor.execute(sBound);
371 except Exception as oXcpt:
372 cNsElapsed = utils.timestampNano() - nsStart;
373 self._aoTraceBack.append([nsStart, 'oXcpt=%s; Statement: %s' % (oXcpt, sBound), cNsElapsed, 0, sCallerName, None]);
374 if self._fnDPrint is not None:
375 self._fnDPrint('db::execute %u ns, caller %s: oXcpt=%s; Statement: %s'
376 % (cNsElapsed, sCallerName, oXcpt, sBound));
377 raise;
378 cNsElapsed = utils.timestampNano() - nsStart;
379
380 if self._fTransaction is False and not self.isAutoCommitting(): # Even SELECTs starts transactions with psycopg2, see FAQ.
381 self._aoTraceBack.append([nsStart, '[START TRANSACTION]', 0, 0, sCallerName, None]);
382 self._startedTransaction();
383 self._aoTraceBack.append([nsStart, sBound, cNsElapsed, oCursor.rowcount, sCallerName, aasExplain]);
384 if self._fnDPrint is not None:
385 self._fnDPrint('db::execute %u ns, caller %s: "\n%s"' % (cNsElapsed, sCallerName, sBound));
386 if self.isAutoCommitting():
387 self._aoTraceBack.append([nsStart, '[AUTO COMMIT]', 0, 0, sCallerName, None]);
388
389 return oRc;
390
391 def callProcInternal(self, oCursor, sProcedure, aoArgs, sCallerName):
392 """
393 Call a stored procedure.
394
395 Mostly a wrapper around the psycopg2 cursor method 'callproc', but
396 collect data for traceback.
397 """
398 if aoArgs is None:
399 aoArgs = list();
400
401 nsStart = utils.timestampNano();
402 try:
403 oRc = oCursor.callproc(sProcedure, aoArgs);
404 except Exception as oXcpt:
405 cNsElapsed = utils.timestampNano() - nsStart;
406 self._aoTraceBack.append([nsStart, 'oXcpt=%s; Calling: %s(%s)' % (oXcpt, sProcedure, aoArgs),
407 cNsElapsed, 0, sCallerName, None]);
408 if self._fnDPrint is not None:
409 self._fnDPrint('db::callproc %u ns, caller %s: oXcpt=%s; Calling: %s(%s)'
410 % (cNsElapsed, sCallerName, oXcpt, sProcedure, aoArgs));
411 raise;
412 cNsElapsed = utils.timestampNano() - nsStart;
413
414 if self._fTransaction is False and not self.isAutoCommitting(): # Even SELECTs starts transactions with psycopg2, see FAQ.
415 self._aoTraceBack.append([nsStart, '[START TRANSACTION]', 0, 0, sCallerName, None]);
416 self._startedTransaction();
417 self._aoTraceBack.append([nsStart, '%s(%s)' % (sProcedure, aoArgs), cNsElapsed, oCursor.rowcount, sCallerName, None]);
418 if self._fnDPrint is not None:
419 self._fnDPrint('db::callproc %u ns, caller %s: "%s(%s)"' % (cNsElapsed, sCallerName, sProcedure, aoArgs));
420 if self.isAutoCommitting():
421 self._aoTraceBack.append([nsStart, '[AUTO COMMIT]', 0, 0, sCallerName, sCallerName, None]);
422
423 return oRc;
424
425 def insertListInternal(self, oCursor, sInsertSql, aoList, fnEntryFmt, sCallerName):
426 """
427 Optimizes the insertion of a list of values.
428 """
429 oRc = None;
430 asValues = [];
431 for aoEntry in aoList:
432 asValues.append(fnEntryFmt(aoEntry));
433 if len(asValues) > 256:
434 oRc = self.executeInternal(oCursor, sInsertSql + 'VALUES' + ', '.join(asValues), None, sCallerName);
435 asValues = [];
436 if asValues:
437 oRc = self.executeInternal(oCursor, sInsertSql + 'VALUES' + ', '.join(asValues), None, sCallerName);
438 return oRc
439
440 def _fetchOne(self, oCursor):
441 """Wrapper around Psycopg2.cursor.fetchone."""
442 oRow = oCursor.fetchone()
443 if self._fnDPrint is not None:
444 self._fnDPrint('db:fetchOne returns: %s' % (oRow,));
445 return oRow;
446
447 def _fetchMany(self, oCursor, cRows):
448 """Wrapper around Psycopg2.cursor.fetchmany."""
449 return oCursor.fetchmany(cRows if cRows is not None else oCursor.arraysize);
450
451 def _fetchAll(self, oCursor):
452 """Wrapper around Psycopg2.cursor.fetchall."""
453 return oCursor.fetchall()
454
455 def _getRowCountWorker(self, oCursor):
456 """Wrapper around Psycopg2.cursor.rowcount."""
457 return oCursor.rowcount;
458
459
460 #
461 # Default cursor access.
462 #
463
464 def execute(self, sOperation, aoArgs = None):
465 """
466 Execute a query or command.
467
468 Mostly a wrapper around the psycopg2 cursor method with the same name,
469 but collect data for traceback.
470 """
471 return self.executeInternal(self._oCursor, sOperation, aoArgs, utils.getCallerName());
472
473 def callProc(self, sProcedure, aoArgs = None):
474 """
475 Call a stored procedure.
476
477 Mostly a wrapper around the psycopg2 cursor method 'callproc', but
478 collect data for traceback.
479 """
480 return self.callProcInternal(self._oCursor, sProcedure, aoArgs, utils.getCallerName());
481
482 def insertList(self, sInsertSql, aoList, fnEntryFmt):
483 """
484 Optimizes the insertion of a list of values.
485 """
486 return self.insertListInternal(self._oCursor, sInsertSql, aoList, fnEntryFmt, utils.getCallerName());
487
488 def fetchOne(self):
489 """Wrapper around Psycopg2.cursor.fetchone."""
490 return self._oCursor.fetchone();
491
492 def fetchMany(self, cRows = None):
493 """Wrapper around Psycopg2.cursor.fetchmany."""
494 return self._oCursor.fetchmany(cRows if cRows is not None else self._oCursor.arraysize);
495
496 def fetchAll(self):
497 """Wrapper around Psycopg2.cursor.fetchall."""
498 return self._oCursor.fetchall();
499
500 def getRowCount(self):
501 """Wrapper around Psycopg2.cursor.rowcount."""
502 return self._oCursor.rowcount;
503
504 def formatBindArgs(self, sStatement, aoArgs):
505 """Wrapper around Psycopg2.cursor.mogrify."""
506 oRet = self._oCursor.mogrify(sStatement, aoArgs);
507 if sys.version_info[0] >= 3 and not isinstance(oRet, str):
508 oRet = oRet.decode('utf-8');
509 return oRet;
510
511 def copyExpert(self, sSqlCopyStmt, oFile, cbBuf = 8192):
512 """ Wrapper around Psycopg2.cursor.copy_expert. """
513 return self._oCursor.copy_expert(sSqlCopyStmt, oFile, cbBuf);
514
515 def getCurrentTimestamps(self):
516 """
517 Returns the current timestamp and the current timestamp minus one tick.
518 This will start a transaction if necessary.
519 """
520 if self._tsCurrent is None:
521 self.execute('SELECT CURRENT_TIMESTAMP, CURRENT_TIMESTAMP - INTERVAL \'1 microsecond\'');
522 (self._tsCurrent, self._tsCurrentMinusOne) = self.fetchOne();
523 return (self._tsCurrent, self._tsCurrentMinusOne);
524
525 def getCurrentTimestamp(self):
526 """
527 Returns the current timestamp.
528 This will start a transaction if necessary.
529 """
530 if self._tsCurrent is None:
531 self.getCurrentTimestamps();
532 return self._tsCurrent;
533
534 def getCurrentTimestampMinusOne(self):
535 """
536 Returns the current timestamp minus one tick.
537 This will start a transaction if necessary.
538 """
539 if self._tsCurrentMinusOne is None:
540 self.getCurrentTimestamps();
541 return self._tsCurrentMinusOne;
542
543
544 #
545 # Additional cursors.
546 #
547 def openCursor(self):
548 """
549 Opens a new cursor (TMDatabaseCursor).
550 """
551 oCursor = self._oConn.cursor();
552 return TMDatabaseCursor(self, oCursor);
553
554 #
555 # Cache support.
556 #
557 def getCache(self, sType):
558 """ Returns the cache dictionary for this data type. """
559 dRet = self.ddCaches.get(sType, None);
560 if dRet is None:
561 dRet = dict();
562 self.ddCaches[sType] = dRet;
563 return dRet;
564
565
566 #
567 # Utilities.
568 #
569
570 @staticmethod
571 def isTsInfinity(tsValue):
572 """ Checks if tsValue is an infinity timestamp. """
573 return isDbTimestampInfinity(tsValue);
574
575 #
576 # Error stuff.
577 #
578 def integrityException(self, sMessage):
579 """
580 Database integrity reporter and exception factory.
581 Returns an TMDatabaseIntegrityException which the caller can raise.
582 """
583 ## @todo Create a new database connection and log the issue in the SystemLog table.
584 ## Alternatively, rollback whatever is going on and do it using the current one.
585 return TMDatabaseIntegrityException(sMessage);
586
587
588 #
589 # Debugging.
590 #
591
592 def dprint(self, sText):
593 """
594 Debug output.
595 """
596 if not self._fnDPrint:
597 return False;
598 self._fnDPrint(sText);
599 return True;
600
601 def debugHtmlReport(self, tsStart = 0):
602 """
603 Used to get a SQL activity dump as HTML, usually for WuiBase._sDebug.
604 """
605 cNsElapsed = 0;
606 for aEntry in self._aoTraceBack:
607 cNsElapsed += aEntry[2];
608
609 sDebug = '<h3>SQL Debug Log (total time %s ns):</h3>\n' \
610 '<table class="tmsqltable">\n' \
611 ' <tr>\n' \
612 ' <th>No.</th>\n' \
613 ' <th>Timestamp (ns)</th>\n' \
614 ' <th>Elapsed (ns)</th>\n' \
615 ' <th>Rows Returned</th>\n' \
616 ' <th>Command</th>\n' \
617 ' <th>Caller</th>\n' \
618 ' </tr>\n' \
619 % (utils.formatNumber(cNsElapsed, '&nbsp;'),);
620
621 iEntry = 0;
622 for aEntry in self._aoTraceBack:
623 iEntry += 1;
624 sDebug += ' <tr>\n' \
625 ' <td align="right">%s</td>\n' \
626 ' <td align="right">%s</td>\n' \
627 ' <td align="right">%s</td>\n' \
628 ' <td align="right">%s</td>\n' \
629 ' <td><pre>%s</pre></td>\n' \
630 ' <td>%s</td>\n' \
631 ' </tr>\n' \
632 % (iEntry,
633 utils.formatNumber(aEntry[0] - tsStart, '&nbsp;'),
634 utils.formatNumber(aEntry[2], '&nbsp;'),
635 utils.formatNumber(aEntry[3], '&nbsp;'),
636 webutils.escapeElem(aEntry[1]),
637 webutils.escapeElem(aEntry[4]),
638 );
639 if aEntry[5] is not None:
640 sDebug += ' <tr>\n' \
641 ' <td colspan="6"><pre style="white-space: pre-wrap;">%s</pre></td>\n' \
642 ' </tr>\n' \
643 % (webutils.escapeElem('\n'.join([aoRow[0] for aoRow in aEntry[5]])),);
644
645 sDebug += '</table>';
646 return sDebug;
647
648 def debugTextReport(self, tsStart = 0):
649 """
650 Used to get a SQL activity dump as text.
651 """
652 cNsElapsed = 0;
653 for aEntry in self._aoTraceBack:
654 cNsElapsed += aEntry[2];
655
656 sHdr = 'SQL Debug Log (total time %s ns)' % (utils.formatNumber(cNsElapsed),);
657 sDebug = sHdr + '\n' + '-' * len(sHdr) + '\n';
658
659 iEntry = 0;
660 for aEntry in self._aoTraceBack:
661 iEntry += 1;
662 sHdr = 'Query #%s Timestamp: %s ns Elapsed: %s ns Rows: %s Caller: %s' \
663 % ( iEntry,
664 utils.formatNumber(aEntry[0] - tsStart),
665 utils.formatNumber(aEntry[2]),
666 utils.formatNumber(aEntry[3]),
667 aEntry[4], );
668 sDebug += '\n' + sHdr + '\n' + '-' * len(sHdr) + '\n';
669
670 sDebug += aEntry[1];
671 if sDebug[-1] != '\n':
672 sDebug += '\n';
673
674 if aEntry[5] is not None:
675 sDebug += 'Explain:\n' \
676 ' %s\n' \
677 % ( '\n'.join([aoRow[0] for aoRow in aEntry[5]]),);
678
679 return sDebug;
680
681 def debugInfoCallback(self, oGlue, fHtml):
682 """ Called back by the glue code on error. """
683 oGlue.write('\n');
684 if not fHtml: oGlue.write(self.debugTextReport());
685 else: oGlue.write(self.debugHtmlReport());
686 oGlue.write('\n');
687 return True;
688
689 def debugEnableExplain(self):
690 """ Enabled explain. """
691 if self._oExplainConn is None:
692 dArgs = \
693 { \
694 'database': config.g_ksDatabaseName,
695 'user': config.g_ksDatabaseUser,
696 'password': config.g_ksDatabasePassword,
697 # 'application_name': sAppName, - Darn stale debian! :/
698 };
699 if config.g_ksDatabaseAddress is not None:
700 dArgs['host'] = config.g_ksDatabaseAddress;
701 if config.g_ksDatabasePort is not None:
702 dArgs['port'] = config.g_ksDatabasePort;
703 self._oExplainConn = psycopg2.connect(**dArgs); # pylint: disable=W0142
704 self._oExplainCursor = self._oExplainConn.cursor();
705 return True;
706
707 def debugDisableExplain(self):
708 """ Disables explain. """
709 self._oExplainCursor = None;
710 self._oExplainConn = None
711 return True;
712
713 def debugIsExplainEnabled(self):
714 """ Check if explaining of SQL statements is enabled. """
715 return self._oExplainConn is not None;
716
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette