You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

213 lines
5.4 KiB

7 years ago
  1. <?php
  2. /**
  3. * CodeIgniter
  4. *
  5. * An open source application development framework for PHP
  6. *
  7. * This content is released under the MIT License (MIT)
  8. *
  9. * Copyright (c) 2014 - 2017, British Columbia Institute of Technology
  10. *
  11. * Permission is hereby granted, free of charge, to any person obtaining a copy
  12. * of this software and associated documentation files (the "Software"), to deal
  13. * in the Software without restriction, including without limitation the rights
  14. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  15. * copies of the Software, and to permit persons to whom the Software is
  16. * furnished to do so, subject to the following conditions:
  17. *
  18. * The above copyright notice and this permission notice shall be included in
  19. * all copies or substantial portions of the Software.
  20. *
  21. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  22. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  23. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  24. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  25. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  26. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  27. * THE SOFTWARE.
  28. *
  29. * @package CodeIgniter
  30. * @author EllisLab Dev Team
  31. * @copyright Copyright (c) 2008 - 2014, EllisLab, Inc. (https://ellislab.com/)
  32. * @copyright Copyright (c) 2014 - 2017, British Columbia Institute of Technology (http://bcit.ca/)
  33. * @license http://opensource.org/licenses/MIT MIT License
  34. * @link https://codeigniter.com
  35. * @since Version 1.3.0
  36. * @filesource
  37. */
  38. defined('BASEPATH') OR exit('No direct script access allowed');
  39. /**
  40. * MySQLi Utility Class
  41. *
  42. * @package CodeIgniter
  43. * @subpackage Drivers
  44. * @category Database
  45. * @author EllisLab Dev Team
  46. * @link https://codeigniter.com/user_guide/database/
  47. */
  48. class CI_DB_mysqli_utility extends CI_DB_utility {
  49. /**
  50. * List databases statement
  51. *
  52. * @var string
  53. */
  54. protected $_list_databases = 'SHOW DATABASES';
  55. /**
  56. * OPTIMIZE TABLE statement
  57. *
  58. * @var string
  59. */
  60. protected $_optimize_table = 'OPTIMIZE TABLE %s';
  61. /**
  62. * REPAIR TABLE statement
  63. *
  64. * @var string
  65. */
  66. protected $_repair_table = 'REPAIR TABLE %s';
  67. // --------------------------------------------------------------------
  68. /**
  69. * Export
  70. *
  71. * @param array $params Preferences
  72. * @return mixed
  73. */
  74. protected function _backup($params = array())
  75. {
  76. if (count($params) === 0)
  77. {
  78. return FALSE;
  79. }
  80. // Extract the prefs for simplicity
  81. extract($params);
  82. // Build the output
  83. $output = '';
  84. // Do we need to include a statement to disable foreign key checks?
  85. if ($foreign_key_checks === FALSE)
  86. {
  87. $output .= 'SET foreign_key_checks = 0;'.$newline;
  88. }
  89. foreach ( (array) $tables as $table)
  90. {
  91. // Is the table in the "ignore" list?
  92. if (in_array($table, (array) $ignore, TRUE))
  93. {
  94. continue;
  95. }
  96. // Get the table schema
  97. $query = $this->db->query('SHOW CREATE TABLE '.$this->db->escape_identifiers($this->db->database.'.'.$table));
  98. // No result means the table name was invalid
  99. if ($query === FALSE)
  100. {
  101. continue;
  102. }
  103. // Write out the table schema
  104. $output .= '#'.$newline.'# TABLE STRUCTURE FOR: '.$table.$newline.'#'.$newline.$newline;
  105. if ($add_drop === TRUE)
  106. {
  107. $output .= 'DROP TABLE IF EXISTS '.$this->db->protect_identifiers($table).';'.$newline.$newline;
  108. }
  109. $i = 0;
  110. $result = $query->result_array();
  111. foreach ($result[0] as $val)
  112. {
  113. if ($i++ % 2)
  114. {
  115. $output .= $val.';'.$newline.$newline;
  116. }
  117. }
  118. // If inserts are not needed we're done...
  119. if ($add_insert === FALSE)
  120. {
  121. continue;
  122. }
  123. // Grab all the data from the current table
  124. $query = $this->db->query('SELECT * FROM '.$this->db->protect_identifiers($table));
  125. if ($query->num_rows() === 0)
  126. {
  127. continue;
  128. }
  129. // Fetch the field names and determine if the field is an
  130. // integer type. We use this info to decide whether to
  131. // surround the data with quotes or not
  132. $i = 0;
  133. $field_str = '';
  134. $is_int = array();
  135. while ($field = $query->result_id->fetch_field())
  136. {
  137. // Most versions of MySQL store timestamp as a string
  138. $is_int[$i] = in_array(strtolower($field->type),
  139. array('tinyint', 'smallint', 'mediumint', 'int', 'bigint'), //, 'timestamp'),
  140. TRUE);
  141. // Create a string of field names
  142. $field_str .= $this->db->escape_identifiers($field->name).', ';
  143. $i++;
  144. }
  145. // Trim off the end comma
  146. $field_str = preg_replace('/, $/' , '', $field_str);
  147. // Build the insert string
  148. foreach ($query->result_array() as $row)
  149. {
  150. $val_str = '';
  151. $i = 0;
  152. foreach ($row as $v)
  153. {
  154. // Is the value NULL?
  155. if ($v === NULL)
  156. {
  157. $val_str .= 'NULL';
  158. }
  159. else
  160. {
  161. // Escape the data if it's not an integer
  162. $val_str .= ($is_int[$i] === FALSE) ? $this->db->escape($v) : $v;
  163. }
  164. // Append a comma
  165. $val_str .= ', ';
  166. $i++;
  167. }
  168. // Remove the comma at the end of the string
  169. $val_str = preg_replace('/, $/' , '', $val_str);
  170. // Build the INSERT string
  171. $output .= 'INSERT INTO '.$this->db->protect_identifiers($table).' ('.$field_str.') VALUES ('.$val_str.');'.$newline;
  172. }
  173. $output .= $newline.$newline;
  174. }
  175. // Do we need to include a statement to re-enable foreign key checks?
  176. if ($foreign_key_checks === FALSE)
  177. {
  178. $output .= 'SET foreign_key_checks = 1;'.$newline;
  179. }
  180. return $output;
  181. }
  182. }