2019-01-11 13:54:01 +01:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import time
|
|
|
|
import redis
|
|
|
|
import subprocess
|
|
|
|
|
2019-06-11 20:27:39 +02:00
|
|
|
host_redis_stream = os.getenv('D4_REDIS_STREAM_HOST', "localhost")
|
|
|
|
port_redis_stream = int(os.getenv('D4_REDIS_STREAM_PORT', 6379))
|
2019-01-16 10:27:59 +01:00
|
|
|
|
|
|
|
redis_server_stream = redis.StrictRedis(
|
|
|
|
host=host_redis_stream,
|
|
|
|
port=port_redis_stream,
|
2019-01-11 13:54:01 +01:00
|
|
|
db=0)
|
|
|
|
type = 1
|
|
|
|
|
|
|
|
try:
|
2019-01-16 10:27:59 +01:00
|
|
|
redis_server_stream.ping()
|
2019-01-11 13:54:01 +01:00
|
|
|
except redis.exceptions.ConnectionError:
|
2019-06-11 20:27:39 +02:00
|
|
|
print('Error: Redis server {}:{}, ConnectionError'.format(host_redis_stream, port_redis_stream))
|
2019-01-11 13:54:01 +01:00
|
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
stream_name = 'stream:{}'.format(type)
|
2019-01-16 10:27:59 +01:00
|
|
|
redis_server_stream.delete('working_session_uuid:{}'.format(type))
|
2019-01-11 13:54:01 +01:00
|
|
|
|
|
|
|
while True:
|
2019-01-16 10:27:59 +01:00
|
|
|
for session_uuid in redis_server_stream.smembers('session_uuid:{}'.format(type)):
|
2019-01-11 13:54:01 +01:00
|
|
|
session_uuid = session_uuid.decode()
|
2019-01-16 10:27:59 +01:00
|
|
|
if not redis_server_stream.sismember('working_session_uuid:{}'.format(type), session_uuid):
|
2019-01-11 13:54:01 +01:00
|
|
|
|
|
|
|
process = subprocess.Popen(['./worker.py', session_uuid])
|
2019-01-16 10:27:59 +01:00
|
|
|
print('Launching new worker{} ... session_uuid={}'.format(type, session_uuid))
|
2019-01-11 13:54:01 +01:00
|
|
|
|
|
|
|
|
2019-01-14 11:17:18 +01:00
|
|
|
#print('.')
|
2019-01-11 13:54:01 +01:00
|
|
|
time.sleep(10)
|