1 | "use strict";
|
2 | Object.defineProperty(exports, "__esModule", { value: true });
|
3 | var tslib_1 = require("tslib");
|
4 | var helper_1 = require("./util/helper");
|
5 | var aggregate_1 = require("./aggregate");
|
6 | var partition_1 = require("./util/partition");
|
7 | var smartMock_1 = require("./smartMock");
|
8 | var tuple = function () {
|
9 | var args = [];
|
10 | for (var _i = 0; _i < arguments.length; _i++) {
|
11 | args[_i] = arguments[_i];
|
12 | }
|
13 | return args;
|
14 | };
|
15 |
|
16 |
|
17 |
|
18 | exports.AGGREGATION = tuple('sum', 'max', 'min', 'average', 'avg', 'median', 'count', 'distinct', 'countd');
|
19 |
|
20 |
|
21 |
|
22 | exports.CONVERSION = tuple('toString', 'toFloat', 'toInt');
|
23 |
|
24 |
|
25 |
|
26 | exports.FILL = tuple('fillNull', 'removeNull');
|
27 | function isFillNullOptionsBySmart(options) {
|
28 | return options.type === 'bySmart';
|
29 | }
|
30 | exports.isFillNullOptionsBySmart = isFillNullOptionsBySmart;
|
31 | function isFillNullOptionsByAgg(options) {
|
32 | return options.type === 'byAgg';
|
33 | }
|
34 | exports.isFillNullOptionsByAgg = isFillNullOptionsByAgg;
|
35 | function isFillNullOptionsByValue(options) {
|
36 | return options.type === 'byValue';
|
37 | }
|
38 | exports.isFillNullOptionsByValue = isFillNullOptionsByValue;
|
39 | var ACTION_TYPES = { AGGREGATION: exports.AGGREGATION, CONVERSION: exports.CONVERSION, FILL: exports.FILL };
|
40 | function converse(value, type) {
|
41 | if (type === 'toString') {
|
42 | return String(value);
|
43 | }
|
44 | if (type === 'toFloat') {
|
45 | return parseFloat(value);
|
46 | }
|
47 | if (type === 'toInt') {
|
48 | return parseInt(value);
|
49 | }
|
50 | return value;
|
51 | }
|
52 | function fillNull(data, field, fillNullOptions) {
|
53 | var result = null;
|
54 | if (isFillNullOptionsByValue(fillNullOptions)) {
|
55 | result = [];
|
56 | data.forEach(function (row) {
|
57 | var newRow = tslib_1.__assign({}, row);
|
58 | if (!newRow[field]) {
|
59 | newRow[field] = fillNullOptions.cfg.value;
|
60 | }
|
61 | result.push(newRow);
|
62 | });
|
63 | }
|
64 | if (isFillNullOptionsByAgg(fillNullOptions)) {
|
65 | result = [];
|
66 | var tempAgg_1 = aggregate_1.aggregate(data.filter(function (row) { return !!row[field]; }), { as: [field], fields: [field], op: [fillNullOptions.cfg.agg] });
|
67 | data.forEach(function (row) {
|
68 | var newRow = tslib_1.__assign({}, row);
|
69 | if (!newRow[field]) {
|
70 | newRow[field] = tempAgg_1[0][field];
|
71 | }
|
72 | result.push(newRow);
|
73 | });
|
74 | }
|
75 | if (isFillNullOptionsBySmart(fillNullOptions)) {
|
76 | result = [];
|
77 | var cleanField_1 = data.map(function (row) { return row[field]; }).filter(function (e) { return !!e; });
|
78 | data.forEach(function (row) {
|
79 | var newRow = tslib_1.__assign({}, row);
|
80 | if (!newRow[field]) {
|
81 | newRow[field] = smartMock_1.oneMoreValue(cleanField_1);
|
82 | }
|
83 | result.push(newRow);
|
84 | });
|
85 | }
|
86 | return result;
|
87 | }
|
88 | function parseSingleSchema(data, schema) {
|
89 | var groupBy = schema.groupBy, actions = schema.actions;
|
90 | var result = data;
|
91 | if (groupBy) {
|
92 | var group = partition_1.partition(data, tslib_1.__spread(groupBy));
|
93 | result = Object.values(group).map(function (groupArray) {
|
94 | helper_1.assert(Array.isArray(groupArray) && groupArray.length > 0, "Invalid groupby!");
|
95 | var newRow = {};
|
96 | groupBy.forEach(function (g) {
|
97 | newRow[g] = groupArray[0][g];
|
98 | });
|
99 | return newRow;
|
100 | });
|
101 | }
|
102 | actions.forEach(function (action) {
|
103 | var type = action.type, field = action.field, as = action.as, options = action.options;
|
104 | var actionCategory = Object.keys(ACTION_TYPES).find(function (cate) { return ACTION_TYPES[cate].some(function (t) { return t === type; }); });
|
105 | var requiredAs = 'new_field';
|
106 | if (as) {
|
107 | requiredAs = as;
|
108 | }
|
109 | else if (field) {
|
110 | requiredAs = field;
|
111 | }
|
112 | if (actionCategory === 'AGGREGATION') {
|
113 | var aggregateParams = {
|
114 | as: [requiredAs],
|
115 | op: [type],
|
116 | };
|
117 | if (field) {
|
118 | aggregateParams.fields = [field];
|
119 | }
|
120 | if (groupBy) {
|
121 | aggregateParams.groupBy = groupBy;
|
122 | }
|
123 | var tempAgg_2 = aggregate_1.aggregate(data, aggregateParams);
|
124 | result.forEach(function (row, index) {
|
125 | row[requiredAs] = tempAgg_2[index][requiredAs];
|
126 | });
|
127 | }
|
128 | if (actionCategory === 'CONVERSION') {
|
129 | if (field) {
|
130 | var tempCon_1 = data.map(function (row) {
|
131 | var newRow = tslib_1.__assign({}, row);
|
132 | newRow[requiredAs] = converse(row[field], type);
|
133 | return newRow;
|
134 | });
|
135 | result.forEach(function (row, index) {
|
136 | row[requiredAs] = tempCon_1[index][requiredAs];
|
137 | });
|
138 | }
|
139 | }
|
140 | if (actionCategory === 'FILL') {
|
141 | if (type === 'removeNull') {
|
142 | if (field) {
|
143 | result = result.filter(function (row) { return row[field]; });
|
144 | }
|
145 | }
|
146 | if (type === 'fillNull') {
|
147 | if (field && options) {
|
148 | var attempt = fillNull(result, field, options);
|
149 | if (attempt) {
|
150 | result = attempt;
|
151 | }
|
152 | }
|
153 | }
|
154 | }
|
155 | });
|
156 | return result;
|
157 | }
|
158 |
|
159 |
|
160 |
|
161 | function parse(data, schemas) {
|
162 | var result = data;
|
163 | var schemaArray = Array.isArray(schemas) ? schemas : [schemas];
|
164 | for (var i = 0; i < schemaArray.length; i++) {
|
165 | result = parseSingleSchema(result, schemaArray[i]);
|
166 | }
|
167 | return result;
|
168 | }
|
169 | exports.parse = parse;
|