@@ -200,23 +200,6 @@ async def _dequeue_rows(self, cursor: Cursor) -> list[Any]:
200200 return await cursor .fetchall ()
201201
202202
203- class PostgreSqlBrokerQueueBuilder (Builder ):
204- """PostgreSql Broker Queue Builder class."""
205-
206- def with_config (self , config : Config ):
207- """Set config.
208-
209- :param config: The config to be set.
210- :return: This method return the builder instance.
211- """
212- self .kwargs |= config .get_database_by_name ("broker" )
213- self .kwargs |= config .get_interface_by_name ("broker" )["common" ]["queue" ]
214- return super ().with_config (config )
215-
216-
217- PostgreSqlBrokerQueue .set_builder (PostgreSqlBrokerQueueBuilder )
218-
219-
220203class PostgreSqlBrokerQueueQueryFactory (ABC ):
221204 """PostgreSql Broker Queue Query Factory class."""
222205
@@ -341,3 +324,20 @@ def __lt__(self, other: Any) -> bool:
341324 return isinstance (other , type (self )) and self .data < other .data
342325 except Exception :
343326 return False
327+
328+
329+ class PostgreSqlBrokerQueueBuilder (Builder ):
330+ """PostgreSql Broker Queue Builder class."""
331+
332+ def with_config (self , config : Config ):
333+ """Set config.
334+
335+ :param config: The config to be set.
336+ :return: This method return the builder instance.
337+ """
338+ self .kwargs |= config .get_database_by_name ("broker" )
339+ self .kwargs |= config .get_interface_by_name ("broker" )["common" ]["queue" ]
340+ return super ().with_config (config )
341+
342+
343+ PostgreSqlBrokerQueue .set_builder (PostgreSqlBrokerQueueBuilder )
0 commit comments