secondary.py

import os
import time
from typing import Optional
import logging
from fastapi import FastAPI, Response, HTTPException
from http import HTTPStatus
from pydantic import BaseModel

logger = logging.getLogger("master")

class MessageModel(BaseModel):
    message: str
    id: int
    is_blocked: bool = False


if (os.environ.get('BREAK', None) != None):
    print('terminating the program because flag was passed')
    exit(1)
    pass
else:
    app = FastAPI(debug=True)


INMEMORY_MESSAGE_LIST = []
DELAY = None if 'DELAY' not in os.environ else float(os.environ['DELAY'])


ERROR_K = 0 if 'ERROR_K' not in os.environ else float(os.environ['ERROR_K'])
def error_simulation(is_blocked):
    global ERROR_K
    if ERROR_K > 0 and is_blocked:
        ERROR_K -= 1
        msg = f'Okay, we got an error, retry your request pls. You will get {ERROR_K} more errors'
        logging.info(msg)
        raise Exception(msg)


@app.get("/ping")
def ping():
    if DELAY is not None and DELAY > 0:
        time.sleep(DELAY)
    return Response(status_code=HTTPStatus.NO_CONTENT.value) 


@app.get("/message")
def get_message():
    res = []
    for m in INMEMORY_MESSAGE_LIST:
        if m == None: 
            break
        res.append(m)
    return res


@app.post("/__message")
def post_message(message: MessageModel):
    logging.info(message)

    try:
        error_simulation(message.is_blocked)

        if DELAY is not None and DELAY > 0:
            print(f'Introducing delay {DELAY} sec..')
            time.sleep(DELAY)
            print(f'Delay completed!')

        if len(INMEMORY_MESSAGE_LIST) <= message.id:
            INMEMORY_MESSAGE_LIST.extend([None for i in range(0, 1 + message.id - len(INMEMORY_MESSAGE_LIST))])

        INMEMORY_MESSAGE_LIST[message.id] = message.message

        print(INMEMORY_MESSAGE_LIST)
        return Response(status_code=HTTPStatus.NO_CONTENT.value)
    except:
        raise HTTPException(status_code=500, detail="Item not found")