44from fastapi import FastAPI
55from servicelib .logging_utils import log_context
66from servicelib .rabbitmq import (
7- RabbitMQClient ,
87 RabbitMQRPCClient ,
98 wait_till_rabbitmq_responsive ,
109)
@@ -22,16 +21,12 @@ async def on_startup() -> None:
2221 logging .INFO ,
2322 msg = "Storage startup Rabbitmq" ,
2423 ):
25- app .state .rabbitmq_client = None
2624 rabbit_settings : RabbitSettings | None = app .state .settings .STORAGE_RABBITMQ
2725 if not rabbit_settings :
2826 raise ConfigurationError (
2927 msg = "RabbitMQ client is de-activated in the settings"
3028 )
3129 await wait_till_rabbitmq_responsive (rabbit_settings .dsn )
32- app .state .rabbitmq_client = RabbitMQClient (
33- client_name = "storage" , settings = rabbit_settings
34- )
3530 app .state .rabbitmq_rpc_server = await RabbitMQRPCClient .create (
3631 client_name = "storage_rpc_server" , settings = rabbit_settings
3732 )
@@ -42,23 +37,13 @@ async def on_shutdown() -> None:
4237 logging .INFO ,
4338 msg = "Storage shutdown Rabbitmq" ,
4439 ):
45- if app .state .rabbitmq_client :
46- await app .state .rabbitmq_client .close ()
4740 if app .state .rabbitmq_rpc_server :
4841 await app .state .rabbitmq_rpc_server .close ()
4942
5043 app .add_event_handler ("startup" , on_startup )
5144 app .add_event_handler ("shutdown" , on_shutdown )
5245
5346
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-
6247def get_rabbitmq_rpc_server (app : FastAPI ) -> RabbitMQRPCClient :
6348 assert app .state .rabbitmq_rpc_server # nosec
6449 return cast (RabbitMQRPCClient , app .state .rabbitmq_rpc_server )
0 commit comments