public function testRewriteQueryRegex()
{
$original_query = "CREATE\n trigger t_bu on t before update";
$mapping_rules = array(array("mapping_type" => "regex", "mapping_key" => "/CREATE[\\s]+TRIGGER/i", "mapping_value" => "CREATE DEFINER='root'@'localhost' TRIGGER"));
$query = rewrite_query($original_query, $mapping_rules);
$this->assertEquals("CREATE DEFINER='root'@'localhost' TRIGGER t_bu on t before update", $query);
}
/**
* This is the real thing: run a given instance-deployment request. This means applying the script on the instance,
* which in turn means executing script queries one by one.
* The request must be in the 'not_started' state.
*
* @param unknown $propagate_script_instance_deployment_id
* @param unknown $submitter
* @param unknown $user
* @param unknown $password
* @throws Exception
*/
public function execute_propagate_script_instance_deployment($propagate_script_instance_deployment_id, $force_manual, $restart_script, $run_single_query, $submitter, $credentials)
{
if ($force_manual) {
$submitter_mask = empty($submitter) ? '%' : $submitter;
$this->get_database()->query("\n\t \t\t\tUPDATE \n\t \t\t\t\tpropagate_script_instance_deployment \n\t \t\t\tSET \n\t \t\t\t\tmanual_approved=1 \n\t \t\t\tWHERE \n\t \t\t\t\tpropagate_script_instance_deployment_id = " . $this->get_database()->quote($propagate_script_instance_deployment_id) . " \n\t \t\t\t\tAND submitted_by LIKE " . $this->get_database()->quote($submitter_mask) . "\n\t \t\t\t");
}
$datas = $this->get_database()->query("\n \t\tSELECT\n \t\t\t*\n \t\tFROM\n \t\t\tpropagate_script_instance_deployment \n \t\t\tJOIN database_instance USING (database_instance_id)\n \t\t\tWHERE\n \t\t\tpropagate_script_instance_deployment_id = " . $this->get_database()->quote($propagate_script_instance_deployment_id) . "\n \t\t\tAND deployment_status IN ('not_started', 'awaiting_guinea_pig', 'paused', 'failed', 'awaiting_dba_approval')\n \t\t")->fetchAll();
if (empty($datas)) {
throw new Exception("Internal error: cannot read instance deployment info: propagate_script_instance_deployment_id=" . $propagate_script_instance_deployment_id);
}
$propagate_script_instance_deployment = $datas[0];
if (array_key_exists('two_step_approval_environments', $this->conf) && !empty($this->conf['two_step_approval_environments']) && in_array($propagate_script_instance_deployment["environment"], $this->conf['two_step_approval_environments'])) {
// This deployment has to further get approval of dba.
if ($submitter != '') {
// normal user.
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "awaiting_dba_approval", "A DBA must approve this deployment", $submitter);
return;
}
}
if ($propagate_script_instance_deployment['deployment_type'] == 'manual' && !$propagate_script_instance_deployment['manual_approved']) {
// Do nothing: this is a manual deployment, and force_manual was not provided.
return;
}
// Check for guinea pigs:
// - A guinea pig can start off right away
// - A non-guinea pig must wait on at least one guinea pig to succeed
// + - unless there is no guinea pig, in which case it is free to go
if (!$propagate_script_instance_deployment['is_guinea_pig']) {
$guinea_pig_deployment_status = $this->get_propagate_script_guinea_pig_deployment_status($propagate_script_instance_deployment['propagate_script_id'], $submitter);
if ($guinea_pig_deployment_status['count_guinea_pigs'] > 0) {
if ($guinea_pig_deployment_status['count_guinea_pigs'] == $guinea_pig_deployment_status['count_failed_guinea_pigs']) {
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "awaiting_approval", "All guinea pigs failed. Will not deploy", $submitter);
// No point in polling anymore... Wait for manual intervention.
return;
}
if ($guinea_pig_deployment_status['count_complete_guinea_pigs'] == 0) {
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "awaiting_guinea_pig", "No deployed guinea pigs yet. Awaiting", $submitter);
// Will not continue. Wait for next poll.
return;
}
}
}
// Begin with status updates
$this->get_database()->query("UPDATE propagate_script_instance_deployment SET processing_start_time=NOW(), processing_end_time=NULL WHERE propagate_script_instance_deployment_id = " . $this->get_database()->quote($propagate_script_instance_deployment_id));
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "deploying", "", $submitter);
// Grab host connection and start issuing queries
try {
$propagate_script = $this->get_propagate_script($propagate_script_instance_deployment['propagate_script_id'], $submitter);
$general_query_mapping = $this->get_general_query_mapping();
$database_role_query_mapping = $this->get_database_role_query_mapping($propagate_script["database_role_id"]);
$database_instance_query_mapping = $this->get_database_instance_query_mapping($propagate_script_instance_deployment["database_instance_id"]);
$deploy_schema = empty($propagate_script_instance_deployment["deploy_schema"]) ? "information_schema" : $propagate_script_instance_deployment["deploy_schema"];
$database_role = $this->get_database_role($propagate_script['database_role_id']);
if ($credentials->is_empty() && $this->instance_credentials) {
$stored_credentials = $this->get_instance_credentials($propagate_script_instance_deployment['host'], $propagate_script_instance_deployment['port']);
if ($stored_credentials) {
$credentials = $stored_credentials;
}
}
$database_wrapper = new DatabaseWrapper(array('database_type' => $database_role['database_type'], 'default_schema' => $deploy_schema, 'host' => $propagate_script_instance_deployment['host'], 'port' => $propagate_script_instance_deployment['port'], 'user' => $credentials->get_username(), 'password' => $credentials->get_password()));
if ($restart_script) {
$start_from = 0;
} else {
$start_from = intval($propagate_script_instance_deployment["current_propagate_script_query_id"]);
}
$queries = $this->get_propagate_script_query($propagate_script_instance_deployment['propagate_script_id'], $submitter, $start_from);
$query_counter = 0;
foreach ($queries as $query) {
$query_counter++;
$this->update_propagate_script_instance_deployment_current_query_id($propagate_script_instance_deployment_id, $submitter, $query['propagate_script_query_id']);
if ($run_single_query && $query_counter > 1) {
break;
}
$sql_code = $query['sql_code'];
$sql_code = rewrite_query($sql_code, $general_query_mapping);
$sql_code = rewrite_query($sql_code, $database_role_query_mapping);
$sql_code = rewrite_query($sql_code, $database_instance_query_mapping);
$database_wrapper->execute($sql_code);
}
if ($run_single_query && $query_counter > 1) {
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "paused", "User initiated a step-single-query", $submitter);
} else {
// Weepee!
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "passed", "Script executed successfully", $submitter);
}
} catch (Exception $e) {
// Bummer
$this->update_propagate_script_instance_deployment_status($propagate_script_instance_deployment_id, "failed", $e->getMessage(), $submitter);
}
$this->get_database()->query("UPDATE propagate_script_instance_deployment SET processing_end_time=NOW() WHERE propagate_script_instance_deployment_id = " . $this->get_database()->quote($propagate_script_instance_deployment_id));
}
