@qrvey/formula-lang
Version:
QFormula support for qrvey projects
67 lines • 2.31 kB
JavaScript
import { AST_PRIMITIVES, OPERATION_SCOPE } from '../constants';
import { isNumberParam } from '../utils';
import { isBetweenInclusiveRange } from '../utils/isBetweenInclusiveRange';
import { isInteger } from '../utils/isInteger';
/**
* `ROUNDUP` Rounds a number up to a specified number of digits.
*/
export const ROUNDUP = {
identifier: 'ROUNDUP',
operationScope: OPERATION_SCOPE.RAW,
functionScope: [OPERATION_SCOPE.RAW, OPERATION_SCOPE.AGGREGATE],
parameters: [
{
identifier: 'NUM_1',
optional: false,
expectedPrimitive: AST_PRIMITIVES.NUMBER,
validator: [isNumberParam],
},
{
identifier: 'DIGITS',
optional: false,
expectedPrimitive: AST_PRIMITIVES.NUMBER,
validator: [
isNumberParam,
isBetweenInclusiveRange(0, 10),
isInteger,
],
},
],
transpiler: {
elasticsearch,
snowflake,
redshift,
postgresql,
databricks,
},
primitiveResult: AST_PRIMITIVES.NUMBER,
};
function elasticsearch(value, digits) {
return `ROUNDUP(${value}, ROUNDDOWN(${digits}, 0))`;
}
function SQL(value, digits) {
return `((CASE
WHEN ${value} IS NULL OR ${digits} IS NULL THEN NULL
WHEN ${digits} > 10 THEN CEIL(ABS(${value}) * POWER(10, 10)) / POWER(10, 10)
WHEN ${digits} < 0 THEN CEIL(ABS(${value}))
ELSE CEIL(ABS(${value}) * POWER(10, TRUNC(${digits})::INTEGER)) / POWER(10, TRUNC(${digits})::INTEGER)
END) * (CASE WHEN ${value} < 0 THEN -1 ELSE 1 END))`;
}
function snowflake(value, digits) {
return SQL(value, digits);
}
function redshift(value, digits) {
return SQL(value, digits);
}
function postgresql(value, digits) {
return SQL(value, digits);
}
function databricks(value, digits) {
return `((CASE
WHEN ${value} IS NULL OR ${digits} IS NULL THEN NULL
WHEN ${digits} > 10 THEN CEIL(ABS(${value}) * POWER(10, 10)) / POWER(10, 10)
WHEN ${digits} < 0 THEN CEIL(ABS(${value}))
ELSE CEIL(ABS(${value}) * POWER(10, CAST(${digits} AS INTEGER))) / POWER(10, CAST(${digits} AS INTEGER))
END) * (CASE WHEN ${value} < 0 THEN -1 ELSE 1 END))`;
}
//# sourceMappingURL=roundup.js.map