layla/defend-together
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Serverless scaling and work to authorizer

Browse Source
This commit is contained in:
Layla
2020-05-21 17:32:27 -04:00
parent 92d0fa6476
commit 835d0f2833
10 changed files with 162 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
infrastructure/lambda/task_queue_manager/lambda_function.py
View File

@ -1,4 +1,15 @@
import json
import redis
import json, os
def lambda_handler(event, context):
print(json.dumps(event))
r = redis.Redis(host=os.environ['REDIS_HOST'], port=6379, db=0)
if event["detail"]["group"] == "service:" + os.environ["ECS_SERVICE"]:
desired = event["detail"]["desiredStatus"]
last = event["detail"]["lastStatus"]
if desired == "RUNNING" and desired == last:
print("Added task: " + event["detail"]["taskArn"])
r.lpush("tasks", event["detail"]["taskArn"])
elif desired == "STOPPED" or last == "STOPPED":
r.lrem("tasks", event["detail"]["taskArn"], 1)
print("Removed task: " + event["detail"]["taskArn"], 1)