72 lines
1.3 KiB
JavaScript
72 lines
1.3 KiB
JavaScript
import mongoose from 'mongoose';
|
|
|
|
import UserModel from './user.model';
|
|
import logger from '../logger';
|
|
import config from '../config';
|
|
|
|
|
|
const options = {
|
|
useNewUrlParser: true,
|
|
user: config.database.user,
|
|
pass: config.database.password,
|
|
};
|
|
|
|
mongoose.connect(config.database.host, options);
|
|
mongoose.set('useFindAndModify', false);
|
|
|
|
const db = mongoose.connection;
|
|
|
|
db.on('error', (err) => logger.error(err));
|
|
db.once('open', () => logger.info('Database connected'));
|
|
|
|
|
|
export const initUser = (id) => (
|
|
new Promise((resolve, reject) => {
|
|
UserModel.findOneAndUpdate({ id }, { id }, { upsert: true }, (err, doc) => {
|
|
if (err) {
|
|
logger.error(err);
|
|
reject(err);
|
|
}
|
|
|
|
resolve(doc);
|
|
});
|
|
})
|
|
);
|
|
|
|
|
|
export const setCity = (userId, cityId) => {
|
|
const document = {
|
|
city: {
|
|
id: cityId,
|
|
},
|
|
};
|
|
|
|
return new Promise((resolve, reject) => {
|
|
UserModel.findOneAndUpdate({ id: userId }, document, { new: true }, (err, doc) => {
|
|
if (err) {
|
|
logger.error(err);
|
|
reject(err);
|
|
}
|
|
|
|
resolve(doc);
|
|
});
|
|
});
|
|
};
|
|
|
|
|
|
export const findAssociedCity = (id) => (
|
|
new Promise((resolve, reject) => {
|
|
UserModel.findOne({ id }, (err, doc) => {
|
|
if (err) {
|
|
logger.error(err);
|
|
reject(err);
|
|
}
|
|
|
|
resolve(doc.city.id);
|
|
});
|
|
})
|
|
);
|
|
|
|
|
|
export { UserModel };
|