Refactor DockerHandler and NginxHandler classes

This commit is contained in:
Janis
2023-11-24 22:27:07 +01:00
parent 3649c6b49c
commit 07ed97a6c3
3 changed files with 31 additions and 36 deletions

View File

@@ -9,27 +9,24 @@ class RequestHandler(threading.Thread):
super().__init__()
self.port = port
self.docker_handler = docker_handler
# Create a TCP/IP socket
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Bind the socket to the port
server_address = ('localhost', self.port)
print('starting up on {} port {}'.format(*server_address))
print(f'Starting up on {server_address[0]} port {server_address[1]}')
self.sock.bind(server_address)
# Listen for incoming connections
self.sock.listen(1)
def run(self):
while True:
print('waiting for a connection on port {}'.format(self.port))
print(f'Waiting for a connection on port {self.port}')
self.connection, self.client_address = self.sock.accept()
try:
print('connection from', self.client_address)
print('Connection from', self.client_address)
self.handle_request()
finally:
self.connection.close()
def handle_request(self):
print('handling request on port {}'.format(self.port))
print(f'Handling request on port {self.port}')
container_ip = docker_container_mapping().get(str(self.port))
if container_ip:
container = self.docker_handler.get_container_by_ip(
@@ -38,29 +35,31 @@ class RequestHandler(threading.Thread):
request = self.connection.recv(1024)
if request[0] == 0x10:
if b'\x01' in request:
print('ping')
print(f'Detected ping request for {container_ip}')
self.forward_request_to_server(request, isStarting)
elif b'\x02' in request:
print('join')
print(f'Detected join/login request for {container_ip}')
if isStarting:
print('container is starting, waiting for 5 seconds')
print(
f'Container {container_ip} is already starting...')
self.forward_request_to_server(request, isStarting)
else:
print('starting docker container {}'.format(container_ip))
print(f'Starting container {container_ip}')
container.start()
elif request[0] == 0xFE:
print('legacy server list ping')
print(f'Detected legacy ping request for {container_ip}')
self.forward_request_to_server(request, isStarting)
else:
print('no docker container mapped to this port')
print(f'No container mapped to port {self.port}')
def forward_request_to_server(self, request, isStarting=False):
print('Forwarding request to placeholder server')
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:
ip = os.environ.get('PLACEHOLDER_SERVER_SLEEPING_IP')
if isStarting:
print('----------------container is starting, waiting for 5 seconds')
print('Container is starting. Using starting placeholder IP')
ip = os.environ.get('PLACEHOLDER_SERVER_STARTING_IP')
server_socket.connect(