mirror of https://github.com/nocodb/nocodb
Pranav C
3 years ago
committed by
Pranav C
8 changed files with 222 additions and 13 deletions
@ -0,0 +1,26 @@ |
|||||||
|
<template> |
||||||
|
<v-text-field |
||||||
|
v-model="formula.value" |
||||||
|
outlined |
||||||
|
class="caption" |
||||||
|
hide-details="auto" |
||||||
|
label="Formula" |
||||||
|
persistent-hint |
||||||
|
hint="Available formulas are ADD, AVG, CONCAT, +, -, /" |
||||||
|
/> |
||||||
|
</template> |
||||||
|
|
||||||
|
<script> |
||||||
|
export default { |
||||||
|
name: 'FormulaOptions', |
||||||
|
props: ['nodes', 'column', 'meta', 'isSQLite', 'alias'], |
||||||
|
data: () => ({ |
||||||
|
formula: {}, |
||||||
|
formulas: ['AVERAGE()', 'COUNT()', 'COUNTA()', 'COUNTALL()', 'SUM()', 'MIN()', 'MAX()', 'AND()', 'OR()', 'TRUE()', 'FALSE()', 'NOT()', 'XOR()', 'ISERROR()', 'IF()', 'LEN()', 'MID()', 'LEFT()', 'RIGHT()', 'FIND()', 'CONCATENATE()', 'T()', 'VALUE()', 'ARRAYJOIN()', 'ARRAYUNIQUE()', 'ARRAYCOMPACT()', 'ARRAYFLATTEN()', 'ROUND()', 'ROUNDUP()', 'ROUNDDOWN()', 'INT()', 'EVEN()', 'ODD()', 'MOD()', 'LOG()', 'EXP()', 'POWER()', 'SQRT()', 'CEILING()', 'FLOOR()', 'ABS()', 'RECORD_ID()', 'CREATED_TIME()', 'ERROR()', 'BLANK()', 'YEAR()', 'MONTH()', 'DAY()', 'HOUR()', 'MINUTE()', 'SECOND()', 'TODAY()', 'NOW()', 'WORKDAY()', 'DATETIME_PARSE()', 'DATETIME_FORMAT()', 'SET_LOCALE()', 'SET_TIMEZONE()', 'DATESTR()', 'TIMESTR()', 'TONOW()', 'FROMNOW()', 'DATEADD()', 'WEEKDAY()', 'WEEKNUM()', 'DATETIME_DIFF()', 'WORKDAY_DIFF()', 'IS_BEFORE()', 'IS_SAME()', 'IS_AFTER()', 'REPLACE()', 'REPT()', 'LOWER()', 'UPPER()', 'TRIM()', 'SUBSTITUTE()', 'SEARCH()', 'SWITCH()', 'LAST_MODIFIED_TIME()', 'ENCODE_URL_COMPONENT()', 'REGEX_EXTRACT()', 'REGEX_MATCH()', 'REGEX_REPLACE()'] |
||||||
|
}) |
||||||
|
} |
||||||
|
</script> |
||||||
|
|
||||||
|
<style scoped> |
||||||
|
|
||||||
|
</style> |
||||||
@ -0,0 +1,77 @@ |
|||||||
|
import {expect} from 'chai'; |
||||||
|
import 'mocha'; |
||||||
|
import knex from '../lib/dataMapper/lib/sql/CustomKnex'; |
||||||
|
import formulaQueryBuilder from "../lib/dataMapper/lib/sql/formulaQueryBuilder"; |
||||||
|
|
||||||
|
process.env.TEST = 'test'; |
||||||
|
|
||||||
|
describe('{Auth, CRUD, HasMany, Belongs} Tests', () => { |
||||||
|
|
||||||
|
let knexMysqlRef; |
||||||
|
let knexPgRef; |
||||||
|
let knexMssqlRef; |
||||||
|
let knexSqliteRef; |
||||||
|
|
||||||
|
|
||||||
|
// Called once before any of the tests in this block begin.
|
||||||
|
before(function (done) { |
||||||
|
knexMysqlRef = knex({client:'mysql2'}) |
||||||
|
knexMssqlRef = knex({client:'mssql'}) |
||||||
|
knexPgRef = knex({client:'pg'}) |
||||||
|
knexSqliteRef = knex({client:'sqlite3'}) |
||||||
|
done() |
||||||
|
}); |
||||||
|
|
||||||
|
|
||||||
|
after((done) => { |
||||||
|
done(); |
||||||
|
}); |
||||||
|
|
||||||
|
|
||||||
|
describe('Formulas', function () { |
||||||
|
|
||||||
|
it('Simple formula', function (done) { |
||||||
|
expect(formulaQueryBuilder("concat(city, ' _ ',city_id+country_id)", 'a',knexMysqlRef).toQuery()).eq('concat(`city`,\' _ \',`city_id` + `country_id`) as a') |
||||||
|
expect(formulaQueryBuilder("concat(city, ' _ ',city_id+country_id)", 'a',knexPgRef).toQuery()).eq('concat("city",\' _ \',"city_id" + "country_id") as a') |
||||||
|
expect(formulaQueryBuilder("concat(city, ' _ ',city_id+country_id)", 'a',knexMssqlRef).toQuery()).eq('concat([city],\' _ \',[city_id] + [country_id]) as a') |
||||||
|
expect(formulaQueryBuilder("concat(city, ' _ ',city_id+country_id)", 'a',knexSqliteRef).toQuery()).eq('`city` || (\' _ \' || (`city_id` + `country_id`)) as a') |
||||||
|
done() |
||||||
|
}); |
||||||
|
it('Addition', function (done) { |
||||||
|
expect(formulaQueryBuilder("ADD(city_id,country_id,2,3,4,5,4)", 'a',knexMysqlRef).toQuery()).eq('`city_id` + (`country_id` + (2 + (3 + (4 + (5 + 4))))) as a') |
||||||
|
expect(formulaQueryBuilder("ADD(city_id,country_id,2,3,4,5,4)", 'a',knexPgRef).toQuery()).eq('"city_id" + ("country_id" + (2 + (3 + (4 + (5 + 4))))) as a') |
||||||
|
expect(formulaQueryBuilder("ADD(city_id,country_id,2,3,4,5,4)", 'a',knexMssqlRef).toQuery()).eq('[city_id] + ([country_id] + (2 + (3 + (4 + (5 + 4))))) as a') |
||||||
|
expect(formulaQueryBuilder("ADD(city_id,country_id,2,3,4,5,4)", 'a',knexSqliteRef).toQuery()).eq('`city_id` + (`country_id` + (2 + (3 + (4 + (5 + 4))))) as a') |
||||||
|
done() |
||||||
|
}); |
||||||
|
it('Average', function (done) { |
||||||
|
expect(formulaQueryBuilder("AVG(city_id,country_id,2,3,4,5,4)", 'a',knexMysqlRef).toQuery()).eq('(`city_id` + (`country_id` + (2 + (3 + (4 + (5 + 4)))))) / 7 as a') |
||||||
|
expect(formulaQueryBuilder("AVG(city_id,country_id,2,3,4,5,4)", 'a',knexPgRef).toQuery()).eq('("city_id" + ("country_id" + (2 + (3 + (4 + (5 + 4)))))) / 7 as a') |
||||||
|
expect(formulaQueryBuilder("AVG(city_id,country_id,2,3,4,5,4)", 'a',knexMssqlRef).toQuery()).eq('([city_id] + ([country_id] + (2 + (3 + (4 + (5 + 4)))))) / 7 as a') |
||||||
|
expect(formulaQueryBuilder("AVG(city_id,country_id,2,3,4,5,4)", 'a',knexSqliteRef).toQuery()).eq('(`city_id` + (`country_id` + (2 + (3 + (4 + (5 + 4)))))) / 7 as a') |
||||||
|
done() |
||||||
|
}); |
||||||
|
}); |
||||||
|
|
||||||
|
});/** |
||||||
|
* @copyright Copyright (c) 2021, Xgene Cloud Ltd |
||||||
|
* |
||||||
|
* @author Naveen MR <oof1lab@gmail.com> |
||||||
|
* @author Pranav C Balan <pranavxc@gmail.com> |
||||||
|
* |
||||||
|
* @license GNU AGPL version 3 or any later version |
||||||
|
* |
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU Affero General Public License as |
||||||
|
* published by the Free Software Foundation, either version 3 of the |
||||||
|
* License, or (at your option) any later version. |
||||||
|
*"concat(city, ' _ ',city_id+country_id)", 'a' |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU Affero General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU Affero General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. |
||||||
|
* |
||||||
|
*/ |
||||||
@ -0,0 +1,70 @@ |
|||||||
|
import jsep from 'jsep'; |
||||||
|
|
||||||
|
|
||||||
|
// todo: switch function based on database
|
||||||
|
|
||||||
|
export default function formulaQueryBuilder(str, alias, knex) { |
||||||
|
const fn = (pt, a?, nestedBinary?) => { |
||||||
|
const colAlias = a ? ` as ${a}` : ''; |
||||||
|
if (pt.type === 'CallExpression') { |
||||||
|
switch (pt.callee.name) { |
||||||
|
case 'ADD': |
||||||
|
case 'SUM': |
||||||
|
if (pt.arguments.length > 1) { |
||||||
|
return fn({ |
||||||
|
type: 'BinaryExpression', |
||||||
|
operator: '+', |
||||||
|
left: pt.arguments[0], |
||||||
|
right: {...pt, arguments: pt.arguments.slice(1)} |
||||||
|
}, a, nestedBinary) |
||||||
|
} else { |
||||||
|
return fn(pt.arguments[0], a, nestedBinary) |
||||||
|
} |
||||||
|
break; |
||||||
|
case 'AVG': |
||||||
|
if (pt.arguments.length > 1) { |
||||||
|
return fn({ |
||||||
|
type: 'BinaryExpression', |
||||||
|
operator: '/', |
||||||
|
left: {...pt, callee: {name: 'SUM'}}, |
||||||
|
right: {type: 'Literal', value: pt.arguments.length} |
||||||
|
}, a, nestedBinary) |
||||||
|
} else { |
||||||
|
return fn(pt.arguments[0], a, nestedBinary) |
||||||
|
} |
||||||
|
break; |
||||||
|
case 'concat': |
||||||
|
case 'CONCAT': |
||||||
|
if (knex.clientType() === 'sqlite3') { |
||||||
|
if (pt.arguments.length > 1) { |
||||||
|
return fn({ |
||||||
|
type: 'BinaryExpression', |
||||||
|
operator: '||', |
||||||
|
left: pt.arguments[0], |
||||||
|
right: {...pt, arguments: pt.arguments.slice(1)} |
||||||
|
}, a, nestedBinary) |
||||||
|
} else { |
||||||
|
return fn(pt.arguments[0], a, nestedBinary) |
||||||
|
} |
||||||
|
} |
||||||
|
break; |
||||||
|
} |
||||||
|
|
||||||
|
return knex.raw(`${pt.callee.name}(${pt.arguments.map(arg => fn(arg).toQuery()).join()})${colAlias}`) |
||||||
|
} else if (pt.type === 'Literal') { |
||||||
|
return knex.raw(`?${colAlias}`, [pt.value]); |
||||||
|
} else if (pt.type === 'Identifier') { |
||||||
|
return knex.raw(`??${colAlias}`, [pt.name]); |
||||||
|
} else if (pt.type === 'BinaryExpression') { |
||||||
|
const query = knex.raw(`${fn(pt.left, null, true).toQuery()} ${pt.operator} ${fn(pt.right, null, true).toQuery()}${colAlias}`) |
||||||
|
if (nestedBinary) { |
||||||
|
query.wrap('(', ')') |
||||||
|
} |
||||||
|
return query; |
||||||
|
} |
||||||
|
}; |
||||||
|
return fn(jsep(str), alias) |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Loading…
Reference in new issue