4
4
from fastapi import FastAPI
5
5
from servicelib .logging_utils import log_context
6
6
from servicelib .rabbitmq import (
7
- RabbitMQClient ,
8
7
RabbitMQRPCClient ,
9
8
wait_till_rabbitmq_responsive ,
10
9
)
@@ -22,16 +21,12 @@ async def on_startup() -> None:
22
21
logging .INFO ,
23
22
msg = "Storage startup Rabbitmq" ,
24
23
):
25
- app .state .rabbitmq_client = None
26
24
rabbit_settings : RabbitSettings | None = app .state .settings .STORAGE_RABBITMQ
27
25
if not rabbit_settings :
28
26
raise ConfigurationError (
29
27
msg = "RabbitMQ client is de-activated in the settings"
30
28
)
31
29
await wait_till_rabbitmq_responsive (rabbit_settings .dsn )
32
- app .state .rabbitmq_client = RabbitMQClient (
33
- client_name = "storage" , settings = rabbit_settings
34
- )
35
30
app .state .rabbitmq_rpc_server = await RabbitMQRPCClient .create (
36
31
client_name = "storage_rpc_server" , settings = rabbit_settings
37
32
)
@@ -42,23 +37,13 @@ async def on_shutdown() -> None:
42
37
logging .INFO ,
43
38
msg = "Storage shutdown Rabbitmq" ,
44
39
):
45
- if app .state .rabbitmq_client :
46
- await app .state .rabbitmq_client .close ()
47
40
if app .state .rabbitmq_rpc_server :
48
41
await app .state .rabbitmq_rpc_server .close ()
49
42
50
43
app .add_event_handler ("startup" , on_startup )
51
44
app .add_event_handler ("shutdown" , on_shutdown )
52
45
53
46
54
- def get_rabbitmq_client (app : FastAPI ) -> RabbitMQClient :
55
- if not app .state .rabbitmq_client :
56
- raise ConfigurationError (
57
- msg = "RabbitMQ client is not available. Please check the configuration."
58
- )
59
- return cast (RabbitMQClient , app .state .rabbitmq_client )
60
-
61
-
62
47
def get_rabbitmq_rpc_server (app : FastAPI ) -> RabbitMQRPCClient :
63
48
assert app .state .rabbitmq_rpc_server # nosec
64
49
return cast (RabbitMQRPCClient , app .state .rabbitmq_rpc_server )
0 commit comments