UNPKG

@golemio/pid

Version:
98 lines 5.18 kB
"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.DelayComputationRepository = void 0; const RedisJsonParser_1 = require("../../../../helpers/RedisJsonParser"); const const_1 = require("../../../../schema-definitions/vehicle-positions/redis/const"); 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 DelayComputationRepository = class DelayComputationRepository { constructor(redisConnector, log) { this.redisConnector = redisConnector; this.log = log; } async getDelayComputationCacheProperty(gtfsTripId, property) { if (!this.redisConnector.isConnected()) { await this.redisConnector.connect(); } const redisKey = `${const_1.GTFS_DELAY_COMPUTATION_NAMESPACE_PREFIX}:${gtfsTripId}`; const connection = this.redisConnector.getConnection(); try { const jsonResult = await connection.call("JSON.GET", redisKey, `$.${property}`); if (!jsonResult) { this.log.warn(`${this.constructor.name}: Property '${property}' not found in Redis key '${redisKey}'`); return null; } return RedisJsonParser_1.RedisJsonParser.parseSingle(jsonResult, this.constructor.name, property); } catch (err) { throw new golemio_errors_1.GeneralError(`${this.constructor.name}: Failed to fetch '${property}' from delay computation cache`, this.constructor.name, err); } } customJsonReducer(fieldCould) { return function reduceListOfFields(prev, curr, idx) { if (idx % fieldCould === 0) { // zero-indexed is the beginning of new-list of properties prev.push([curr]); } else { // all next are pushed into the new-list prev[prev.length - 1].push(curr); } return prev; }; } async getJsonPathFieldsAsListOfValues(tripIds, jsonPath, expectingKeysCount) { if (!this.redisConnector.isConnected()) { await this.redisConnector.connect(); } const mapTripToStops = new Map(); if (!tripIds.length) return mapTripToStops; const redisKeys = tripIds.map((t) => `${const_1.GTFS_DELAY_COMPUTATION_NAMESPACE_PREFIX}:${t}`); const connection = this.redisConnector.getConnection(); try { const jsonResult = await connection.call("JSON.MGET", redisKeys.concat(jsonPath)); if (!jsonResult) { this.log.info(`${this.constructor.name}: Cannot read custom jsonPath '${jsonPath}' in Redis keys for trip ids: '${tripIds}'`); return null; } const reducerFn = this.customJsonReducer(expectingKeysCount); for (const [index, resultLine] of jsonResult.entries()) { if (!resultLine) { mapTripToStops.set(tripIds[index], []); } else { const parsedLine = JSON.parse(resultLine); mapTripToStops.set(tripIds[index], parsedLine.reduce(reducerFn, [])); } } return mapTripToStops; } catch (err) { throw new golemio_errors_1.GeneralError(`${this.constructor.name}: Failed to read custom jsonPath '${jsonPath}' from delay computation cache`, this.constructor.name, err); } } }; exports.DelayComputationRepository = DelayComputationRepository; exports.DelayComputationRepository = DelayComputationRepository = __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]) ], DelayComputationRepository); //# sourceMappingURL=DelayComputationRepository.js.map