dataservices-api/server/extension/sql/90_isochrone.sql

22 lines
1.3 KiB
MySQL
Raw Normal View History

2016-02-11 05:26:32 +08:00
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_isochrone(username TEXT, orgname TEXT, source geometry(Geometry, 4326), mode TEXT, range integer[], options text[] DEFAULT array[]::text[])
2016-02-11 02:59:00 +08:00
RETURNS SETOF cdb_dataservices_server.isoline AS $$
2016-02-11 00:17:53 +08:00
plpy.execute("SELECT cdb_dataservices_server._connect_to_redis('{0}')".format(username))
redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection']
plpy.execute("SELECT cdb_dataservices_server._get_isolines_routing_config({0}, {1})".format(plpy.quote_nullable(username), plpy.quote_nullable(orgname)))
user_isolines_config = GD["user_isolines_routing_config_{0}".format(username)]
2016-02-11 00:17:53 +08:00
type = 'isochrone'
if user_isolines_config.google_services_user:
plpy.error('This service is not available for google service users.')
2016-02-11 02:59:00 +08:00
here_plan = plpy.prepare("SELECT cdb_dataservices_server._cdb_here_routing_isolines($1, $2, $3, $4, $5, $6, $7) as isoline; ", ["text", "text", "text", "geometry(Geometry, 4326)", "text", "integer[]", "text[]"])
2016-02-11 05:26:32 +08:00
result = plpy.execute(here_plan, [username, orgname, type, source, mode, range, options])
isolines = []
for element in result:
isoline = element['isoline']
isoline = isoline.translate(None, "()").split(',')
isolines.append(isoline)
2016-02-11 00:17:53 +08:00
2016-02-11 05:26:32 +08:00
return isolines
2016-02-11 00:17:53 +08:00
$$ LANGUAGE plpythonu;