Add databse type to files repo spans

kevinbarbour · Mar 12, 2024 · 2c39cba · 2c39cba
1 parent fd35e8a
commit 2c39cba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/files/repo_files.go b/internal/files/repo_files.go
@@ -39,6 +39,7 @@ type sqlRepository struct {
 func (r *sqlRepository) Record(ctx context.Context, file AcceptedFile) error {
 	ctx, span := telemetry.StartSpan(ctx, "files-record", trace.WithAttributes(
 		attribute.String("achgateway.file_id", file.FileID),
+		attribute.String("achgateway.database", "mysql"),
 	))
 	defer span.End()
 
@@ -58,6 +59,7 @@ func (r *sqlRepository) Record(ctx context.Context, file AcceptedFile) error {
 func (r *sqlRepository) Cancel(ctx context.Context, fileID string) error {
 	ctx, span := telemetry.StartSpan(ctx, "files-cancel", trace.WithAttributes(
 		attribute.String("achgateway.file_id", fileID),
+		attribute.String("achgateway.database", "mysql"),
 	))
 	defer span.End()
 
@@ -81,6 +83,7 @@ type spannerRepository struct {
 func (r *spannerRepository) Record(ctx context.Context, file AcceptedFile) error {
 	ctx, span := telemetry.StartSpan(ctx, "files-record", trace.WithAttributes(
 		attribute.String("achgateway.file_id", file.FileID),
+		attribute.String("achgateway.database", "spanner"),
 	))
 	defer span.End()
 
@@ -98,6 +101,7 @@ func (r *spannerRepository) Record(ctx context.Context, file AcceptedFile) error
 func (r *spannerRepository) Cancel(ctx context.Context, fileID string) error {
 	ctx, span := telemetry.StartSpan(ctx, "files-cancel", trace.WithAttributes(
 		attribute.String("achgateway.file_id", fileID),
+		attribute.String("achgateway.database", "spanner"),
 	))
 	defer span.End()