能否在无Tomcat/Spring的AWS Lambda中用AWS-XRay追踪MySQL查询？

Got it, let's tackle this problem head-on. Since you're working in a serverless Lambda environment, we don't need the bulky Spring/Tomcat configurations the docs mention—we can set up X-Ray SQL tracing with minimal changes. Here's how to do it step by step:

1. Add X-Ray SDK Dependencies #

First, you'll need the X-Ray Java SDK and its JDBC instrumentation module. If you're using Maven, add these to your pom.xml:

<dependency>
    <groupId>com.amazonaws</groupId>
    <artifactId>aws-xray-recorder-sdk-java</artifactId>
    <version>2.16.0</version> <!-- Use the latest stable version -->
</dependency>
<dependency>
    <groupId>com.amazonaws</groupId>
    <artifactId>aws-xray-recorder-sdk-jdbc</artifactId>
    <version>2.16.0</version>
</dependency>

For Gradle users, add this to your build.gradle:

implementation 'com.amazonaws:aws-xray-recorder-sdk-java:2.16.0'
implementation 'com.amazonaws:aws-xray-recorder-sdk-jdbc:2.16.0'

2. Instrument Your Database Connection #

You have two simple ways to wrap your JDBC connection for X-Ray tracing—pick whichever fits your existing code better:

Option A: Wrap Your Existing Connection #

Modify your getDatabaseConnection method to wrap the raw JDBC connection with X-Ray's TracingConnection. This lets you keep your original connection logic intact:

public Connection getDatabaseConnection(String jdbcUrl, String dbUser, String dbPassword) throws SQLException {
    // Get your original connection using your existing code
    Connection originalConnection = DriverManager.getConnection(jdbcUrl, dbUser, dbPassword);
    // Wrap it with X-Ray's tracing wrapper to capture SQL details
    return new com.amazonaws.xray.sql.TracingConnection(originalConnection);
}

Option B: Use the X-Ray JDBC URL Prefix #

An even cleaner approach is to prepend jdbc:xray: to your existing JDBC URL. X-Ray's JDBC driver will automatically proxy your original driver and handle tracing:

public Connection getDatabaseConnection(String jdbcUrl, String dbUser, String dbPassword) throws SQLException {
    // Add the X-Ray prefix to your JDBC URL
    String xrayEnabledUrl = "jdbc:xray:" + jdbcUrl.substring("jdbc:".length());
    // Get the connection as usual—X-Ray takes care of the rest
    return DriverManager.getConnection(xrayEnabledUrl, dbUser, dbPassword);
}

3. Configure Lambda IAM Permissions #

Your Lambda execution role needs permission to send trace data to X-Ray. Attach the AWSXRayDaemonWriteAccess managed policy to your role, or add these custom permissions:

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "xray:PutTraceSegments",
                "xray:PutTelemetryRecords"
            ],
            "Resource": "*"
        }
    ]
}

4. Enable X-Ray Tracing for Your Lambda #

Head to the Lambda console, navigate to your function, and go to Configuration > Monitoring and operations tools > X-Ray. Toggle on "Active tracing"—this tells Lambda to create a root X-Ray segment for each invocation, which your SQL traces will attach to as child segments.

5. Test and Verify #

Run your Lambda function, then open the X-Ray console. Check the Service Map—you should see a node for your database, and clicking into any trace will show detailed SQL information like:

• The executed query string
• Execution time and latency
• Any errors or exceptions thrown
• Database connection details

Extra Tips for Edge Cases #

• Connection Pools: If you're using a connection pool (like HikariCP), wrap your data source with com.amazonaws.xray.sql.TracingDataSource instead of individual connections. This ensures every connection from the pool is traced.
• Custom Segments: If you want to group SQL calls under a specific sub-segment, you can manually create one in your handler before executing queries:

  Segment subSegment = AWSXRay.beginSubsegment("Database Operations");
  try {
      // Run your SQL queries here
  } finally {
      AWSXRay.endSubsegment();
  }
  

• Version Compatibility: Make sure your X-Ray SDK version is compatible with your JDBC driver—most common drivers (MySQL, PostgreSQL, Aurora) work seamlessly with the latest SDK versions.

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