Get tokio_tungstenite::WebSocketStream from axum::WebSocket

[mdeloof]

Hi

I was wondering if there is a way to get a tokio_tungstenite::WebSocketStream from a axum::WebSocket? If I understand correctly from looking at the source code, the latter is a wrapper around the former? But I couldn't find a way to access the inner WebSocketStream. Or would I be better off writing a custom extractor in this case?

Thanks! :)

[davidpdrsn]

That's intentionally not possible as that would make tungstenite part of axum's public api, which we don't want.

I'd recommend abstracting over the Stream and Sink trait instead if possible.

[mdeloof]

Ah okay!

I am trying to make it generic over Sink and Stream, but I'm not sure what type Item should be set to as Message is not the same for axum and tungstenite?

[davidpdrsn]

You should be able to copy this function into your code to map from axum's message type to tungstenite's.