From d2c5f075aa1e85bf92f7558b8db6dce3cdb93add Mon Sep 17 00:00:00 2001
From: Geoffrey Ragot <geoffrey.ragot@gmail.com>
Date: Sun, 8 Sep 2024 13:38:22 +0200
Subject: [PATCH] feat: add traces

---
 components/ledger/internal/controller/ledger/resolver.go | 3 ++-
 components/ledger/internal/storage/bucket/migrations.go  | 4 ++++
 components/ledger/internal/storage/ledger/migrations.go  | 4 ++++
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/components/ledger/internal/controller/ledger/resolver.go b/components/ledger/internal/controller/ledger/resolver.go
index 9c32d15208..b9fa9e89f9 100644
--- a/components/ledger/internal/controller/ledger/resolver.go
+++ b/components/ledger/internal/controller/ledger/resolver.go
@@ -63,7 +63,8 @@ func (r *Resolver) GetLedger(ctx context.Context, name string) (*Controller, err
 		return nil, err
 	}
 
-	r.metricsRegistry.ActiveLedgers().Add(ctx, +1)
+	// todo: add only once
+	//r.metricsRegistry.ActiveLedgers().Add(ctx, +1)
 
 	return New(
 		name,
diff --git a/components/ledger/internal/storage/bucket/migrations.go b/components/ledger/internal/storage/bucket/migrations.go
index c8c3da4030..fab0a97141 100644
--- a/components/ledger/internal/storage/bucket/migrations.go
+++ b/components/ledger/internal/storage/bucket/migrations.go
@@ -5,6 +5,7 @@ import (
 	"context"
 	"embed"
 	_ "embed"
+	"github.com/formancehq/ledger/internal/opentelemetry/tracer"
 	"github.com/formancehq/stack/libs/go-libs/migrations"
 	"github.com/uptrace/bun"
 	"text/template"
@@ -32,5 +33,8 @@ func getMigrator(name string) *migrations.Migrator {
 }
 
 func Migrate(ctx context.Context, db bun.IDB, name string) error {
+	ctx, span := tracer.Start(ctx, "Migrate bucket")
+	defer span.End()
+
 	return getMigrator(name).Up(ctx, db)
 }
diff --git a/components/ledger/internal/storage/ledger/migrations.go b/components/ledger/internal/storage/ledger/migrations.go
index 6b96a028b5..fd0ae17772 100644
--- a/components/ledger/internal/storage/ledger/migrations.go
+++ b/components/ledger/internal/storage/ledger/migrations.go
@@ -6,6 +6,7 @@ import (
 	"embed"
 	_ "embed"
 	"fmt"
+	"github.com/formancehq/ledger/internal/opentelemetry/tracer"
 	"github.com/formancehq/stack/libs/go-libs/migrations"
 	"github.com/uptrace/bun"
 	"text/template"
@@ -37,5 +38,8 @@ func getMigrator(bucketName, ledgerName string) *migrations.Migrator {
 }
 
 func Migrate(ctx context.Context, db bun.IDB, bucketName, ledgerName string) error {
+	ctx, span := tracer.Start(ctx, "Migrate ledger")
+	defer span.End()
+
 	return getMigrator(bucketName, ledgerName).Up(ctx, db)
 }