fix (dot/telemetry): NoTelemetry flag stops telemetry

dot/node.go

@@ -341,7 +341,7 @@ func NewNode(cfg *Config, ks *keystore.GlobalKeystore, stopFunc func()) (*Node,
 	}
 
 	if cfg.Global.NoTelemetry {
-		return node, nil
+		telemetry.GetInstance().KillInstance()
 	}
 
 	telemetry.GetInstance().AddConnections(gd.TelemetryEndpoints)

dot/telemetry/telemetry.go

@@ -45,6 +45,14 @@ type Handler struct {
 	log         log.Logger
 }
 
+// Instance functions that telemetry handler instance needs to implement
+type Instance interface {
+	AddConnections(conns []*genesis.TelemetryEndpoint)
+	SendMessage(msg *Message) error
+	startListening()
+	KillInstance()
+}
+
 // KeyValue object to hold key value pairs used in telemetry messages
 type KeyValue struct {
 	key   string
@@ -53,11 +61,11 @@ type KeyValue struct {
 
 var (
 	once            sync.Once
-	handlerInstance *Handler
+	handlerInstance Instance
 )
 
 // GetInstance singleton pattern to for accessing TelemetryHandler
-func GetInstance() *Handler { //nolint
+func GetInstance() Instance { //nolint
 	if handlerInstance == nil {
 		once.Do(
 			func() {
@@ -134,6 +142,11 @@ func (h *Handler) startListening() {
 	}
 }
 
+// KillInstance replaces current telemetry handler instance with no op telemetry handler instance
+func (h *Handler) KillInstance() {
+	handlerInstance = &NoopHandler{}
+}
+
 type response struct {
 	ID        int                    `json:"id"`
 	Payload   map[string]interface{} `json:"payload"`
@@ -152,3 +165,20 @@ func msgToBytes(message Message) []byte {
 	}
 	return resB
 }
+
+// NoopHandler struct no op handling (ignoring) telemetry messages
+type NoopHandler struct {
+}
+
+func (h *NoopHandler) startListening() {}
+
+// SendMessage no op for telemetry send message function
+func (h *NoopHandler) SendMessage(msg *Message) error {
+	return nil
+}
+
+// AddConnections no op for telemetry add connections function
+func (h *NoopHandler) AddConnections(conns []*genesis.TelemetryEndpoint) {}
+
+// KillInstance no op for telemetry kill instance function
+func (h *NoopHandler) KillInstance() {}

dot/telemetry/telemetry_test.go

@@ -139,6 +139,42 @@ func TestListenerConcurrency(t *testing.T) {
 	}
 }
 
+func TestKillInstance(t *testing.T) {
+	const qty = 1000
+	var wg sync.WaitGroup
+	wg.Add(qty)
+
+	resultCh = make(chan []byte)
+	for i := 0; i < qty; i++ {
+		if i == qty/2 {
+			GetInstance().KillInstance()
+		}
+		go func() {
+			GetInstance().SendMessage(NewTelemetryMessage(
+				NewKeyValue("best", "hash"),
+				NewKeyValue("height", big.NewInt(2)),
+				NewKeyValue("msg", "block.import"),
+				NewKeyValue("origin", "NetworkInitialSync")))
+			wg.Done()
+		}()
+	}
+	wg.Wait()
+	counter := 0
+	tk := time.NewTicker(time.Second * 2)
+main:
+	for {
+		select {
+		case <-tk.C:
+			break main
+		case <-resultCh:
+			counter++
+		}
+	}
+	tk.Stop()
+
+	require.LessOrEqual(t, counter, qty/2)
+}
+
 func listen(w http.ResponseWriter, r *http.Request) {
 	c, err := upgrader.Upgrade(w, r, nil)
 	if err != nil {