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.

151 lines
4.3 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. * MS SQL Forge 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_mssql_forge extends CI_DB_forge {
  49. /**
  50. * CREATE TABLE IF statement
  51. *
  52. * @var string
  53. */
  54. protected $_create_table_if = "IF NOT EXISTS (SELECT * FROM sysobjects WHERE ID = object_id(N'%s') AND OBJECTPROPERTY(id, N'IsUserTable') = 1)\nCREATE TABLE";
  55. /**
  56. * DROP TABLE IF statement
  57. *
  58. * @var string
  59. */
  60. protected $_drop_table_if = "IF EXISTS (SELECT * FROM sysobjects WHERE ID = object_id(N'%s') AND OBJECTPROPERTY(id, N'IsUserTable') = 1)\nDROP TABLE";
  61. /**
  62. * UNSIGNED support
  63. *
  64. * @var array
  65. */
  66. protected $_unsigned = array(
  67. 'TINYINT' => 'SMALLINT',
  68. 'SMALLINT' => 'INT',
  69. 'INT' => 'BIGINT',
  70. 'REAL' => 'FLOAT'
  71. );
  72. // --------------------------------------------------------------------
  73. /**
  74. * ALTER TABLE
  75. *
  76. * @param string $alter_type ALTER type
  77. * @param string $table Table name
  78. * @param mixed $field Column definition
  79. * @return string|string[]
  80. */
  81. protected function _alter_table($alter_type, $table, $field)
  82. {
  83. if (in_array($alter_type, array('ADD', 'DROP'), TRUE))
  84. {
  85. return parent::_alter_table($alter_type, $table, $field);
  86. }
  87. $sql = 'ALTER TABLE '.$this->db->escape_identifiers($table).' ALTER COLUMN ';
  88. $sqls = array();
  89. for ($i = 0, $c = count($field); $i < $c; $i++)
  90. {
  91. $sqls[] = $sql.$this->_process_column($field[$i]);
  92. }
  93. return $sqls;
  94. }
  95. // --------------------------------------------------------------------
  96. /**
  97. * Field attribute TYPE
  98. *
  99. * Performs a data type mapping between different databases.
  100. *
  101. * @param array &$attributes
  102. * @return void
  103. */
  104. protected function _attr_type(&$attributes)
  105. {
  106. if (isset($attributes['CONSTRAINT']) && strpos($attributes['TYPE'], 'INT') !== FALSE)
  107. {
  108. unset($attributes['CONSTRAINT']);
  109. }
  110. switch (strtoupper($attributes['TYPE']))
  111. {
  112. case 'MEDIUMINT':
  113. $attributes['TYPE'] = 'INTEGER';
  114. $attributes['UNSIGNED'] = FALSE;
  115. return;
  116. case 'INTEGER':
  117. $attributes['TYPE'] = 'INT';
  118. return;
  119. default: return;
  120. }
  121. }
  122. // --------------------------------------------------------------------
  123. /**
  124. * Field attribute AUTO_INCREMENT
  125. *
  126. * @param array &$attributes
  127. * @param array &$field
  128. * @return void
  129. */
  130. protected function _attr_auto_increment(&$attributes, &$field)
  131. {
  132. if ( ! empty($attributes['AUTO_INCREMENT']) && $attributes['AUTO_INCREMENT'] === TRUE && stripos($field['type'], 'int') !== FALSE)
  133. {
  134. $field['auto_increment'] = ' IDENTITY(1,1)';
  135. }
  136. }
  137. }