moved sigterm handler out of the library

alessio-perugini · May 13, 2020 · 3a6db79 · 3a6db79
1 parent 058760c
commit 3a6db79
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -8,6 +8,7 @@ import (
 	"log"
 	"net/url"
 	"os"
+	"os/signal"
 	"time"
 )
 
@@ -123,4 +124,8 @@ func main() {
 
 	peng := p.New(&config)
 	peng.Start()
+	sig := make(chan os.Signal, 1024)
+	signal.Notify(sig, os.Interrupt)
+	<-sig
+	log.Println("Quitting Peng, bye!")
 }
diff --git a/peng.go b/peng.go
@@ -7,8 +7,6 @@ import (
 	_ "github.com/google/gopacket/layers" //Used to init internal struct
 	"github.com/google/gopacket/pcap"
 	"log"
-	"os"
-	"os/signal"
 	"time"
 )
 
@@ -71,11 +69,6 @@ func (p *Peng) Start() {
 			p.inspect(packet)
 		}
 	}()
-
-	sig := make(chan os.Signal, 1024)
-	signal.Notify(sig, os.Interrupt)
-	<-sig
-	log.Println("Quitting Peng, bye!")
 }
 
 func (p *Peng) PrintAllInfo() {