Skip to content

Commit 7696d7a

Browse files
committed
fixed starlette/fastapi example.
1 parent 7267c67 commit 7696d7a

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

examples/server_aiohttp_websocket.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import asyncclick as click
55
from aiohttp import web
66

7-
from examples.fixtures import generate_certificate_and_key
7+
from fixtures import generate_certificate_and_key
88
from rsocket.helpers import create_future
99
from rsocket.local_typing import Awaitable
1010
from rsocket.payload import Payload

examples/server_fastapi_websocket.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ async def request_response(self, payload: Payload) -> Awaitable[Payload]:
2121
async def endpoint(websocket: WebSocket):
2222
await websocket.accept()
2323
transport = Http3TransportWebsocket(websocket)
24-
RSocketServer(transport=transport)
24+
RSocketServer(transport=transport, handler_factory=Handler)
2525
await transport.wait_for_disconnect()
2626

2727

rsocket/transports/http3_transport.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def __init__(self, websocket: Union[WebSocket, ClientWebSocket]):
164164
async def send_frame(self, frame: Frame):
165165
with wrap_transport_exception():
166166
try:
167-
data = serialize_with_frame_size_header(frame)
167+
data = frame.serialize()
168168
try:
169169
await self._websocket.send_bytes(data)
170170
except LocalProtocolError as exception:
@@ -190,7 +190,7 @@ async def incoming_data_listener(self):
190190
self._disconnect_event.set()
191191
break
192192

193-
async for frame in self._frame_parser.receive_data(data):
193+
async for frame in self._frame_parser.receive_data(data, 0):
194194
self._incoming_frame_queue.put_nowait(frame)
195195

196196
except asyncio.CancelledError:

0 commit comments

Comments
 (0)