Refactor to handle multiple traces in repository

internal/repository/interfaces.go

@@ -6,5 +6,5 @@ import (
 )
 
 type Repository interface {
-	SaveTrace(ctx context.Context, trace *models.Trace) error
+	SaveTraces(ctx context.Context, traces []*models.Trace) error
 }

internal/repository/mongodb.go

@@ -25,7 +25,16 @@ func NewMongoRepository(uri string) (*MongoRepository, error) {
 	}, nil
 }
 
-func (r *MongoRepository) SaveTrace(ctx context.Context, trace *models.Trace) error {
-	_, err := r.traceCollection.InsertOne(ctx, trace)
+func (r *MongoRepository) SaveTraces(ctx context.Context, traces []*models.Trace) error {
+	if len(traces) == 0 {
+		return nil
+	}
+
+	documents := make([]interface{}, len(traces))
+	for i, trace := range traces {
+		documents[i] = trace
+	}
+
+	_, err := r.traceCollection.InsertMany(ctx, documents)
 	return err
 }

internal/server/trace_server.go

@@ -23,9 +23,9 @@ func NewTraceServer(traceService *service.TraceService) *TraceServer {
 
 func (s *TraceServer) Export(ctx context.Context, req *collectorpb.ExportTraceServiceRequest) (*collectorpb.ExportTraceServiceResponse, error) {
 	for _, resourceSpans := range req.ResourceSpans {
-		trace := telemetry.ConvertResourceSpansToTrace(resourceSpans)
-		if trace != nil {
-			err := s.traceService.ProcessTrace(ctx, trace)
+		traces := telemetry.ConvertResourceSpansToTraces(resourceSpans)
+		if traces != nil {
+			err := s.traceService.ProcessTrace(ctx, traces)
 			if err != nil {
 				return nil, status.Errorf(codes.Internal, "Failed to process trace: %v", err)
 			}

internal/service/trace_service.go

@@ -14,6 +14,6 @@ func NewTraceService(repo repository.Repository) *TraceService {
 	return &TraceService{repo: repo}
 }
 
-func (s *TraceService) ProcessTrace(ctx context.Context, trace *models.Trace) error {
-	return s.repo.SaveTrace(ctx, trace)
+func (s *TraceService) ProcessTrace(ctx context.Context, traces []*models.Trace) error {
+	return s.repo.SaveTraces(ctx, traces)
 }