Jacek Kowalski
2020-06-21 ccf3da83dcbfe47ece72526f41e2a34a4bdebd44
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
<?php
class Database {
    private $pdo;
    private $addStatement;
    
    private $cacheId;
    private $cacheNum;
    
    public function __construct($file) {
        $this->pdo = new PDO('sqlite:'.$file);
        $this->pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
        
        $this->pdo->query('CREATE TABLE IF NOT EXISTS vehicles2 (
            id INT PRIMARY KEY,
            num INT UNIQUE,
            weight INT,
            line VARCHAR,
            date INT
        )');
        try {
            $this->beginTransaction();
            $this->pdo->query('INSERT INTO vehicles2 SELECT id, num, weight, \'?\', \''.time().'\' FROM vehicles');
            $this->commit();
            $this->pdo->query('DROP TABLE vehicles');
        } catch(PDOException $e) {
            $this->rollback();
        }
        
        $this->addStatement = $this->pdo->prepare('INSERT OR REPLACE INTO vehicles2 (id, num, weight, line, date) VALUES (:id, :num, :weight, :line, :date)');
        
        $this->_cacheClear();
    }
    
    public function beginTransaction() {
        $this->pdo->beginTransaction();
    }
    
    public function commit() {
        $this->pdo->commit();
    }
    
    public function rollback() {
        $this->pdo->rollback();
    }
    
    protected function _cachePopulate() {
        if($this->cacheId === NULL) {
            $st = $this->pdo->prepare('SELECT * FROM vehicles2');
            $st->execute();
            $result = $st->fetchAll(PDO::FETCH_ASSOC);
            
            $this->cacheId = [];
            $this->cacheNum = [];
            foreach($result as $vehicle) {
                $this->_cacheAdd($vehicle);
            }
        }
    }
    
    protected function _cacheAdd($vehicle) {
        $this->_cachePopulate();
        $this->cacheId[$vehicle['id']] = $vehicle;
        $this->cacheNum[$vehicle['num']] = $vehicle;
    }
    
    protected function _cacheClear() {
        $this->cacheId = NULL;
        $this->cacheNum = NULL;
    }
    
    public function getAllById() {
        $this->_cachePopulate();
        return $this->cacheId;
    }
    
    public function getAllByNum() {
        $this->_cachePopulate();
        return $this->cacheNum;
    }
    
    public function getById($id) {
        $this->_cachePopulate();
        return $this->cacheId[$id] ?? NULL;
    }
    
    public function getByNum($num) {
        $this->_cachePopulate();
        return $this->cacheNum[$num] ?? NULL;
    }
    
    public function clear() {
        $this->pdo->query('DELETE FROM vehicles2');
        $this->_cacheClear();
    }
    
    public function add($id, $num, $weight, $line = NULL, $date = NULL) {
        $vehicle = [
            'id' => (string)$id,
            'num' => (string)$num,
            'weight' => (string)$weight,
            'line' => (string)($line ?? ''),
            'date' => (string)($date ?? time()),
        ];
        $this->addStatement->execute($vehicle);
        $this->_cacheAdd($vehicle);
    }
    
    public function addMapping($vehiclesMapping, Mapper $mapper) {
        $this->beginTransaction();
        $weight = count($vehiclesMapping);
        foreach($vehiclesMapping as $id => $num) {
            $trip = $mapper->getTTSSVehicleTrip($id);
            $this->add($id, $num, $weight, ($trip['line'] ?? '?') . ' ' . ($trip['direction'] ?? '?'));
        }
        $this->commit();
    }
}