@golemio/pid
Version:
Golemio PID Module
166 lines • 8.43 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.JISInfotextsRepository = void 0;
const ropid_gtfs_1 = require("../../ropid-gtfs");
const const_1 = require("../../../schema-definitions/const");
const JISInfotextsModel_1 = require("../../../schema-definitions/jis/models/JISInfotextsModel");
const AbstractValidatableRepository_1 = require("@golemio/core/dist/helpers/data-access/postgres/repositories/AbstractValidatableRepository");
const CoreToken_1 = require("@golemio/core/dist/helpers/ioc/CoreToken");
const golemio_errors_1 = require("@golemio/core/dist/shared/golemio-errors");
const golemio_validator_1 = require("@golemio/core/dist/shared/golemio-validator");
const sequelize_1 = require("@golemio/core/dist/shared/sequelize");
const tsyringe_1 = require("@golemio/core/dist/shared/tsyringe");
const RepositoryUtils_1 = require("../../shared/RepositoryUtils");
const JISContainerToken_1 = require("../ioc/JISContainerToken");
const JISInfotextsPresetsRepository_1 = require("./JISInfotextsPresetsRepository");
const JISInfotextsRopidGTFSStopsRepository_1 = require("./JISInfotextsRopidGTFSStopsRepository");
let JISInfotextsRepository = class JISInfotextsRepository extends AbstractValidatableRepository_1.AbstractValidatableRepository {
constructor(connector, logger, infotextStopRepository, infotextPresetsRepository, gtfsStopRepository) {
super(connector, logger);
this.logger = logger;
this.infotextStopRepository = infotextStopRepository;
this.infotextPresetsRepository = infotextPresetsRepository;
this.gtfsStopRepository = gtfsStopRepository;
this.schema = const_1.PG_SCHEMA;
this.tableName = JISInfotextsModel_1.JISInfotextsModel.tableName;
this.validator = new golemio_validator_1.JSONSchemaValidator("JISInfotextsRepository", JISInfotextsModel_1.JISInfotextsModel.jsonSchema);
this.sequelizeModel = connector
.getConnection()
.define(this.tableName, JISInfotextsModel_1.JISInfotextsModel.attributeModel, { schema: this.schema });
this.sequelizeModel.belongsToMany(this.gtfsStopRepository.sequelizeModel, {
through: this.infotextStopRepository["sequelizeModel"],
foreignKey: "infotext_id",
otherKey: "stop_id",
as: "stops",
});
this.gtfsStopRepository.sequelizeModel.belongsToMany(this.sequelizeModel, {
through: this.infotextStopRepository["sequelizeModel"],
foreignKey: "stop_id",
otherKey: "infotext_id",
});
this.infotextStopRepository["sequelizeModel"].belongsTo(this.sequelizeModel, {
targetKey: "id",
foreignKey: "infotext_id",
});
this.infotextStopRepository["sequelizeModel"].belongsTo(this.gtfsStopRepository.sequelizeModel, {
targetKey: "stop_id",
foreignKey: "stop_id",
});
this.sequelizeModel.hasMany(this.infotextStopRepository["sequelizeModel"], {
sourceKey: "id",
foreignKey: "infotext_id",
});
this.gtfsStopRepository.sequelizeModel.hasMany(this.infotextStopRepository["sequelizeModel"], {
sourceKey: "stop_id",
foreignKey: "stop_id",
});
this.sequelizeModel.hasMany(this.infotextPresetsRepository["sequelizeModel"], {
sourceKey: "id",
foreignKey: "infotext_id",
as: "presets",
});
this.infotextPresetsRepository["sequelizeModel"].belongsTo(this.sequelizeModel, {
targetKey: "id",
foreignKey: "infotext_id",
});
}
/**
* Refresh all data from VYMI and delete old data
*
* @param data The data to be upserted
* @param options Options for the operation
*/
async refreshData(data, options) {
try {
const currentInfotexts = await this.sequelizeModel.findAll({ transaction: options?.transaction });
const currentInfotextsMap = new Map();
for (const infotext of currentInfotexts) {
currentInfotextsMap.set(infotext.id, infotext);
}
const toBeUpserted = [];
for (const infotext of data) {
const info = currentInfotextsMap.get(infotext.id);
if (!info ||
info.updated_at <= infotext.updated_timestamp ||
info.active_period_end?.getTime() !== infotext.active_period_end?.getTime() ||
info.active_period_start.getTime() !== infotext.active_period_start.getTime()) {
toBeUpserted.push(infotext);
}
currentInfotextsMap.delete(infotext.id);
}
const toBeDeleted = [];
for (const { id } of currentInfotextsMap.values()) {
toBeDeleted.push(id);
}
const [upserted] = await Promise.all([
this.sequelizeModel.bulkCreate(toBeUpserted, {
updateOnDuplicate: this.getUpdateAttributes(),
transaction: options?.transaction,
}),
this.sequelizeModel.destroy({
where: {
id: { [sequelize_1.Op.in]: toBeDeleted },
},
transaction: options?.transaction,
}),
]);
return upserted;
}
catch (err) {
RepositoryUtils_1.RepositoryUtils.mapSequelizeError(err, this.constructor.name, "refreshData");
}
}
getUpdateAttributes() {
return Object.keys(JISInfotextsModel_1.JISInfotextsModel.attributeModel).filter((attribute) => !["created_at"].includes(attribute));
}
/**
* @param options
* @returns The number of deleted items
*/
async deleteAll(options) {
try {
return await this.sequelizeModel.destroy({ where: {}, transaction: options?.transaction });
}
catch (err) {
throw new golemio_errors_1.GeneralError("Error in deleteAll", this.constructor.name, err);
}
}
async findAll() {
return await this.sequelizeModel.findAll({
include: [
{
model: this.gtfsStopRepository.sequelizeModel,
as: "stops",
attributes: ["stop_id"],
through: { attributes: [] },
required: true,
},
],
});
}
};
exports.JISInfotextsRepository = JISInfotextsRepository;
exports.JISInfotextsRepository = JISInfotextsRepository = __decorate([
(0, tsyringe_1.injectable)(),
__param(0, (0, tsyringe_1.inject)(CoreToken_1.CoreToken.PostgresConnector)),
__param(1, (0, tsyringe_1.inject)(CoreToken_1.CoreToken.Logger)),
__param(2, (0, tsyringe_1.inject)(JISContainerToken_1.JISContainerToken.JISInfotextsRopidGTFSStopsRepository)),
__param(3, (0, tsyringe_1.inject)(JISContainerToken_1.JISContainerToken.JISInfotextsPresetsRepository)),
__param(4, (0, tsyringe_1.inject)(JISContainerToken_1.JISContainerToken.RopidGTFSStopsRepository)),
__metadata("design:paramtypes", [Object, Object, JISInfotextsRopidGTFSStopsRepository_1.JISInfotextsRopidGTFSStopsRepository,
JISInfotextsPresetsRepository_1.JISInfotextsPresetsRepository,
ropid_gtfs_1.RopidGTFSStopsModel])
], JISInfotextsRepository);
//# sourceMappingURL=JISInfotextsRepository.js.map