Django获取查询执行时间

问题描述

有没有办法在不使用django-debug-toolbar的情况下测量Django查询时间？它不是用来调试/记录的，我需要它像about # results (# seconds)那样显示在网页上。

编辑：我需要在生产模式下使用此功能，因此DEBUG=True不是选项

---

解决方案

虽然django.db.connection.queries会给出查询次数，但仅当DEBUG=True时才起作用。为了将其用于一般用途，我们可以使用Database instrumentation [Django docs]并编写一个包装器来计时我们的查询。下面是直接来自上述文档的示例，我想说它似乎符合您的需求：

import time

class QueryLogger:

    def __init__(self):
        self.queries = []

    def __call__(self, execute, sql, params, many, context):
        current_query = {'sql': sql, 'params': params, 'many': many}
        start = time.monotonic()
        try:
            result = execute(sql, params, many, context)
        except Exception as e:
            current_query['status'] = 'error'
            current_query['exception'] = e
            raise
        else:
            current_query['status'] = 'ok'
            return result
        finally:
            duration = time.monotonic() - start
            current_query['duration'] = duration
            self.queries.append(current_query)

要使用它，您应该以这样的方式进行查询：

from django.db import connection

query_logger = QueryLogger()

with connection.execute_wrapper(query_logger):
    # Make your queries here

for query in query_logger.queries:
    print(query['sql'], query['duration'])