]>
Commit | Line | Data |
---|---|---|
1 | <?php | |
2 | class DbUpdater { | |
3 | ||
4 | private $pdo; | |
5 | private $db_type; | |
6 | private $need_version; | |
7 | ||
8 | function __construct($pdo, $db_type, $need_version) { | |
9 | $this->pdo = Db::pdo(); //$pdo; | |
10 | $this->db_type = $db_type; | |
11 | $this->need_version = (int) $need_version; | |
12 | } | |
13 | ||
14 | function getSchemaVersion() { | |
15 | $row = $this->pdo->query("SELECT schema_version FROM ttrss_version")->fetch(); | |
16 | return (int) $row['schema_version']; | |
17 | } | |
18 | ||
19 | function isUpdateRequired() { | |
20 | return $this->getSchemaVersion() < $this->need_version; | |
21 | } | |
22 | ||
23 | function getSchemaLines($version) { | |
24 | $filename = "schema/versions/".$this->db_type."/$version.sql"; | |
25 | ||
26 | if (file_exists($filename)) { | |
27 | return explode(";", preg_replace("/[\r\n]/", "", file_get_contents($filename))); | |
28 | } else { | |
29 | user_error("DB Updater: schema file for version $version is not found."); | |
30 | return false; | |
31 | } | |
32 | } | |
33 | ||
34 | function performUpdateTo($version, $html_output = true) { | |
35 | if ($this->getSchemaVersion() == $version - 1) { | |
36 | ||
37 | $lines = $this->getSchemaLines($version); | |
38 | ||
39 | if (is_array($lines)) { | |
40 | ||
41 | $this->pdo->beginTransaction(); | |
42 | ||
43 | foreach ($lines as $line) { | |
44 | if (strpos($line, "--") !== 0 && $line) { | |
45 | if (!$this->pdo->query($line)) { | |
46 | if ($html_output) { | |
47 | print_notice("Query: $line"); | |
48 | print_error("Error: " . implode(", ", $this->pdo->errorInfo())); | |
49 | } else { | |
50 | _debug("Query: $line"); | |
51 | _debug("Error: " . implode(", ", $this->pdo->errorInfo())); | |
52 | } | |
53 | ||
54 | return false; | |
55 | } | |
56 | } | |
57 | } | |
58 | ||
59 | $db_version = $this->getSchemaVersion(); | |
60 | ||
61 | if ($db_version == $version) { | |
62 | $this->pdo->commit(); | |
63 | return true; | |
64 | } else { | |
65 | $this->pdo->rollBack(); | |
66 | return false; | |
67 | } | |
68 | } else { | |
69 | return false; | |
70 | } | |
71 | } else { | |
72 | return false; | |
73 | } | |
74 | } | |
75 | ||
76 | } |