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)); }