chore(relayer): close db connection before stop service. (#17709)

packages/relayer/event.go

@@ -128,6 +128,7 @@ type FindAllByAddressOpts struct {
 
 // EventRepository is used to interact with events in the store
 type EventRepository interface {
+	Close() error
 	Save(ctx context.Context, opts *SaveEventOpts) (*Event, error)
 	UpdateStatus(ctx context.Context, id int, status EventStatus) error
 	UpdateFeesAndProfitability(ctx context.Context, id int, opts *UpdateFeesAndProfitabilityOpts) error

packages/relayer/indexer/indexer.go

@@ -252,6 +252,11 @@ func (i *Indexer) Name() string {
 // context is stopped externally by cmd/main.go shutdown.
 func (i *Indexer) Close(ctx context.Context) {
 	i.wg.Wait()
+
+	// Close db connection.
+	if err := i.eventRepo.Close(); err != nil {
+		slog.Error("Failed to close db connection", "err", err)
+	}
 }
 
 // Start starts the indexer, which should initialize the queue, add to wait groups,

packages/relayer/pkg/http/server.go

@@ -2,6 +2,7 @@ package http
 
 import (
 	"context"
+	"log/slog"
 	"math/big"
 	"net/http"
 	"os"
@@ -130,6 +131,11 @@ func (srv *Server) Start(address string) error {
 
 // Shutdown shuts down the HTTP server
 func (srv *Server) Shutdown(ctx context.Context) error {
+	// Close db connection.
+	if err := srv.eventRepo.Close(); err != nil {
+		slog.Error("Failed to close db connection", "err", err)
+	}
+
 	return srv.echo.Shutdown(ctx)
 }
 

packages/relayer/pkg/mock/event_repository.go

@@ -22,6 +22,11 @@ func NewEventRepository() *EventRepository {
 		events: make([]*relayer.Event, 0),
 	}
 }
+
+func (r *EventRepository) Close() error {
+	return nil
+}
+
 func (r *EventRepository) Save(ctx context.Context, opts *relayer.SaveEventOpts) (*relayer.Event, error) {
 	r.events = append(r.events, &relayer.Event{
 		ID:           rand.Int(), // nolint: gosec

packages/relayer/pkg/repo/event.go

@@ -30,6 +30,16 @@ func NewEventRepository(dbHandler db.DB) (*EventRepository, error) {
 	}, nil
 }
 
+// Close closes the database connection.
+func (r *EventRepository) Close() error {
+	sqlDB, err := r.db.DB()
+	if err != nil {
+		return err
+	}
+
+	return sqlDB.Close()
+}
+
 func (r *EventRepository) Save(ctx context.Context, opts *relayer.SaveEventOpts) (*relayer.Event, error) {
 	e := &relayer.Event{
 		Data:                   datatypes.JSON(opts.Data),

packages/relayer/processor/processor.go

@@ -383,6 +383,11 @@ func (p *Processor) Close(ctx context.Context) {
 	p.cancel()
 
 	p.wg.Wait()
+
+	// Close db connection.
+	if err := p.eventRepo.Close(); err != nil {
+		slog.Error("Failed to close db connection", "err", err)
+	}
 }
 
 func (p *Processor) Start() error {

packages/relayer/watchdog/watchdog.go

@@ -202,6 +202,11 @@ func (w *Watchdog) Close(ctx context.Context) {
 	w.cancel()
 
 	w.wg.Wait()
+
+	// Close db connection.
+	if err := w.eventRepo.Close(); err != nil {
+		slog.Error("Failed to close db connection", "err", err)
+	}
 }
 
 func (w *Watchdog) Start() error {