Integrating APItoolkit with MySQL

Java Example (using JDBC)

import io.opentelemetry.api.GlobalOpenTelemetry;
import io.opentelemetry.api.trace.Span;
import io.opentelemetry.api.trace.Tracer;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;

public class MySQLExample {
    private final Tracer tracer = GlobalOpenTelemetry.getTracer("mysql-instrumentation");
    private final String jdbcUrl = "jdbc:mysql://localhost:3306/mydatabase";
    private final String username = "user";
    private final String password = "password";

    public List<Product> getProducts() {
        Span span = tracer.spanBuilder("mysql.query").startSpan();
        try {
            span.setAttribute("db.system", "mysql");
            span.setAttribute("db.name", "mydatabase");
            span.setAttribute("db.operation", "SELECT");
            span.setAttribute("db.statement", "SELECT id, name, price FROM products");

            List<Product> products = new ArrayList<>();

            try (Connection conn = DriverManager.getConnection(jdbcUrl, username, password);
                 PreparedStatement stmt = conn.prepareStatement("SELECT id, name, price FROM products");
                 ResultSet rs = stmt.executeQuery()) {

                while (rs.next()) {
                    Product product = new Product();
                    product.setId(rs.getLong("id"));
                    product.setName(rs.getString("name"));
                    product.setPrice(rs.getBigDecimal("price"));
                    products.add(product);
                }
            }

            span.setAttribute("db.result_count", products.size());
            return products;
        } catch (Exception e) {
            span.recordException(e);
            throw new RuntimeException("Failed to query products", e);
        } finally {
            span.end();
        }
    }

    public void createProduct(Product product) {
        Span span = tracer.spanBuilder("mysql.insert").startSpan();
        try {
            span.setAttribute("db.system", "mysql");
            span.setAttribute("db.name", "mydatabase");
            span.setAttribute("db.operation", "INSERT");
            span.setAttribute("db.statement", "INSERT INTO products (name, price) VALUES (?, ?)");

            try (Connection conn = DriverManager.getConnection(jdbcUrl, username, password);
                 PreparedStatement stmt = conn.prepareStatement("INSERT INTO products (name, price) VALUES (?, ?)")) {

                stmt.setString(1, product.getName());
                stmt.setBigDecimal(2, product.getPrice());
                int rowsAffected = stmt.executeUpdate();

                span.setAttribute("db.rows_affected", rowsAffected);
            }
        } catch (Exception e) {
            span.recordException(e);
            throw new RuntimeException("Failed to create product", e);
        } finally {
            span.end();
        }
    }
}

Node.js Example (using mysql2)

const mysql = require('mysql2/promise');
const opentelemetry = require('@opentelemetry/api');

const tracer = opentelemetry.trace.getTracer('mysql-instrumentation');

const pool = mysql.createPool({
  host: 'localhost',
  user: 'user',
  password: 'password',
  database: 'mydatabase',
  waitForConnections: true,
  connectionLimit: 10,
  queueLimit: 0
});

async function getProducts() {
  const span = tracer.startSpan('mysql.query');

  try {
    span.setAttribute('db.system', 'mysql');
    span.setAttribute('db.name', 'mydatabase');
    span.setAttribute('db.operation', 'SELECT');
    span.setAttribute('db.statement', 'SELECT id, name, price FROM products');

    const [rows] = await pool.query('SELECT id, name, price FROM products');

    span.setAttribute('db.result_count', rows.length);
    return rows;
  } catch (error) {
    span.recordException(error);
    throw error;
  } finally {
    span.end();
  }
}

async function createProduct(name, price) {
  const span = tracer.startSpan('mysql.insert');

  try {
    span.setAttribute('db.system', 'mysql');
    span.setAttribute('db.name', 'mydatabase');
    span.setAttribute('db.operation', 'INSERT');
    span.setAttribute('db.statement', 'INSERT INTO products (name, price) VALUES (?, ?)');

    const [result] = await pool.query(
      'INSERT INTO products (name, price) VALUES (?, ?)',
      [name, price]
    );

    span.setAttribute('db.rows_affected', result.affectedRows);
    span.setAttribute('db.mysql.insert_id', result.insertId);

    return result;
  } catch (error) {
    span.recordException(error);
    throw error;
  } finally {
    span.end();
  }
}

Java Auto-Instrumentation (using Java Agent)

1. Download the OpenTelemetry Java agent:

wget https://github.com/open-telemetry/opentelemetry-java-instrumentation/releases/latest/download/opentelemetry-javaagent.jar

2. Run your application with the agent:

java -javaagent:path/to/opentelemetry-javaagent.jar \
     -Dotel.service.name=mysql-service \
     -Dotel.exporter.otlp.endpoint=http://otelcol.apitoolkit.io:4317 \
     -Dotel.resource.attributes=at-project-key=YOUR_API_KEY \
     -jar your-application.jar

Node.js Auto-Instrumentation

1. Install required packages:

npm install @opentelemetry/sdk-node @opentelemetry/auto-instrumentations-node @opentelemetry/exporter-trace-otlp-proto

2. Set up auto-instrumentation in your application:

const { NodeSDK } = require('@opentelemetry/sdk-node');
const { getNodeAutoInstrumentations } = require('@opentelemetry/auto-instrumentations-node');
const { OTLPTraceExporter } = require('@opentelemetry/exporter-trace-otlp-proto');

const sdk = new NodeSDK({
  traceExporter: new OTLPTraceExporter({
    url: 'http://otelcol.apitoolkit.io:4317/v1/traces',
  }),
  instrumentations: [getNodeAutoInstrumentations()],
  resourceAttributes: {
    'at-project-key': 'YOUR_API_KEY',
    'service.name': 'mysql-service',
  },
});

// Initialize the SDK before importing MySQL
sdk.start();

// Now it's safe to import and use MySQL
const mysql = require('mysql2/promise');