From 64994ed50bf27f21ec4a17c241f72bc333ed1afe Mon Sep 17 00:00:00 2001
From: Jacek Kowalski <Jacek@jacekk.info>
Date: Wed, 20 Feb 2019 09:00:24 +0000
Subject: [PATCH] Use transaction to add mapping to the DB
---
parse.php | 80 ++++++++++++++++++++++++++++++++++------
1 files changed, 68 insertions(+), 12 deletions(-)
diff --git a/parse.php b/parse.php
index 02e5f28..f438ba6 100644
--- a/parse.php
+++ b/parse.php
@@ -1,4 +1,5 @@
<?php
+require_once(__DIR__.'/lib/database.php');
require_once(__DIR__.'/lib/fetch.php');
require_once(__DIR__.'/lib/mapper.php');
@@ -10,6 +11,7 @@
'gtfs_file' => 'VehiclePositions_A.pb',
'ttss' => 'http://91.223.13.70/internetservice/geoserviceDispatcher/services/vehicleinfo/vehicles',
'ttss_file' => 'vehicles_A.json',
+ 'database' => 'mapping_A.sqlite3',
'result' => 'mapping_A.json',
],
];
@@ -17,32 +19,86 @@
foreach($sources as $name => $source) {
$logger = new Monolog\Logger('fetch_'.$name);
try {
+ foreach(['gtfs_file', 'ttss_file', 'database', 'result'] as $field) {
+ $source[$field] = __DIR__.'/data/'.$source[$field];
+ }
+ $source['result_temp'] = $source['result'].'.tmp';
+
$logger->info('Fetching '.$name.' position data from FTP...');
- $updated = ftp_fetch_if_newer($source['gtfs'], __DIR__.'/data/'.$source['gtfs_file']);
+ $updated = ftp_fetch_if_newer($source['gtfs'], $source['gtfs_file']);
if(!$updated) {
$logger->info('Nothing to do, remote file not newer than local one');
continue;
}
$logger->info('Fetching '.$name.' positions from TTSS...');
- fetch($source['ttss'], __DIR__.'/data/'.$source['ttss_file']);
+ fetch($source['ttss'],$source['ttss_file']);
$logger->info('Loading data...');
$mapper = new Mapper();
- $mapper->loadTTSS(__DIR__.'/data/'.$source['ttss_file']);
- $mapper->loadGTFS(__DIR__.'/data/'.$source['gtfs_file']);
+ $mapper->loadTTSS($source['ttss_file']);
+ $mapper->loadGTFS($source['gtfs_file']);
+
+ $db = new Database($source['database']);
$logger->info('Finding correct offset...');
$offset = $mapper->findOffset();
- if($offset) {
- $logger->info('Got offset '.$offset.', creating mapping...');
- $mapping = $mapper->getMapping($offset);
- $json = json_encode($mapping);
- if(!file_put_contents(__DIR__.'/data/'.$source['result'].'.tmp', $json)) {
- throw new Exception('Result save failed');
- }
- rename(__DIR__.'/data/'.$source['result'].'.tmp', __DIR__.'/data/'.$source['result']);
+ if(!$offset) {
+ throw new Exception('Offset not found');
}
+
+ $logger->info('Got offset '.$offset.', creating mapping...');
+ $mapping = $mapper->getMapping($offset);
+
+ $logger->info('Checking the data for correctness...');
+ $weight = count($mapping);
+
+ $correct = 0;
+ $incorrect = 0;
+ $old = 0;
+ $maxWeight = 0;
+ foreach($mapping as $id => $vehicle) {
+ $dbVehicle = $db->getById($id);
+ if($dbVehicle) {
+ $maxWeight = max($maxWeight, $dbVehicle['weight']);
+ if((int)substr($vehicle['num'], 2) == (int)$dbVehicle['num']) {
+ $correct += 1;
+ } else {
+ $incorrect += 1;
+ }
+ continue;
+ }
+
+ $dbVehicle = $db->getByNum($vehicle['num']);
+ if($dbVehicle && $dbVehicle['id'] != $id) {
+ $old += 1;
+ }
+ }
+ $logger->info('Weight: '.$weight.', correct: '.$correct.', incorrect: '.$incorrect.', old: '.$old);
+
+ $previousMapping = NULL;
+ if($incorrect > $correct && $maxWeight > $weight) {
+ throw new Exception('Ignoring result due to better data already present');
+ } elseif($old > $correct) {
+ $logger->warn('Replacing DB data with the new mapping');
+ $db->clear();
+ } else {
+ $previousMapping = @json_decode(@file_get_contents($source['result']), TRUE);
+ }
+
+ $db->addMapping($mapping);
+
+ if(is_array($previousMapping)) {
+ $logger->info('Merging previous data with current mapping');
+ $mapping = $mapping + $previousMapping;
+ ksort($mapping);
+ }
+
+ $json = json_encode($mapping);
+ if(!file_put_contents($source['result_temp'], $json)) {
+ throw new Exception('Result save failed');
+ }
+ rename($source['result_temp'], $source['result']);
$logger->info('Finished');
} catch(Throwable $e) {
$logger->error($e->getMessage(), ['exception' => $e, 'exception_string' => (string)$e]);
--
Gitblit v1.9.1