使用Django定位数据库查询来源的可行方案咨询

嗨，这个问题绝对是Django开发者排查数据库性能瓶颈时的高频痛点！我来分享几个靠谱的解决方案，比你手动补丁游标要更稳妥：

1. 现成的官方方案：用connection.execute_wrapper钩子 #

Django其实已经内置了专门用于拦截SQL执行的API——execute_wrapper，完全不需要你手动去修改游标类的execute方法，兼容性和安全性都更高。

你可以在项目启动时注册一个包装函数，自动给每个执行的SQL追加调用栈注释，示例代码如下：

import traceback
from django.db import connection

def annotate_sql_with_traceback(execute, sql, params, many, context):
    # 过滤掉Django内部的调用栈，只保留用户业务代码的部分
    stack_trace = traceback.format_stack()
    user_code_stack = [
        line.strip() for line in stack_trace
        if 'django/db' not in line and 'site-packages' not in line
    ]
    # 把栈信息转成单行注释，避免换行破坏SQL结构，只保留前5行核心信息
    stack_comment = ' -- ' + ' | '.join(user_code_stack[:5])
    # 执行修改后的SQL
    return execute(sql + stack_comment, params, many, context)

# 在项目初始化时注册这个包装器，比如放在app的ready()方法里，或者settings.py末尾
connection.execute_wrapper(annotate_sql_with_traceback)

这个方法的好处是：完全基于Django官方API，不会因为框架内部游标实现变化而失效；还能灵活过滤掉无用的框架内部调用栈，只保留你关心的业务代码位置。

2. 第三方工具：django-sql-commenter #

如果不想自己写代码，还有现成的第三方库django-sql-commenter，它可以自动给每个SQL查询添加结构化注释，默认就会包含生成该SQL的模型、视图、函数等信息，你还可以通过配置扩展添加调用栈。

安装后只需在settings.py里添加配置：

INSTALLED_APPS = [
    # ... 其他app
    'sql_commenter',
]

SQL_COMMENTER_CONFIG = {
    'INCLUDE_MODEL': True,
    'INCLUDE_VIEW': True,
    # 可以自定义添加调用栈信息
    'CUSTOM_COMMENTS': lambda context: {
        'traceback': ' | '.join([line.strip() for line in traceback.format_stack()[:5] if 'django/db' not in line])
    }
}

这样数据库日志里的SQL就会自动带上类似/* model: User, view: user_list, traceback: ... */的注释，直接就能定位代码来源。

比你当前方案更优的点 #

你手动补丁游标类的方法虽然可行，但有几个潜在问题：

• 直接修改游标类型的execute方法，可能会和Django内部的其他游标包装器（比如CursorDebugWrapper）冲突；
• 无法灵活过滤框架内部的调用栈，导致SQL注释里全是无用的Django代码路径，反而干扰排查；
• 兼容性差，Django版本更新后游标实现可能变化，你的补丁可能失效。

而用execute_wrapper或者django-sql-commenter则完全避开了这些问题，更适合长期使用。

补充一句：既然你已经在用Django Silk这类工具，它们其实也能记录每个SQL的调用栈，但如果是从数据库日志反向溯源，必须把代码位置嵌入到SQL本身里，上面的方案正好解决这个需求。

内容的提问来源于stack exchange，提问作者Grey Panther