Mock Windows signal handler (#276)

Author: svkirillov

internal/scanner/scanner.go

@@ -10,8 +10,6 @@ import (
 	"math/rand"
 	"net/http"
 	"net/url"
-	"os"
-	"os/signal"
 	"regexp"
 	"strings"
 	"sync"
@@ -321,32 +319,11 @@ func (s *Scanner) Run(ctx context.Context) error {
 	// separately.
 	var requestsCounter uint64
 
-	userSignal := make(chan os.Signal, 1)
-	signal.Notify(userSignal, syscall.SIGUSR1)
-	defer func() {
-		signal.Stop(userSignal)
-		close(userSignal)
-	}()
-
-	go func() {
-		for {
-			select {
-			case _, ok := <-userSignal:
-				if !ok {
-					return
-				}
-
-				s.logger.
-					WithFields(logrus.Fields{
-						"sent":  atomic.LoadUint64(&requestsCounter),
-						"total": s.db.NumberOfTests,
-					}).Info("Testing status")
-
-			case <-ctx.Done():
-				return
-			}
-		}
-	}()
+	closeListener, err := s.testStatusSignalHandler(ctx, &requestsCounter)
+	if err != nil {
+		return err
+	}
+	defer closeListener()
 
 	for e := 0; e < gn; e++ {
 		go func() {

internal/scanner/scanner_signal_handler_unix.go

@@ -0,0 +1,41 @@
+package scanner
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"sync/atomic"
+	"syscall"
+
+	"github.com/sirupsen/logrus"
+)
+
+func (s *Scanner) testStatusSignalHandler(ctx context.Context, requestsCounter *uint64) (func(), error) {
+	userSignal := make(chan os.Signal, 1)
+	signal.Notify(userSignal, syscall.SIGUSR1)
+
+	go func() {
+		for {
+			select {
+			case _, ok := <-userSignal:
+				if !ok {
+					return
+				}
+
+				s.logger.
+					WithFields(logrus.Fields{
+						"sent":  atomic.LoadUint64(requestsCounter),
+						"total": s.db.NumberOfTests,
+					}).Info("Testing status")
+
+			case <-ctx.Done():
+				return
+			}
+		}
+	}()
+
+	return func() {
+		signal.Stop(userSignal)
+		close(userSignal)
+	}, nil
+}

internal/scanner/scanner_signal_handler_windows.go

@@ -0,0 +1,9 @@
+package scanner
+
+import (
+	"context"
+)
+
+func (s *Scanner) testStatusSignalHandler(_ context.Context, _ *uint64) (func(), error) {
+	return func() {}, nil
+}