@golemio/pid
Version:
Golemio PID Module
258 lines • 12.2 kB
JavaScript
"use strict";
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
return function (target, key) { decorator(target, key, paramIndex); }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.PublicVehiclePositionsRepository = void 0;
const IoRedisConnector_1 = require("@golemio/core/dist/helpers/data-access/redis/IoRedisConnector");
const CoreToken_1 = require("@golemio/core/dist/helpers/ioc/CoreToken");
const ioc_1 = require("@golemio/core/dist/output-gateway/ioc");
const golemio_errors_1 = require("@golemio/core/dist/shared/golemio-errors");
const tsyringe_1 = require("@golemio/core/dist/shared/tsyringe");
let PublicVehiclePositionsRepository = class PublicVehiclePositionsRepository {
constructor(redisConnector, log) {
this.redisConnector = redisConnector;
this.log = log;
this.setName = undefined; // loaded via pub sub
}
setCurrentSetName(name) {
this.setName = name;
}
async getAllVehicleIds(boundingBox) {
if (!this.setName) {
this.log.debug("Empty setName for public vehicle positions api.");
return [];
}
if (!this.redisConnector.isConnected()) {
await this.redisConnector.connect();
}
const connection = this.redisConnector.getConnection();
let vehicleIds = [];
try {
vehicleIds = (await connection.geosearch(this.setName, "FROMLONLAT", boundingBox.centerPoint.longitude, boundingBox.centerPoint.latitude, "BYBOX", boundingBox.width, boundingBox.height, "m", "ASC"));
}
catch (error) {
throw new golemio_errors_1.GeneralError("Cannot get vehicle ids from cache", this.constructor.name, error);
}
return vehicleIds;
}
async getAllVehiclePositions(ids, shouldGetFuturePositions) {
if (!this.setName) {
this.log.debug("Empty setName for public vehicle positions api.");
return [];
}
const connection = this.redisConnector.getConnection();
let vehiclePositions = [];
try {
const keys = [];
if (shouldGetFuturePositions) {
keys.push(...ids.map((vehicleIdWithTrip) => `${this.setName}:future-vehicle-${vehicleIdWithTrip}`));
}
else {
keys.push(...ids.map((vehicleId) => `${this.setName}:vehicle-${vehicleId}`));
}
if (keys.length > 0) {
vehiclePositions = await connection.mget(keys);
}
}
catch (error) {
throw new golemio_errors_1.GeneralError("Cannot get vehicle positions from cache", this.constructor.name, error);
}
const parsedVehiclePositions = [];
for (const vehiclePosition of vehiclePositions) {
if (vehiclePosition === null) {
continue;
}
try {
parsedVehiclePositions.push(JSON.parse(vehiclePosition));
}
catch (error) {
throw new golemio_errors_1.GeneralError("Cannot parse vehicle position", this.constructor.name, error);
}
}
return parsedVehiclePositions;
}
async getAllVehiclePositionsForMultipleTrips(tripIds, omitFutureTrips) {
const positionsByTrip = new Map();
if (!this.setName || tripIds.length === 0) {
this.log.debug("Empty setName for public vehicle positions api or empty trip ids.");
return positionsByTrip;
}
if (!this.redisConnector.isConnected()) {
await this.redisConnector.connect();
}
const [tripToVehicleIdsCurrent, tripToVehicleIdsFuture] = await Promise.all([
this.getVehicleIdsForMultiple(tripIds, "trip"),
omitFutureTrips ? new Map() : this.getVehicleIdsForMultiple(tripIds, "future-trip"),
]);
if (tripToVehicleIdsCurrent.size + tripToVehicleIdsFuture.size > 0) {
await this.pushPositionsIntoMapByTrip(positionsByTrip, tripToVehicleIdsCurrent, tripToVehicleIdsFuture);
if (!omitFutureTrips) {
await this.pushPositionsIntoMapByTrip(positionsByTrip, tripToVehicleIdsCurrent, tripToVehicleIdsFuture, true);
}
}
// Refactor/Change order here? See the proposed flow...
// First: Load the canceled trips, remove those canceled gtfs-trip_ids from tripIds list,
// Second step: Search for less trips => less vehicleIds => less Current Trips + Future trips
const canceledTrips = await this.getCanceledTrips(tripIds);
for (const canceledTrip of canceledTrips) {
positionsByTrip.set(canceledTrip.gtfs_trip_id, [canceledTrip]);
}
return positionsByTrip;
}
async pushPositionsIntoMapByTrip(positionsByTrip, tripToVehicleIdsCurrent, tripToVehicleIdsFuture, shouldGetFuturePositions) {
const vehicleIds = new Set();
for (const vehicles of tripToVehicleIdsCurrent.values()) {
vehicles.forEach((vehicleId) => {
vehicleIds.add(vehicleId);
});
}
for (const [futureTripId, vehicles] of tripToVehicleIdsFuture.entries()) {
vehicles.forEach((vehicleId) => {
vehicleIds.add(`${vehicleId}-${futureTripId}`);
});
}
const vehicleIdsList = Array.from(vehicleIds);
const allPositions = await this.getAllVehiclePositions(vehicleIdsList, shouldGetFuturePositions === true);
for (const position of allPositions) {
if (!positionsByTrip.has(position.gtfs_trip_id)) {
positionsByTrip.set(position.gtfs_trip_id, []);
}
positionsByTrip.get(position.gtfs_trip_id).push(position);
}
}
async getTripsWithUntrackedVehicles(tripIds) {
const untrackedTrips = new Set();
if (!this.setName || tripIds.length === 0) {
this.log.debug("Empty setName for public vehicle positions api or empty trip ids.");
return untrackedTrips;
}
if (!this.redisConnector.isConnected()) {
await this.redisConnector.connect();
}
const tripsToVehicleIds = await this.getVehicleIdsForMultiple(tripIds, "trip");
const vehicleIds = Array.from(new Set(Array.from(tripsToVehicleIds.values()).flat()));
try {
// TODO optimization point here - use JSON.GET once it's available in the redis/valkey server
const allPositions = await this.getAllVehiclePositions(vehicleIds);
// trips that HAVE positions
const tripsWithPositions = new Set();
for (const position of allPositions) {
tripsWithPositions.add(position.gtfs_trip_id);
if (position.delay === null) {
untrackedTrips.add(position.gtfs_trip_id);
}
}
// trips that DON'T have any position (they are also untracked )
for (const tripId of tripIds) {
if (!tripsWithPositions.has(tripId)) {
untrackedTrips.add(tripId);
}
}
}
catch (error) {
throw new golemio_errors_1.GeneralError("Cannot get vehicle positions from cache", this.constructor.name, error);
}
return untrackedTrips;
}
// Current trips are saved in the cache with key made from vehicleId only. For future trips tripId has to be also provided.
async getDetailedVehiclePosition(vehicleId, tripId) {
if (!this.setName) {
this.log.debug("Empty setName for public vehicle positions api.");
return null;
}
if (!this.redisConnector.isConnected()) {
await this.redisConnector.connect();
}
const connection = this.redisConnector.getConnection();
let vehiclePosition = null;
try {
const key = tripId ? `${this.setName}:future-trip-${vehicleId}-${tripId}` : `${this.setName}:vehicle-${vehicleId}`;
vehiclePosition = await connection.get(key);
}
catch (error) {
this.log.error(error, "Cannot get vehicle position from cache");
throw new golemio_errors_1.GeneralError("Cannot get vehicle position from cache", this.constructor.name, error);
}
if (vehiclePosition === null) {
this.log.info(`${this.constructor.name}: Cannot find vehicle_id ${vehicleId}`);
return null;
}
try {
return JSON.parse(vehiclePosition);
}
catch (error) {
this.log.error(error, `Cannot parse ${vehicleId}`);
throw new golemio_errors_1.GeneralError(`Cannot parse ${vehicleId}`, this.constructor.name, error);
}
}
async getVehicleIdsForMultiple(tripIds, keyPrefix) {
const tripToVehicleIds = new Map();
const connection = this.redisConnector.getConnection();
const pipeline = connection.pipeline();
for (const tripId of tripIds) {
pipeline.lrange(`${this.setName}:${keyPrefix}-${tripId}`, 0, -1);
}
try {
const trips = await pipeline.exec();
let i = -1;
for (const [error, vehicleIds] of trips) {
i++;
if (error) {
throw new golemio_errors_1.GeneralError(`Cannot get vehicle ids for ${keyPrefix} from cache`, this.constructor.name, error);
}
const tripId = tripIds[i];
const existingVehicleIds = tripToVehicleIds.get(tripId) || [];
tripToVehicleIds.set(tripId, existingVehicleIds.concat(vehicleIds));
}
}
catch (error) {
if (error instanceof golemio_errors_1.GeneralError) {
throw error;
}
throw new golemio_errors_1.GeneralError(`Cannot get vehicle ids for ${keyPrefix} from cache`, this.constructor.name, error);
}
return tripToVehicleIds;
}
async getCanceledTrips(tripIds) {
const connection = this.redisConnector.getConnection();
const parsedCanceledTrips = [];
if (tripIds.length === 0) {
return [];
}
const keys = [];
for (const tripId of tripIds) {
keys.push(`${this.setName}:canceled-trips-${tripId}`);
}
const canceledTrips = await connection.mget(keys);
for (const canceledTrip of canceledTrips) {
if (canceledTrip === null) {
continue;
}
try {
parsedCanceledTrips.push(JSON.parse(canceledTrip));
}
catch (error) {
throw new golemio_errors_1.GeneralError("Cannot parse canceled trip", this.constructor.name, error);
}
}
return parsedCanceledTrips;
}
};
exports.PublicVehiclePositionsRepository = PublicVehiclePositionsRepository;
exports.PublicVehiclePositionsRepository = PublicVehiclePositionsRepository = __decorate([
(0, tsyringe_1.injectable)(),
__param(0, (0, tsyringe_1.inject)(ioc_1.ContainerToken.RedisConnector)),
__param(1, (0, tsyringe_1.inject)(CoreToken_1.CoreToken.Logger)),
__metadata("design:paramtypes", [IoRedisConnector_1.IoRedisConnector, Object])
], PublicVehiclePositionsRepository);
//# sourceMappingURL=PublicVehiclePositionsRepository.js.map