capture ctrl-c and close GroupConsumer

command/cmd/group-consumer.go

@@ -5,6 +5,9 @@ import (
 	"errors"
 	"fmt"
 	"math"
+	"os"
+	"os/signal"
+	"syscall"
 
 	"github.com/childe/healer"
 	"github.com/spf13/cobra"
@@ -59,25 +62,30 @@ var groupConsumerCmd = &cobra.Command{
 		if err != nil {
 			return err
 		}
+		sigChan := make(chan os.Signal, 1)
+		signal.Notify(sigChan, os.Interrupt, syscall.SIGINT)
+
 		defer consumer.Close()
 
 		messages, err := consumer.Consume(nil)
 		if err != nil {
 			return err
 		}
 
-		if maxMessages <= 0 {
-			for {
-				message := <-messages
-				fmt.Printf("%d: %s\n", message.Message.Offset, message.Message.Value)
-			}
-		} else {
-			for i := 0; i < maxMessages; i++ {
+		i := 0
+		for {
+			select {
+			case <-sigChan:
+				return nil
+			default:
 				message := <-messages
 				fmt.Printf("%d: %s\n", message.Message.Offset, message.Message.Value)
+				i++
+				if maxMessages > 0 && i >= maxMessages {
+					return nil
+				}
 			}
 		}
-		return nil
 	},
 }