Hoe krijg ik psycopg2-registratie van de uitvoeringstijd van de query?

Eenvoudig genoeg om een ​​tijdstempel in te stellen aan het begin van de uitvoering en de duur aan het einde te berekenen. U hebt uw eigen eenvoudige subklassen van LoggingConnection en LoggingCursor nodig. Zie mijn voorbeeldcode.

Dit is gebaseerd op de bron van MinTimeLoggingConnection die u kunt vinden in psycopg2/extras.py bron.

import time
import psycopg2
import psycopg2.extensions
from psycopg2.extras import LoggingConnection, LoggingCursor
import logging

logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger(__name__)

# MyLoggingCursor simply sets self.timestamp at start of each query                                                                 
class MyLoggingCursor(LoggingCursor):
    def execute(self, query, vars=None):
        self.timestamp = time.time()
        return super(MyLoggingCursor, self).execute(query, vars)

    def callproc(self, procname, vars=None):
        self.timestamp = time.time()
        return super(MyLoggingCursor, self).callproc(procname, vars)

# MyLogging Connection:                                                                                                             
#   a) calls MyLoggingCursor rather than the default                                                                                
#   b) adds resulting execution (+ transport) time via filter()                                                                     
class MyLoggingConnection(LoggingConnection):
    def filter(self, msg, curs):
        return msg + "   %d ms" % int((time.time() - curs.timestamp) * 1000)

    def cursor(self, *args, **kwargs):
        kwargs.setdefault('cursor_factory', MyLoggingCursor)
        return LoggingConnection.cursor(self, *args, **kwargs)

db_settings = {
    ....
}

query_txt = "[query_text_from file]"

conn = psycopg2.connect(connection_factory=MyLoggingConnection, **db_settings)
conn.initialize(logger)

cur = conn.cursor()
cur.execute(query_text)

en je krijgt:

DEBUG: __main__:[query]     3 ms

binnen uw filter() je kunt de opmaak wijzigen, of ervoor kiezen om niet weer te geven, als het minder dan een bepaalde waarde is.