395 lines
12 KiB
JavaScript
395 lines
12 KiB
JavaScript
"use strict";
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.getAttrRefInExpression = exports.execOp = exports.opMultipleParams = exports.isExpression = exports.isAggrExpression = exports.isStringExpression = exports.isMathExpression = exports.isCompareExpression = exports.isBoolExpression = exports.isLogicExpression = exports.isDateExpression = exports.isGeoExpression = void 0;
|
||
const tslib_1 = require("tslib");
|
||
const assert_1 = tslib_1.__importDefault(require("assert"));
|
||
const dayjs_1 = tslib_1.__importDefault(require("dayjs"));
|
||
const weekOfYear_1 = tslib_1.__importDefault(require("dayjs/plugin/weekOfYear"));
|
||
const dayOfYear_1 = tslib_1.__importDefault(require("dayjs/plugin/dayOfYear"));
|
||
const geo_1 = require("../utils/geo");
|
||
dayjs_1.default.extend(weekOfYear_1.default);
|
||
dayjs_1.default.extend(dayOfYear_1.default);
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
;
|
||
function isGeoExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$contains', '$distance'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isGeoExpression = isGeoExpression;
|
||
function isDateExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$year', '$month', '$weekday', '$weekOfYear', '$day', '$dayOfMonth',
|
||
'$dayOfWeek', '$dayOfYear', '$dateDiff', '$dateCeil', '$dateFloor'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isDateExpression = isDateExpression;
|
||
function isLogicExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$and', '$or', '$not'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isLogicExpression = isLogicExpression;
|
||
function isBoolExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$true', '$false'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isBoolExpression = isBoolExpression;
|
||
function isCompareExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$gt', '$lt', '$gte', '$lte', '$eq', '$ne',
|
||
'$startsWith', '$endsWith', '$includes'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isCompareExpression = isCompareExpression;
|
||
function isMathExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$add', '$subtract', '$multiply', '$divide', '$abs', '$pow',
|
||
'$round', '$floor', '$ceil', '$mod'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isMathExpression = isMathExpression;
|
||
function isStringExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$concat'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isStringExpression = isStringExpression;
|
||
function isAggrExpression(expression) {
|
||
if (Object.keys(expression).length == 1) {
|
||
const op = Object.keys(expression)[0];
|
||
if (['$$max', '$$min', '$$sum', '$$avg', '$$count'].includes(op)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
exports.isAggrExpression = isAggrExpression;
|
||
function isExpression(expression) {
|
||
return typeof expression === 'object' && Object.keys(expression).length === 1 && Object.keys(expression)[0].startsWith('$');
|
||
}
|
||
exports.isExpression = isExpression;
|
||
function opMultipleParams(op) {
|
||
return !['$year', '$month', '$weekday', '$weekOfYear', '$day', '$dayOfMonth',
|
||
'$dayOfWeek', '$dayOfYear', '$not', '$true', '$false', '$abs',
|
||
'$round', '$floor', '$ceil', '$$max', '$$min', '$$sum', '$$avg', '$$count'].includes(op);
|
||
}
|
||
exports.opMultipleParams = opMultipleParams;
|
||
function execOp(op, params) {
|
||
switch (op) {
|
||
case '$gt': {
|
||
return params[0] > params[1];
|
||
}
|
||
case '$lt': {
|
||
return params[0] < params[1];
|
||
}
|
||
case '$gte': {
|
||
return params[0] >= params[1];
|
||
}
|
||
case '$lte': {
|
||
return params[0] <= params[1];
|
||
}
|
||
case '$eq': {
|
||
return params[0] === params[1];
|
||
}
|
||
case '$ne': {
|
||
return params[0] !== params[1];
|
||
}
|
||
case '$startsWith': {
|
||
return params[0].startsWith(params[1]);
|
||
}
|
||
case '$endsWith': {
|
||
return params[0].endsWith(params[1]);
|
||
}
|
||
case '$includes': {
|
||
return params[0].includes(params[1]);
|
||
}
|
||
case '$add': {
|
||
if (typeof params[0] === 'number') {
|
||
let result = 0;
|
||
params.forEach((ele) => result += ele);
|
||
return result;
|
||
}
|
||
else {
|
||
let result = '';
|
||
params.forEach((ele) => result += ele);
|
||
return result;
|
||
}
|
||
}
|
||
case '$subtract': {
|
||
return params[0] - params[1];
|
||
}
|
||
case '$multiply': {
|
||
let result = 1;
|
||
params.forEach((ele) => result = result * ele);
|
||
return result;
|
||
}
|
||
case '$divide': {
|
||
return params[0] / params[1];
|
||
}
|
||
case '$abs': {
|
||
return Math.abs(params);
|
||
}
|
||
case '$round': {
|
||
return Math.round(params);
|
||
}
|
||
case '$ceil': {
|
||
return Math.ceil(params);
|
||
}
|
||
case '$mod': {
|
||
return params[0] % params[1];
|
||
}
|
||
case '$floor': {
|
||
return Math.floor(params);
|
||
}
|
||
case '$round': {
|
||
return Math.round(params);
|
||
}
|
||
case '$pow': {
|
||
return Math.pow(params[0], params[1]);
|
||
}
|
||
case '$true': {
|
||
return !!params;
|
||
}
|
||
case '$false':
|
||
case '$not': {
|
||
return !params;
|
||
}
|
||
case '$and': {
|
||
for (const p of params) {
|
||
if (!p) {
|
||
return false;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
case '$or': {
|
||
for (const p of params) {
|
||
if (!!p) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
case '$year': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.year();
|
||
}
|
||
case '$month': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.month();
|
||
}
|
||
case '$weekday': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.day(); // 0~6
|
||
}
|
||
case '$weekOfYear': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.week();
|
||
}
|
||
case '$day':
|
||
case '$dayOfMonth': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.date();
|
||
}
|
||
case '$dayOfWeek': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.day(); // 0~6
|
||
}
|
||
case '$dayOfYear': {
|
||
const value = (0, dayjs_1.default)(params);
|
||
return value.dayOfYear(); // 0~6
|
||
}
|
||
case '$dateDiff': {
|
||
const value1 = (0, dayjs_1.default)(params[0]);
|
||
const value2 = (0, dayjs_1.default)(params[1]);
|
||
switch (params[2]) {
|
||
case 'y':
|
||
case 'M':
|
||
case 'd':
|
||
case 'h':
|
||
case 'm':
|
||
case 's': {
|
||
return value1.diff(value2, params[2]);
|
||
}
|
||
default: {
|
||
(0, assert_1.default)(false);
|
||
}
|
||
}
|
||
}
|
||
case '$dateCeil': {
|
||
const value = (0, dayjs_1.default)(params[0]);
|
||
switch (params[1]) {
|
||
case 'y': {
|
||
return value.startOf('year').millisecond();
|
||
}
|
||
case 'M': {
|
||
return value.startOf('month').millisecond();
|
||
}
|
||
case 'd': {
|
||
return value.startOf('day').millisecond();
|
||
}
|
||
case 'h': {
|
||
return value.startOf('hour').millisecond();
|
||
}
|
||
case 'm': {
|
||
return value.startOf('minute').millisecond();
|
||
}
|
||
case 's': {
|
||
return value.startOf('second').millisecond();
|
||
}
|
||
default: {
|
||
(0, assert_1.default)(false);
|
||
}
|
||
}
|
||
}
|
||
case '$dateFloor': {
|
||
const value = (0, dayjs_1.default)(params[0]);
|
||
switch (params[1]) {
|
||
case 'y': {
|
||
return value.endOf('year').millisecond();
|
||
}
|
||
case 'M': {
|
||
return value.endOf('month').millisecond();
|
||
}
|
||
case 'd': {
|
||
return value.endOf('day').millisecond();
|
||
}
|
||
case 'h': {
|
||
return value.endOf('hour').millisecond();
|
||
}
|
||
case 'm': {
|
||
return value.endOf('minute').millisecond();
|
||
}
|
||
case 's': {
|
||
return value.endOf('second').millisecond();
|
||
}
|
||
default: {
|
||
(0, assert_1.default)(false);
|
||
}
|
||
}
|
||
}
|
||
case '$distance': {
|
||
const [geo1, geo2] = params;
|
||
const { type: type1, coordinate: coordinate1 } = geo1;
|
||
const { type: type2, coordinate: coordinate2 } = geo2;
|
||
if (type1 !== 'point' || type2 !== 'point') {
|
||
throw new Error('目前只支持point类型的距离运算');
|
||
}
|
||
return (0, geo_1.getDistanceBetweenPoints)(coordinate1[1], coordinate1[0], coordinate2[1], coordinate2[0]);
|
||
}
|
||
case '$contains': {
|
||
throw new Error('$contains类型未实现');
|
||
}
|
||
case '$concat': {
|
||
return params.join('');
|
||
}
|
||
default: {
|
||
(0, assert_1.default)(false, `不能识别的expression运算符：${op}`);
|
||
}
|
||
}
|
||
}
|
||
exports.execOp = execOp;
|
||
/**
|
||
* 检查一个表达式，并分析其涉及到的属性
|
||
* @param expression
|
||
* @returns {
|
||
* '#current': [当前结点涉及的属性]
|
||
* 'node-1': [node-1结点上涉及的属性]
|
||
* }
|
||
*/
|
||
function getAttrRefInExpression(expression) {
|
||
const result = {
|
||
['#current']: [],
|
||
};
|
||
const check = (node) => {
|
||
if (node['#attr']) {
|
||
result['#current'].push(node['#attr']);
|
||
}
|
||
else if (node['#refAttr']) {
|
||
if (result[node['#refId']]) {
|
||
result[node['#refId']].push(node['#refAttr']);
|
||
}
|
||
else {
|
||
Object.assign(result, {
|
||
[node['#refId']]: [node['#refAttr']],
|
||
});
|
||
}
|
||
}
|
||
else if (node instanceof Array) {
|
||
for (const subNode of node) {
|
||
check(subNode);
|
||
}
|
||
}
|
||
else if (typeof node === 'object') {
|
||
for (const attr in node) {
|
||
check(node[attr]);
|
||
}
|
||
}
|
||
};
|
||
check(expression);
|
||
return result;
|
||
}
|
||
exports.getAttrRefInExpression = getAttrRefInExpression;
|