added code to run main loop of service
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
import time
|
||||
import logging
|
||||
import time
|
||||
import os
|
||||
from initialise_app import initialise_app
|
||||
from bandwidth import measure as measure_bandwidth
|
||||
from database import connect
|
||||
|
||||
def event():
|
||||
logging.info('started event')
|
||||
# get env vars
|
||||
replicates = int(os.getenv('REPLICATES'))
|
||||
# setup database connection
|
||||
db = connect()
|
||||
# run event
|
||||
for rep_num in range(replicates):
|
||||
logging.info(f'running replicate {rep_num}')
|
||||
# do measurement
|
||||
try:
|
||||
res = measure_bandwidth()
|
||||
except Exception as e:
|
||||
logging.error(f'failed to measure bandwidth: {e}')
|
||||
continue
|
||||
# upload result to database
|
||||
payload_dict = {
|
||||
'user_id': os.getenv('USER_ID'),
|
||||
'data': res
|
||||
}
|
||||
try:
|
||||
db_id = db.insert_one(payload_dict).inserted_id
|
||||
except Exception as e:
|
||||
logging.error(f'failed sending results to database: {e}')
|
||||
continue
|
||||
logging.debug(f'data sent to database received db_id: {db_id}')
|
||||
time.sleep(1)
|
||||
logging.info('finished event')
|
||||
|
||||
if __name__ == '__main__':
|
||||
initialise_app()
|
||||
trigger_time = time.time()
|
||||
while True:
|
||||
if time.time() >= trigger_time:
|
||||
# set new trigger time
|
||||
trigger_time += int(os.getenv('TRIGGER_INTERVAL_SECONDS'))
|
||||
# run event
|
||||
try:
|
||||
event()
|
||||
except Exception as e:
|
||||
logging.error(e)
|
||||
time.sleep(1)
|
||||
|
||||
Reference in New Issue
Block a user