lavatech
/
rana-cli
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

send batches of at most 10 heartbeats to api

This commit is contained in:
Alan Hamlett 2018-09-20 10:14:25 -07:00
parent 8809c55765
commit b488cd537c
3 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_offlinequeue.py
View File

@ -239,12 +239,12 @@ class OfflineQueueTestCase(TestCase):
queue = Queue(None, None)
self.assertEquals(queue._get_db_file(), fh.name)
saved_heartbeats = queue.pop_many()
saved_heartbeats = next(queue.pop_many())
self.assertNothingPrinted()
self.assertNothingLogged(logs)
# make sure only heartbeats with error code responses were saved
self.assertEquals(len(saved_heartbeats), 2)
self.assertEquals(sum(1 for x in saved_heartbeats), 2)
self.assertEquals(saved_heartbeats[0].entity, os.path.realpath(os.path.join(tempdir, entities[1])))
self.assertEquals(saved_heartbeats[1].entity, os.path.realpath(os.path.join(tempdir, entities[3])))
@ -295,7 +295,7 @@ class OfflineQueueTestCase(TestCase):
queue = Queue(None, None)
self.assertEquals(queue._get_db_file(), fh.name)
saved_heartbeats = queue.pop_many()
saved_heartbeats = next(queue.pop_many())
self.assertNothingPrinted()
# make sure all offline heartbeats were sent, so queue should only have 1 heartbeat left from the second 500 response
@ -346,7 +346,7 @@ class OfflineQueueTestCase(TestCase):
queue = Queue(None, None)
self.assertEquals(queue._get_db_file(), fh.name)
saved_heartbeats = queue.pop_many()
saved_heartbeats = next(queue.pop_many())
self.assertNothingPrinted()
expected = "WakaTime WARNING Missing 4 results from api.\nWakaTime WARNING Missing 2 results from api."
@ -398,7 +398,7 @@ class OfflineQueueTestCase(TestCase):
# offline queue should still have saved heartbeats
queue = Queue(None, None)
saved_heartbeats = queue.pop_many()
saved_heartbeats = next(queue.pop_many())
self.assertNothingPrinted()
self.assertEquals(len(saved_heartbeats), 2)
@ -444,7 +444,7 @@ class OfflineQueueTestCase(TestCase):
# offline queue should still have saved heartbeats
queue = Queue(None, None)
saved_heartbeats = queue.pop_many()
saved_heartbeats = next(queue.pop_many())
self.assertNothingPrinted()
self.assertEquals(len(saved_heartbeats), 2)

3
wakatime/main.py
View File

@ -66,8 +66,7 @@ def execute(argv=None):
retval = send_heartbeats(heartbeats, args, configs)
if retval == SUCCESS:
queue = Queue(args, configs)
offline_heartbeats = queue.pop_many(args.sync_offline_activity)
if len(offline_heartbeats) > 0:
for offline_heartbeats in queue.pop_many(args.sync_offline_activity):
retval = send_heartbeats(offline_heartbeats, args, configs)
return retval

6
wakatime/offlinequeue.py
View File

@ -115,8 +115,12 @@ class Queue(object):
break
heartbeats.append(heartbeat)
count += 1
if count % 10 == 0:
yield heartbeats
heartbeats = []
return heartbeats
if heartbeats:
yield heartbeats
def _get_db_file(self):
home = '~'