I'm using a federation gateway based on the federation-example in the repository.
I just tried to update the code to use the new websocket.Handle (which in turn is based on the UniversalProtocolHandler) since the legacy handler is more or less deprecated.
It seems like there is not really a way to propagate things provided in the init-func payload to my subgraphs now. Previously I could put stuff in the context which was returned from the init-func but that context is no longer used outside the init message and setting up the heartbeat.
Any hints on how I should go about this?