mqtt_broker.py

import asyncio
from gmqtt import Client as MQTTClient

class MQTTBroker:
    def __init__(self, host='127.0.0.1', port=1883):
        self.host = host
        self.port = port
        self.clients = {}

    async def start(self):
        server = await asyncio.start_server(self.handle_client, self.host, self.port)
        async with server:
            await server.serve_forever()

    async def handle_client(self, reader, writer):
        # Simple broker logic here
        addr = writer.get_extra_info('peername')
        print(f"Connection from {addr}")

        while True:
            data = await reader.read(100)
            if not data:
                break
            print(f"Received {data} from {addr}")

        writer.close()
        await writer.wait_closed()

if __name__ == '__main__':
    broker = MQTTBroker()
    asyncio.run(broker.start())