gtfs-via-postgres/lib/stats_by_agency_route_stop_hour.js

Process GTFS using PostgreSQL.

'use strict'

const afterAll = (opt) => {
	let materialized = false
	if (opt.statsByAgencyIdAndRouteIdAndStopAndHour === 'materialized-view') {
		materialized = true
	} else if (opt.statsByAgencyIdAndRouteIdAndStopAndHour !== 'view') {
		throw new Error('invalid opt.statsByAgencyIdAndRouteIdAndStopAndHour, must be one of these: none, view, materialized-view.')
	}
	const createViewCmd = materialized
		? `CREATE MATERIALIZED VIEW`
		: `CREATE OR REPLACE VIEW`

	return `\
${createViewCmd} "${opt.schema}".stats_by_agency_route_stop_hour AS
SELECT DISTINCT ON (agency_id, route_id, stop_id, effective_hour)
	agency_id, route_id, stop_id, station_id,
	"date" as service_date,
	date_trunc('hour', t_arrival) AS effective_hour,
	count(*) OVER (PARTITION BY route_id, stop_id, date_trunc('hour', t_arrival)) AS nr_of_arrs
FROM "${opt.schema}".arrivals_departures;

${materialized ? `\
CREATE INDEX ON "${opt.schema}".stats_by_agency_route_stop_hour (route_id);
CREATE INDEX ON "${opt.schema}".stats_by_agency_route_stop_hour (stop_id);
CREATE INDEX ON "${opt.schema}".stats_by_agency_route_stop_hour (station_id);
CREATE INDEX ON "${opt.schema}".stats_by_agency_route_stop_hour (effective_hour);
` : ''}

${opt.postgraphile ? `\
COMMENT ON${materialized ? ' MATERIALIZED' : ''} VIEW "${opt.schema}".stats_by_agency_route_stop_hour IS E'@name hourlyStats\\n@primaryKey route_id,stop_id,effective_hour\\n@foreignKey (route_id) references routes|@fieldName route|@foreignFieldName statsByStopIdAndHour\\n@foreignKey (stop_id) references stops|@fieldName stop|@foreignFieldName statsByRouteIdAndHour';
` : ''}
`
}

module.exports = {
	afterAll,
}