Improve interruption handling.
This commit is contained in:
parent
f19ca0e8a5
commit
aa474d3623
9
main.go
9
main.go
|
@ -20,6 +20,7 @@ import (
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"os/signal"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -30,10 +31,16 @@ func main() {
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() {
|
go func() {
|
||||||
|
notifier := make(chan os.Signal)
|
||||||
|
signal.Notify(notifier, os.Interrupt)
|
||||||
|
<-notifier
|
||||||
m.Close()
|
m.Close()
|
||||||
|
os.Exit(0)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
defer m.Close()
|
||||||
|
|
||||||
results := make(chan retrieval.Result, 4096)
|
results := make(chan retrieval.Result, 4096)
|
||||||
|
|
||||||
t := &retrieval.Target{
|
t := &retrieval.Target{
|
||||||
|
|
Loading…
Reference in New Issue