Method add

src/data-classes/train-timetables.js:29–49

(timetable)

    }

    add(timetable) {
        const {array, trainLookup, railwayDirectionLookup} = this,
            trainId = timetable.t,
            timetablesByTrain = trainLookup.get(trainId),
            railwayDirectionId = `${timetable.r.id}:${timetable.d.id}`,
            timetablesByDirection = railwayDirectionLookup.get(railwayDirectionId);

        array.push(timetable);

        if (Array.isArray(timetablesByTrain)) {
            timetablesByTrain.push(timetable);
        } else {
            trainLookup.set(trainId, timetablesByTrain ? [timetablesByTrain, timetable] : timetable);
        }

        if (timetablesByDirection) {
            timetablesByDirection.push(timetable);
        } else {
            railwayDirectionLookup.set(railwayDirectionId, [timetable]);
        }
    }

    getByTrainId(trainId) {
        const timetables = this.trainLookup.get(trainId);