Enable otel for sql connections (#1389)

Author: djeebus

iac/provider-gcp/nomad/configs/otel-collector.yaml

@@ -139,6 +139,7 @@ processors:
           - "orchestrator.*"
           - "template.*"
           - "api.*"
+          - "db.sql.connection.*"
           - "vault.*"
           - "client_proxy.*"
           - "Click*"
@@ -308,4 +309,4 @@ service:
         - otlp
       processors: [batch]
       exporters:
-        - otlphttp/grafana_cloud
+        - otlphttp/grafana_cloud

packages/api/go.mod

@@ -83,6 +83,7 @@ require (
 	github.com/Masterminds/goutils v1.1.1 // indirect
 	github.com/Masterminds/sprig/v3 v3.2.3 // indirect
 	github.com/Workiva/go-datastructures v1.1.0 // indirect
+	github.com/XSAM/otelsql v0.40.0 // indirect
 	github.com/agext/levenshtein v1.2.3 // indirect
 	github.com/air-verse/air v1.61.7 // indirect
 	github.com/alecthomas/units v0.0.0-20211218093645-b94a6e3cc137 // indirect

packages/api/go.sum

@@ -12,9 +12,12 @@ require (
 require (
 	ariga.io/atlas v0.15.0 // indirect
 	entgo.io/ent v0.12.5 // indirect
+	github.com/XSAM/otelsql v0.40.0 // indirect
 	github.com/agext/levenshtein v1.2.3 // indirect
 	github.com/apparentlymart/go-textseg/v15 v15.0.0 // indirect
 	github.com/dchest/uniuri v1.2.0 // indirect
+	github.com/go-logr/logr v1.4.3 // indirect
+	github.com/go-logr/stdr v1.2.2 // indirect
 	github.com/go-openapi/inflect v0.21.0 // indirect
 	github.com/go-test/deep v1.0.8 // indirect
 	github.com/google/go-cmp v0.7.0 // indirect
@@ -23,6 +26,10 @@ require (
 	github.com/lib/pq v1.10.9 // indirect
 	github.com/mitchellh/go-wordwrap v1.0.1 // indirect
 	github.com/zclconf/go-cty v1.14.1 // indirect
+	go.opentelemetry.io/auto/sdk v1.2.0 // indirect
+	go.opentelemetry.io/otel v1.38.0 // indirect
+	go.opentelemetry.io/otel/metric v1.38.0 // indirect
+	go.opentelemetry.io/otel/trace v1.38.0 // indirect
 	golang.org/x/mod v0.27.0 // indirect
 	golang.org/x/sync v0.17.0 // indirect
 	golang.org/x/text v0.29.0 // indirect

packages/docker-reverse-proxy/go.mod

@@ -21,6 +21,7 @@ require (
 	entgo.io/ent v0.12.5 // indirect
 	github.com/Azure/go-ansiterm v0.0.0-20250102033503-faa5f7b0171c // indirect
 	github.com/Microsoft/go-winio v0.6.2 // indirect
+	github.com/XSAM/otelsql v0.40.0 // indirect
 	github.com/agext/levenshtein v1.2.3 // indirect
 	github.com/apparentlymart/go-textseg/v15 v15.0.0 // indirect
 	github.com/cenkalti/backoff/v4 v4.3.0 // indirect

packages/docker-reverse-proxy/go.sum

@@ -24,6 +24,7 @@ require (
 	cloud.google.com/go/storage v1.50.0
 	connectrpc.com/connect v1.18.1
 	entgo.io/ent v0.12.5
+	github.com/XSAM/otelsql v0.40.0
 	github.com/aws/aws-sdk-go-v2 v1.36.3
 	github.com/aws/aws-sdk-go-v2/config v1.29.14
 	github.com/aws/aws-sdk-go-v2/feature/s3/manager v1.17.74

packages/local-dev/go.mod

@@ -6,8 +6,10 @@ import (
 	"time"
 
 	"entgo.io/ent/dialect"
-	"entgo.io/ent/dialect/sql"
+	entsql "entgo.io/ent/dialect/sql"
+	"github.com/XSAM/otelsql"
 	_ "github.com/lib/pq"
+	semconv "go.opentelemetry.io/otel/semconv/v1.28.0"
 
 	"github.com/e2b-dev/infra/packages/shared/pkg/models"
 )
@@ -22,17 +24,24 @@ func NewClient(maxConns, maxIdle int) (*DB, error) {
 		return nil, fmt.Errorf("database URL is empty")
 	}
 
-	drv, err := sql.Open(dialect.Postgres, databaseURL)
+	db, err := otelsql.Open(dialect.Postgres, databaseURL, otelsql.WithAttributes(
+		semconv.DBSystemPostgreSQL,
+	))
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to open db: %w", err)
+	}
+
+	if err = otelsql.RegisterDBStatsMetrics(db, otelsql.WithAttributes(semconv.DBSystemPostgreSQL)); err != nil {
+		return nil, fmt.Errorf("failed to register db stats metrics: %w", err)
 	}
 
 	// Get the underlying sql.DB object of the driver.
-	db := drv.DB()
 	db.SetMaxOpenConns(maxConns)
 	db.SetMaxIdleConns(maxIdle)
 	db.SetConnMaxLifetime(time.Minute * 30)
 
+	drv := entsql.OpenDB(dialect.Postgres, db)
+
 	client := models.NewClient(models.Driver(drv))
 
 	return &DB{Client: client}, nil