|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
/** |
|
4
|
|
|
* Starts a capistrano script for either retrieving data from an environment, |
|
5
|
|
|
* or pushing data into an environment. Initiated by {@link DNDataTransfer}. |
|
6
|
|
|
* |
|
7
|
|
|
* @package deploynaut |
|
8
|
|
|
* @subpackage jobs |
|
9
|
|
|
*/ |
|
10
|
|
|
class DataTransferJob extends DeploynautJob { |
|
11
|
|
|
|
|
12
|
|
|
/** |
|
13
|
|
|
* set by a resque worker |
|
14
|
|
|
*/ |
|
15
|
|
|
public $args = array(); |
|
16
|
|
|
|
|
17
|
|
|
public function setUp() { |
|
18
|
|
|
$this->updateStatus('Started'); |
|
19
|
|
|
chdir(BASE_PATH); |
|
20
|
|
|
} |
|
21
|
|
|
|
|
22
|
|
|
public function perform() { |
|
23
|
|
|
echo "[-] DataTransferJob starting" . PHP_EOL; |
|
24
|
|
|
$log = new DeploynautLogFile($this->args['logfile']); |
|
25
|
|
|
$dataTransfer = DNDataTransfer::get()->byID($this->args['dataTransferID']); |
|
26
|
|
|
$environment = $dataTransfer->Environment(); |
|
27
|
|
|
$backupDataTransfer = null; |
|
28
|
|
|
|
|
29
|
|
View Code Duplication |
if(!empty($this->args['backupBeforePush']) && $dataTransfer->Direction == 'push') { |
|
|
|
|
|
|
30
|
|
|
$backupDataTransfer = DNDataTransfer::create(); |
|
31
|
|
|
$backupDataTransfer->EnvironmentID = $environment->ID; |
|
32
|
|
|
$backupDataTransfer->Direction = 'get'; |
|
33
|
|
|
$backupDataTransfer->Mode = $dataTransfer->Mode; |
|
34
|
|
|
$backupDataTransfer->ResqueToken = $dataTransfer->ResqueToken; |
|
35
|
|
|
$backupDataTransfer->AuthorID = $dataTransfer->AuthorID; |
|
36
|
|
|
$backupDataTransfer->write(); |
|
37
|
|
|
|
|
38
|
|
|
$dataTransfer->BackupDataTransferID = $backupDataTransfer->ID; |
|
39
|
|
|
$dataTransfer->write(); |
|
40
|
|
|
} |
|
41
|
|
|
|
|
42
|
|
|
// This is a bit icky, but there is no easy way of capturing a failed run by using the PHP Resque |
|
43
|
|
|
try { |
|
44
|
|
|
// Disallow concurrent jobs (don't rely on queuing implementation to restrict this) |
|
45
|
|
|
// Only consider data transfers started in the last 30 minutes (older jobs probably got stuck) |
|
46
|
|
|
$runningTransfers = DNDataTransfer::get() |
|
47
|
|
|
->filter(array( |
|
48
|
|
|
'EnvironmentID' => $environment->ID, |
|
49
|
|
|
'Status' => array('Queued', 'Started'), |
|
50
|
|
|
'Created:GreaterThan' => strtotime('-30 minutes') |
|
51
|
|
|
)) |
|
52
|
|
|
->exclude('ID', $dataTransfer->ID); |
|
53
|
|
|
|
|
54
|
|
|
if($runningTransfers->count()) { |
|
55
|
|
|
$runningTransfer = $runningTransfers->first(); |
|
56
|
|
|
$message = sprintf( |
|
57
|
|
|
'Error: another transfer is in progress (started at %s by %s)', |
|
58
|
|
|
$runningTransfer->dbObject('Created')->Nice(), |
|
59
|
|
|
$runningTransfer->Author()->Title |
|
60
|
|
|
); |
|
61
|
|
|
$log->write($message); |
|
62
|
|
|
throw new \RuntimeException($message); |
|
63
|
|
|
} |
|
64
|
|
|
|
|
65
|
|
View Code Duplication |
if($backupDataTransfer) { |
|
|
|
|
|
|
66
|
|
|
$log->write('Backing up existing data'); |
|
67
|
|
|
try { |
|
68
|
|
|
$environment->Backend()->dataTransfer($backupDataTransfer, $log); |
|
69
|
|
|
$this->updateBackupStatus('Finished'); |
|
70
|
|
|
} catch(Exception $e) { |
|
71
|
|
|
$this->updateBackupStatus('Failed'); |
|
72
|
|
|
throw $e; |
|
73
|
|
|
} |
|
74
|
|
|
} |
|
75
|
|
|
|
|
76
|
|
|
$environment->Backend()->dataTransfer( |
|
77
|
|
|
$dataTransfer, |
|
78
|
|
|
$log |
|
79
|
|
|
); |
|
80
|
|
|
} catch(RuntimeException $exc) { |
|
81
|
|
|
$log->write($exc->getMessage()); |
|
82
|
|
|
|
|
83
|
|
|
echo "[-] DataTransferJob failed" . PHP_EOL; |
|
84
|
|
|
throw $exc; |
|
85
|
|
|
} |
|
86
|
|
|
|
|
87
|
|
|
$this->updateStatus('Finished'); |
|
88
|
|
|
echo "[-] DataTransferJob finished" . PHP_EOL; |
|
89
|
|
|
} |
|
90
|
|
|
|
|
91
|
|
|
/** |
|
92
|
|
|
* @param string $status |
|
93
|
|
|
* @global array $databaseConfig |
|
94
|
|
|
*/ |
|
95
|
|
|
protected function updateStatus($status) { |
|
96
|
|
|
global $databaseConfig; |
|
97
|
|
|
DB::connect($databaseConfig); |
|
98
|
|
|
$transfer = DNDataTransfer::get()->byID($this->args['dataTransferID']); |
|
99
|
|
|
$transfer->Status = $status; |
|
100
|
|
|
$transfer->write(); |
|
101
|
|
|
} |
|
102
|
|
|
|
|
103
|
|
|
/** |
|
104
|
|
|
* @param string $status Status of DNDataTransfer |
|
105
|
|
|
* @global array $databaseConfig |
|
106
|
|
|
*/ |
|
107
|
|
View Code Duplication |
protected function updateBackupStatus($status) { |
|
|
|
|
|
|
108
|
|
|
global $databaseConfig; |
|
109
|
|
|
DB::connect($databaseConfig); |
|
110
|
|
|
$deployment = DNDeployment::get()->byID($this->args['deploymentID']); |
|
111
|
|
|
$backup = $deployment->BackupDataTransfer(); |
|
112
|
|
|
if($backup && $backup->exists()) { |
|
113
|
|
|
$backup->Status = $status; |
|
114
|
|
|
$backup->write(); |
|
115
|
|
|
} |
|
116
|
|
|
} |
|
117
|
|
|
|
|
118
|
|
|
/** |
|
119
|
|
|
* @return DNData |
|
120
|
|
|
*/ |
|
121
|
|
|
protected function DNData() { |
|
122
|
|
|
return DNData::inst(); |
|
123
|
|
|
} |
|
124
|
|
|
} |
|
125
|
|
|
|
silverstripe /
deploynaut
Loading history...
Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.
You can also find more detailed suggestions in the “Code” section of your repository.