d1bc6a816f
* -- Intermediary Commit -- Just want to commit my code over the weekend, in case I spill coffee on my workstation. * Create DB models ready for write. * Handle SSE events * Update ref for stack-orchestrator * Use env in one place only. * Boot Application on PR * Update syntax * Update syntax * Correct command * Use bash instead of sh * Use until instead of while * Make linter happy and check sse subscription err * Handle Reorgs - Untested
35 lines
1019 B
Go
35 lines
1019 B
Go
package shutdown
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"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, notifierCh chan os.Signal, waitTime time.Duration, DB sql.Database, BC *beaconclient.BeaconClient) error {
|
|
successCh, errCh := gracefulshutdown.Shutdown(ctx, notifierCh, waitTime, map[string]gracefulshutdown.Operation{
|
|
// Combining DB shutdown with BC because BC needs DB open to cleanly shutdown.
|
|
"beaconClient": func(ctx context.Context) error {
|
|
defer DB.Close()
|
|
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
|
|
}
|
|
}
|