Add looping call manager to analytics manager
This commit is contained in:
parent
ddb88eb1c9
commit
417a8f719f
1 changed files with 16 additions and 8 deletions
|
@ -1,3 +1,5 @@
|
||||||
|
from lbrynet.core import looping_call_manager
|
||||||
|
|
||||||
from twisted.internet import task
|
from twisted.internet import task
|
||||||
|
|
||||||
import constants
|
import constants
|
||||||
|
@ -8,18 +10,24 @@ class Manager(object):
|
||||||
self.analytics_api = analytics_api
|
self.analytics_api = analytics_api
|
||||||
self.events_generator = events_generator
|
self.events_generator = events_generator
|
||||||
self.track = track
|
self.track = track
|
||||||
self.send_heartbeat = task.LoopingCall(self._send_heartbeat)
|
self.looping_call_manager = self.setup_looping_calls()
|
||||||
self.update_tracked_metrics = task.LoopingCall(self._update_tracked_metrics)
|
|
||||||
|
def setup_looping_calls(self):
|
||||||
|
call_manager = looping_call_manager.LoopingCallManager()
|
||||||
|
looping_calls = [
|
||||||
|
('send_heartbeat', self._send_heartbeat),
|
||||||
|
('update_tracked_metrics', self._update_tracked_metrics),
|
||||||
|
]
|
||||||
|
for name, fn in looping_calls:
|
||||||
|
call_manager.register_looping_call(name, task.LoopingCall(fn))
|
||||||
|
return call_manager
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
self.send_heartbeat.start(60)
|
self.looping_call_manager.start('send_heartbeat', 60)
|
||||||
self.update_tracked_metrics.start(300)
|
self.looping_call_manager.start('update_tracked_metrics', 300)
|
||||||
|
|
||||||
def shutdown(self):
|
def shutdown(self):
|
||||||
if self.send_heartbeat.running:
|
self.looping_call_manager.shutdown()
|
||||||
self.send_heartbeat.stop()
|
|
||||||
if self.update_tracked_metrics.running:
|
|
||||||
self.update_tracked_metrics.stop()
|
|
||||||
|
|
||||||
def send_download_started(self, name, stream_info=None):
|
def send_download_started(self, name, stream_info=None):
|
||||||
event = self.events_generator.download_started(name, stream_info)
|
event = self.events_generator.download_started(name, stream_info)
|
||||||
|
|
Loading…
Reference in a new issue