default batch size fix

README.md

@@ -53,7 +53,7 @@ class NotifyHandler(Handler):
         # users = await UserRepository.find_all([r.user_id for r in requests])
         # await mailer.send_all([construct_message(user) for user in users])
 
-    def batch_size(self) -> None:
+    def batch_size(self) -> Optional[int]:
         '''
         Optionally, you can ovveride global batch size per handler
         '''

sharded_queue/__init__.py

@@ -47,8 +47,8 @@ async def handle(self, *requests: T) -> None:
     async def stop(self) -> None:
         pass
 
-    def batch_size(self) -> int:
-        return self.settings.batch_size
+    def batch_size(self) -> Optional[int]:
+        pass
 
 
 class Tube(NamedTuple):
@@ -231,7 +231,10 @@ async def process(self, tube: Tube, limit: Optional[int] = None) -> int:
                 if tube.handler is RecurrentHandler:
                     batch_size = self.settings.recurrent_tasks_limit
                 else:
-                    batch_size = min(limit, instance.batch_size())
+                    batch_size = instance.batch_size()
+                    if not batch_size:
+                        batch_size = self.settings.batch_size
+                    batch_size = min(limit, batch_size)
                 processed = False
                 for pipe in pipes:
                     msgs = await storage.range(pipe, batch_size)