diff --git a/kontor-scripts/read_queue.py b/kontor-scripts/read_queue.py index 22dc8f6..5561f73 100644 --- a/kontor-scripts/read_queue.py +++ b/kontor-scripts/read_queue.py @@ -1,37 +1,41 @@ +import logging + import stomp import json import time +import msgspec from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter -from config import get_logger parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter) parser.add_argument('--verbose', '-v', action='count', default=0) -parser.add_argument('--config', '-c', default='kontor-docker') +parser.add_argument("--server", "-s", default="127.0.0.1") args = parser.parse_args() +logger = logging.getLogger(__name__) + +class Link(msgspec.Struct): + url: str class MyListener(stomp.ConnectionListener): - def __init__(self, log): - self.log = log + def __init__(self): + pass def on_error(self, frame): - self.log.info(f"received an error {frame.body}") + logger.info("received an error %s", frame.body) def on_message(self, frame): - self.log.info(f"received a message '{frame.body}'") - data = json.loads(frame.body) - url = data['url'] - self.log.info(f"found link: {url}") + logger.info("received a message %s", frame.body) + link = msgspec.json.decode(frame.body, type=Link) + self.log.info("found link: %s", link.url) if __name__ == '__main__': - log = get_logger(args.verbose, args.config) - log.info("kontor.read_queue started") - host = [('127.0.0.1', 61616)] + logger.setLevel(logging.INFO) + logger.info("kontor.read_queue started") + host = [(args.server, 61616)] conn = stomp.Connection(host_and_ports=host) - conn.set_listener('', MyListener(log)) + conn.set_listener('', MyListener()) conn.connect(username='artemis', passcode='artemis', wait=True) - conn.subscribe(destination='KontorMediaFile::add_link_file', id=1, ack='auto', headers={}) + conn.subscribe(destination='add_link', id=1, ack='auto', headers={}) time.sleep(5) conn.disconnect() - log.info("kontor.read_queue finished") - + logger.info("kontor.read_queue finished")