From cac95695ca6dd9475dac969b8489fb50d6dd6920 Mon Sep 17 00:00:00 2001 From: capossele <angelocapossele@gmail.com> Date: Tue, 7 Apr 2020 17:26:51 +0100 Subject: [PATCH] :recycle: refactors livefeed --- plugins/spa/livefeed.go | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/plugins/spa/livefeed.go b/plugins/spa/livefeed.go index a9405a01..bf2ab1af 100644 --- a/plugins/spa/livefeed.go +++ b/plugins/spa/livefeed.go @@ -19,8 +19,8 @@ var liveFeedWorkerPool *workerpool.WorkerPool func configureLiveFeed() { liveFeedWorkerPool = workerpool.New(func(task workerpool.Task) { - task.Param(0).(*message.CachedMessage).Consume(func(transaction *message.Message) { - sendToAllWSClient(&msg{MsgTypeTx, &tx{transaction.Id().String(), 0}}) + task.Param(0).(*message.CachedMessage).Consume(func(message *message.Message) { + sendToAllWSClient(&msg{MsgTypeTx, &tx{message.Id().String(), 0}}) }) task.Return(nil) @@ -28,26 +28,26 @@ func configureLiveFeed() { } func runLiveFeed() { - newTxRateLimiter := time.NewTicker(time.Second / 10) - notifyNewTx := events.NewClosure(func(tx *message.CachedMessage, metadata *tangle.CachedMessageMetadata) { + newMsgRateLimiter := time.NewTicker(time.Second / 10) + notifyNewMsg := events.NewClosure(func(message *message.CachedMessage, metadata *tangle.CachedMessageMetadata) { metadata.Release() select { - case <-newTxRateLimiter.C: - liveFeedWorkerPool.TrySubmit(tx) + case <-newMsgRateLimiter.C: + liveFeedWorkerPool.TrySubmit(message) default: - tx.Release() + message.Release() } }) - daemon.BackgroundWorker("SPA[TxUpdater]", func(shutdownSignal <-chan struct{}) { - messagelayer.Tangle.Events.TransactionAttached.Attach(notifyNewTx) + daemon.BackgroundWorker("SPA[MsgUpdater]", func(shutdownSignal <-chan struct{}) { + messagelayer.Tangle.Events.TransactionAttached.Attach(notifyNewMsg) liveFeedWorkerPool.Start() <-shutdownSignal - log.Info("Stopping SPA[TxUpdater] ...") - messagelayer.Tangle.Events.TransactionAttached.Detach(notifyNewTx) - newTxRateLimiter.Stop() + log.Info("Stopping SPA[MsgUpdater] ...") + messagelayer.Tangle.Events.TransactionAttached.Detach(notifyNewMsg) + newMsgRateLimiter.Stop() liveFeedWorkerPool.Stop() - log.Info("Stopping SPA[TxUpdater] ... done") + log.Info("Stopping SPA[MsgUpdater] ... done") }, shutdown.ShutdownPrioritySPA) } -- GitLab