fix: Flush sinks in grpc services when shutting down (#143)

* fix: Flush sinks in grpc services when shutting down * chore: Add log message when stopping module
ethpandaops · Jul 17, 2023 · 693a530 · 693a530
1 parent bd929ad
commit 693a530
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/pkg/output/stdout/stdout.go b/pkg/output/stdout/stdout.go
@@ -61,7 +61,7 @@ func (h *StdOut) Start(ctx context.Context) error {
 }
 
 func (h *StdOut) Stop(ctx context.Context) error {
-	return nil
+	return h.proc.Shutdown(ctx)
 }
 
 func (h *StdOut) HandleNewDecoratedEvent(ctx context.Context, event *xatu.DecoratedEvent) error {

diff --git a/pkg/server/service/event-ingester/ingester.go b/pkg/server/service/event-ingester/ingester.go
@@ -54,6 +54,12 @@ func (e *Ingester) Start(ctx context.Context, grpcServer *grpc.Server) error {
 func (e *Ingester) Stop(ctx context.Context) error {
 	e.log.Info("stopping module")
 
+	for _, sink := range e.sinks {
+		if err := sink.Stop(ctx); err != nil {
+			return err
+		}
+	}
+
 	return nil
 }