tracing.Shutdown

Author: vodolaz095

tracing/exporter.go

@@ -12,7 +12,7 @@ const defaultTimeout = 10 * time.Second
 
 var exp trace.SpanExporter
 
-// Wait allows span exporter to inhibit application shutdown until it sends all traces
+// Wait allows span exporter to inhibit application shutdown until it sends all traces.
 func Wait(ctx context.Context) (err error) {
 	if exp != nil {
 		<-ctx.Done()
@@ -28,3 +28,19 @@ func Wait(ctx context.Context) (err error) {
 	}
 	return nil
 }
+
+// Shutdown forces exporter to save all spans and turn off.
+func Shutdown(ctx context.Context) (err error) {
+	if exp != nil {
+		shutdownContext, cancel := context.WithTimeout(ctx, defaultTimeout)
+		defer cancel()
+		err = exp.Shutdown(shutdownContext)
+		if err != nil {
+			log.Error().Err(err).Msgf("Error saving telemetry spans: %s", err)
+			return err
+		}
+		log.Debug().Msgf("All spans are saved")
+		return nil
+	}
+	return nil
+}