-
Notifications
You must be signed in to change notification settings - Fork 0
/
middleware.py
70 lines (57 loc) · 2.05 KB
/
middleware.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import threading
from queue import Queue
from flask import Flask, request
app = Flask(__name__)
# Create a queue to hold incoming requests
request_queue = Queue()
# Define a function to handle each request in a separate thread
def process_request():
while True:
# Get the next request from the queue
request = request_queue.get()
# Your middleware logic goes here
# You can modify the request, perform authentication, logging, rate limiting, etc.
# Forward the modified request to the real server
response = requests.request(
method=request.method,
url=f'https://realserver.com/{request.path}',
headers=request.headers,
data=request.get_data(),
cookies=request.cookies,
allow_redirects=False
)
# Your middleware logic on the response goes here
# You can modify the response, add headers, log the response, etc.
# Return the modified response to the customer
response_tuple = (
response.content,
response.status_code,
response.headers.items()
)
# Put the response back in the queue
request_queue.task_done()
request.response_queue.put(response_tuple)
# Create a pool of worker threads to handle requests
for i in range(5):
t = threading.Thread(target=process_request)
t.daemon = True
t.start()
@app.route('/', defaults={'path': ''})
@app.route('/<path:path>')
def catch_all(path):
# Create a new request object and add it to the queue
request_obj = {
'method': request.method,
'path': path,
'headers': request.headers,
'data': request.get_data(),
'cookies': request.cookies,
'response_queue': Queue()
}
request_queue.put(request_obj)
# Wait for the response to be processed by a worker thread
response_tuple = request_obj['response_queue'].get()
# Return the modified response to the customer
return response_tuple
if __name__ == '__main__':
app.run(debug=True)