From 1d0ee54b1f0f57e120a4bd59ef8d94c701a2987b Mon Sep 17 00:00:00 2001 From: DeOwl Date: Sat, 25 Oct 2025 20:24:06 +0300 Subject: [PATCH] added rabbitmq --- dockerfile | 6 ++++-- main.py | 20 ++++++++++---------- requirements.txt | 2 +- source/rabbitmq_test.py | 34 +++++++++++++++++++++++++--------- 4 files changed, 40 insertions(+), 22 deletions(-) diff --git a/dockerfile b/dockerfile index 1baae46..3d6ba60 100644 --- a/dockerfile +++ b/dockerfile @@ -1,5 +1,7 @@ FROM pennylaneai/pennylane:latest-lightning-qubit WORKDIR /app -COPY . . +COPY ./source ./source +COPY ./main.py ./ +COPY ./requirements.txt ./ RUN pip install --no-cache-dir -r requirements.txt -CMD ["python", "app.py"] \ No newline at end of file +CMD ["python", "main.py"] \ No newline at end of file diff --git a/main.py b/main.py index 39d7e56..5259bb9 100644 --- a/main.py +++ b/main.py @@ -3,24 +3,24 @@ from time import sleep, time from multiprocessing import Process, Queue import source.vqe as vqe -import project.source.archive.mqtt as mqtt +#import project.source.archive.mqtt as mqtt if __name__ == "__main__": broker_address = "mqtt.deowl.ru" # Example public broker broker_port = 1883 - mqttBroker = mqtt.vqeMqttBroker(broker_address, broker_port, max_connection_attempts=5) - print(mqttBroker.get_status()) - mqttBroker.connect_to_server() + #mqttBroker = mqtt.vqeMqttBroker(broker_address, broker_port, max_connection_attempts=5) + #print(mqttBroker.get_status()) + #mqttBroker.connect_to_server() - while True: - print(mqttBroker.get_status()) - sleep(1) + #while True: + # print(mqttBroker.get_status()) + # sleep(1) # GLOBAL SYSTEM PARAMETERS - active_electrons=2 - active_orbitals=2 + #active_electrons=2 + '''active_orbitals=2 max_iterations = 500 conv_tol = 1e-04 step_size = 0.05 @@ -36,4 +36,4 @@ if __name__ == "__main__": sleep(0.1) client.loop_stop() # Stop the background loop thread if used - client.disconnect() \ No newline at end of file + client.disconnect()''' \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index a7f735d..d8df59e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ basis-set-exchange==0.11 -paho-mqtt==2.1.0 \ No newline at end of file +pika==1.3.2 \ No newline at end of file diff --git a/source/rabbitmq_test.py b/source/rabbitmq_test.py index bb570ca..c3c9bdb 100644 --- a/source/rabbitmq_test.py +++ b/source/rabbitmq_test.py @@ -1,20 +1,36 @@ +import os +import sys import pika import ssl -print - context = ssl.create_default_context() context.verify_mode = ssl.CERT_REQUIRED ssl_options= pika.SSLOptions(context=context, server_hostname="rabbitmq.deowl.ru") -credential = pika.PlainCredentials("admin", "admin") +credential = pika.PlainCredentials("test", "test") -try: + +def main(): connection = pika.BlockingConnection(pika.ConnectionParameters('rabbitmq.deowl.ru', virtual_host='/', port=5671, ssl_options=ssl_options, credentials=credential)) channel = connection.channel() - print("good") -except pika.exceptions.AMQPConnectionError as e: - print(f"Connection error: {e}") -except Exception as e: - print(f"An unexpected error occurred: {e}") \ No newline at end of file + + channel.queue_declare(queue='hello') + + def callback(ch, method, properties, body): + print(f" [x] Received {body}") + + channel.basic_consume(queue='hello', on_message_callback=callback, auto_ack=True) + + print(' [*] Waiting for messages. To exit press CTRL+C') + channel.start_consuming() + +if __name__ == '__main__': + try: + main() + except KeyboardInterrupt: + print('Interrupted') + try: + sys.exit(0) + except SystemExit: + os._exit(0) \ No newline at end of file