Passport.js + mongoose объединяем две коллекции

Добрый день! Решил поделиться своими небольшими достижениями в использовании популярной библиотеки Passport.js. Задача была такой: использовать две локальные стратегии passport.js при том, что пользователи находятся в двух разных коллекциях Mongo Atlas.

Я искал ответы в интернете и встретил подобные вопросы, но либо ответа нет вообще, либо все не то:

• Тут ответа не нашлось

• Тут ответ близок и даже наверно рабочий, но он работает за счет дополнительного запроса к БД

Но из этих двух вопросов мне понравился больше второй, на его основе и рассмотрим как же реализовать локальные стратегии Passport.js если пользователи находятся в разных коллекциях.

И так, Дано: две коллекции пользователей в Mongo Atlas, и две локальные стратеги в Passport.js.

В данной статье я не буду описывать как сделать свою локальную стратегию. Это хорошо описано в этой статье, я как раз опирался на нее, когда писал свою, рекомендую.

Задача, реализовать аутентификацию учителей по логину и паролю, учеников по Фамилии и номеру класса. Тут кстати в вопросе за одно и объясняется, почему мы используем две локальных стратегии, именно потому, что способы идентификации пользователя разные (учителя- логин/пароль, ученики — Фамилия/класс)

Давайте рассмотрим, что мы имеем.

Коллекция учителей:

teachers = { login: "Zayka-English", password: "qwerty", klass: "7Б", ... }

Коллекция ученики:

students = { surname: "Иванов", klass: "7Б", category: "Ботан", ... }

Стратегия passport.js для входа учителей:

passport.use('teachers_login', new LocalStrategy({     // Переписываем переменные из того что пришло из формы     usernameField: 'login',     passReqToCallback: true },     async function (req, username, password, done) {         // Поиск пользователя по логину         await Teachers.findOne({ login: username }, function (err, user) {             if (err) { return done(err); }             // проверяем результат поиска             if (!user) {                 return done(null, false, req.flash('msg', ['Ошибка', ': Пользователь не найден']));             } else {                 // Проверяем пароль                 if (user.password == password) {                     console.log("Пользователь выполнил вход");                     return done(null, user);                 } else {                     console.log('Неправильно введен пароль!');                     return done(null, false, req.flash('msg', ['Ошибка', ': Неправильно введен пароль']));                 }             }         });     } ))

Стратегия passport.js для входа учеников:

passport.use('students_login', new LocalStrategy({     // Переписываем переменные из того что пришло из формы     usernameField: 'surname',   	userpasswordField: 'klass',     passReqToCallback: true },     async function (req, username, password, done) {         // Поиск пользователя по Фамилии         await Students.findOne({ surname: username }, function (err, user) {             if (err) { return done(err); }             // проверяем результат поиска             if (!user) {                 return done(null, false, req.flash('msg', ['Ошибка', ': Пользователь не найден']));             } else {                 // Проверяем пароль                 if (user.password == password) {                     console.log("Пользователь выполнил вход");                     return done(null, user);                 } else {                     return done(null, false, req.flash('msg', ['Ошибка', ': Неправильно введен класс']));                 }             }         });     } ))

Ну и конечно же сериализация и десериализация:

// Записывает id пользователя в cookie passport.serializeUser(function (user, done) {     done(null, user.id); });  // Проверяет пользователя по id из cookie passport.deserializeUser(function(id, done) {   User.findById(id, function(err, user) {     done(err, user);   }); });

Сериализация и десериализация взяты с сайта passport.js и требуют поправки, чем мы сейчас и займемся. Вся проблема нашей задачи состоит в том, что во время входа пользователя сериализация сработает и запишет в cookie id пользователя, но при десериализации нам нужно реализовать поиск в нужной коллекции. В одном из ответов в интернете было предложено поочередно проверять обе коллекции. То есть сначала ищем id в коллекции с учителями, если там нет. то ищем в коллекции с учениками. Но такой подход использует несколько запросов к БД, что увеличивает время на десериализацию. А что если коллекций не две, а больше? Я где-то видел спрашивалось о трех коллекциях, или представим, что коллекций 5! В этом случае десериализация может происходить достаточно долго. Что же делать? А ответ оказался достаточно прост. Нам необходимо дать понять, в какой из коллекций искать id пользователя при десериализации. Давайте сделаем это.

Для того, что бы это реализовать для начала в наших коллекциях добавим к объектам с пользователями некоторые флаги:

Коллекция учителей:

teachers = {    login: "Zayka-English",    password: "qwerty",    klass: "7Б",    flag: "teacher", // в значении флага напишем к какой категории относится пользователь   ... }

Коллекция учеников:

students = { 	surname: "Иванов", 	klass: "7Б", 	category: "Ботан", 	flag: "student", //тут делаем аналогично 	...}

Теперь во время входа пользователя мы запишем в cookie не только id пользователя, но и флаг, который есть в нашем объекте User. Это делается при сериализации

passport.serializeUser(function (user, done) { // На данный этапе в переменной user находится объект с данными пользователя // который мы получили при его идентификации в наших стратегиях     done(null, [user.flag, user.id]); });

Теперь после того как пользователь «представился» мы внесли в cookie массив, в котором первый элемент — это flag (категория к которой относится пользователь), а второй — это id пользователя.

Теперь при десериализации нам необходимо просто проверить к какой из категорий относится пользователь и проверять его id в конкретной коллекции:

passport.deserializeUser(function(id, done) {   // На этом этапе переменная id содержит наш массив   // Если флаг "учитель", ищем id в коллекции "Учителя"   if (id[0] == "teacher"){ 	  Teacher.findById(id[1], function(err, user) {     	done(err, user);   	})   // Если иное, ищем id в коллекции "Ученики"   }else{   	Student.findById(id[1], function(err, user) {     	done(err, user);   	})   } });

При такой реализации десериализация будет происходить именно в той коллекции, к которой относится пользователь, то есть будет происходить только один запрос к БД к конкретной коллекции.

Так как у нас всего две коллекции, то выбор коллекции при десериализации мы осуществили через if/else, если у вас больше коллекций, конечно же будет уместней использовать конструкцию switch

На этом все дорогие друзья, в следующей статье я напишу как же сделать авторизацию учителя, что бы в ответе получить объект с данными учителя и с массивом объектов учеников)

Использовал при написании:

Пожалуй, лучший учебник по JavaSkript

Официальная документация по Passport.js

Статья о том, как создать свою локальную коллекцию в passport.js