-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
45 lines (37 loc) · 1.2 KB
/
index.js
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
module.exports = {
register(objection, options = {}) {
const {
columnName = 'deleted_at',
deletedValue= new Date(),
notDeletedValue = null
} = options;
class SoftDeleteQueryBuilder extends objection.QueryBuilder {
constructor(modelClass) {
super(modelClass)
if (modelClass.softDelete) {
this.onBuild(q => {
const isUpsert = q._operations.filter((operator) => operator.name === 'upsertGraph').length > 0;
if (!isUpsert && q.isFind() && !q.context().includeDeleted) {
q.where(`${q.tableRefFor(modelClass)}.${columnName}`, notDeletedValue);
}
});
}
}
withDeleted() {
return this.context({ includeDeleted: true });
}
delete(...args) {
if (this.modelClass().softDelete) {
return super.patch({ [columnName]: deletedValue });
}
return super.delete(...args);
}
forceDelete(...args) {
return super.delete(...args);
}
}
objection.QueryBuilder = SoftDeleteQueryBuilder;
objection.Model.QueryBuilder = SoftDeleteQueryBuilder;
objection.Model.RelatedQueryBuilder = SoftDeleteQueryBuilder;
}
};