File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change 3737import os
3838import pickle
3939import subprocess
40+ import sys
4041import time
4142import tempfile
4243import threading
@@ -808,10 +809,13 @@ def send_heartbeat(self):
808809 'queue' : self .rabbitmq_queue ,
809810 'extractor_info' : self .extractor_info
810811 }
812+ next_heartbeat = time .time ()
811813 while self .thread :
812814 try :
813- self .channel .basic_publish (exchange = 'extractors' , routing_key = '' , body = json .dumps (message ))
814- time .sleep (self .heartbeat )
815+ self .channel .connection .process_data_events ()
816+ if time .time () >= next_heartbeat :
817+ self .channel .basic_publish (exchange = 'extractors' , routing_key = '' , body = json .dumps (message ))
818+ next_heartbeat = time .time () + self .heartbeat
815819 except SystemExit :
816820 raise
817821 except KeyboardInterrupt :
@@ -820,6 +824,8 @@ def send_heartbeat(self):
820824 raise
821825 except Exception : # pylint: disable=broad-except
822826 logging .getLogger (__name__ ).exception ("Error while sending heartbeat." )
827+ sys .exit (- 1 )
828+ time .sleep (1 )
823829
824830
825831class RabbitMQHandler (Connector ):
You can’t perform that action at this time.
0 commit comments