first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
@@ -0,0 +1,120 @@
+'use strict';
+
+var _ = require('underscore');
+
+var pg  = require('./../pg');
+const errorHandlerFactory = require('../../../services/error_handler_factory');
+
+function GeoJsonFormat() {
+    this.buffer = '';
+}
+
+GeoJsonFormat.prototype = new pg('geojson');
+
+GeoJsonFormat.prototype._contentType = "application/json; charset=utf-8";
+
+GeoJsonFormat.prototype.getQuery = function(sql, options) {
+    var gn = options.gn;
+    var dp = options.dp;
+    return 'SELECT *, ST_AsGeoJSON(' + gn + ',' + dp + ') as the_geom FROM (' + sql + ') as foo';
+};
+
+GeoJsonFormat.prototype.startStreaming = function() {
+    this.total_rows = 0;
+    if (this.opts.beforeSink) {
+        this.opts.beforeSink();
+    }
+    if (this.opts.callback) {
+        this.buffer += this.opts.callback + '(';
+    }
+    this.buffer += '{"type": "FeatureCollection", "features": [';
+    this._streamingStarted = true;
+};
+
+GeoJsonFormat.prototype.handleQueryRow = function(row) {
+
+    if ( ! this._streamingStarted ) {
+        this.startStreaming();
+    }
+
+    var geojson = [
+        '{',
+        '"type":"Feature",',
+        '"geometry":' + row[this.opts.gn] + ',',
+        '"properties":'
+    ];
+    delete row[this.opts.gn];
+    delete row.the_geom_webmercator;
+    geojson.push(JSON.stringify(row));
+    geojson.push('}');
+
+    this.buffer += (this.total_rows++ ? ',' : '') + geojson.join('');
+
+    if (this.total_rows % (this.opts.bufferedRows || 1000)) {
+        this.opts.sink.write(this.buffer);
+        this.buffer = '';
+    }
+};
+
+GeoJsonFormat.prototype.handleQueryEnd = function(/*result*/) {
+    if (this.error && !this._streamingStarted) {
+        this.callback(this.error);
+        return;
+    }
+
+    if ( this.opts.profiler ) {
+        this.opts.profiler.done('gotRows');
+    }
+
+    if ( ! this._streamingStarted ) {
+        this.startStreaming();
+    }
+
+    this.buffer += ']'; // end of features
+
+    if (this.error) {
+        this.buffer += ',"error":' + JSON.stringify(errorHandlerFactory(this.error).getResponse().error);
+    }
+
+    this.buffer += '}'; // end of root object
+
+    if (this.opts.callback) {
+        this.buffer += ')';
+    }
+
+    this.opts.sink.write(this.buffer);
+    this.opts.sink.end();
+    this.buffer = '';
+
+    this.callback();
+};
+
+function _toGeoJSON(data, gn, callback){
+  try {
+    var out = {
+      type: "FeatureCollection",
+      features: []
+    };
+
+    _.each(data.rows, function(ele){
+      var _geojson = {
+          type: "Feature",
+          properties: { },
+          geometry: { }
+      };
+      _geojson.geometry = JSON.parse(ele[gn]);
+      delete ele[gn];
+      delete ele.the_geom_webmercator; // TODO: use skipfields
+      _geojson.properties = ele;
+      out.features.push(_geojson);
+    });
+
+    // return payload
+    callback(null, out);
+  } catch (err) {
+    callback(err,null);
+  }
+}
+
+module.exports = GeoJsonFormat;
+module.exports.toGeoJSON = _toGeoJSON;