您可以创建一些实现start 方法的Web 套接字服务器包装器,该方法将负责创建和运行WSServer,以及使用SubscriptionManager 创建一个SubscriptionServer
// in subscription.js
import { PubSub, SubscriptionManager } from 'graphql-subscriptions';
const pubSub = new PubSub();
let subscriptionManagerOptions = {
schema: schema, // this is your graphql schema
setupFunctions: {
// here come your setup functions
},
pubSub: pubSub
};
const subscriptionManager = new SubscriptionManager(subscriptionManagerOptions);
export { pubSub, subscriptionManager };
创建subscriptionManager 后,我们现在可以实现WSServer
import { createServer } from 'http';
import { SubscriptionServer } from 'subscription-transport-ws';
import { subscriptionManager } from './subscription';
const webSocketServerWrapper = {
start: function(port){
const webSocketServer = createServer((request, response) => {
response.writeHead(404);
response.end();
});
webSocketServer.listen(port, () => {
console.log('WSServer listening on port ' + port);
});
new SubscriptionServer({
subscriptionManager,
onSubscribe: (message, options, request) => {
return Promise.resolve(Object.assign({}, options, {}));
}
}, webSocketServer);
}
};
export default webSocketServerWrapper;
现在您可以在初始化文件中导入webSocketServerWrapper,例如index.js,然后运行webSocketServerWrapper.start(PORT);
Here,我写的第二个答案,你可以找到负责创建示例订阅的代码以及如何处理它。