source: trunk/web/punbb/include/dblayer/pgsql.php @ 6

Last change on this file since 6 was 6, checked in by dj3c1t, 13 years ago

mise a jour du trunk
File size: 12.2 KB
RevLine 
[6]1<?php
2
3/**
4 * Copyright (C) 2008-2011 FluxBB
5 * based on code by Rickard Andersson copyright (C) 2002-2008 PunBB
6 * License: http://www.gnu.org/licenses/gpl.html GPL version 2 or higher
7 */
8
9// Make sure we have built in support for PostgreSQL
10if (!function_exists('pg_connect'))
11        exit('This PHP environment doesn\'t have PostgreSQL support built in. PostgreSQL support is required if you want to use a PostgreSQL database to run this forum. Consult the PHP documentation for further assistance.');
12
13
14class DBLayer
15{
16        var $prefix;
17        var $link_id;
18        var $query_result;
19        var $last_query_text = array();
20        var $in_transaction = 0;
21
22        var $saved_queries = array();
23        var $num_queries = 0;
24
25        var $error_no = false;
26        var $error_msg = 'Unknown';
27
28        var $datatype_transformations = array(
29                '%^(TINY|SMALL)INT( )?(\\([0-9]+\\))?( )?(UNSIGNED)?$%i'                        =>      'SMALLINT',
30                '%^(MEDIUM)?INT( )?(\\([0-9]+\\))?( )?(UNSIGNED)?$%i'                           =>      'INTEGER',
31                '%^BIGINT( )?(\\([0-9]+\\))?( )?(UNSIGNED)?$%i'                                         =>      'BIGINT',
32                '%^(TINY|MEDIUM|LONG)?TEXT$%i'                                                                          =>      'TEXT',
33                '%^DOUBLE( )?(\\([0-9,]+\\))?( )?(UNSIGNED)?$%i'                                        =>      'DOUBLE PRECISION',
34                '%^FLOAT( )?(\\([0-9]+\\))?( )?(UNSIGNED)?$%i'                                          =>      'REAL'
35        );
36
37
38        function DBLayer($db_host, $db_username, $db_password, $db_name, $db_prefix, $p_connect)
39        {
40                $this->prefix = $db_prefix;
41
42                if ($db_host)
43                {
44                        if (strpos($db_host, ':') !== false)
45                        {
46                                list($db_host, $dbport) = explode(':', $db_host);
47                                $connect_str[] = 'host='.$db_host.' port='.$dbport;
48                        }
49                        else
50                                $connect_str[] = 'host='.$db_host;
51                }
52
53                if ($db_name)
54                        $connect_str[] = 'dbname='.$db_name;
55
56                if ($db_username)
57                        $connect_str[] = 'user='.$db_username;
58
59                if ($db_password)
60                        $connect_str[] = 'password='.$db_password;
61
62                if ($p_connect)
63                        $this->link_id = @pg_pconnect(implode(' ', $connect_str));
64                else
65                        $this->link_id = @pg_connect(implode(' ', $connect_str));
66
67                if (!$this->link_id)
68                        error('Unable to connect to PostgreSQL server', __FILE__, __LINE__);
69
70                // Setup the client-server character set (UTF-8)
71                if (!defined('FORUM_NO_SET_NAMES'))
72                        $this->set_names('utf8');
73
74                return $this->link_id;
75        }
76
77
78        function start_transaction()
79        {
80                ++$this->in_transaction;
81
82                return (@pg_query($this->link_id, 'BEGIN')) ? true : false;
83        }
84
85
86        function end_transaction()
87        {
88                --$this->in_transaction;
89
90                if (@pg_query($this->link_id, 'COMMIT'))
91                        return true;
92                else
93                {
94                        @pg_query($this->link_id, 'ROLLBACK');
95                        return false;
96                }
97        }
98
99
100        function query($sql, $unbuffered = false) // $unbuffered is ignored since there is no pgsql_unbuffered_query()
101        {
102                if (strrpos($sql, 'LIMIT') !== false)
103                        $sql = preg_replace('%LIMIT ([0-9]+),([ 0-9]+)%', 'LIMIT \\2 OFFSET \\1', $sql);
104
105                if (defined('PUN_SHOW_QUERIES'))
106                        $q_start = get_microtime();
107
108                @pg_send_query($this->link_id, $sql);
109                $this->query_result = @pg_get_result($this->link_id);
110
111                if (pg_result_status($this->query_result) != PGSQL_FATAL_ERROR)
112                {
113                        if (defined('PUN_SHOW_QUERIES'))
114                                $this->saved_queries[] = array($sql, sprintf('%.5f', get_microtime() - $q_start));
115
116                        ++$this->num_queries;
117
118                        $this->last_query_text[$this->query_result] = $sql;
119
120                        return $this->query_result;
121                }
122                else
123                {
124                        if (defined('PUN_SHOW_QUERIES'))
125                                $this->saved_queries[] = array($sql, 0);
126
127                        $this->error_no = false;
128                        $this->error_msg = @pg_result_error($this->query_result);
129
130                        if ($this->in_transaction)
131                                @pg_query($this->link_id, 'ROLLBACK');
132
133                        --$this->in_transaction;
134
135                        return false;
136                }
137        }
138
139
140        function result($query_id = 0, $row = 0, $col = 0)
141        {
142                return ($query_id) ? @pg_fetch_result($query_id, $row, $col) : false;
143        }
144
145
146        function fetch_assoc($query_id = 0)
147        {
148                return ($query_id) ? @pg_fetch_assoc($query_id) : false;
149        }
150
151
152        function fetch_row($query_id = 0)
153        {
154                return ($query_id) ? @pg_fetch_row($query_id) : false;
155        }
156
157
158        function num_rows($query_id = 0)
159        {
160                return ($query_id) ? @pg_num_rows($query_id) : false;
161        }
162
163
164        function affected_rows()
165        {
166                return ($this->query_result) ? @pg_affected_rows($this->query_result) : false;
167        }
168
169
170        function insert_id()
171        {
172                $query_id = $this->query_result;
173
174                if ($query_id && $this->last_query_text[$query_id] != '')
175                {
176                        if (preg_match('%^INSERT INTO ([a-z0-9\_\-]+)%is', $this->last_query_text[$query_id], $table_name))
177                        {
178                                // Hack (don't ask)
179                                if (substr($table_name[1], -6) == 'groups')
180                                        $table_name[1] .= '_g';
181
182                                $temp_q_id = @pg_query($this->link_id, 'SELECT currval(\''.$table_name[1].'_id_seq\')');
183                                return ($temp_q_id) ? intval(@pg_fetch_result($temp_q_id, 0)) : false;
184                        }
185                }
186
187                return false;
188        }
189
190
191        function get_num_queries()
192        {
193                return $this->num_queries;
194        }
195
196
197        function get_saved_queries()
198        {
199                return $this->saved_queries;
200        }
201
202
203        function free_result($query_id = false)
204        {
205                if (!$query_id)
206                        $query_id = $this->query_result;
207
208                return ($query_id) ? @pg_free_result($query_id) : false;
209        }
210
211
212        function escape($str)
213        {
214                return is_array($str) ? '' : pg_escape_string($str);
215        }
216
217
218        function error()
219        {
220                $result['error_sql'] = @current(@end($this->saved_queries));
221                $result['error_no'] = $this->error_no;
222                $result['error_msg'] = $this->error_msg;
223
224                return $result;
225        }
226
227
228        function close()
229        {
230                if ($this->link_id)
231                {
232                        if ($this->in_transaction)
233                        {
234                                if (defined('PUN_SHOW_QUERIES'))
235                                        $this->saved_queries[] = array('COMMIT', 0);
236
237                                @pg_query($this->link_id, 'COMMIT');
238                        }
239
240                        if ($this->query_result)
241                                @pg_free_result($this->query_result);
242
243                        return @pg_close($this->link_id);
244                }
245                else
246                        return false;
247        }
248
249
250        function get_names()
251        {
252                $result = $this->query('SHOW client_encoding');
253                return strtolower($this->result($result)); // MySQL returns lowercase so lets be consistent
254        }
255
256
257        function set_names($names)
258        {
259                return $this->query('SET NAMES \''.$this->escape($names).'\'');
260        }
261
262
263        function get_version()
264        {
265                $result = $this->query('SELECT VERSION()');
266
267                return array(
268                        'name'          => 'PostgreSQL',
269                        'version'       => preg_replace('%^[^0-9]+([^\s,-]+).*$%', '\\1', $this->result($result))
270                );
271        }
272
273
274        function table_exists($table_name, $no_prefix = false)
275        {
276                $result = $this->query('SELECT 1 FROM pg_class WHERE relname = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\'');
277                return $this->num_rows($result) > 0;
278        }
279
280
281        function field_exists($table_name, $field_name, $no_prefix = false)
282        {
283                $result = $this->query('SELECT 1 FROM pg_class c INNER JOIN pg_attribute a ON a.attrelid = c.oid WHERE c.relname = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\' AND a.attname = \''.$this->escape($field_name).'\'');
284                return $this->num_rows($result) > 0;
285        }
286
287
288        function index_exists($table_name, $index_name, $no_prefix = false)
289        {
290                $result = $this->query('SELECT 1 FROM pg_index i INNER JOIN pg_class c1 ON c1.oid = i.indrelid INNER JOIN pg_class c2 ON c2.oid = i.indexrelid WHERE c1.relname = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\' AND c2.relname = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_'.$this->escape($index_name).'\'');
291                return $this->num_rows($result) > 0;
292        }
293
294
295        function create_table($table_name, $schema, $no_prefix = false)
296        {
297                if ($this->table_exists($table_name, $no_prefix))
298                        return true;
299
300                $query = 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$table_name." (\n";
301
302                // Go through every schema element and add it to the query
303                foreach ($schema['FIELDS'] as $field_name => $field_data)
304                {
305                        $field_data['datatype'] = preg_replace(array_keys($this->datatype_transformations), array_values($this->datatype_transformations), $field_data['datatype']);
306
307                        $query .= $field_name.' '.$field_data['datatype'];
308
309                        // The SERIAL datatype is a special case where we don't need to say not null
310                        if (!$field_data['allow_null'] && $field_data['datatype'] != 'SERIAL')
311                                $query .= ' NOT NULL';
312
313                        if (isset($field_data['default']))
314                                $query .= ' DEFAULT '.$field_data['default'];
315
316                        $query .= ",\n";
317                }
318
319                // If we have a primary key, add it
320                if (isset($schema['PRIMARY KEY']))
321                        $query .= 'PRIMARY KEY ('.implode(',', $schema['PRIMARY KEY']).'),'."\n";
322
323                // Add unique keys
324                if (isset($schema['UNIQUE KEYS']))
325                {
326                        foreach ($schema['UNIQUE KEYS'] as $key_name => $key_fields)
327                                $query .= 'UNIQUE ('.implode(',', $key_fields).'),'."\n";
328                }
329
330                // We remove the last two characters (a newline and a comma) and add on the ending
331                $query = substr($query, 0, strlen($query) - 2)."\n".')';
332
333                $result = $this->query($query) ? true : false;
334
335                // Add indexes
336                if (isset($schema['INDEXES']))
337                {
338                        foreach ($schema['INDEXES'] as $index_name => $index_fields)
339                                $result &= $this->add_index($table_name, $index_name, $index_fields, false, $no_prefix);
340                }
341
342                return $result;
343        }
344
345
346        function drop_table($table_name, $no_prefix = false)
347        {
348                if (!$this->table_exists($table_name, $no_prefix))
349                        return true;
350
351                return $this->query('DROP TABLE '.($no_prefix ? '' : $this->prefix).$table_name) ? true : false;
352        }
353
354
355        function rename_table($old_table, $new_table, $no_prefix = false)
356        {
357                // If there new table exists and the old one doesn't, then we're happy
358                if ($this->table_exists($new_table, $no_prefix) && !$this->table_exists($old_table, $no_prefix))
359                        return true;
360
361                return $this->query('ALTER TABLE '.($no_prefix ? '' : $this->prefix).$old_table.' RENAME TO '.($no_prefix ? '' : $this->prefix).$new_table) ? true : false;
362        }
363
364
365        function add_field($table_name, $field_name, $field_type, $allow_null, $default_value = null, $after_field = null, $no_prefix = false)
366        {
367                if ($this->field_exists($table_name, $field_name, $no_prefix))
368                        return true;
369
370                $field_type = preg_replace(array_keys($this->datatype_transformations), array_values($this->datatype_transformations), $field_type);
371
372                $result = $this->query('ALTER TABLE '.($no_prefix ? '' : $this->prefix).$table_name.' ADD '.$field_name.' '.$field_type) ? true : false;
373
374                if ($default_value !== null)
375                {
376                        if (!is_int($default_value) && !is_float($default_value))
377                                $default_value = '\''.$this->escape($default_value).'\'';
378
379                        $result &= $this->query('ALTER TABLE '.($no_prefix ? '' : $this->prefix).$table_name.' ALTER '.$field_name.' SET DEFAULT '.$default_value) ? true : false;
380                        $result &= $this->query('UPDATE '.($no_prefix ? '' : $this->prefix).$table_name.' SET '.$field_name.'='.$default_value) ? true : false;
381                }
382
383                if (!$allow_null)
384                        $result &= $this->query('ALTER TABLE '.($no_prefix ? '' : $this->prefix).$table_name.' ALTER '.$field_name.' SET NOT NULL') ? true : false;
385
386                return $result;
387        }
388
389
390        function alter_field($table_name, $field_name, $field_type, $allow_null, $default_value = null, $after_field = null, $no_prefix = false)
391        {
392                if (!$this->field_exists($table_name, $field_name, $no_prefix))
393                        return true;
394
395                $field_type = preg_replace(array_keys($this->datatype_transformations), array_values($this->datatype_transformations), $field_type);
396
397                $result = $this->add_field($table_name, 'tmp_'.$field_name, $field_type, $allow_null, $default_value, $after_field, $no_prefix);
398                $result &= $this->query('UPDATE '.($no_prefix ? '' : $this->prefix).$table_name.' SET tmp_'.$field_name.' = '.$field_name) ? true : false;
399                $result &= $this->drop_field($table_name, $field_name, $no_prefix);
400                $result &= $this->query('ALTER TABLE '.($no_prefix ? '' : $this->prefix).$table_name.' RENAME COLUMN tmp_'.$field_name.' TO '.$field_name) ? true : false;
401
402                return $result;
403        }
404
405
406        function drop_field($table_name, $field_name, $no_prefix = false)
407        {
408                if (!$this->field_exists($table_name, $field_name, $no_prefix))
409                        return true;
410
411                return $this->query('ALTER TABLE '.($no_prefix ? '' : $this->prefix).$table_name.' DROP '.$field_name) ? true : false;
412        }
413
414
415        function add_index($table_name, $index_name, $index_fields, $unique = false, $no_prefix = false)
416        {
417                if ($this->index_exists($table_name, $index_name, $no_prefix))
418                        return true;
419
420                return $this->query('CREATE '.($unique ? 'UNIQUE ' : '').'INDEX '.($no_prefix ? '' : $this->prefix).$table_name.'_'.$index_name.' ON '.($no_prefix ? '' : $this->prefix).$table_name.'('.implode(',', $index_fields).')') ? true : false;
421        }
422
423
424        function drop_index($table_name, $index_name, $no_prefix = false)
425        {
426                if (!$this->index_exists($table_name, $index_name, $no_prefix))
427                        return true;
428
429                return $this->query('DROP INDEX '.($no_prefix ? '' : $this->prefix).$table_name.'_'.$index_name) ? true : false;
430        }
431
432        function truncate_table($table_name, $no_prefix = false)
433        {
434                return $this->query('DELETE FROM '.($no_prefix ? '' : $this->prefix).$table_name) ? true : false;
435        }
436}
Note: See TracBrowser for help on using the repository browser.