feat: ctrl+c handling

chainservice/chainservice_test.go

@@ -9,9 +9,9 @@ import (
 	"github.com/chainbound/apollo/types"
 )
 
-const (
-	rpcUrl = "wss://arb-mainnet.g.alchemy.com/v2/5_JWUuiS1cewWFpLzRxdjgZM0yLA4Uqp"
-)
+// const (
+// 	rpcUrl = "wss://arb-mainnet.g.alchemy.com/v2/5_JWUuiS1cewWFpLzRxdjgZM0yLA4Uqp"
+// )
 
 func newChainService() *ChainService {
 	// ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)

main.go

@@ -4,7 +4,9 @@ import (
 	"context"
 	"fmt"
 	"os"
+	"os/signal"
 	"path"
+	"syscall"
 	"time"
 
 	_ "embed"
@@ -136,6 +138,7 @@ func Run(opts types.ApolloOpts) error {
 	defaultTimeout := time.Second * 30
 
 	service := chainservice.NewChainService(defaultTimeout, opts.RateLimit, cfg.Rpc)
+	setupCloseHandler(service)
 
 	out := output.NewOutputHandler()
 
@@ -193,3 +196,14 @@ func Run(opts types.ApolloOpts) error {
 
 	return nil
 }
+
+func setupCloseHandler(svc *chainservice.ChainService) {
+	c := make(chan os.Signal)
+	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+	go func() {
+		<-c
+		logger.Warn().Msg("ctrl+c pressed, exiting...")
+		svc.DumpMetrics()
+		os.Exit(0)
+	}()
+}