56 lines
2.2 KiB
MySQL
56 lines
2.2 KiB
MySQL
|
CREATE TYPE cdb_dataservices_server.simple_route AS (
|
||
|
|
shape geometry(LineString,4326),
|
||
|
|
length real,
|
||
|
|
duration integer
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
CREATE OR REPLACE FUNCTION cdb_dataservices_server._cdb_mapzen_route_point_to_point(
|
||
|
|
username TEXT,
|
||
|
|
orgname TEXT,
|
||
|
|
origin geometry(Point, 4326),
|
||
|
|
destination geometry(Point, 4326),
|
||
|
|
mode TEXT,
|
||
|
|
options text[] DEFAULT ARRAY[]::text[],
|
||
|
|
units text DEFAULT 'kilometers')
|
||
|
|
RETURNS cdb_dataservices_server.simple_route AS $$
|
||
|
|
import json
|
||
|
|
from cartodb_services.mapzen import MapzenRouting, MapzenRoutingResponse
|
||
|
|
from cartodb_services.mapzen.types import polyline_to_linestring
|
||
|
|
from cartodb_services.metrics import QuotaService
|
||
|
|
from cartodb_services.tools import Coordinate
|
||
|
|
|
||
|
|
redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection']
|
||
|
|
user_routing_config = GD["user_routing_config_{0}".format(username)]
|
||
|
|
|
||
|
|
quota_service = QuotaService(user_routing_config, redis_conn)
|
||
|
|
|
||
|
|
try:
|
||
|
|
client = MapzenRouting(user_routing_config.mapzen_app_key)
|
||
|
|
|
||
|
|
orig_lat = plpy.execute("SELECT ST_Y('%s') AS lat" % origin)[0]['lat']
|
||
|
|
orig_lon = plpy.execute("SELECT ST_X('%s') AS lon" % origin)[0]['lon']
|
||
|
|
origin_coordinates = Coordinate(orig_lon, orig_lat)
|
||
|
|
dest_lat = plpy.execute("SELECT ST_Y('%s') AS lat" % destination)[0]['lat']
|
||
|
|
dest_lon = plpy.execute("SELECT ST_X('%s') AS lon" % destination)[0]['lon']
|
||
|
|
dest_coordinates = Coordinate(dest_lon, dest_lat)
|
||
|
|
|
||
|
|
resp = client.calculate_route_point_to_point(origin_coordinates, dest_coordinates, mode, options, units)
|
||
|
|
|
||
|
|
if resp:
|
||
|
|
shape_linestring = polyline_to_linestring(resp.shape)
|
||
|
|
quota_service.increment_success_geocoder_use()
|
||
|
|
return [shape_linestring, resp.length, resp.duration]
|
||
|
|
else:
|
||
|
|
quota_service.increment_empty_geocoder_use()
|
||
|
|
except BaseException as e:
|
||
|
|
import sys, traceback
|
||
|
|
type_, value_, traceback_ = sys.exc_info()
|
||
|
|
quota_service.increment_failed_geocoder_use()
|
||
|
|
error_msg = 'There was an error trying to obtain route using mapzen provider: {0}'.format(e)
|
||
|
|
plpy.notice(traceback.format_tb(traceback_))
|
||
|
|
plpy.error(error_msg)
|
||
|
|
finally:
|
||
|
|
quota_service.increment_total_geocoder_use()
|
||
|
|
$$ LANGUAGE plpythonu SECURITY DEFINER;
|