hongbo-miao/hongbomiao.com

import { GraphQLObjectType, GraphQLString } from 'graphql';
import { PubSub } from 'graphql-subscriptions';

const pubsub = new PubSub();

const subscription = new GraphQLObjectType({
  name: 'Subscription',
  fields: {
    ping: {
      type: GraphQLString,
      resolve: (source) => {
        if (source instanceof Error) {
          throw source;
        }
        return source.ping;
      },
      subscribe: () => {
        return pubsub.asyncIterator('ping');
      },
    },
  },
});

setInterval(() => {
  pubsub.publish('ping', {
    ping: 'pong',
  });
}, 5e3).unref(); // 5s

export default subscription;