maplibre/martin
1
1
Fork 0
mirror of https://github.com/maplibre/martin.git synced 2024-12-19 12:51:37 +03:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

feat: add geom function

Browse Source
This commit is contained in:
Andrey Bakhvalov 2018-10-19 16:51:31 +03:00
parent 4a9095cb39
commit ffb02794dc
2 changed files with 50 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
docker-entrypoint-initdb/01.sql
View File

@ -47,4 +47,51 @@ end;
$$
;
-- TODO: add martin function
create or replace function get_trips(z integer, x integer, y integer, query_params json) returns bytea
stable
strict
parallel safe
language plpgsql
as $$
DECLARE
bounds GEOMETRY(POLYGON, 3857) := TileBBox(z, x, y, 3857);
date_from DATE := (query_params->>'date_from')::DATE;
date_to DATE := (query_params->>'date_to')::DATE;
in_hour INTEGER := (query_params->>'hour')::INTEGER;
in_dow INTEGER[];
res BYTEA;
BEGIN
WITH sel_zones AS (
SELECT locationid, geom
FROM taxi_zones
WHERE geom && bounds
),
tile AS (
SELECT
sel_zones.locationid,
coalesce(sum(trips_by_hour.trips_count), 0)::integer AS trips,
coalesce(round(avg(trips_by_hour.trips_price)), 0)::integer AS trips_price,
coalesce(round(avg(trips_by_hour.trips_duration)), 0)::integer AS trips_duration,
ST_AsMVTGeom(min(sel_zones.geom), bounds, 4096, 1024, TRUE) AS geom
FROM
sel_zones LEFT JOIN trips_by_hour ON (
sel_zones.locationid = trips_by_hour.pulocationid
AND cast(trips_by_hour.pickup_datetime AS DATE) >= date_from
AND cast(trips_by_hour.pickup_datetime AS DATE) <= date_to
AND ((extract (HOUR FROM trips_by_hour.pickup_datetime) = in_hour) OR (in_hour IS NULL))
)
GROUP BY
sel_zones.locationid
)
SELECT INTO res ST_AsMVT(tile, 'trips', 4096, 'geom')
FROM tile
WHERE geom IS NOT NULL;
RETURN res;
END;
$$
;
alter function get_trips(integer, integer, integer, json) owner to postgres
;

4
docker-entrypoint-initdb/02.sh
View File

@ -2,8 +2,8 @@
wget -O ${HOME}/taxi_zones.zip https://s3.amazonaws.com/nyc-tlc/misc/taxi_zones.zip
unzip ${HOME}/taxi_zones.zip -d ${HOME}
ogr2ogr -f PostgreSQL PG:dbname=db -nln taxi_zones -nlt MULTIPOLYGON ${HOME}/taxi_zones.shp
geomColumn=wkb_geometry
geomColumn=geom
ogr2ogr -f PostgreSQL PG:dbname=db -nln taxi_zones -nlt MULTIPOLYGON -lco GEOMETRY_NAME=${geomColumn} ${HOME}/taxi_zones.shp
psql -U postgres -d db -c "alter table taxi_zones alter column ${geomColumn} type geometry(multipolygon, 3857) using st_transform(${geomColumn}, 3857); \
create index idx_taxi_zones_geom on taxi_zones using gist(${geomColumn});"