Merge pull request #12 from adhocore/ctrl-c

Ctrl C

Author: adhocore

cmd/fast/main.go

@@ -3,7 +3,10 @@ package main
 import (
 	"flag"
 	"log"
+	"os"
+	"os/signal"
 	"sync"
+	"syscall"
 	"time"
 
 	"github.com/adhocore/chin"
@@ -23,9 +26,17 @@ func main() {
 	s := chin.New().WithWait(&wg)
 	go s.Start()
 
+	var sig = make(chan os.Signal, 1)
+	signal.Notify(sig, os.Interrupt, syscall.SIGTERM, syscall.SIGABRT)
+
 	wg.Add(1)
 	go doFast(s, &wg, noUp)
 
+	go func() {
+		<-sig
+		s.Stop()
+	}()
+
 	wg.Wait()
 }
 

fast.go

@@ -61,7 +61,7 @@ func Measure(noUp bool) (*Fast, error) {
 
 	var err error
 	for _, browser := range browsers {
-		err = doMeasure(browser, fast, cmds...)
+		err = doMeasure(browser, cmds...)
 		if err == nil || !errors.Is(err, exec.ErrNotFound) {
 			break
 		}
@@ -70,7 +70,7 @@ func Measure(noUp bool) (*Fast, error) {
 	return fast, err
 }
 
-func doMeasure(browser string, fast *Fast, cmds ...chromedp.Action) error {
+func doMeasure(browser string, cmds ...chromedp.Action) error {
 	opts := chromedp.DefaultExecAllocatorOptions[:]
 	if browser != "" {
 		opts = append(opts, chromedp.ExecPath(browser))