152 lines
4.0 KiB
JavaScript
152 lines
4.0 KiB
JavaScript
var _ = require('underscore');
|
|
var BaseWidget = require('./base');
|
|
var debug = require('debug')('windshaft:widget:formula');
|
|
|
|
var dot = require('dot');
|
|
dot.templateSettings.strip = false;
|
|
|
|
var columnTypeQueryTpl = dot.template(
|
|
'SELECT pg_typeof({{=it.column}})::oid FROM ({{=it.query}}) _cdb_histogram_column_type limit 1'
|
|
);
|
|
|
|
var formulaQueryTpl = dot.template([
|
|
'SELECT',
|
|
' {{=it._operation}}({{=it._column}}) AS result,',
|
|
' (SELECT count(1) FROM ({{=it._query}}) _cdb_formula_nulls WHERE {{=it._column}} IS NULL) AS nulls_count',
|
|
' {{?it._isFloatColumn}},(SELECT count(1) FROM ({{=it._query}}) _cdb_formula_nulls',
|
|
' WHERE {{=it._column}} = \'infinity\'::float OR {{=it._column}} = \'-infinity\'::float) AS infinities_count',
|
|
' ,(SELECT count(1) FROM ({{=it._query}}) _cdb_formula_nulls',
|
|
' WHERE {{=it._column}} = \'NaN\'::float) AS nans_count{{?}}',
|
|
'FROM ({{=it._query}}) _cdb_formula',
|
|
'{{?it._isFloatColumn}}WHERE',
|
|
' {{=it._column}} != \'infinity\'::float',
|
|
'AND',
|
|
' {{=it._column}} != \'-infinity\'::float',
|
|
'AND',
|
|
' {{=it._column}} != \'NaN\'::float{{?}}'
|
|
].join('\n'));
|
|
|
|
var VALID_OPERATIONS = {
|
|
count: true,
|
|
avg: true,
|
|
sum: true,
|
|
min: true,
|
|
max: true
|
|
};
|
|
|
|
var FLOAT_OIDS = {
|
|
700: true,
|
|
701: true
|
|
};
|
|
|
|
var TYPE = 'formula';
|
|
|
|
/**
|
|
{
|
|
type: 'formula',
|
|
options: {
|
|
column: 'name',
|
|
operation: 'count' // count, sum, avg
|
|
}
|
|
}
|
|
*/
|
|
function Formula(query, options) {
|
|
if (!_.isString(options.operation)) {
|
|
throw new Error('Formula expects `operation` in widget options');
|
|
}
|
|
|
|
if (!VALID_OPERATIONS[options.operation]) {
|
|
throw new Error("Formula does not support '" + options.operation + "' operation");
|
|
}
|
|
|
|
if (options.operation !== 'count' && !_.isString(options.column)) {
|
|
throw new Error('Formula expects `column` in widget options');
|
|
}
|
|
|
|
BaseWidget.apply(this);
|
|
|
|
this.query = query;
|
|
this.column = options.column || '1';
|
|
this.operation = options.operation;
|
|
this._isFloatColumn = null;
|
|
}
|
|
|
|
Formula.prototype = new BaseWidget();
|
|
Formula.prototype.constructor = Formula;
|
|
|
|
module.exports = Formula;
|
|
|
|
Formula.prototype.sql = function(psql, override, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = override;
|
|
override = {};
|
|
}
|
|
|
|
var _query = this.query;
|
|
|
|
var columnTypeQuery = columnTypeQueryTpl({
|
|
column: this.column, query: _query
|
|
});
|
|
|
|
if (this._isFloatColumn === null) {
|
|
var readOnlyTransaction = true;
|
|
psql.query(columnTypeQuery, function(err, result) {
|
|
self._isFloatColumn = false;
|
|
if (!err && !!result.rows[0]) {
|
|
var pgType = result.rows[0].pg_typeof;
|
|
if (FLOAT_OIDS.hasOwnProperty(pgType)) {
|
|
self._isFloatColumn = true;
|
|
}
|
|
}
|
|
self.sql(psql, override, callback);
|
|
}, readOnlyTransaction);
|
|
return null;
|
|
}
|
|
|
|
var formulaSql = formulaQueryTpl({
|
|
_isFloatColumn: this._isFloatColumn,
|
|
_query: _query,
|
|
_operation: this.operation,
|
|
_column: this.column
|
|
});
|
|
|
|
debug(formulaSql);
|
|
|
|
return callback(null, formulaSql);
|
|
};
|
|
|
|
Formula.prototype.format = function(result) {
|
|
var formattedResult = {
|
|
operation: this.operation,
|
|
result: 0,
|
|
nulls: 0,
|
|
nans: 0,
|
|
infinities: 0
|
|
};
|
|
|
|
if (result.rows.length) {
|
|
formattedResult.operation = this.operation;
|
|
formattedResult.result = result.rows[0].result;
|
|
formattedResult.nulls = result.rows[0].nulls_count;
|
|
formattedResult.nans = result.rows[0].nans_count;
|
|
formattedResult.infinities = result.rows[0].infinities_count;
|
|
}
|
|
|
|
return formattedResult;
|
|
};
|
|
|
|
Formula.prototype.getType = function() {
|
|
return TYPE;
|
|
};
|
|
|
|
Formula.prototype.toString = function() {
|
|
return JSON.stringify({
|
|
_type: TYPE,
|
|
_query: this.query,
|
|
_column: this.column,
|
|
_operation: this.operation
|
|
});
|
|
};
|