-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
implement custom query expression for attributes (#166)
* implement custom query expression for attributes * fix regex warnings * rm unused regexp group * rm unused regexp group * rm unused regexp group * rm unused regexp group * rm unused regexp group * 2.6.56
- Loading branch information
1 parent
8d93c16
commit 65924d4
Showing
8 changed files
with
789 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,194 @@ | ||
| const {Args, DataError} = require('@themost/common'); | ||
| const {hasOwnProperty} = require('./has-own-property'); | ||
| const {QueryEntity, QueryExpression, QueryField} = require('@themost/query'); | ||
| class DataFieldQueryResolver { | ||
| /** | ||
| * @param {import("./data-model").DataModel} target | ||
| */ | ||
| constructor(target) { | ||
| this.target = target; | ||
| } | ||
|
|
||
| /** | ||
| * | ||
| * @param {string} value | ||
| * @returns {string} | ||
| */ | ||
| formatName(value) { | ||
| if (/^\$/.test(value)) { | ||
| return value.replace(/(\$?(\w+)?)/g, '$2').replace(/\.(\w+)/g, '.$1') | ||
| } | ||
| return value; | ||
| } | ||
|
|
||
| nameReplacer(key, value) { | ||
| if (typeof value === 'string') { | ||
| if (/^\$\w+$/.test(value)) { | ||
| const baseModel = this.target.base(); | ||
| const name = value.replace(/^\$/, ''); | ||
| let field = null; | ||
| let collection = null; | ||
| // try to find if field belongs to base model | ||
| if (baseModel) { | ||
| field = baseModel.getAttribute(name); | ||
| collection = baseModel.viewAdapter; | ||
| } | ||
| if (field == null) { | ||
| collection = this.target.sourceAdapter; | ||
| field = this.target.getAttribute(name); | ||
| } | ||
| if (field) { | ||
| return { | ||
| $name: collection + '.' + name | ||
| } | ||
| } | ||
| throw new DataError('An expression contains an attribute that cannot be found', null, this.target.name, name); | ||
| } else { // noinspection RegExpUnnecessaryNonCapturingGroup | ||
| if (/^\$\w+\.\w+$/.test(value)) { | ||
| return { | ||
| $name: value.replace(/^\$/, '') | ||
| } | ||
| } | ||
| } | ||
| } | ||
| return value; | ||
| } | ||
|
|
||
| /** | ||
| * @param {import("./types").DataField} field | ||
| * @returns {{$select?: import("@themost/query").QueryField, $expand?: import("@themost/query").QueryEntity[]}|null} | ||
| */ | ||
| resolve(field) { | ||
| Args.check(field != null, new DataError('E_FIELD','Field may not be null', null, this.target.name)); | ||
| if (Array.isArray(field.query) === false) { | ||
| return { | ||
| select: null, | ||
| expand: [] | ||
| }; | ||
| } | ||
| let expand = []; | ||
| let select = null; | ||
| const self = this; | ||
| // get base model | ||
| const baseModel = this.target.base(); | ||
| for (const stage of field.query) { | ||
| if (stage.$lookup) { | ||
| // get from model | ||
| const from = stage.$lookup.from; | ||
| const fromModel = this.target.context.model(from); | ||
| if (stage.$lookup.pipeline && stage.$lookup.pipeline.length) { | ||
| stage.$lookup.pipeline.forEach(function(pipelineStage) { | ||
| if (pipelineStage.$match && pipelineStage.$match.$expr) { | ||
| const q = new QueryExpression().select('*').from(self.target.sourceAdapter); | ||
| // get expression as string | ||
| const exprString = JSON.stringify(pipelineStage.$match.$expr, function(key, value) { | ||
| if (typeof value === 'string') { | ||
| if (/\$\$\w+/.test(value)) { | ||
| let localField = /\$\$(\w+)/.exec(value)[1]; | ||
| let localFieldAttribute = self.target.getAttribute(localField); | ||
| if (localFieldAttribute && localFieldAttribute.model === self.target.name) { | ||
| return { | ||
| $name: self.target.sourceAdapter + '.' + localField | ||
| } | ||
| } | ||
| if (baseModel) { | ||
| localFieldAttribute = baseModel.getAttribute(localField); | ||
| if (localFieldAttribute) { | ||
| return { | ||
| $name: baseModel.viewAdapter + '.' + localField | ||
| } | ||
| } | ||
| } | ||
| throw new DataError('E_FIELD', 'Data field cannot be found', null, self.target.name, localField); | ||
| } | ||
| } | ||
| return self.nameReplacer(key, value); | ||
| }); | ||
| const joinCollection = new QueryEntity(fromModel.viewAdapter).as(stage.$lookup.as).left(); | ||
| Object.defineProperty(joinCollection, 'model', { | ||
| configurable: true, | ||
| enumerable: false, | ||
| writable: true, | ||
| value: fromModel.name | ||
| }); | ||
| const joinExpression = Object.assign(new QueryExpression(), { | ||
| $where: JSON.parse(exprString) | ||
| }); | ||
| q.join(joinCollection).with(joinExpression); | ||
| const appendExpand = [].concat(q.$expand); | ||
| expand.push.apply(expand, appendExpand); | ||
| } | ||
| }); | ||
| } else { | ||
| let localField = this.formatName(stage.$lookup.localField); | ||
| if (/\./.test(localField) === false) { | ||
| // get local field expression | ||
| let localFieldAttribute = this.target.getAttribute(localField); | ||
| if (localFieldAttribute && localFieldAttribute.model === this.target.name) { | ||
| localField = `${this.target.sourceAdapter}.${localField}`; | ||
| } else { | ||
| // get base model | ||
| const baseModel = this.target.base(); | ||
| if (baseModel) { | ||
| localFieldAttribute = baseModel.getAttribute(localField); | ||
| if (localFieldAttribute) { | ||
| localField = `${baseModel.viewAdapter}.${localField}`; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| const foreignField = this.formatName(stage.$lookup.foreignField); | ||
| const q = new QueryExpression().select('*').from(this.target.sourceAdapter); | ||
| Args.check(fromModel != null, new DataError('E_MODEL', 'Data model cannot be found', null, from)); | ||
| const joinCollection = new QueryEntity(fromModel.viewAdapter).as(stage.$lookup.as).left(); | ||
| Object.defineProperty(joinCollection, 'model', { | ||
| configurable: true, | ||
| enumerable: false, | ||
| writable: true, | ||
| value: fromModel.name | ||
| }); | ||
| q.join(joinCollection).with( | ||
| new QueryExpression().where(new QueryField(localField)) | ||
| .equal(new QueryField(foreignField).from(stage.$lookup.as)) | ||
| ); | ||
| const appendExpand = [].concat(q.$expand); | ||
| expand.push.apply(expand, appendExpand); | ||
| } | ||
| } | ||
| const name = field.property || field.name; | ||
| if (stage.$project) { | ||
| Args.check(hasOwnProperty(stage.$project, name), new DataError('E_QUERY', 'Field projection expression is missing.', null, this.target.name, field.name)); | ||
| const expr = Object.getOwnPropertyDescriptor(stage.$project, name).value; | ||
| if (typeof expr === 'string') { | ||
| select = new QueryField(this.formatName(expr)).as(name) | ||
| } else { | ||
| const expr1 = Object.defineProperty({}, name, { | ||
| configurable: true, | ||
| enumerable: true, | ||
| writable: true, | ||
| value: expr | ||
| }); | ||
| // Important note: Field references e.g. $customer.email | ||
| // are not supported by @themost/query@Formatter | ||
| // and should be replaced by name references e.g. { "$name": "customer.email" } | ||
| // A workaround is being used here is a regular expression replacer which | ||
| // will try to replace "$customer.email" with { "$name": "customer.email" } | ||
| // but this operation is definitely a feature request for @themost/query | ||
| const finalExpr = JSON.parse(JSON.stringify(expr1, function(key, value) { | ||
| return self.nameReplacer(key, value); | ||
| })); | ||
| select = Object.assign(new QueryField(), finalExpr); | ||
| } | ||
| } | ||
| } | ||
| return { | ||
| $select: select, | ||
| $expand: expand | ||
| } | ||
| } | ||
|
|
||
| } | ||
|
|
||
| module.exports = { | ||
| DataFieldQueryResolver | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.