mirror of https://github.com/D4-project/d4-core
36 lines
972 B
Python
36 lines
972 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
import redis
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
redis_server = redis.StrictRedis(
|
||
|
|
host="localhost",
|
||
|
|
port=6379,
|
||
|
|
db=0)
|
||
|
|
type = 1
|
||
|
|
|
||
|
|
try:
|
||
|
|
redis_server.ping()
|
||
|
|
except redis.exceptions.ConnectionError:
|
||
|
|
print('Error: Redis server {}:{}, ConnectionError'.format(host_redis, port_redis))
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
stream_name = 'stream:{}'.format(type)
|
||
|
|
redis_server.delete('working_session_uuid:{}'.format(type))
|
||
|
|
|
||
|
|
while True:
|
||
|
|
for session_uuid in redis_server.smembers('session_uuid:{}'.format(type)):
|
||
|
|
session_uuid = session_uuid.decode()
|
||
|
|
if not redis_server.sismember('working_session_uuid:{}'.format(type), session_uuid):
|
||
|
|
|
||
|
|
process = subprocess.Popen(['./worker.py', session_uuid])
|
||
|
|
print('New worker launched: {}'.format(session_uuid))
|
||
|
|
|
||
|
|
|
||
|
|
#print('.')
|
||
|
|
time.sleep(10)
|