mirror of
https://github.com/aljazceru/crawler_v2.git
synced 2025-12-17 07:24:21 +01:00
extracted handle signals
This commit is contained in:
@@ -8,7 +8,10 @@ import (
|
||||
"github/pippellia-btc/crawler/pkg/redb"
|
||||
"github/pippellia-btc/crawler/pkg/walks"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"sync/atomic"
|
||||
"syscall"
|
||||
"time"
|
||||
)
|
||||
|
||||
@@ -210,3 +213,13 @@ func Promote(db redb.RedisDB, node graph.ID) error {
|
||||
|
||||
return db.Promote(ctx, node)
|
||||
}
|
||||
|
||||
// HandleSignals listens for OS signals and triggers context cancellation.
|
||||
func HandleSignals(cancel context.CancelFunc) {
|
||||
signals := make(chan os.Signal, 1)
|
||||
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-signals
|
||||
|
||||
log.Println("signal received. shutting down...")
|
||||
cancel()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user