aws_backend_simulation.py

import concurrent.futures
import os
import random
import time
import requests
import json

path = '/tmp/token_9238572973'

with open(path, 'r') as f:
    token = f.read().splitlines()[0]

endpoint = 'http://localhost:8888/AWSMonitor'

print(token)

START = lambda npart: requests.put(endpoint, params={'token': token}, data=json.dumps({'msg': 'START', 'npart': npart}))
FINISHED = lambda: requests.put(endpoint, params={'token': token}, data=json.dumps({'msg': 'FIN'}))
INVOKED = lambda: requests.put(endpoint, params={'token': token}, data=json.dumps({'msg': 'INV'}))

def lambda_sim(idx):
    t = random.randint(1, 3)
    t1 = random.randint(1, 10)

    time.sleep(t)
    INVOKED()

    time.sleep(t1)

    FINISHED()

    return (t, idx, True)

def ProcessAndMerge(npart=32):
    START(npart)
    print("elo")
    with concurrent.futures.ThreadPoolExecutor(max_workers=npart) as ex:
        futures = [ex.submit(lambda_sim, i) for i in range(npart)]
        for future in concurrent.futures.as_completed(futures):
            pass


if __name__ == '__main__':
    ProcessAndMerge()