parent
a6bca59ffe
commit
c87c97c90f
|
@ -1,9 +1,9 @@
|
|||
import { IsBoolean, IsInt, IsOptional, IsPositive } from 'class-validator';
|
||||
import { getConnection } from 'typeorm';
|
||||
import { RunnerNotFoundError } from '../../../errors/RunnerErrors';
|
||||
import { ScanStationNotFoundError } from '../../../errors/ScanStationErrors';
|
||||
import { TrackNotFoundError } from '../../../errors/TrackErrors';
|
||||
import { Runner } from '../../entities/Runner';
|
||||
import { ScanStation } from '../../entities/ScanStation';
|
||||
import { Track } from '../../entities/Track';
|
||||
import { TrackScan } from '../../entities/TrackScan';
|
||||
|
||||
/**
|
||||
|
@ -38,7 +38,7 @@ export abstract class UpdateTrackScan {
|
|||
*/
|
||||
@IsInt()
|
||||
@IsPositive()
|
||||
public station: number;
|
||||
public track: number;
|
||||
|
||||
/**
|
||||
* Update a TrackScan entity based on this.
|
||||
|
@ -47,8 +47,7 @@ export abstract class UpdateTrackScan {
|
|||
public async update(scan: TrackScan): Promise<TrackScan> {
|
||||
scan.valid = this.valid;
|
||||
scan.runner = await this.getRunner();
|
||||
scan.station = await this.getStation();
|
||||
scan.track = scan.station.track;
|
||||
scan.track = await this.getTrack();
|
||||
|
||||
return scan;
|
||||
}
|
||||
|
@ -67,11 +66,11 @@ export abstract class UpdateTrackScan {
|
|||
/**
|
||||
* Gets a runner based on the runner id provided via this.runner.
|
||||
*/
|
||||
public async getStation(): Promise<ScanStation> {
|
||||
const station = await getConnection().getRepository(ScanStation).findOne({ id: this.station }, { relations: ['track'] });
|
||||
if (!station) {
|
||||
throw new ScanStationNotFoundError();
|
||||
public async getTrack(): Promise<Track> {
|
||||
const track = await getConnection().getRepository(Track).findOne({ id: this.track });
|
||||
if (!track) {
|
||||
throw new TrackNotFoundError();
|
||||
}
|
||||
return station;
|
||||
return track;
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue