Add log tracer

authzed · Jun 21, 2021 · e73cd23 · e73cd23
1 parent 266aa0c
commit e73cd23
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/internal/datastore/postgres/log_tracer.go b/internal/datastore/postgres/log_tracer.go
@@ -0,0 +1,17 @@
+package postgres
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v4"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/trace"
+)
+
+type tracingLogger struct {
+}
+
+func (tl tracingLogger) Log(ctx context.Context, level pgx.LogLevel, msg string, data map[string]interface{}) {
+	span := trace.SpanFromContext(ctx)
+	span.AddEvent(msg, trace.WithAttributes(attribute.Stringer("level", level)))
+}
diff --git a/internal/datastore/postgres/options.go b/internal/datastore/postgres/options.go
@@ -18,7 +18,7 @@ type postgresOptions struct {
 
 	enablePrometheusStats bool
 
-	driver string
+	logger *tracingLogger
 }
 
 const (
@@ -33,7 +33,6 @@ func generateConfig(options []PostgresOption) (postgresOptions, error) {
 	computed := postgresOptions{
 		gcWindow:          24 * time.Hour,
 		watchBufferLength: defaultWatchBufferLength,
-		driver:            "postgres",
 	}
 
 	for _, option := range options {
@@ -108,6 +107,6 @@ func EnablePrometheusStats() PostgresOption {
 
 func EnableTracing() PostgresOption {
 	return func(po *postgresOptions) {
-		po.driver = tracingDriverName
+		po.logger = &tracingLogger{}
 	}
 }