yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-07-25 09:56:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add other hooks

Browse Source 
Signed-off-by: Alexandre Bodin <bodin.alex@gmail.com>
This commit is contained in:
Alexandre Bodin 2020-04-21 18:41:36 +02:00
parent ebee707ec4
commit de5d1d6d54
1 changed files with 27 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
packages/strapi-database/lib/queries/create-query.js
View File

@ -26,6 +26,12 @@ class Query {
return this.model.associations; return this.model.associations;
} }
async executeHook(hook, ...args) {
if (_.has(this.model, hook)) {
await this.model[hook](...args);
}
}
/** /**
* Run custom database logic * Run custom database logic
*/ */
@ -65,12 +71,6 @@ class Query {
return result; return result;
} }
async executeHook(hook, ...args) {
if (_.has(this.model, hook)) {
await this.model[hook](...args);
}
}
async create(params = {}, ...args) { async create(params = {}, ...args) {
const newParams = replaceIdByPrimaryKey(params, this.model); const newParams = replaceIdByPrimaryKey(params, this.model);
@ -101,18 +101,33 @@ class Query {
return entry; return entry;
} }
count(params = {}, ...args) { async count(params = {}, ...args) {
const newParams = replaceIdByPrimaryKey(params, this.model); const newParams = replaceIdByPrimaryKey(params, this.model);
return this.connectorQuery.count(newParams, ...args);
await this.executeHook('beforeCount', newParams, ...args);
const count = await this.connectorQuery.count(newParams, ...args);
await this.executeHook('afterCount', count);
return count;
} }
search(params = {}, ...args) { async search(params = {}, ...args) {
const newParams = replaceIdByPrimaryKey(params, this.model); const newParams = replaceIdByPrimaryKey(params, this.model);
return this.connectorQuery.search(newParams, ...args);
await this.executeHook('beforeSearch', newParams, ...args);
const results = await this.connectorQuery.search(newParams, ...args);
await this.executeHook('afterSearch', results);
return results;
} }
countSearch(params = {}, ...args) { async countSearch(params = {}, ...args) {
const newParams = replaceIdByPrimaryKey(params, this.model); const newParams = replaceIdByPrimaryKey(params, this.model);
return this.connectorQuery.countSearch(newParams, ...args);
await this.executeHook('beforeCountSearch', newParams, ...args);
const count = await this.connectorQuery.countSearch(newParams, ...args);
await this.executeHook('afterCountSearch', count);
return count;
} }
} }