Check if there is any connected scraper

handlers/scraper/remote.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"fmt"
 	"net"
+	"sync/atomic"
 	"syscall"
 	"time"
 
@@ -16,9 +17,11 @@ type remoteResult struct {
 	outChan   chan error
 }
 
+var connStream atomic.Int32
 var inChan chan remoteResult
 
 func init() {
+	connStream.Store(0)
 	inChan = make(chan remoteResult)
 
 	ln, err := net.Listen("tcp", "localhost:4444")
@@ -53,6 +56,8 @@ func handleConnection(conn net.Conn) {
 
 		buf := make([]byte, 64*1024)
 		go func() {
+			connStream.Add(1)
+			defer connStream.Add(-1)
 			for {
 				select {
 				case rm := <-inChan:
@@ -86,6 +91,10 @@ func handleConnection(conn net.Conn) {
 }
 
 func ScrapeRemote(i *InstaData) error {
+	if connStream.Load() == 0 {
+		return errors.New("Remote scraper is not running")
+	}
+
 	remoteRes := remoteResult{
 		instaData: i,
 		outChan:   make(chan error),