39 lines
1.0 KiB
Go
39 lines
1.0 KiB
Go
package shutdown
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/beaconclient"
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/database/sql"
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/gracefulshutdown"
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/loghelper"
|
|
)
|
|
|
|
// Shutdown all the internal services for the application.
|
|
func ShutdownServices(ctx context.Context, waitTime time.Duration, DB sql.Database, BC *beaconclient.BeaconClient) error {
|
|
successCh, errCh := gracefulshutdown.Shutdown(ctx, waitTime, map[string]gracefulshutdown.Operation{
|
|
"database": func(ctx context.Context) error {
|
|
err := DB.Close()
|
|
if err != nil {
|
|
loghelper.LogError(err).Error("Unable to close the DB")
|
|
}
|
|
return err
|
|
},
|
|
"beaconClient": func(ctx context.Context) error {
|
|
err := BC.StopHeadTracking()
|
|
if err != nil {
|
|
loghelper.LogError(err).Error("Unable to trigger shutdown of head tracking")
|
|
}
|
|
return err
|
|
},
|
|
})
|
|
|
|
select {
|
|
case _ = <-successCh:
|
|
return nil
|
|
case err := <-errCh:
|
|
return err
|
|
}
|
|
}
|