some code improvements

topfreegames · Feb 19, 2019 · 1a53280 · 1a53280
1 parent b88a08d
commit 1a53280
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/feedback/invalid_token.go b/feedback/invalid_token.go
@@ -248,6 +248,9 @@ func (i *InvalidTokenHandler) deleteTokensFromGame(tokens []string, game, platfo
 			MetricsTokensDeleteNonexistent, int64(len(tokens)-res.RowsAffected()),
 			game, platform)
 
+		statsReporterReportMetricCount(i.StatsReporter,
+			MetricsTokensDeleteSuccess, int64(res.RowsAffected()), game, platform)
+
 		return nil
 	}
 

diff --git a/feedback/listener.go b/feedback/listener.go
@@ -93,13 +93,13 @@ func (l *Listener) configure(statsdClientrOrNil interfaces.StatsDClient) error {
 	}
 	l.Queue = q
 
-	broker, err := NewBroker(l.Logger, l.Config, nil, q.MessagesChannel(), l.Queue.PendingMessagesWaitGroup())
+	broker, err := NewBroker(l.Logger, l.Config, l.StatsReporters, q.MessagesChannel(), l.Queue.PendingMessagesWaitGroup())
 	if err != nil {
 		return fmt.Errorf("error creating new broker: %s", err.Error())
 	}
 	l.Broker = broker
 
-	handler, err := NewInvalidTokenHandler(l.Logger, l.Config, nil, l.Broker.InvalidTokenOutChan)
+	handler, err := NewInvalidTokenHandler(l.Logger, l.Config, l.StatsReporters, l.Broker.InvalidTokenOutChan)
 	if err != nil {
 		return fmt.Errorf("error creating new invalid token handler: %s", err.Error())
 	}