Clean up scheduler and rest of store on CTRL C

services/store.go

@@ -1,6 +1,11 @@
 package services
 
 import (
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/smartcontractkit/chainlink-go/logger"
 	"github.com/smartcontractkit/chainlink-go/models"
 )
 
@@ -9,6 +14,8 @@ type Store struct {
 	Scheduler *Scheduler
 	Config    Config
 	KeyStore  *KeyStore
+	sigs      chan os.Signal
+	Exiter    func(int)
 }
 
 func NewStore(config Config) *Store {
@@ -18,14 +25,23 @@ func NewStore(config Config) *Store {
 		Scheduler: NewScheduler(orm),
 		Config:    config,
 		KeyStore:  NewKeyStore(config.KeysDir()),
+		Exiter:    os.Exit,
 	}
 }
 
 func (self *Store) Start() error {
+	self.sigs = make(chan os.Signal, 1)
+	signal.Notify(self.sigs, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		<-self.sigs
+		self.Close()
+		self.Exiter(1)
+	}()
 	return self.Scheduler.Start()
 }
 
 func (self *Store) Close() {
+	logger.Info("Gracefully exiting...")
 	self.Scheduler.Stop()
 	self.ORM.Close()
 }

services/store_test.go

@@ -0,0 +1,28 @@
+package services_test
+
+import (
+	"syscall"
+	"testing"
+
+	. "github.com/onsi/gomega"
+	"github.com/smartcontractkit/chainlink-go/internal/cltest"
+)
+
+func TestGracefulShutdown(t *testing.T) {
+	t.Parallel()
+	RegisterTestingT(t)
+	store := cltest.Store()
+	defer store.Close()
+
+	var completed bool
+	store.Exiter = func(code int) {
+		completed = true
+	}
+
+	store.Start()
+	syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
+
+	Eventually(func() bool {
+		return completed
+	}).Should(BeTrue())
+}