CartoDB-SQL-API/lib/models/formats/pg/arraybuffer.js
Daniel García Aubert 5a96dbb59c Run eslint --fix
2019-12-23 18:19:08 +01:00

89 lines
2.5 KiB
JavaScript

'use strict';
var _ = require('underscore');
var Pg = require('./../pg');
var ArrayBufferSer = require('../../bin-encoder');
function BinaryFormat () {}
BinaryFormat.prototype = new Pg('arraybuffer');
BinaryFormat.prototype._contentType = 'application/octet-stream';
BinaryFormat.prototype._extractTypeFromName = function (name) {
var g = name.match(/.*__(uintclamp|uint|int|float)(8|16|32)/i);
if (g && g.length === 3) {
var typeName = g[1] + g[2];
return ArrayBufferSer.typeNames[typeName];
}
};
// jshint maxcomplexity:12
BinaryFormat.prototype.transform = function (result, options, callback) {
var total_rows = result.rowCount;
var rows = result.rows;
// get headers
if (!total_rows) {
callback(null, new Buffer(0));
return;
}
var headersNames = Object.keys(rows[0]);
var headerTypes = [];
if (_.contains(headersNames, 'the_geom')) {
callback(new Error('geometry types are not supported'), null);
return;
}
try {
var i;
var r;
var n;
var t;
// get header types (and guess from name)
for (i = 0; i < headersNames.length; ++i) {
r = rows[0];
n = headersNames[i];
if (typeof (r[n]) === 'string') {
headerTypes.push(ArrayBufferSer.STRING);
} else if (typeof (r[n]) === 'object') {
t = this._extractTypeFromName(n);
t = t || ArrayBufferSer.FLOAT32;
headerTypes.push(ArrayBufferSer.BUFFER + t);
} else {
t = this._extractTypeFromName(n);
headerTypes.push(t || ArrayBufferSer.FLOAT32);
}
}
// pack the data
var header = new ArrayBufferSer(ArrayBufferSer.STRING, headersNames);
var data = [header];
for (i = 0; i < headersNames.length; ++i) {
var d = [];
n = headersNames[i];
for (r = 0; r < total_rows; ++r) {
var row = rows[r][n];
if (headerTypes[i] > ArrayBufferSer.BUFFER) {
row = new ArrayBufferSer(headerTypes[i] - ArrayBufferSer.BUFFER, row);
}
d.push(row);
}
var b = new ArrayBufferSer(headerTypes[i], d);
data.push(b);
}
// create the final buffer
var all = new ArrayBufferSer(ArrayBufferSer.BUFFER, data);
callback(null, all.buffer);
} catch (e) {
callback(e, null);
}
};
module.exports = BinaryFormat;