/
RestoreDbTask.php
100 lines (85 loc) · 2.82 KB
/
RestoreDbTask.php
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
<?php
/**
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; under version 2
* of the License (non-upgradable).
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
* Copyright (c) 2015 (original work) Open Assessment Technologies SA (under the project TAO-PRODUCT);
* @author Mikhail Kamarouski, <kamarouski@1pt.com>
*/
require_once "phing/Task.php";
require_once "phing/tasks/system/ExecTask.php";
class RestoreDbTask extends Task
{
private $dir;
/**
* @var TaoDbConfig
*/
private $taoDbConfig;
public function addTaoDbConfig(TaoDbConfig $db)
{
$this->taoDbConfig = $db;
}
/**
* The main entry point method.
*/
public function main()
{
$command = null;
$execTask = new ExecTask();
$execTask->setProject($this->getProject());
$restoreGZ = $this->getDir() . DIRECTORY_SEPARATOR . $this->taoDbConfig->getDbName() . '.gz';
switch ($this->taoDbConfig->getDbDriver()) {
case 'pdo_pgsql':
$command = vsprintf("gunzip -c %s | psql %s --host=%s --username=%s --no-password",
[
$restoreGZ,
$this->taoDbConfig->getDbName(),
$this->taoDbConfig->getDbHost(),
$this->taoDbConfig->getDbUser(),
]);
break;
case 'pdo_mysql':
$command = vsprintf("gunzip -c %s | mysql -u %s -p%s %s",
[
$restoreGZ,
$this->taoDbConfig->getDbUser(),
$this->taoDbConfig->getDbPass(),
$this->taoDbConfig->getDbName(),
]);
break;
default:
throw new BuildException('DB Driver is not supported');
break;
}
if ($command) {
$execTask->setCommand($command);
$execTask->main();
$this->log(sprintf('Backup for %s restored at %s', $this->taoDbConfig->getDbName(), $restoreGZ));
}
}
/**
* @return mixed
*/
public function getDir()
{
return $this->dir;
}
/**
* @param mixed $dir
*/
public function setDir($dir)
{
$this->dir = $dir;
}
}