import os
import config
import time
from flask import Flask, request
from aws_xray_sdk.core import patch_all, xray_recorder
from aws_xray_sdk.ext.flask.middleware import XRayMiddleware

app = Flask(__name__)

xray_recorder.configure(
    context_missing='LOG_ERROR',
    service=config.XRAY_APP_NAME,
)
patch_all()

XRayMiddleware(app, xray_recorder)

@app.route('/defaultroute')
def default():
    print('----------------')
    print(request.headers)
    print('----------------')
    time.sleep(config.TIMEOUT_VALUE)
    return config.WHO_AM_I

@app.route('/timeoutroute')
def timeout():
    print('----------------')
    print(request.headers)
    print('----------------')
    time.sleep(config.TIMEOUT_VALUE)
    return config.WHO_AM_I

@app.route('/tlsroute')
def tlsroute():
    print('----------------')
    print(request.headers)
    print('----------------')
    return config.WHO_AM_I

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=config.PORT, debug=config.DEBUG_MODE)