Spaces:
Runtime error
Runtime error
File size: 2,124 Bytes
56b6519 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
var mongoose = require('mongoose');
var Schema = mongoose.Schema;
var CompanySchema = new Schema(
{
name: { type: String, required: true, unique: true },
shortName: String,
logo: String,
},
{ timestamps: true },
);
/*
*** Statics ***
*/
// Get all companies
CompanySchema.statics.getAll = () => {
return new Promise((resolve, reject) => {
var query = Company.find();
query.select('name shortName logo');
query
.exec()
.then(rows => {
resolve(rows);
})
.catch(err => {
reject(err);
});
});
};
// Create company
CompanySchema.statics.create = company => {
return new Promise((resolve, reject) => {
var query = new Company(company);
query
.save(company)
.then(row => {
resolve({ _id: row._id, name: row.name });
})
.catch(err => {
if (err.code === 11000)
reject({
fn: 'BadParameters',
message: 'Company name already exists',
});
else reject(err);
});
});
};
// Update company
CompanySchema.statics.update = (companyId, company) => {
return new Promise((resolve, reject) => {
var query = Company.findOneAndUpdate({ _id: companyId }, company);
query
.exec()
.then(rows => {
if (rows) resolve(rows);
else reject({ fn: 'NotFound', message: 'Company Id not found' });
})
.catch(err => {
if (err.code === 11000)
reject({
fn: 'BadParameters',
message: 'Company name already exists',
});
else reject(err);
});
});
};
// Delete company
CompanySchema.statics.delete = companyId => {
return new Promise((resolve, reject) => {
var query = Company.findOneAndDelete({ _id: companyId });
query
.exec()
.then(rows => {
if (rows) resolve(rows);
else reject({ fn: 'NotFound', message: 'Company Id not found' });
})
.catch(err => {
reject(err);
});
});
};
/*
*** Methods ***
*/
var Company = mongoose.model('Company', CompanySchema);
module.exports = Company;
|