source: trunk/lib/DB.inc.php @ 42

Last change on this file since 42 was 42, checked in by scdev, 18 years ago

detabbed all files ;P
File size: 11.3 KB
Line 
1<?php
2/**
3 * DB.inc.php
4 * code by strangecode :: www.strangecode.com :: this document contains copyrighted information
5 *
6 * DB abstraction layer.
7 *
8 * @author  Quinn Comendant <quinn@strangecode.com>
9 * @version 1.0.1
10 */
11
12class DB {
13
14    // If DB::connect has successfully opened a db connection.
15    var $_connected = false;
16
17    // Database handler.
18    var $dbh;
19
20    // Hash of DB parameters.
21    var $_params = array();
22
23    // Default parameters.
24    var $_param_defaults = array(
25
26        // DB passwords should be set as apache environment variables in httpd.conf, readable only by root.
27        'db_server' => 'localhost',
28        'db_name' => null,
29        'db_user' => null,
30        'db_pass' => null,
31
32        // Debugging.
33        'db_always_debug' => false, // TRUE = display all SQL queries.
34        'db_debug' => false, // TRUE = display db errors.
35        'db_die_on_failure' => false, // TRUE = script stops on db error.
36    );
37
38    // Translate between HTML and MySQL character set names.
39    var $mysql_character_sets = array(
40        'utf-8' => 'utf8',
41        'iso-8859-1' => 'latin1',
42    );
43
44    // Caches.
45    var $existing_tables;
46    var $table_columns;
47
48    /**
49     * This method enforces the singleton pattern for this class.
50     *
51     * @return  object  Reference to the global SessionCache object.
52     * @access  public
53     * @static
54     */
55    function &getInstance()
56    {
57        static $instance = null;
58
59        if ($instance === null) {
60            $instance = new DB();
61        }
62
63        return $instance;
64    }
65
66    /**
67     * Constructor.
68     */
69    function DB()
70    {
71        // Initialize default params.
72        if (isset($params) && is_array($params)) {
73            // Merge new parameters with old overriding only those passed.
74            $this->_params = array_merge($this->_params, $params);
75        }
76    }
77
78    /**
79     * Set (or overwrite existing) parameters by passing an array of new parameters.
80     *
81     * @access public
82     *
83     * @param  array    $params     Array of parameters (key => val pairs).
84     */
85    function setParam($params)
86    {
87        if (!isset($this) || !is_a($this, 'DB')) {
88            $this =& DB::getInstance();
89        }
90
91        if (isset($params) && is_array($params)) {
92            // Merge new parameters with old overriding only those passed.
93            $this->_params = array_merge($this->_params, $params);
94        } else {
95            App::logMsg(sprintf('Parameters are not an array: %s', $params), LOG_ERR, __FILE__, __LINE__);
96        }
97    }
98
99    /**
100     * Return the value of a parameter.
101     *
102     * @access  public
103     *
104     * @param   string  $param      The key of the parameter to return.
105     *
106     * @return  mixed               Parameter value.
107     */
108    function getParam($param)
109    {
110        if (!isset($this) || !is_a($this, 'DB')) {
111            $this =& DB::getInstance();
112        }
113
114        if (isset($this->_params[$param])) {
115            return $this->_params[$param];
116        } else {
117            App::logMsg(sprintf('Parameter is not set: %s', $param), LOG_DEBUG, __FILE__, __LINE__);
118            return null;
119        }
120    }
121
122    /**
123     * Connect to database with credentials in params.
124     *
125     * @access  public
126     * @author  Quinn Comendant <quinn@strangecode.com>
127     * @since   28 Aug 2005 14:02:49
128     */
129    function connect()
130    {
131        if (!isset($this) || !is_a($this, 'DB')) {
132            $this =& DB::getInstance();
133        }
134
135        if (!$this->getParam('db_name') || !$this->getParam('db_user') || !$this->getParam('db_pass')) {
136            App::logMsg('Database credentials missing.', LOG_EMERG, __FILE__, __LINE__);
137            return false;
138        }
139
140        // Connect to database. Always create a new link to the server.
141        if ($this->dbh = mysql_connect($this->getParam('db_server'), $this->getParam('db_user'), $this->getParam('db_pass'), true)) {
142            // Select database
143            mysql_select_db($this->getParam('db_name'), $this->dbh);
144        }
145
146        // Test for connection errors.
147        if (!$this->dbh || mysql_error($this->dbh)) {
148            $mysql_error_msg = $this->dbh ? 'Codebase MySQL error: (' . mysql_errno($this->dbh) . ') ' . mysql_error($this->dbh) : 'Codebase MySQL error: Could not connect to server.';
149            App::logMsg($mysql_error_msg, LOG_EMERG, __FILE__, __LINE__);
150
151            // Print helpful or pretty error?
152            if ($this->getParam('db_debug')) {
153                echo $mysql_error_msg . "\n";
154            } else {
155                echo _("This page is temporarily unavailable. It should be back up in a few minutes.");
156            }
157
158            // Die or continue without connection?
159            if ($this->getParam('db_die_on_failure')) {
160                echo "\n\n<!-- Script execution stopped out of embarrassment. -->";
161                die;
162            } else {
163                return false;
164            }
165        }
166
167        // DB connection success!
168        $this->_connected = true;
169
170        // Tell MySQL what character set we're useing. Available only on MySQL verions > 4.01.01.
171        $this->query("/*!40101 SET NAMES '" . $this->mysql_character_sets[strtolower(App::getParam('character_set'))] . "' */");
172
173        return true;
174    }
175
176    /**
177     * Close db connection.
178     *
179     * @access  public
180     * @author  Quinn Comendant <quinn@strangecode.com>
181     * @since   28 Aug 2005 14:32:01
182     */
183    function close()
184    {
185        if (!isset($this) || !is_a($this, 'DB')) {
186            $this =& DB::getInstance();
187        }
188
189        if (!$this->_connected) {
190            return false;
191        }
192
193        mysql_close($this->dbh);
194    }
195
196    /**
197     * Return the current database handler.
198     *
199     * @access  public
200     * @return  resource Current value of $this->dbh.
201     * @author  Quinn Comendant <quinn@strangecode.com>
202     * @since   20 Aug 2005 13:50:36
203     */
204    function getDBH()
205    {
206        if (!isset($this) || !is_a($this, 'DB')) {
207            $this =& DB::getInstance();
208        }
209
210        if (!$this->_connected) {
211            return false;
212        }
213
214        return $this->dbh;
215    }
216
217    /**
218     * Returns connection status
219     *
220     * @access  public
221     * @author  Quinn Comendant <quinn@strangecode.com>
222     * @since   28 Aug 2005 14:58:09
223     */
224    function isConnected()
225    {
226        return $this->_connected;
227    }
228
229    /**
230     * A wrapper for mysql_query. Allows us to set the database link_identifier,
231     * to trap errors and ease debugging.
232     *
233     * @param  string  $query   The SQL query to execute
234     * @param  bool    $debug   If true, prints debugging info
235     * @return resource         Query identifier
236     */
237    function query($query, $debug=false)
238    {
239        static $_query_count = 0;
240
241        if (!isset($this) || !is_a($this, 'DB')) {
242            $this =& DB::getInstance();
243        }
244
245        if (!$this->_connected) {
246           return false;
247        }
248
249        $_query_count++;
250        $debugqry = preg_replace("/\n[\t ]+/", "\n", $query);
251        if ($this->getParam('db_always_debug') || $debug) {
252            echo "<!-- ----------------- Query $_query_count ---------------------\n$debugqry\n-->\n";
253        }
254
255        // Execute!
256        $qid = mysql_query($query, $this->dbh);
257
258        // Error checking.
259        if (!$qid || mysql_error($this->dbh)) {
260            if ($this->getParam('db_debug')) {
261                echo '<pre style="padding:2em; background:#ddd; font:9px monaco;">' . wordwrap(mysql_error($this->dbh)) . '<hr>' . htmlspecialchars($debugqry) . '</pre>';
262            } else {
263                echo _("This page is temporarily unavailable. It should be back up in a few minutes.");
264            }
265            App::logMsg(sprintf('MySQL error %s: %s in query: %s', mysql_errno($this->dbh), mysql_error($this->dbh), $debugqry), LOG_EMERG, __FILE__, __LINE__);
266            if ($this->getParam('db_die_on_failure')) {
267                echo "\n\n<!-- Script execution stopped out of embarrassment. -->";
268                die;
269            }
270        }
271
272        return $qid;
273    }
274
275    /**
276     * Loads a list of tables in the current database into an array, and returns
277     * true if the requested table is found. Use this function to enable/disable
278     * funtionality based upon the current available db tables or to dynamically
279     * create tables if missing.
280     *
281     * @param  string $table    The name of the table to search.
282     * @param  bool   $strict   Get fresh table info (in case DB changed).
283     * @return bool    true if given $table exists.
284     */
285    function tableExists($table, $use_cached_results=true)
286    {
287        if (!isset($this) || !is_a($this, 'DB')) {
288            $this =& DB::getInstance();
289        }
290
291        if (!$this->_connected) {
292            return false;
293        }
294
295        if (!isset($this->existing_tables) || !$use_cached_results) {
296            $this->existing_tables = array();
297            $qid = $this->query("SHOW TABLES");
298            while (list($row) = mysql_fetch_row($qid)) {
299                $this->existing_tables[] = $row;
300            }
301        }
302        if (in_array($table, $this->existing_tables)) {
303            return true;
304        } else {
305            App::logMsg(sprintf('nonexistent DB table: %s.%s', $this->getParam('db_name'), $table), LOG_ALERT, __FILE__, __LINE__);
306            return false;
307        }
308    }
309
310    /**
311     * Tests if the given array of columns exists in the specified table.
312     *
313     * @param  string $table    The name of the table to search.
314     * @param  array  $columns  An array of column names.
315     * @param  bool   $strict   Exact schema match, or are additional fields in the table okay?
316     * @param  bool   $strict   Get fresh table info (in case DB changed).
317     * @return bool    true if given $table exists.
318     */
319    function columnExists($table, $columns, $strict=true, $use_cached_results=true)
320    {
321        if (!isset($this) || !is_a($this, 'DB')) {
322            $this =& DB::getInstance();
323        }
324
325        if (!$this->_connected) {
326            return false;
327        }
328
329        // Ensure the table exists.
330        if (!$this->tableExists($table, $use_cached_results)) {
331            return false;
332        }
333
334        // For single-value columns.
335        if (!is_array($columns)) {
336            $columns = array($columns);
337        }
338
339        if (!isset($this->table_columns[$table]) || !$use_cached_results) {
340            // Populate and cache array of current columns for this table.
341            $this->table_columns[$table] = array();
342            $qid = $this->query("DESCRIBE $table");
343            while ($row = mysql_fetch_row($qid)) {
344                $this->table_columns[$table][] = $row[0];
345            }
346        }
347
348        if ($strict) {
349            // Do an exact comparison of table schemas.
350            sort($columns);
351            sort($this->table_columns[$table]);
352            return $this->table_columns[$table] == $columns;
353        } else {
354            // Only check that the specified columns are available in the table.
355            $match_columns = array_intersect($this->table_columns[$table], $columns);
356            sort($columns);
357            sort($match_columns);
358            return $match_columns == $columns;
359        }
360    }
361
362    /**
363     * Reset cached items.
364     *
365     * @access  public
366     * @author  Quinn Comendant <quinn@strangecode.com>
367     * @since   28 Aug 2005 22:10:50
368     */
369    function resetCache()
370    {
371        $this->existing_tables = null;
372        $this->table_columns = null;
373    }
374
375} // End.
376
377?>
Note: See TracBrowser for help on using the repository browser.