Skip to content

Commit 6f34a94

Browse files
author
Anze
committed
Make jobs' refresh interval configurable
1 parent ef74049 commit 6f34a94

File tree

3 files changed

+7
-4
lines changed

3 files changed

+7
-4
lines changed

.env.example

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
BACKEND_URL=https://grafolean.com/api
22
BOT_TOKEN=
3+
JOBS_REFRESH_INTERVAL=120

collector.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -156,11 +156,12 @@ def run_job(job, jobstore_alias, run_times, logger_name):
156156

157157

158158
class Collector(object):
159-
__slots__ = 'backend_url', 'bot_token', 'scheduler', 'known_jobs'
159+
__slots__ = 'backend_url', 'bot_token', 'scheduler', 'known_jobs', 'jobs_refresh_interval'
160160

161-
def __init__(self, backend_url, bot_token):
161+
def __init__(self, backend_url, bot_token, jobs_refresh_interval):
162162
self.backend_url = backend_url
163163
self.bot_token = bot_token
164+
self.jobs_refresh_interval = jobs_refresh_interval
164165
self.known_jobs = {}
165166

166167
@abstractmethod
@@ -278,7 +279,7 @@ def execute(self):
278279
self.refresh_jobs()
279280
except:
280281
logging.exception("Error refreshing jobs.")
281-
time.sleep(120)
282+
time.sleep(self.jobs_refresh_interval)
282283

283284
except KeyboardInterrupt:
284285
logging.info("Got exit signal, exiting.")

snmpcollector.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@ def jobs(self):
209209
bot_token = os.environ.get('BOT_TOKEN')
210210
if not backend_url or not bot_token:
211211
raise Exception("Please specify BACKEND_URL and BOT_TOKEN env vars.")
212+
jobs_refresh_interval = int(os.environ.get('JOBS_REFRESH_INTERVAL', 120))
212213

213-
c = SNMPCollector(backend_url, bot_token)
214+
c = SNMPCollector(backend_url, bot_token, jobs_refresh_interval)
214215
c.execute()

0 commit comments

Comments
 (0)