54 lines
1.6 KiB
JavaScript
54 lines
1.6 KiB
JavaScript
|
module.exports = Self => {
|
||
|
|
Self.rewriteDbError = function(replaceErrFunc) {
|
||
|
|
this.once('attached', () => {
|
||
|
|
let realUpsert = this.upsert;
|
||
|
|
this.upsert = async(data, options, cb) => {
|
||
|
|
if (options instanceof Function) {
|
||
|
|
cb = options;
|
||
|
|
options = null;
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
await realUpsert.call(this, data, options);
|
||
|
|
if (cb) cb();
|
||
|
|
} catch (err) {
|
||
|
|
let myErr = replaceErr(err, replaceErrFunc);
|
||
|
|
if (cb)
|
||
|
|
cb(myErr);
|
||
|
|
else
|
||
|
|
throw myErr;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
let realCreate = this.create;
|
||
|
|
this.create = async(data, options, cb) => {
|
||
|
|
if (options instanceof Function) {
|
||
|
|
cb = options;
|
||
|
|
options = null;
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
await realCreate.call(this, data, options);
|
||
|
|
if (cb) cb();
|
||
|
|
} catch (err) {
|
||
|
|
let myErr = replaceErr(err, replaceErrFunc);
|
||
|
|
if (cb)
|
||
|
|
cb(myErr);
|
||
|
|
else
|
||
|
|
throw myErr;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
function replaceErr(err, replaceErrFunc) {
|
||
|
|
if (Array.isArray(err)) {
|
||
|
|
let errs = [];
|
||
|
|
for (let e of err)
|
||
|
|
errs.push(replaceErrFunc(e));
|
||
|
|
return errs;
|
||
|
|
}
|
||
|
|
return replaceErrFunc(err);
|
||
|
|
}
|
||
|
|
};
|