DEV Community

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

Posted on

2 2

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!

Hostinger image

Get n8n VPS hosting 3x cheaper than a cloud solution

Get fast, easy, secure n8n VPS hosting from $4.99/mo at Hostinger. Automate any workflow using a pre-installed n8n application and no-code customization.

Start now

Top comments (0)

Billboard image

The Next Generation Developer Platform

Coherence is the first Platform-as-a-Service you can control. Unlike "black-box" platforms that are opinionated about the infra you can deploy, Coherence is powered by CNC, the open-source IaC framework, which offers limitless customization.

Learn more

👋 Kindness is contagious

Please leave a ❤️ or a friendly comment on this post if you found it helpful!

Okay