GoldenPogacsa/consumer-scheduler
GoldenPogacsa/consumer-scheduler
Archived
7
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Small fixes

Browse Source
This commit is contained in:
Pünkösd Marcell 2020-03-29 19:26:52 +02:00
parent 9870174874
commit 04edb91a39
2 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
app.py
View File

@ -1,7 +1,8 @@
#!/usr/bin/env python #!/usr/bin/env python
import sentry_sdk import sentry_sdk
import time import time
import request import requests
import requests.exceptions
import os import os
import redis import redis
import json import json
@ -15,9 +16,10 @@ __copyright__ = "Copyright 2020, GoldenPogácsa Team"
__module_name__ = "app" __module_name__ = "app"
__version__text__ = "1" __version__text__ = "1"
sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10") sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10")
if __name__ == "__main__": def main():
r = redis.Redis(host='localhost', port=6379, db=0) r = redis.Redis(host='localhost', port=6379, db=0)
ip_list = os.environ['INITIAL_SERVERS'].split(',') ip_list = os.environ['INITIAL_SERVERS'].split(',')
@ -26,11 +28,14 @@ if __name__ == "__main__":
temp_dict = {} temp_dict = {}
for i in ip_list: for ip in ip_list:
response = request.post(http=i + '/sync', json={'uuid': os.environ['LOCAL_UUID']}) try:
response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']})
except requests.exceptions.ConnectionError:
continue
if response.status_code == 200: if response.status_code == 200:
temp_dict[response.json()['uuid']] = {'ip': i} temp_dict[response.json()['uuid']] = {'ip': ip}
ip_list_redis.update(temp_dict) ip_list_redis.update(temp_dict)
@ -40,14 +45,20 @@ if __name__ == "__main__":
ip_list_redis = json.loads(r.get('consumer_list').decode('utf-8')) ip_list_redis = json.loads(r.get('consumer_list').decode('utf-8'))
for i in ip_list_redis: for ip in ip_list_redis:
response = request.post(http=i + '/sync', json={'uuid': os.environ['LOCAL_UID']}) try:
response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']})
except requests.exceptions.ConnectionError:
continue
if response.status_code == 200: if response.status_code == 200:
temp_dict[response.json()['uuid']] = {'ip': i} temp_dict[response.json()['uuid']] = {'ip': ip}
ip_list_redis.update(temp_dict) ip_list_redis.update(temp_dict)
r.set('consumer_list', json.dumps(ip_list_redis).encode('utf-8')) r.set('consumer_list', json.dumps(ip_list_redis).encode('utf-8'))
time.sleep(30) time.sleep(30)
print("Consumer-Scheduler")
if __name__ == "__main__":
main()

2
requirements.txt
View File

@ -1,3 +1,3 @@
sentry_sdk sentry_sdk
request requests
redis redis