switch to embeded data model

drillprop · Feb 27, 2019 · 2e2561f · 2e2561f
1 parent 6264fa5
commit 2e2561f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/src/graphql/resolvers/Mutation.js b/src/graphql/resolvers/Mutation.js
@@ -1,9 +1,8 @@
-import fetch from 'node-fetch';
 import bcrypt from 'bcryptjs';
+import 'dotenv/config';
+import jwt from 'jsonwebtoken';
 import Album from '../../models/Album';
 import User from '../../models/User';
-import jwt from 'jsonwebtoken';
-import 'dotenv/config';
 
 const Mutation = {
   createCd: async (parent, args, ctx, info) => {
@@ -18,8 +17,10 @@ const Mutation = {
       artist,
       image
     });
+    await album.save();
     await user.albums.push(album);
-    user.save();
+    await user.save();
+
     return album;
   },
   signup: async (parent, args, ctx, info) => {

diff --git a/src/models/Album.js b/src/models/Album.js
@@ -1,6 +1,6 @@
-import { Schema, model } from 'mongoose';
+import { model, Schema } from 'mongoose';
 
-const albumSchema = new Schema({
+export const albumSchema = new Schema({
   artist: String,
   title: String,
   image: String

diff --git a/src/models/User.js b/src/models/User.js
@@ -1,12 +1,13 @@
-import { Schema, model } from 'mongoose';
+import { model, Schema } from 'mongoose';
+import { albumSchema } from './Album';
 
 const userSchema = new Schema({
   name: { type: String, unique: true, required: true },
   email: { type: String, unique: true, required: true },
   password: { type: String, required: true },
   date: { type: Date, default: Date.now },
   avatar: String,
-  albums: []
+  albums: [albumSchema]
 });
 
 const User = model('Users', userSchema);