fix: flush everything when quitting

Signed-off-by: Vincent Boutour <bob@vibioh.fr>
ViBiOh · Jan 28, 2024 · 8f93404 · 8f93404
1 parent 7fdddd5
commit 8f93404
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/telemetry/telemetry.go b/pkg/telemetry/telemetry.go
@@ -158,12 +158,20 @@ func (s Service) Middleware(name string) func(next http.Handler) http.Handler {
 
 func (s Service) Close(ctx context.Context) {
 	if s.tracerProvider != nil {
+		if err := s.tracerProvider.ForceFlush(ctx); err != nil {
+			slog.LogAttrs(ctx, slog.LevelError, "flush trace provider", slog.Any("error", err))
+		}
+
 		if err := s.tracerProvider.Shutdown(ctx); err != nil {
 			slog.LogAttrs(ctx, slog.LevelError, "shutdown trace provider", slog.Any("error", err))
 		}
 	}
 
 	if s.meterProvider != nil {
+		if err := s.meterProvider.ForceFlush(ctx); err != nil {
+			slog.LogAttrs(ctx, slog.LevelError, "flush meter provider", slog.Any("error", err))
+		}
+
 		if err := s.meterProvider.Shutdown(ctx); err != nil {
 			slog.LogAttrs(ctx, slog.LevelError, "shutdown meter provider", slog.Any("error", err))
 		}