update subscriptions

Author: nikolasburk

package.json

@@ -9,7 +9,7 @@
   "dependencies": {
     "@prisma/client": "^2.0.0-beta.2",
     "bcryptjs": "2.4.3",
-    "graphql-yoga": "1.17.4",
+    "graphql-yoga": "1.18.3",
     "jsonwebtoken": "8.5.1"
   },
   "devDependencies": {

prisma/dev.db

@@ -6,6 +6,9 @@ const Subscription = require('./resolvers/Subscription')
 const User = require('./resolvers/User')
 const Link = require('./resolvers/Link')
 const Vote = require('./resolvers/Vote')
+const { PubSub } = require('graphql-yoga')
+
+const pubsub = new PubSub()
 
 const prisma = new PrismaClient({
   errorFormat: 'minimal',
@@ -27,6 +30,7 @@ const server = new GraphQLServer({
     return {
       ...request,
       prisma,
+      pubsub
     }
   },
 })

src/index.js

@@ -4,13 +4,17 @@ const { APP_SECRET, getUserId } = require('../utils')
 
 function post(parent, args, context, info) {
   const userId = getUserId(context)
-  return context.prisma.link.create({
+
+  const newLink = context.prisma.link.create({
     data: {
       url: args.url,
       description: args.description,
       postedBy: { connect: { id: userId } },
     }
   })
+  context.pubsub.publish("NEW_LINK", newLink)
+
+  return newLink
 }
 
 async function signup(parent, args, context, info) {
@@ -59,12 +63,15 @@ async function vote(parent, args, context, info) {
     throw new Error(`Already voted for link: ${args.linkId}`)
   }
 
-  return context.prisma.vote.create({
+  const newVote = context.prisma.vote.create({
     data: {
       user: { connect: { id: userId } },
       link: { connect: { id: Number(args.linkId) } },
     }
   })
+  context.pubsub.publish("NEW_VOTE", newVote)
+
+  return newVote
 }
 
 module.exports = {

src/resolvers/Mutation.js

@@ -1,5 +1,5 @@
 function newLinkSubscribe(parent, args, context, info) {
-  return context.prisma.$subscribe.link({ mutation_in: ['CREATED'] }).node()
+  return context.pubsub.asyncIterator("NEW_LINK")
 }
 
 const newLink = {
@@ -10,7 +10,7 @@ const newLink = {
 }
 
 function newVoteSubscribe(parent, args, context, info) {
-  return context.prisma.$subscribe.vote({ mutation_in: ['CREATED'] }).node()
+  return context.pubsub.asyncIterator("NEW_VOTE")
 }
 
 const newVote = {
@@ -22,5 +22,5 @@ const newVote = {
 
 module.exports = {
   newLink,
-  newVote,
-}
+  newVote
+}