source: pro-violet-viettel/sourcecode/system/database/drivers/mysql/mysql_forge.php @ 866

Last change on this file since 866 was 289, checked in by dungnv, 11 years ago
File size: 6.3 KB
Line 
1<?php  if ( ! defined('BASEPATH')) exit('No direct script access allowed');
2/**
3 * CodeIgniter
4 *
5 * An open source application development framework for PHP 5.1.6 or newer
6 *
7 * @package             CodeIgniter
8 * @author              ExpressionEngine Dev Team
9 * @copyright   Copyright (c) 2008 - 2011, EllisLab, Inc.
10 * @license             http://codeigniter.com/user_guide/license.html
11 * @link                http://codeigniter.com
12 * @since               Version 1.0
13 * @filesource
14 */
15
16// ------------------------------------------------------------------------
17
18/**
19 * MySQL Forge Class
20 *
21 * @category    Database
22 * @author              ExpressionEngine Dev Team
23 * @link                http://codeigniter.com/user_guide/database/
24 */
25class CI_DB_mysql_forge extends CI_DB_forge {
26
27        /**
28         * Create database
29         *
30         * @access      private
31         * @param       string  the database name
32         * @return      bool
33         */
34        function _create_database($name)
35        {
36                return "CREATE DATABASE ".$name;
37        }
38
39        // --------------------------------------------------------------------
40
41        /**
42         * Drop database
43         *
44         * @access      private
45         * @param       string  the database name
46         * @return      bool
47         */
48        function _drop_database($name)
49        {
50                return "DROP DATABASE ".$name;
51        }
52
53        // --------------------------------------------------------------------
54
55        /**
56         * Process Fields
57         *
58         * @access      private
59         * @param       mixed   the fields
60         * @return      string
61         */
62        function _process_fields($fields)
63        {
64                $current_field_count = 0;
65                $sql = '';
66
67                foreach ($fields as $field=>$attributes)
68                {
69                        // Numeric field names aren't allowed in databases, so if the key is
70                        // numeric, we know it was assigned by PHP and the developer manually
71                        // entered the field information, so we'll simply add it to the list
72                        if (is_numeric($field))
73                        {
74                                $sql .= "\n\t$attributes";
75                        }
76                        else
77                        {
78                                $attributes = array_change_key_case($attributes, CASE_UPPER);
79
80                                $sql .= "\n\t".$this->db->_protect_identifiers($field);
81
82                                if (array_key_exists('NAME', $attributes))
83                                {
84                                        $sql .= ' '.$this->db->_protect_identifiers($attributes['NAME']).' ';
85                                }
86
87                                if (array_key_exists('TYPE', $attributes))
88                                {
89                                        $sql .=  ' '.$attributes['TYPE'];
90
91                                        if (array_key_exists('CONSTRAINT', $attributes))
92                                        {
93                                                switch ($attributes['TYPE'])
94                                                {
95                                                        case 'decimal':
96                                                        case 'float':
97                                                        case 'numeric':
98                                                                $sql .= '('.implode(',', $attributes['CONSTRAINT']).')';
99                                                        break;
100
101                                                        case 'enum':
102                                                        case 'set':
103                                                                $sql .= '("'.implode('","', $attributes['CONSTRAINT']).'")';
104                                                        break;
105
106                                                        default:
107                                                                $sql .= '('.$attributes['CONSTRAINT'].')';
108                                                }
109                                        }
110                                }
111
112                                if (array_key_exists('UNSIGNED', $attributes) && $attributes['UNSIGNED'] === TRUE)
113                                {
114                                        $sql .= ' UNSIGNED';
115                                }
116
117                                if (array_key_exists('DEFAULT', $attributes))
118                                {
119                                        $sql .= ' DEFAULT \''.$attributes['DEFAULT'].'\'';
120                                }
121
122                                if (array_key_exists('NULL', $attributes) && $attributes['NULL'] === TRUE)
123                                {
124                                        $sql .= ' NULL';
125                                }
126                                else
127                                {
128                                        $sql .= ' NOT NULL';
129                                }
130
131                                if (array_key_exists('AUTO_INCREMENT', $attributes) && $attributes['AUTO_INCREMENT'] === TRUE)
132                                {
133                                        $sql .= ' AUTO_INCREMENT';
134                                }
135                        }
136
137                        // don't add a comma on the end of the last field
138                        if (++$current_field_count < count($fields))
139                        {
140                                $sql .= ',';
141                        }
142                }
143
144                return $sql;
145        }
146
147        // --------------------------------------------------------------------
148
149        /**
150         * Create Table
151         *
152         * @access      private
153         * @param       string  the table name
154         * @param       mixed   the fields
155         * @param       mixed   primary key(s)
156         * @param       mixed   key(s)
157         * @param       boolean should 'IF NOT EXISTS' be added to the SQL
158         * @return      bool
159         */
160        function _create_table($table, $fields, $primary_keys, $keys, $if_not_exists)
161        {
162                $sql = 'CREATE TABLE ';
163
164                if ($if_not_exists === TRUE)
165                {
166                        $sql .= 'IF NOT EXISTS ';
167                }
168
169                $sql .= $this->db->_escape_identifiers($table)." (";
170
171                $sql .= $this->_process_fields($fields);
172
173                if (count($primary_keys) > 0)
174                {
175                        $key_name = $this->db->_protect_identifiers(implode('_', $primary_keys));
176                        $primary_keys = $this->db->_protect_identifiers($primary_keys);
177                        $sql .= ",\n\tPRIMARY KEY ".$key_name." (" . implode(', ', $primary_keys) . ")";
178                }
179
180                if (is_array($keys) && count($keys) > 0)
181                {
182                        foreach ($keys as $key)
183                        {
184                                if (is_array($key))
185                                {
186                                        $key_name = $this->db->_protect_identifiers(implode('_', $key));
187                                        $key = $this->db->_protect_identifiers($key);
188                                }
189                                else
190                                {
191                                        $key_name = $this->db->_protect_identifiers($key);
192                                        $key = array($key_name);
193                                }
194
195                                $sql .= ",\n\tKEY {$key_name} (" . implode(', ', $key) . ")";
196                        }
197                }
198
199                $sql .= "\n) DEFAULT CHARACTER SET {$this->db->char_set} COLLATE {$this->db->dbcollat};";
200
201                return $sql;
202        }
203
204        // --------------------------------------------------------------------
205
206        /**
207         * Drop Table
208         *
209         * @access      private
210         * @return      string
211         */
212        function _drop_table($table)
213        {
214                return "DROP TABLE IF EXISTS ".$this->db->_escape_identifiers($table);
215        }
216
217        // --------------------------------------------------------------------
218
219        /**
220         * Alter table query
221         *
222         * Generates a platform-specific query so that a table can be altered
223         * Called by add_column(), drop_column(), and column_alter(),
224         *
225         * @access      private
226         * @param       string  the ALTER type (ADD, DROP, CHANGE)
227         * @param       string  the column name
228         * @param       array   fields
229         * @param       string  the field after which we should add the new field
230         * @return      object
231         */
232        function _alter_table($alter_type, $table, $fields, $after_field = '')
233        {
234                $sql = 'ALTER TABLE '.$this->db->_protect_identifiers($table)." $alter_type ";
235
236                // DROP has everything it needs now.
237                if ($alter_type == 'DROP')
238                {
239                        return $sql.$this->db->_protect_identifiers($fields);
240                }
241
242                $sql .= $this->_process_fields($fields);
243
244                if ($after_field != '')
245                {
246                        $sql .= ' AFTER ' . $this->db->_protect_identifiers($after_field);
247                }
248
249                return $sql;
250        }
251
252        // --------------------------------------------------------------------
253
254        /**
255         * Rename a table
256         *
257         * Generates a platform-specific query so that a table can be renamed
258         *
259         * @access      private
260         * @param       string  the old table name
261         * @param       string  the new table name
262         * @return      string
263         */
264        function _rename_table($table_name, $new_table_name)
265        {
266                $sql = 'ALTER TABLE '.$this->db->_protect_identifiers($table_name)." RENAME TO ".$this->db->_protect_identifiers($new_table_name);
267                return $sql;
268        }
269
270}
271
272/* End of file mysql_forge.php */
273/* Location: ./system/database/drivers/mysql/mysql_forge.php */
Note: See TracBrowser for help on using the repository browser.