simon987/od-database-crawler
1
0
Fork 0
mirror of https://github.com/terorie/od-database-crawler.git synced 2025-04-18 01:46:43 +00:00
Code Issues Projects Releases Wiki Activity

Fix too long sleep

Browse Source
This commit is contained in:
Richard Patel 2019-02-03 16:28:43 +01:00
parent e919323169
commit ef7d17cad4
No known key found for this signature in database
GPG Key ID: C268B2BBDA2ABECB
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
main.go
View File

@ -97,7 +97,7 @@ func cmdBase(_ *cobra.Command, _ []string) {
if err != nil {
logrus.WithError(err).
Error("Failed to get new task")
time.Sleep(viper.GetDuration(ConfCooldown))
sleep(viper.GetDuration(ConfCooldown), appCtx)
continue
}
if t == nil {
@ -189,3 +189,12 @@ func hardShutdown(c context.Context) {
<-c.Done()
os.Exit(1)
}
func sleep(d time.Duration, c context.Context) {
select {
case <-time.After(d):
break
case <-c.Done():
break
}
}