eunomia-bpf项目：MySQL 8.0.40版本下SQL查询追踪的技术实践

在数据库性能分析和问题排查过程中，对SQL查询的执行情况进行实时监控是一项非常有价值的工作。本文将详细介绍如何使用eunomia-bpf项目中的BPF技术来追踪MySQL 8.0.40版本的SQL查询执行情况。

背景与挑战

MySQL作为最流行的关系型数据库之一，其内部执行机制随着版本迭代不断变化。在MySQL 8.0.40版本中，我们发现传统的dispatch_command函数符号名称发生了变化，且参数结构也有所调整，这给基于BPF的追踪带来了新的挑战。

技术实现方案

1. 符号名称解析

在MySQL 8.0.40版本中，dispatch_command函数的符号名称经过C++名称修饰后变为_Z16dispatch_commandP3THDPK8COM_DATA19enum_server_command。通过c++filt工具可以将其还原为可读形式：

echo 'uprobe:/usr/sbin/mysqld:_Z16dispatch_commandP3THDPK8COM_DATA19enum_server_command' | c++filt

输出结果为：

uprobe:/usr/sbin/mysqld:dispatch_command(THD*, COM_DATA const*, enum_server_command)

2. 参数结构分析

MySQL 8.0.40版本中，dispatch_command函数的第二个参数是一个指向COM_DATA联合体的指针。这个联合体包含了不同类型的命令数据，其中com_query成员专门用于存储SQL查询信息。

我们需要在BPF程序中定义相应的数据结构：

struct COM_QUERY_DATA {
  const char *query;    // 查询字符串指针
  unsigned int length;  // 查询字符串长度
};

union COM_DATA {
    struct COM_QUERY_DATA com_query;
    // 其他命令类型的数据结构
};

3. BPF程序实现

基于上述分析，我们实现了完整的BPF追踪程序：

#!/usr/bin/env bpftrace

// 定义COM_QUERY_DATA结构
struct COM_QUERY_DATA {
  const char *query;
  unsigned int length;
};

// 定义COM_DATA联合
union COM_DATA {
    struct COM_QUERY_DATA com_query;
    // 其他类型省略
};

// 跟踪MySQL中的dispatch_command函数
uprobe:/usr/sbin/mysqld:_Z16dispatch_commandP3THDPK8COM_DATA19enum_server_command
{
    // 记录开始时间
    @start_times[tid] = nsecs;
    
    // 获取COM_DATA指针
    $com_data_ptr = (union COM_DATA *)arg1;

    // 提取SQL查询字符串
    $query = str($com_data_ptr->com_query.query);

    // 输出查询信息
    printf("MySQL command executed by PID %d: SQL Query: %s\n", pid, $query);
}

// 跟踪函数返回
uprobe:/usr/sbin/mysqld:_Z16dispatch_commandP3THDPK8COM_DATA19enum_server_command
{
    // 计算执行时间
    $start = @start_times[tid];
    $delta = (nsecs - $start) / 1000000;
    
    // 输出执行时间
    printf("Latency: %u ms\n", $delta);
    
    // 清理map
    delete(@start_times[tid]);
}

技术要点解析

1. 符号名称处理：MySQL 8.0.40使用C++名称修饰，需要通过c++filt工具解析出实际函数签名。

2. 参数访问：COM_DATA是一个联合体，需要正确解析其内部结构才能获取SQL查询字符串。

3. 性能监控：通过记录函数进入和退出的时间戳，可以精确计算SQL查询的执行时间。

4. 线程安全：使用线程ID(tid)作为map的key，确保在多线程环境下的正确性。

实际应用价值

该技术方案可以应用于：

1. 生产环境SQL查询监控
2. 慢查询分析
3. 数据库性能优化
4. 异常SQL检测
5. 数据库审计

总结

通过对MySQL 8.0.40内部机制的深入分析和BPF技术的灵活运用，我们成功实现了对SQL查询的实时监控。这一技术方案不仅解决了新版本MySQL的兼容性问题，还为数据库性能分析提供了强有力的工具。eunomia-bpf项目的这一实践展示了BPF技术在数据库领域的强大应用潜力。