Как добавить данные в массив MongoDB?

Question

[Айнур Ярулин]

Здравствуйте, подскажите как добавить данные в массив MongoDB.
Если быть конкретнее, то если пользователь перешел по реферальной ссылке, то его нужно
добавить в массив (referrals = []) человека который пригласил этого самого реферала.

Вот так выглядит схема пользователя:

userSchema

const userSchema = new Schema({
    login: {
        type: String,
        required: true
    },
    email: {
        type: String,
        required: true
    },
    password: {
        type: String,
        required: true
    },
    referrals: []
})

// Так я ищу человека, по ссылке которого перешли
const referrer = await User.findOne({ _id: req.query.ref })

// Так я сохраняю нового пользователя в базу данных
const user = new User({ 
    login: ...,
    email: ...,
    password: ...
})
await user.save()

// А это я пытался добавить нового реферала в массив, пользователь которого его пригласил.
referrer.update({
    referrals: user
})
// Но моя татарская логика меня подвела и такой код не сработал

Можете подсказать, что нужно делать? Спасибо.

Answer 1

[Алексей Ярков]

Как-то так:

const userSchema = new Schema({
    login: {
        type: String,
        required: true
    },
    email: {
        type: String,
        required: true
    },
    password: {
        type: String,
        required: true
    },
    referrals: [{
      type: Schema.Types.ObjectId,
      ref: 'User',
    }]
})

referrer.update({
    $push: {
        referrals: user._id
    }
})

Comments

[Айнур Ярулин]

Алексей Ярков,

referrals: user._id

А обязательно писал на конце _id?

[Алексей Ярков]

squadbrodyaga, что вас останавливает от попытки сделать так и так и сравнить?

[Айнур Ярулин]

Алексей Ярков, ну да кстати, спасибо за помощь)

[Алексей Ярков]

squadbrodyaga, пожалуйста )

[Айнур Ярулин]

Алексей Ярков, Блин, у меня вылезла ошибка TypeError: Cannot read property 'update'

[Айнур Ярулин]

Алексей Ярков, А ой, забыл поменять код в UserSchema

[Алексей Ярков]

squadbrodyaga, хочу напомнить, что надо будет вызывать метод populate, чтобы получить полноценный массив referrals. Ну или плагин юзать.

[Айнур Ярулин]

Алексей Ярков, так, этот плагин нужен, чтобы у меня всё работало? Просто сейчас не работает,
а как вызвать populate?

[Алексей Ярков]

squadbrodyaga,

как вызвать populate?

Ну не знаю... Может в документации чего пишут?

[Айнур Ярулин]

Алексей Ярков, может, может) Надеюсь это будет моё последнее сообщение и дальше всё будет работать

[Айнур Ярулин]

Алексей Ярков, блин, не хочет работать

[Алексей Ярков]

squadbrodyaga, ну так завтра праздник, а рабочий день уже закончился ))
Ошибка какая хоть?

[Айнур Ярулин]

Алексей Ярков, вот именно, ошибок то нету, просто в базе ничего не появляется.
Может где-то что-то не написал

router.post('/signup', async (req, res) => {

    const { login, email, password } = req.body

    // Если ссылка реферальная, то в переменной будут данные пригласившего. 
    const referrer = await User.findOne({ _id: req.query.ref })
        
    // Добавление пользователя в базу данных
    const user = new User({ 
        login, 
        email, 
        password: password,
    })
    await user.save()
    
    // Добавление нового пользователя в список рефералов пригласившего
    referrer.update({
        $push: {
            referrals: user._id
        }
    })
})

[Айнур Ярулин]

Алексей Ярков, TypeError: Cannot read property 'update'

[Алексей Ярков]

squadbrodyaga, ну логично. В вашей ситуации подойдёт user.update, но тогда непонятно зачем в массив пихать собственный _id. Может быть так?

user.update({
        $push: {
            referrals: referrer._id
        }
    })

[Айнур Ярулин]

Алексей Ярков, да не, стоп.
Мне нужно, чтобы Рефереру в массив referrals добавлялся человек, который зарегистрировался по его ссылке,
а НЕ рефералу в массив добавлялся человек, который его пригласил

[Алексей Ярков]

squadbrodyaga,

await User.update({ _id: req.query.ref }, {
        $push: {
            referrals: user._id
        }
    })

[Айнур Ярулин]

Алексей Ярков, ну сейчас ошибок не показывает, рекомендуют использовать updateOne, updateMany, bulkWrite, потому что update устарело, а пользователь всё равно не показывается в базе.

Может я накосячил в userSchema и неправильно установил плагин mongoose-autopopulate?

const {model, Schema} = require('mongoose')
const userSchema = new Schema({
    login: {
        type: String,
        required: true
    },
    email: {
        type: String,
        required: true
    },
    password: {
        type: String,
        required: true
    },
    referrals: [{
        type: Schema.Types.ObjectId,
        ref: 'User',
        autopopulate: true
    }]
})
userSchema.plugin(require('mongoose-autopopulate'));
module.exports = model('User', userSchema)

[Алексей Ярков]

squadbrodyaga, mongoose-autopopulate на выборку влияет, а не на добавление. Вы как проверяете наличие записи в БД?

[Айнур Ярулин]

Алексей Ярков, На сайте cloud.mongodb.com

[Айнур Ярулин]

Алексей Ярков, когда стрелочку нажимаешь, то ничего не показывает

[Алексей Ярков]

squadbrodyaga, ну вот так я у себя в проекте добавляю элемент в массив, когда создаю страницу.
Добавляю юзеру в массив memorials созданную страницу:

async (req: Request, res: Response) => {
        const memorial = new MemorialModel(filterEmptyPropertiesObject<CreateMemorial>({
          author: req.session?.user._id,
          /*...*/
        }))

        try {
          const savedMemorial = await this.memorialRepo.create(memorial)
          await this.userRepo.update({
            id: req.session?.user.id // id у меня числовой автоинкрементный
          }, {
            $push: {
              memorials: savedMemorial._id
            }
          })

          return res.json({
            memorial: getMemorialResponse(savedMemorial)
          })
        } catch (err) {
          return res.status(500).json({
            message: 'Ошибка при сохранении страницы в базу',
            error: err
          })
        }
      }

[Айнур Ярулин]

Алексей Ярков, блин, у меня вечно всё через jopy) Сегодня у меня будет долгая ночь

[Алексей Ярков]

squadbrodyaga, советую отложить на утро. Правда. На свежую голову решение придёт куда быстрее.

[Айнур Ярулин]

Алексей Ярков, мне не нужен сон, мне нужны ответы)

[Алексей Ярков]

squadbrodyaga, ну тогда удачи )) А я уже стакан холодный из морозилки достал. Пойду деградировать ))