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.

198 lines
4.9 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 2.1.0
  36. * @filesource
  37. */
  38. defined('BASEPATH') OR exit('No direct script access allowed');
  39. /**
  40. * PDO Result Class
  41. *
  42. * This class extends the parent result class: CI_DB_result
  43. *
  44. * @package CodeIgniter
  45. * @subpackage Drivers
  46. * @category Database
  47. * @author EllisLab Dev Team
  48. * @link https://codeigniter.com/user_guide/database/
  49. */
  50. class CI_DB_pdo_result extends CI_DB_result {
  51. /**
  52. * Number of rows in the result set
  53. *
  54. * @return int
  55. */
  56. public function num_rows()
  57. {
  58. if (is_int($this->num_rows))
  59. {
  60. return $this->num_rows;
  61. }
  62. elseif (count($this->result_array) > 0)
  63. {
  64. return $this->num_rows = count($this->result_array);
  65. }
  66. elseif (count($this->result_object) > 0)
  67. {
  68. return $this->num_rows = count($this->result_object);
  69. }
  70. elseif (($num_rows = $this->result_id->rowCount()) > 0)
  71. {
  72. return $this->num_rows = $num_rows;
  73. }
  74. return $this->num_rows = count($this->result_array());
  75. }
  76. // --------------------------------------------------------------------
  77. /**
  78. * Number of fields in the result set
  79. *
  80. * @return int
  81. */
  82. public function num_fields()
  83. {
  84. return $this->result_id->columnCount();
  85. }
  86. // --------------------------------------------------------------------
  87. /**
  88. * Fetch Field Names
  89. *
  90. * Generates an array of column names
  91. *
  92. * @return bool
  93. */
  94. public function list_fields()
  95. {
  96. $field_names = array();
  97. for ($i = 0, $c = $this->num_fields(); $i < $c; $i++)
  98. {
  99. // Might trigger an E_WARNING due to not all subdrivers
  100. // supporting getColumnMeta()
  101. $field_names[$i] = @$this->result_id->getColumnMeta($i);
  102. $field_names[$i] = $field_names[$i]['name'];
  103. }
  104. return $field_names;
  105. }
  106. // --------------------------------------------------------------------
  107. /**
  108. * Field data
  109. *
  110. * Generates an array of objects containing field meta-data
  111. *
  112. * @return array
  113. */
  114. public function field_data()
  115. {
  116. try
  117. {
  118. $retval = array();
  119. for ($i = 0, $c = $this->num_fields(); $i < $c; $i++)
  120. {
  121. $field = $this->result_id->getColumnMeta($i);
  122. $retval[$i] = new stdClass();
  123. $retval[$i]->name = $field['name'];
  124. $retval[$i]->type = $field['native_type'];
  125. $retval[$i]->max_length = ($field['len'] > 0) ? $field['len'] : NULL;
  126. $retval[$i]->primary_key = (int) ( ! empty($field['flags']) && in_array('primary_key', $field['flags'], TRUE));
  127. }
  128. return $retval;
  129. }
  130. catch (Exception $e)
  131. {
  132. if ($this->db->db_debug)
  133. {
  134. return $this->db->display_error('db_unsupported_feature');
  135. }
  136. return FALSE;
  137. }
  138. }
  139. // --------------------------------------------------------------------
  140. /**
  141. * Free the result
  142. *
  143. * @return void
  144. */
  145. public function free_result()
  146. {
  147. if (is_object($this->result_id))
  148. {
  149. $this->result_id = FALSE;
  150. }
  151. }
  152. // --------------------------------------------------------------------
  153. /**
  154. * Result - associative array
  155. *
  156. * Returns the result set as an array
  157. *
  158. * @return array
  159. */
  160. protected function _fetch_assoc()
  161. {
  162. return $this->result_id->fetch(PDO::FETCH_ASSOC);
  163. }
  164. // --------------------------------------------------------------------
  165. /**
  166. * Result - object
  167. *
  168. * Returns the result set as an object
  169. *
  170. * @param string $class_name
  171. * @return object
  172. */
  173. protected function _fetch_object($class_name = 'stdClass')
  174. {
  175. return $this->result_id->fetchObject($class_name);
  176. }
  177. }