deamonize and cursory logging
This commit is contained in:
50
app/main.py
50
app/main.py
@ -1,27 +1,51 @@
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import time
|
||||
|
||||
from services.loft_service import LoftService
|
||||
from helpers import aws_helper
|
||||
from sqs_listener import SqsListener
|
||||
from sqs_listener.daemon import Daemon
|
||||
|
||||
print("Starting Solver Service (v0.3.1)...")
|
||||
|
||||
# # listen to the solver queue
|
||||
while True:
|
||||
msg = aws_helper.receive_message(os.environ['SOLVER_SQS_INGEST_QUEUE'])
|
||||
|
||||
# get the item from the queue
|
||||
for message in msg.get("Messages", []):
|
||||
# for now the solver service only supports Loft types
|
||||
# this is here to allow us to create an extensible way to
|
||||
class ServiceListener(SqsListener):
|
||||
def handle_message(self, body, attributes, messages_attributes):
|
||||
# gather/manage/process data based on the particular needs
|
||||
service = LoftService(message)
|
||||
print(f'{time.localtime()} - message received...')
|
||||
service = LoftService(body)
|
||||
service.process()
|
||||
# log the things
|
||||
|
||||
# delete message once process is complete
|
||||
response = aws_helper.delete_message(os.environ['SOLVER_SQS_INGEST_QUEUE'], message['ReceiptHandle'])
|
||||
class ServiceDaemon(Daemon):
|
||||
def run(self):
|
||||
print("Starting Solver Service (v0.3.2)...")
|
||||
print("Initializing listener")
|
||||
listener = ServiceListener(
|
||||
'measure-development-solver-ingest',
|
||||
region_name=os.environ['SOLVER_AWS_REGION'],
|
||||
aws_access_key=os.environ['SOLVER_AWS_ACCESS_KEY_ID'],
|
||||
aws_secret_key=os.environ['SOLVER_AWS_SECRET_ACCESS_KEY'],
|
||||
queue_url=os.environ['SOLVER_SQS_INGEST_QUEUE']
|
||||
)
|
||||
listener.listen()
|
||||
|
||||
# probably convert to logging, but maybe keep this to some extent
|
||||
print("MESSAGE PROCESSED: ", message['MessageId'])
|
||||
if __name__ == "__main__":
|
||||
daemon = ServiceDaemon('/var/run/sqs_daemon.pid',stdout='/app/logs/stdout', stderr='/app/logs/stderr', stdin='/app/logs/stdin')
|
||||
if len(sys.argv) == 2:
|
||||
if 'start' == sys.argv[1]:
|
||||
print("Starting listener daemon")
|
||||
daemon.start()
|
||||
elif 'stop' == sys.argv[1]:
|
||||
print("Attempting to stop the daemon")
|
||||
daemon.stop()
|
||||
elif 'restart' == sys.argv[1]:
|
||||
daemon.restart()
|
||||
else:
|
||||
print("Unknown command")
|
||||
sys.exit(2)
|
||||
sys.exit(0)
|
||||
else:
|
||||
print("usage: %s start|stop|restart" % sys.argv[0])
|
||||
sys.exit(2)
|
||||
|
Reference in New Issue
Block a user