DEV Community

Cover image for sqltrace: A low-code intrusion library that provides SQL tracing capabilities
chenquan
chenquan

Posted on

sqltrace: A low-code intrusion library that provides SQL tracing capabilities

sqltrace

A low-code intrusion library that provides SQL tracing capabilities, suitable for any relational database (Sqlite3, MySQL, Oracle, SQL Server, PostgreSQL, TiDB, etc.) and ORM libraries for various relational database (gorm, xorm, sqlx, etc.)

GitHub logo chenquan / sqltrace

A low-code intrusion library that provides SQL tracing capabilities, suitable for any relational database (Sqlite3, MySQL, Oracle, SQL Server, PostgreSQL, TiDB, etc.) and ORM libraries for various relational database (gorm, xorm, sqlx, etc.)

sqltrace

A low-code intrusion library that provides SQL tracing capabilities, suitable for any relational database (Sqlite3, MySQL, Oracle, SQL Server PostgreSQL, TiDB, etc.) and ORM libraries for various relational database (gorm, xorm, sqlx, etc.)

😜installation

go get -u github.com/chenquan/sqltrace
Enter fullscreen mode Exit fullscreen mode

👏how to use

native

package main
import (
    "context"
    "database/sql"
    "fmt"

    "github.com/chenquan/sqltrace"
    "github.com/mattn/go-sqlite3"
    _ "github.com/mattn/go-sqlite3"
    "go.opentelemetry.io/otel"
    "go.opentelemetry.io/otel/trace"
)

func main() {
    // Create a sqlite3 driver with link tracking
    driver := sqltrace.NewDriver(sqltrace.Config{
        Name:           "sqlite3_trace",
        DataSourceName: "sqlite3",
        Endpoint:       "http://localhost:14268/api/traces",
        Sampler:        1,
        Batcher:        "jaeger",
    }, &sqlite3.SQLiteDriver{})
    defer sqltrace.StopAgent()

    // register new driver
    sql.Register("sqlite3_trace", driver)

    // open database
    db, err := sql.Open("sqlite3_trace", "identifier.sqlite")
    if err != nil {
        panic(err)
    }

    tracer := otel.
Enter fullscreen mode Exit fullscreen mode

😜installation

go get -u github.com/chenquan/sqltrace
Enter fullscreen mode Exit fullscreen mode

👏how to use

native

package main

import (
    "context"
    "database/sql"
    "fmt"

    "github.com/chenquan/sqltrace"
    "github.com/mattn/go-sqlite3"
    _ "github.com/mattn/go-sqlite3"
    "go.opentelemetry.io/otel"
    "go.opentelemetry.io/otel/trace"
)

func main() {
    // Create a sqlite3 driver with link tracking
    driver := sqltrace.NewDriver(sqltrace.Config{
        Name:           "sqlite3_trace",
        DataSourceName: "sqlite3",
        Endpoint:       "http://localhost:14268/api/traces",
        Sampler:        1,
        Batcher:        "jaeger",
    }, &sqlite3.SQLiteDriver{})
    defer sqltrace.StopAgent()

    // register new driver
    sql.Register("sqlite3_trace", driver)

    // open database
    db, err := sql.Open("sqlite3_trace", "identifier.sqlite")
    if err != nil {
        panic(err)
    }

    tracer := otel.GetTracerProvider().Tracer("sqlite3_trace")
    ctx, span := tracer.Start(context.Background(),
        "test",
        trace.WithSpanKind(trace.SpanKindClient),
    )
    defer span.End()

    db.ExecContext(ctx, `CREATE TABLE t
(
    age  integer,
    name TEXT
)`)
    db.ExecContext(ctx, "insert into t values (?,?)", 1, "chenquan")

    // transaction
    tx, err := db.BeginTx(ctx, nil)
    stmt, err := tx.PrepareContext(ctx, "select age+1 as age,name from t where age = ?;")
    stmt.QueryContext(ctx, 1)
    tx.Commit()

    rows, err := db.QueryContext(ctx, "select  age+1 as age,name from t;")
    for rows.Next() {
        var age int
        var name string
        err := rows.Scan(&age, &name)
        if err != nil {
            fmt.Println(err)
        }

        fmt.Println(age, name)
    }
}
Enter fullscreen mode Exit fullscreen mode

native

gorm

package main

import (
    "context"
    "database/sql"
    "fmt"

    "github.com/chenquan/sqltrace"
    "github.com/mattn/go-sqlite3"
    "go.opentelemetry.io/otel"
    "go.opentelemetry.io/otel/trace"
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

func main() {
    // Create a sqlite3 driver with link tracking
    driver := sqltrace.NewDriver(sqltrace.Config{
        Name:           "sqlite3_trace",
        DataSourceName: "sqlite3",
        Endpoint:       "http://localhost:14268/api/traces",
        Sampler:        1,
        Batcher:        "jaeger",
    }, &sqlite3.SQLiteDriver{})
    defer sqltrace.StopAgent()

    // register new driver
    sql.Register("sqlite3_trace", driver)

    db, err := gorm.Open(sqlite.Dialector{
        DriverName: "sqlite3_trace",
        DSN:        "identifier.sqlite",
    }, &gorm.Config{})
    if err != nil {
        panic(err)
    }
    tracer := otel.GetTracerProvider().Tracer("sqlite3_trace")
    ctx, span := tracer.Start(context.Background(),
        "test",
        trace.WithSpanKind(trace.SpanKindClient),
    )
    defer span.End()

    type T struct {
        Age  int    `db:"age"`
        Name string `db:"name"`
    }
    db = db.WithContext(ctx)
    db.Migrator().AutoMigrate(&T{})
    db.Create(&T{
        Age:  1,
        Name: "chenquan",
    })

    rows, err := db.Raw("select  age+1 as age,name from ts;").Rows()
    if err != nil {
        panic(err)
    }

    for rows.Next() {
        var age int
        var name string
        err := rows.Scan(&age, &name)
        if err != nil {
            fmt.Println(err)
        }

        fmt.Println(age, name)
    }
    db.Commit()
}


Enter fullscreen mode Exit fullscreen mode

gorm

⭐star

If you like or are using this project to learn or start your solution, please give it a star⭐. Thanks!

Top comments (0)