Gracefully close consumers

Author: anhthii

cmd/mpcium/main.go

@@ -201,6 +201,14 @@ func runNode(ctx context.Context, c *cli.Command) error {
 		<-sigChan
 		logger.Warn("Shutdown signal received, canceling context...")
 		cancel()
+
+		// Gracefully close consumers
+		if err := keygenConsumer.Close(); err != nil {
+			logger.Error("Failed to close keygen consumer", err)
+		}
+		if err := signingConsumer.Close(); err != nil {
+			logger.Error("Failed to close signing consumer", err)
+		}
 	}()
 
 	var wg sync.WaitGroup

examples/generate/main.go

@@ -61,6 +61,7 @@ func main() {
 
 	// STEP 2: Register the result handler AFTER all walletIDs are stored
 	err = mpcClient.OnWalletCreationResult(func(event event.KeygenResultEvent) {
+		logger.Info("Received wallet creation result", "event", event)
 		now := time.Now()
 		startTimeAny, ok := walletStartTimes.Load(event.WalletID)
 		if ok {
@@ -80,7 +81,6 @@ func main() {
 		} else {
 			logger.Warn("Received wallet result but no start time found", "walletID", event.WalletID)
 		}
-		logger.Info("Wallet creation result received", "event", event)
 		wg.Done()
 	})
 	if err != nil {