assign albums to user only

drillprop · Feb 26, 2019 · 6264fa5 · 6264fa5
1 parent 94d3ffc
commit 6264fa5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/graphql/resolvers/Mutation.js b/src/graphql/resolvers/Mutation.js
@@ -12,8 +12,14 @@ const Mutation = {
       throw new Error('Sign in to add a cd');
     }
     let { title, artist, image } = args;
-    const album = new Album({ title, artist, image });
-    await album.save();
+    console.log(user._id);
+    const album = new Album({
+      title,
+      artist,
+      image
+    });
+    await user.albums.push(album);
+    user.save();
     return album;
   },
   signup: async (parent, args, ctx, info) => {

diff --git a/src/models/User.js b/src/models/User.js
@@ -6,7 +6,7 @@ const userSchema = new Schema({
   password: { type: String, required: true },
   date: { type: Date, default: Date.now },
   avatar: String,
-  albums: [{ type: Schema.Types.ObjectId, ref: 'Album' }]
+  albums: []
 });
 
 const User = model('Users', userSchema);