From c7f02fea1526d04583e9d348dca0a069418f90aa Mon Sep 17 00:00:00 2001
From: Jusong Yu <jusong.yeu@gmail.com>
Date: Fri, 20 Dec 2024 09:41:23 +0100
Subject: [PATCH] Use aiormq with closing fix that gracefully teardown RMQ
 connection

---
 pyproject.toml | 4 ++++
 uv.lock        | 8 +++-----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 9461b7a46..3c2831eba 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,6 +35,7 @@ dependencies = [
   'numpy~=1.21',
   'paramiko~=3.0',
   'plumpy~=0.22.3',
+  'aiormq',
   'pgsu~=0.3.0',
   'psutil~=5.6',
   'psycopg[binary]~=3.0',
@@ -509,3 +510,6 @@ passenv =
     AIIDA_TEST_WORKERS
 commands = molecule {posargs:test}
 """
+
+[tool.uv.sources]
+aiormq = {git = "https://github.com/unkcpz/aiormq", rev = "f4da0adbddfec4b984f44093e8d04021fb163464"}
diff --git a/uv.lock b/uv.lock
index f56926877..d785dd5ef 100644
--- a/uv.lock
+++ b/uv.lock
@@ -23,6 +23,7 @@ name = "aiida-core"
 version = "2.6.2.post0"
 source = { editable = "." }
 dependencies = [
+    { name = "aiormq" },
     { name = "alembic" },
     { name = "archive-path" },
     { name = "circus" },
@@ -157,6 +158,7 @@ tui = [
 requires-dist = [
     { name = "aiida-core", extras = ["atomic-tools", "rest", "tests", "tui"], marker = "extra == 'pre-commit'" },
     { name = "aiida-export-migration-tests", marker = "extra == 'tests'", specifier = "==0.9.0" },
+    { name = "aiormq", git = "https://github.com/unkcpz/aiormq?rev=f4da0adbddfec4b984f44093e8d04021fb163464" },
     { name = "alembic", specifier = "~=1.2" },
     { name = "archive-path", specifier = "~=0.4.2" },
     { name = "ase", marker = "extra == 'atomic-tools'", specifier = "~=3.18" },
@@ -261,15 +263,11 @@ wheels = [
 [[package]]
 name = "aiormq"
 version = "6.8.1"
-source = { registry = "https://pypi.org/simple" }
+source = { git = "https://github.com/unkcpz/aiormq?rev=f4da0adbddfec4b984f44093e8d04021fb163464#f4da0adbddfec4b984f44093e8d04021fb163464" }
 dependencies = [
     { name = "pamqp" },
     { name = "yarl" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/a4/79/5397756a8782bf3d0dce392b48260c3ec81010f16bef8441ff03505dccb4/aiormq-6.8.1.tar.gz", hash = "sha256:a964ab09634be1da1f9298ce225b310859763d5cf83ef3a7eae1a6dc6bd1da1a", size = 30528 }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/2e/be/1a613ae1564426f86650ff58c351902895aa969f7e537e74bfd568f5c8bf/aiormq-6.8.1-py3-none-any.whl", hash = "sha256:5da896c8624193708f9409ffad0b20395010e2747f22aa4150593837f40aa017", size = 31174 },
-]
 
 [[package]]
 name = "alabaster"