Support REQUEST_TIMEOUT status code for a batch fetch with no_wait=True (#618)

diorcety · web-flow · commit 1ecc5e15eb45 · 2024-10-03T11:05:40.000-07:00
* Fixes #617
diff --git a/nats/js/client.py b/nats/js/client.py
@@ -1244,7 +1244,7 @@ async def _fetch_n(
                         )
                         msg = await self._sub.next_msg(timeout=deadline)
                         status = JetStreamContext.is_status_msg(msg)
-                        if status == api.StatusCode.NO_MESSAGES:
+                        if status == api.StatusCode.NO_MESSAGES or status == api.StatusCode.REQUEST_TIMEOUT:
                             # No more messages after this so fallthrough
                             # after receiving the rest.
                             break

Original file line number	Diff line number	Diff line change
`@@ -1244,7 +1244,7 @@ async def _fetch_n(`
`1244`	`1244`	`)`
`1245`	`1245`	`msg = await self._sub.next_msg(timeout=deadline)`
`1246`	`1246`	`status = JetStreamContext.is_status_msg(msg)`
`1247`		`- if status == api.StatusCode.NO_MESSAGES:`
	`1247`	`+ if status == api.StatusCode.NO_MESSAGES or status == api.StatusCode.REQUEST_TIMEOUT:`
`1248`	`1248`	`# No more messages after this so fallthrough`
`1249`	`1249`	`# after receiving the rest.`
`1250`	`1250`	`break`