Created
August 22, 2021 10:17
-
-
Save jamby1100/916b201e2a8739818ab3442a07d74631 to your computer and use it in GitHub Desktop.
This snippet is based off the work of Mathew Wachter (https://github.com/matthewwachter/py-tcp-threaded-server). I added modifications to allow it to communicate to Lambda via requests
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/python3 | |
from datetime import datetime | |
import json | |
from pprint import pprint | |
import socket | |
import os | |
from threading import Thread | |
import requests | |
class ThreadedServer(Thread): | |
def __init__(self, host, port, tcp_server_base_url, timeout=60, debug=False): | |
self.host = host | |
self.port = port | |
self.timeout = timeout | |
self.debug = debug | |
self.tcp_server_base_url = tcp_server_base_url | |
Thread.__init__(self) | |
print("THREADED SERVER INIT") | |
# run by the Thread object | |
def run(self): | |
if self.debug: | |
print(datetime.now()) | |
print('SERVER Starting...', '\n') | |
self.listen() | |
def listen(self): | |
print("START LISTENING") | |
# create an instance of socket | |
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | |
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) | |
# bind the socket to its host and port | |
self.sock.bind((self.host, self.port)) | |
if self.debug: | |
print(datetime.now()) | |
print('SERVER Socket Bound', self.host, self.port, '\n') | |
# start listening for a client | |
self.sock.listen(5) | |
if self.debug: | |
print(datetime.now()) | |
print('SERVER Listening...', '\n') | |
while True: | |
# get the client object and address | |
client, address = self.sock.accept() | |
# set a timeout | |
client.settimeout(self.timeout) | |
if self.debug: | |
print(datetime.now()) | |
print('CLIENT Connected:', client, '\n') | |
# start a thread to listen to the client | |
Thread(target=self.listenToClient, args=(client, address)).start() | |
# send the client a connection message | |
# res = { | |
# 'cmd': 'connected', | |
# } | |
# response = dumps(res) | |
# client.send(response.encode('utf-8')) | |
def listenToClient(self, client, address): | |
# set a buffer size ( could be 2048 or 4096 / power of 2 ) | |
size = 1024 | |
while True: | |
try: | |
# try to receive data from the client | |
data = client.recv(size).decode('utf-8') | |
if data: | |
print("AND THE DATA IS") | |
print(data) | |
response = requests.post( | |
self.tcp_server_base_url, | |
data=data, | |
headers={"Content-Type":"application/xml"} | |
) | |
# send a response back to the client | |
res = { | |
'cmd': "Ready", | |
'data': "for the rock" | |
} | |
print("AND THE RESPONSE IS") | |
print(response.text) | |
response = response.text | |
client.send(response.encode('utf-8')) | |
else: | |
raise Exception('Client disconnected') | |
except Exception as error: | |
print("AND THE ERROR IS: ") | |
print(error) | |
if self.debug: | |
print(datetime.now()) | |
print('CLIENT Disconnected:', client, '\n') | |
client.close() | |
return False | |
if __name__ == "__main__": | |
bind_address = os.getenv("TCP_IP_ADDRESS_BIND") | |
port = int(os.getenv("TCP_PORT_NUMBER")) | |
tcp_server_base_url = os.getenv("APIGW_BASE_URL") | |
ThreadedServer(bind_address, port, tcp_server_base_url, timeout=86400, debug=True).start() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment