Sounds great. A more general approach than Apollo is certainly appreciated.
GraphQL over WebSockets - The Guild Blog (the-guild.dev)
Thought I mention this link. The also point out a couple of sec issues and shortcomings with the subscriptions-transport-ws protocoll (that you are probably aware of) and that make a compelling argument for me to stay away from it.