. ** ** dbi.class - UltraFlexible Database Interface 3000 */ class it_dbi { var $_found_rows; # public: number of found rows if CALC_FOUND_ROWS was set var $_data; # semi-public: current result as assoc array # Default configuration of dbi class var $_defaultconfig = array ( 'db' => null, 'server' => "localhost", 'server_update' => null, 'user' => "itools", 'pw' => "", 'safety' => 1, # 0= never die, 1=die if query invalid, 2=die also if no results #'keyfield' => 'ID', # Don't set to null here, filled later by _get_field_info() #'charset' => # client charset (requires MySQL 5.0.7 or later) 'classprefix' => "", 'getfieldinfo' => true, # do not read schema. only select() allowed 'localized_defaultlanguage' => "de", # Localize fields with this suffix, e.g. copy title_de to title on read 'throttle_writes' => 0, # sleep for 'throttle_writes' multiplied by the execution time after every write 'unbuffered' => false, # use MYSQLI_USE_RESULT (WARNING this is not at all equivalent to normal it_dbi WARNING) 'ignored_warnings' => "", # regex of additional mysql warnings numbers to ignore ); var $_key; # Key of currently loaded record or null (public readonly) var $_fields; # Array of name => array(Field,Type,Null,Key,Default,Extra,Length) of fields (public readonly) var $_fieldnames; # All field names in schema for case mismatch alert var $_convertfunc; # Array of name => convert function (currently intval and floatval) for this field's values var $_link; # DB link identifier (private) var $_dbid; # string identifying db connection. used for _state_purgeshared from outside var $_affectedrows;# Affected rows (mysql_affected_rows() gets clobbered) var $_insertid; # Last inserted id (mysqli_insert_id() gets clobbered) var $_writes; # Number of (non-omittable) writes to sql using this instance /** * Constructor: Initialize the DBI interface * @param $p optional array(key => value) of configuration data * @param $query Optional initial query to run */ function __construct($p = array(), $query = null) { # Shortcut: String config means use this table with default values if (!is_array($p)) $p = array('table' => $p); if ($p['home']) $p['db'] = strtr(it::match('/www/([^/]*)', $p['home']), '.-', '__'); # If the global singleton defaults are for this db/server/server_update then use them. $dp = (array)$GLOBALS['it_dbi']->_p; if ((!isset($p['db']) || ($p['db'] == $dp['db'])) && (!isset($p['server']) || ($p['server'] == $dp['server'])) && (!isset($p['server_update']) || ($p['server_update'] == $dp['server_update']))) $p += $dp; # Combine our settings with user's defaults and class defaults $p += (array)$GLOBALS['it_dbi_defaultconfig'] + array('db' => $GLOBALS['ULTRADB']) + $this->_defaultconfig + array('charset' => strtr(strtolower(ini_get('default_charset')), array('iso-8859-1' => 'latin1', 'utf-8' => 'utf8mb4', 'utf8' => 'utf8mb4'))); unset($this->_defaultconfig); # to shorten ED() output $this->_p = $p; if ($p['table']) # Standard use: create a table object { if (!isset($GLOBALS['it_dbi'])) new it_dbi; if ($p['getfieldinfo']) $this->_p += $this->_get_field_info(); # Get $this->_fields and p[keyfield, autoincrement, randomid] if (is_array($query)) { # Call with all arguments except first one $args = func_get_args(); array_shift($args); call_user_func_array(array($this, "select"), $args); } elseif (isset($query)) $this->read($query); } else $GLOBALS['it_dbi'] =& $this; } /** * function Tablename($query) * Constructor. Returns record object from table Tablename. * If $query is set, it encodes a SELECT to execute and store in the returned object, see select() * Note: Old second parameter $config has been deprecated and will be removed */ #:} /** * Factory: Create classes of all database tables. Call statically. * @param $p array(key => value) of configuration data */ static function createclasses($p = array()) { # Make sure singleton exists $dbi = $GLOBALS['it_dbi'] ? $GLOBALS['it_dbi'] : new it_dbi($p); $p += $dbi->_p; $dbid = "{$p['user']}@{$p['server']}:{$p['db']}"; $state = it_dbi::_state_get($dbid); if (!$tables = $state['tables']) { $tables = $dbi->tables($p); $state = it_dbi::_state_get($dbid); # State could have been modified by $db->tables() call $state['tables'] = $tables; it_dbi::_state_put($dbid, $state); } foreach ($tables as $table) { # Either create class in autoloader or manually just below if (!class_exists($p['classprefix'] . $table)) it_dbi::createclass(array('table' => $table) + $p); } } /** * Convert table given by name into a class */ static function createclass($p) { # Shortcut: String config means use this table with default values if (!is_array($p)) $p = array('table' => $p); # Make sure singleton exists $dbi = $GLOBALS['it_dbi'] ? $GLOBALS['it_dbi'] : new it_dbi(array('table' => null) + $p); $p += $dbi->_p; # FIXME: (has to be checked for side effects!) $dbid = $dbi->_dbid = "{$p['user']}@{$p['server']}:{$p['db']}"; if (!isset($dbi->_tables[$dbid])) { $state = it_dbi::_state_get($dbid); $dbi->_tables[$dbid] = array(); if (!($tables = $state['tables'])) { for ($tables = array(), $res = $dbi->query('SHOW TABLES', $p); $row = mysqli_fetch_row($res);) $tables[] = $row[0]; $state = it_dbi::_state_get($dbid); # State could have been modified by query above $state['tables'] = $tables; it_dbi::_state_put($dbid, $state); } foreach ($tables as $table) $dbi->_tables[$dbid][$table] = true; } if ($p['forcecreate'] || $dbi->_tables[$dbid][$p['table']]) # Do not generate classes for non-existant tables (can be overridden by forcecreate => true, used in tests/it_dbi.t) { $classname = $p['classprefix'] . $p['table']; if (substr($classname, 0, 4) != 'PMA_') # It is designed behaviour that an error is generated if this class already exists! { $interface = function_exists("interface_exists") && interface_exists("Iterator", false) ? "implements Iterator" : ""; $code = "class $classname extends it_dbi $interface { function __construct(/* \$query ... */) { \$args = func_get_args(); \$query = array_shift(\$args); # Preserve type (scalar/array) in single parameter case foreach (\$args as \$arg) \$query = array_merge((array)\$query, (array)\$arg); parent::__construct(" . var_export($p, true) . ", \$query); } }"; debug("it_dbi::createclass('{$p['table']}'): creating class $classname, dbid=$dbid", 5); eval($code); } } } /** * INTERNAL: Connect to mysql server and maintain a global link cache */ function _connect($p = array()) { $p += $this->_p; $dbid = "{$p['user']}@{$p['server']}:{$p['db']}"; $state = it_dbi::_state_get($dbid); if ($p['reconnect'] || !($this->_link = $state['link'])) { # Force new link if same server/user was seen before (mysql ignores selected db) if ($GLOBALS['it_dbi']->_connected["{$p['server']}/{$p['user']}"]++) $this->_link = @mysqli_connect($p['server'], $p['user'], $p['pw']); else $this->_link = @mysqli_connect($p['server'], $p['user'], $p['pw']); if (!$this->_link) { # One retry after a short delay it::log('sqllog', "it_dbi(): retrying DB link (mysqli_connect {$p['server']}, {$p['db']}): " . mysqli_connect_error()); sleep(1); $this->_link = @mysqli_connect($p['server'], $p['user'], $p['pw']); } if (!$this->_link) $this->_fatal("_connect(): can't create DB link (mysqli_connect {$p['user']}@{$p['server']}, {$p['db']}): " . mysqli_connect_error()); if (!(@mysqli_select_db($this->_link, $p['db']))) $this->_fatal("_connect(): can't select database \"{$p['db']}\""); # set charset used for this connection if ($p['charset']) if (!mysqli_set_charset($this->_link, $p['charset'])) $this->_fatal("_connect(): can't set charset \"{$p['charset']}\""); # NOTE: This overwrites old state but that is on purpose. New link means we should refetch all info about connection $state['link'] = $this->_link; it_dbi::_state_put($dbid, $state, false); # Store only locally as link is not shared anyway } } /** * INTERNAL: construct SQL SET clause of changed values from tags array. * $force = must write all fields, dont try to optimize */ function _set($tags, $force = false) { $result = array(); $dyndata = $this->_dyndata; foreach((array)$tags as $field => $value) { $f = trim($field, "-"); if (!$this->_fields[$f] && it::match(",$f,", $this->_fieldnames)) it::error("case mismatch on field $f, valid fields $this->_fieldnames"); if ($this->_p['charset'] == "utf8") # NOTE: Mysql charset is simply utf8, not utf-8 $value = it::any2utf8($value, "error in db-field $field"); if (!$this->_fields[$f] && $this->_fields['dyncols']) { if (substr($field, 0, 1) === "-") $newdyns[] = "'\$.$f', " . $value; else if ($force || isset($value) && isset($dyndata[$f]) ? strval($value) !== strval($dyndata[$f]) : $value !== $dyndata[$f] || !array_key_exists($f, $dyndata)) { if (is_null($value)) $deldyns[] = "'\$.$f'"; else $newdyns[] = "'\$.$f', " . $this->escape_string($value); } $alldyns[] = $this->escape_string($f) . ", " . (substr($field, 0, 1) === "-" ? $value : $this->escape_string($value)); $dyndata[$f] = $value; } else if (substr($field, 0, 1) === '-') # Unquoted value (always added) $result[] = substr($field, 1) . "=$value"; else if ($force || (isset($value) && isset($this->_data[$field]) ? strval($value) !== strval($this->_data[$field]) : $value !== $this->_data[$field] || !array_key_exists($field, $this->_data))) $result[] = "`$field`=" . (isset($value) ? $this->escape_string($value) : 'NULL'); } if ($alldyns) { if ($force == "insert") # INSERT/REPLACE $result[] = "dyncols = JSON_OBJECT(" . join(", ", (array)$alldyns) . ")"; else if ($newdyns || $deldyns) { $source = $this->_dyndata ? 'dyncols' : '"{}"'; if ($newdyns) $source = "JSON_SET($source, " . join(', ', $newdyns) . ')'; if ($deldyns) $source = "JSON_REMOVE($source, " . join(', ', $deldyns) . ')'; $result[] = "dyncols = $source"; } } $this->_writes += $result ? 1 : 0; return $result ? 'SET ' . implode(', ', $result) : ''; } /** * INTERNAL: construct SQL FROM clause of fields JOIN and FROM in query params * @see select() */ function _from($params, $omit_from = false) { $result = $this->_p['table']; if (isset($params['JOIN']) || isset($params['FROM'])) # WARNING: this field gets abused for "tablename USE INDEX (fast2) $result = trim($params['FROM'] . " " . $params['JOIN']); return ($omit_from ? "" : "FROM ") . $result; } /** * Create an SQL query (the stuff after 'WHERE') according to an array of selection criteria. * * Example: * $sql = $table->_where(array('Type' => 'bar', * 'Date >=' => '1999-01-01', '-Date <' => 'NOW()', * 'Status' => array('foo', 'bar', 'qux'), # same as 'Status IN' => ... * 'User NI' => 'chris'), 'ORDER BY Date'); * * @param $params optional array of fieldname => value tupels. These are ANDed to form a WHERE clause. * fieldname can contain an operator (separated by space), the default operator is '='. * The special operator 'NI' specifies that the argument must be contained in a comma-separated list. * @param $link DB link used to escape values (not used anymore) * @param $omit_where Do not add 'WHERE ' to result * @return The generated SQL clause * @see select() * @see iterate() */ function _where($params) { $dyncols_enabled = !strlen($params['JOIN'] . $params['FROM']); unset($params['JOIN'], $params['FROM']); if (is_array($params) && (count($params) > 0)) { $query = ''; $stringquery = ''; $sep = ''; foreach($params as $field => $value) { if (is_int($field)) /* no key specified; just append */ { if (strcasecmp($value, 'OR')) $stringquery .= " $value"; else $sep = ' OR '; } else if ($field == "LIMIT") { if ($value !== false) # only false no null; uninitialized values should not unintentionally omit LIMIT $stringquery .= " LIMIT " . (it::match('^[ ,\d]+$', $value) ?? it::error(['title' => "invalid LIMIT $value", 'body' => $params]) + 0); } else { $needs_where = true; if (!isset($value)) { $op = 'IS'; $qval = 'NULL'; } else { if (preg_match('/^(\S+)\s+(\S.*)$/', $field, $regs)) { $field = $regs[1]; $op = strtoupper($regs[2]); } else $op = is_array($value) ? 'IN' : '='; # If field name starts with '-', the raw value is taken, no escaping is done and no quotes are put around it. if (substr($field, 0, 1) == '-') { $field = substr($field, 1); # Strip that '-' sign $qval = $value; } else if (!is_array($value)) $qval = $this->escape_string((string)$value); } if ($dyncols_enabled && $this->_fields['dyncols'] && !$this->_fields[$field] && strpos($field, '(') === false) $field = "JSON_EXTRACT(dyncols, " . $this->escape_string('$.' . $field) . ")"; switch ($op) { case 'NI': if ($value) { $parts = array(); foreach ((array)$value as $val) $parts[] = "CONCAT(',',$field,',') LIKE " . $this->escape_string("%,$val,%"); $query .= $sep . "$field IS NOT NULL AND (" . join(" OR ", $parts) . ")"; # Check for IS NOT NULL to take advantage of index } else $query .= $sep . "1"; break; case 'MATCH': $qval = join(' ', (array)$value); $query .= "$sep$op ($field) AGAINST (" . $this->escape_string($qval) . " IN BOOLEAN MODE)"; break; case 'IN': case 'NOT IN': if (is_array($value)) { if ($value) { $qvals = array(); foreach ($value as $val) $qvals[] = $this->escape_string($val); $query .= "$sep$field $op (" . join(",", $qvals) . ")"; # null is mapped to '' } else $query .= $sep . (($op == 'IN') ? "0" : "1"); break; } /* FALLTHROUGH */ default: if (isset($qval)) $query .= "$sep$field $op $qval"; else it::fatal('Undefined $qval when constructing query due to invalid $value (array)'); break; } $sep = ' AND '; } } $query .= $stringquery; if ($needs_where) $query = "WHERE $query"; } return $query; } /** * Internal: Output class name::error message and terminate execution. */ function _fatal($text, $body = null) { $text = get_class($this).'::'.$text; if ($this->_link && ($errstr = mysqli_error($this->_link))) $text = "\"$errstr\" in $text [errno " . mysqli_errno($this->_link) . "]"; if ($this->_link && ($res = @mysqli_fetch_row(mysqli_query($this->_link, 'select database()')))) # dont create extra errs $text .= ", DB: " . $res[0]; it::fatal(['title' => $text . ", Server: " . $this->_p['server'], 'body' => $body]); /* NOT REACHED */ } /** * Post-process data after reading a record. * This is a stub-function that can be overloaded. */ function _read_post_process() { } /** * Pre-process data before writing a record. * This is a stub-function that can be overloaded. * @param $tags Reference to update/create tags, can be modified as needed * @param $command SQL command ('INSERT', 'REPLACE' or 'UPDATE') */ function _write_pre_process(&$tags, $command) # NOPHPLINT FIXME 2020-12 CS Remove legacy hooks now replace by pure and static versions { } /** * Hook to postprocess data after reading a record. * This is a stub-function that can be overloaded. * @param $data Data of record read from db * @return Record data including postprocess modifications */ static function _read_postprocess($data) { return $data; } /** * Hook to preprocess data before writing the record. * This is a stub-function that can be overloaded. * @param $data Update/create data tags * @return Record data to be written including preprocess modifications/additions/deletions */ static function _write_preprocess($data) { return $data; } /** * Return an array of all tables of this database */ function tables($p = array()) { for ($qr = $this->query('SHOW TABLES', $p); $row = mysqli_fetch_row($qr);) $result[] = $row[0]; return (array)$result; } /** * Clear record */ function clear() { foreach ((array)$this->_fields + (array)$this->_localizedfields + (array)$this->_data as $field => $dummy) unset($this->$field); unset($this->_key, $this->_data); } /** * Semi-internal: send a raw SQL query and return mysql result value * @param $query complete SQL query string * @return MySQL result which is false for errors. May die on error if safety is big enough */ function query($query, $p = array()) { $p += $this->_p; $start = gettimeofday(true); if (($writing = !it::match('^(EXPLAIN|SELECT|SHOW)', $query, array('utf8' => false)))) { if ($p['server_update']) { debug("switching to update server \"{$p['server_update']}\"", 5); $this->_p['server'] = $p['server'] = $p['server_update']; unset($this->_p['server_update'], $p['server_update'], $this->_link); } else if ($p['server'] == "localhost" && $p['db'] == $GLOBALS['ULTRADB'] && preg_grep('/replicate-do/', (array)@it::file($GLOBALS['ULTRAHOME'] . "/etc/my.cnf"))) if (($t = @it::file($GLOBALS['ULTRAHOME'] . "/doc/machines.txt")) && preg_grep("/^" . gethostname() . "/", array_slice($t, 2))) it::error("local mysql write on a replication slave machine?"); } $this->_connect($p); # must be called after update server switching code debug("{$p['user']}@{$p['server']}:{$p['db']}" . '.' . get_class($this) . "::query(\"$query\")", 4); if (!($result = mysqli_query($this->_link, $query, $p['unbuffered'] ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT)) && $p['safety']) { $errno = mysqli_errno($this->_link); if (($p['safety'] < 2) && ($errno == 1062)) # Duplicate entry return false; if ($errno == 2006) # mysql server has gone away: retry { it::log('sqllog', "it_dbi(): reconnecting mysqli_connect {$p['server']}, {$p['db']}"); $this->_connect(array('reconnect' => true)); $result = mysqli_query($this->_link, $query, $p['unbuffered'] ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT); } if (!$result) $this->_fatal("query(\"$query\") failed"); } else if (it::match('^(CREATE|ALTER|DROP) ', $query, array('utf8' => false))) { # Purge cache for schema changes (after modifying table) $dbid = "{$p['user']}@{$p['server']}:{$p['db']}"; it_dbi::_state_purgeshared($dbid); } $this->_affectedrows = $this->_link->affected_rows; # get_warnings() clobbers this $this->_insertid = mysqli_insert_id($this->_link); # get_warnings() clobbers this if (($warning = $this->_link->get_warnings())) { do { if (!it::match(trim($this->_p['ignored_warnings'] . "|1364|1261|1051|1062", "|"), $warning->errno)) $messages[] = $warning->message . " [error $warning->errno]"; } while ($warning->next() && ++$checked < 20); if ($messages) it::error(['title' => "Mysql warning: " . $messages[0], 'body' => "$query\n\n" . join("\n", $messages) . "\n"]); } if ($writing && $this->_p['throttle_writes']) usleep(1000000 * (gettimeofday(true) - $start) * $this->_p['throttle_writes']); $msec = round(1000 * (gettimeofday(true) - $start)); $slow = $msec >= 2000; if ($GLOBALS['debug_sqllog'] || $GLOBALS['debug_sqltrace'] || $slow) { $backtrace = (EDC('sqltrace') || $slow) ? it_debug::backtrace(1) : null; $truncquery = strlen($query) > 1000 ? mb_substr($query, 0, 1000) . '...' : $query; it::log('sqllog', "$msec\t$truncquery\t$backtrace\t" . $this->_p['server'] . ($slow ? "\tSLOW" : "")); $this->_sqllog[] = array( 'time' => $msec, 'query' => $query, ) + ($backtrace ? array('backtrace' => $backtrace) : array()); } return $result; } /** * Read a single record by primary key, not destroying old query result $this->_result * @param $id primary key. If null, record is cleared * @return True if record could be read, false if not. */ function read($id=null) { $old_result = $this->_result; $old_nofetch = $this->_nofetch; $result = $this->select(array($this->_p['keyfield'] => $id)); $this->_result = $old_result; $this->_nofetch = $old_nofetch; return $result; } /** * Select a set of records from table and fetch the first one * @param $query Vararg. Arrays of (field => value) pairs or plain string query parts. Default: Select all * Fields will be joined by AND * Fields can contain a compare operator: 'name LIKE' => "j%" or 'amount >' => 100 * Fields can start with - to prevent quoting of right side: '-modified' => "CURDATE()" * @param $query['SELECT'] expression to be returned, e.g. 'SELECT' => 'DISTINCT foo'. defaults to '*' * @param $query['FROM'] table names to use for joins, e.g. 'FROM' => 'tableA LEFT JOIN tableB ON a=b' * @param $query['JOIN'] like 'FROM' * @param $query['CALC_FOUND_ROWS'] if true, if true member var _found_rows contains number of matching rows before LIMIT * @param $query['LIMIT'] max number of rows to return; false for no limit * @param $query['NOFETCH'] if true the first row is not prefetched (e.g. when directly using _result) * @return Number of matching rows, use iterate() to fetch the next record * @see iterate() * @see _where() */ function select(/* $query = array|string, ... */) { $query = array(); foreach (func_get_args() as $arg) $query = array_merge($query, (array)$arg); $this->_connect(); $result = 0; $calc_found_rows = false; $what = '*'; if (isset($query['SELECT'])) { $what = $query['SELECT']; unset($query['SELECT']); } unset($this->_found_rows); if (isset($query['CALC_FOUND_ROWS']) && $query['CALC_FOUND_ROWS']) { $calc_found_rows = true; $what = 'SQL_CALC_FOUND_ROWS '.$what; } unset($query['CALC_FOUND_ROWS']); # Always unset, so CALC_FOUND_ROWS => false doesn't generate bogus query if (EDC('nocache')) $what = 'SQL_NO_CACHE ' . $what; $nofetch = $this->_nofetch = $query['NOFETCH']; unset($query['NOFETCH']); $this->clear(); if ($this->_result = $this->query($sql = "SELECT $what " . $this->_from($query) . " " . $this->_where($query))) { $result = $this->_p['unbuffered'] ? true : mysqli_num_rows($this->_result); if ($calc_found_rows) { list($count) = mysqli_fetch_row($this->query('SELECT FOUND_ROWS()')); $this->_found_rows = intval($count); } if (!$this->iterate() && ($this->_p['safety'] >= 2)) $this->_fatal("select(): query produced no results: \"$sql\""); } $this->_nofetch = !$nofetch; return $result; } /** * Puts next result from previous select() in member variables * @return true if another record was fetched, false if no more records * @see select() */ function iterate() { if (!$this->_nofetch) { if ($this->_data = mysqli_fetch_assoc($this->_result)) { if ($localizedfields = $this->_localizedfields) foreach ($localizedfields as $field => $dummy) unset($this->$field); foreach ((array)$this->_dyndata as $field => $dummy) unset($this->$field); foreach (($this->_dyndata = ($t = $this->_data['dyncols']) ? (array)json_decode($t, true) : []) as $field => $value) $this->_data[$field] = $value; unset($this->_data['dyncols']); foreach (static::_read_postprocess($this->_data) as $field => $value) { $this->$field = (isset($value) && $this->_convertfunc[$field]) ? ($this->_data[$field] = $this->_convertfunc[$field]($value)) : $value; if (!array_key_exists($field, $this->_data)) # Register fields added by _read_postprocess() so they get clear()ed $this->_data[$field] = $value; } if ($localizedfields) { $lang = T_lang(); foreach ($localizedfields as $field => $dummy) { $value = $this->{$field . "_" . $lang}; if (!isset($value)) $value = $this->{$field . "_" . $this->_p['localized_defaultlanguage']}; if (isset($value)) { if (isset($this->$field)) it::fatal("Field name clash: Overwriting {$this->_p['table']}.$field with {$field}_{$lang}, only use one of those fields"); else $this->$field = $value; } } } if (!empty($this->_p['keyfield'])) $this->_key = $this->_data[$this->_p['keyfield']]; } else { $this->clear(); if ($this->_p['unbuffered']) $this->_result->close(); } $this->_read_post_process(); } else $this->_nofetch = false; return (bool)$this->_data; } /** * Insert a record into table. Values are taken assoc array $tags. Keys in $tags * should contain row names unless "dyncols" exists in schema (they are stored as * dynamic columns then). Keys with - prefix suppress quoting of values. * After inserting, all values are valid (record is read back). * Does not destroy internal state of last select() call * @param $tags key => value pairs to set * @return true for success, false for failure (e.g. duplicate entry for key) */ function insert($tags = array(), $command = "INSERT") { $this->_connect(); /* Pre-processing, $tags is passed by reference and may be modified here */ $tags = static::_write_preprocess($tags); $this->_write_pre_process($tags, $command); if ($this->_p['randomid'] && !isset($tags[$this->_p['keyfield']])) $tags[$this->_p['keyfield']] = bin2hex(random_bytes(16)); $set = $this->_set($tags, "insert"); if ($result = $this->query($query = "$command INTO {$this->_p['table']} " . $set)) { $id = ($this->_p['autoincrement'] && !isset($tags[$this->_p['keyfield']])) ? $this->_insertid : $tags[$this->_p['keyfield']]; if ($this->_p['keyfield'] && !$this->read($id) && $this->_p['safety']) $this->_fatal("insert(): can't read record back (key truncated?), id=\"$id\"", $query); $this->_touchedids[$this->_key] = true; } else $this->clear(); return $result; } /** * Replace a record in a table * @param $tags Additional key => value pairs * Does not destroy internal state of last select() call * @see insert() */ function replace($tags = array()) { return $this->insert($tags, "REPLACE"); } /** * Create a record like replace(), but optimize if similar or identical record already exists * MUST GIVE ALL FIELDS INCLUDING ID * @param $tags key => value pairs to set */ function store($tags = array()) { return $tags[$this->_p['keyfield']] && $this->read($tags[$this->_p['keyfield']]) ? $this->update($tags) : $this->replace($tags); } /** * Update current record or a number of records given by where condition * @param $tags key => value pairs (these have priority over changes in member vars) * @param $where condition to select records to be modified (if not current record) * @return number of modified records (or false on error). WARNING: read LIMIT docs before using it * Does not destroy internal state of last select() call */ function update($tags = array(), $where = null) { $this->_connect(); $result = 0; # in case we optimize away the query /* Pre-processing, $tags is passed by reference and may be modified here */ $tags = static::_write_preprocess($tags); $this->_write_pre_process($tags, 'UPDATE'); if ($set = $this->_set($tags, isset($where))) { if (!isset($where)) $where = array($this->_p['keyfield'] => $this->_data[$this->_p['keyfield']]); if ($result = $this->query("UPDATE " . $this->_from($where, true) . " $set " . $this->_where($where))) { $result = $this->_affectedrows; if (array_key_exists($this->_p['keyfield'], $tags)) # Did we just update the key? $this->_key = $tags[$this->_p['keyfield']]; if (isset($this->_key) && $this->read($this->_key)) $this->_nofetch = false; # So we can do while(iterate()) update(); } } $this->_touchedids[$this->_key] = true; return $result; } /** * Delete a record * Does not destroy query result $this->_result * @param $query optional query for WHERE, default: delete currently loaded record * @return number of deleted records */ function delete($query = null) { $this->_connect(); $result = 0; if (!$query && $this->_p['keyfield']) { $query = array($this->_p['keyfield'] => $this->_key); $this->clear(); } if ($query && $this->query("DELETE " . $this->_from($query) . " " . $this->_where($query))) $result = $this->_affectedrows; return $result; } /** * Delete records matching query which were not touched since creation of this object or last call of this func */ function delete_untouched($query = null) { if ($this->select(['SELECT' => $this->_p['keyfield']] + (array)$query)) while ($this->iterate()) if (($id = $this->_key) && !$this->_touchedids[$id] && $this->delete()) $result[] = $id; unset($this->_touchedids); return (array)$result; } /** * Escapes a string for use in a DB query * @param The string to be quoted * @return The quoted value */ function escape_string($str) { $this->_connect(); return "'" . mysqli_real_escape_string($this->_link, $str) . "'"; } /** * INTERNAL: Store information about a table's fields in $this->_fields, possibly from cache. * @return array(keyfield, autoincrement, randomid) */ function _get_field_info() { $result = array(); $dbid = "{$this->_p['user']}@{$this->_p['server']}:{$this->_p['db']}"; $state = it_dbi::_state_get($dbid); if (!($this->_fields = $state['fields'][$this->_p['table']])) { debug("it_dbi(): no fields for {$dbid}.{$this->_p['table']}, calculating.", 5); for ($res = $this->query('SHOW COLUMNS FROM ' . $this->_p['table']); $res && ($field = mysqli_fetch_assoc($res)); ) { $this->_fields[$field['Field']] = $field + array('Length' => preg_match('/date|time/', $field['Type']) ? 20 : intval(it::match('\d+', $field['Type']))); if (preg_match('/^(tiny|small|medium|)int|^float|^double/', $field['Type'])) $this->_convertfunc[$field['Field']] = it::match('int', $field['Type']) ? "intval" : "floatval"; } $this->_fieldnames = "," . join(",", array_keys((array)$this->_fields)) . ","; # Consider all fields which have _{localized_defaultlanguage} suffix as localized foreach (preg_grep('/_' . $this->_p['localized_defaultlanguage'] . '$/', array_keys((array)$this->_fields)) as $field) $this->_localizedfields[substr($field, 0, -1 - strlen($this->_p['localized_defaultlanguage']))] = true; $state = it_dbi::_state_get($dbid); # State could have been modified by query above $state['fields'][$this->_p['table']] = $this->_fields; $state['convertfunc'][$this->_p['table']] = $this->_convertfunc; $state['localizedfields'][$this->_p['table']] = $this->_localizedfields; it_dbi::_state_put($dbid, $state); } else # Existing _fields, copy other info too { $this->_convertfunc = $state['convertfunc'][$this->_p['table']]; $this->_localizedfields = $state['localizedfields'][$this->_p['table']]; } foreach((array)$this->_fields as $field) { if ($field['Key'] == 'PRI') { $result = array('keyfield' => $field['Field'], 'autoincrement' => (strpos($field['Extra'], "auto_increment") !== false), 'randomid' => (strpos($field['Type'], "char") !== false)); break; } } unset($GLOBALS['it_dbi']->_p['table'], $GLOBALS['it_dbi']->_p['keyfield']); # Remove cruft return $result; } static function _state_get($dbid) { if (!($result = $GLOBALS['it_dbi']->_state[$dbid])) $result = $GLOBALS['it_dbi']->_state[$dbid] = (array)it_cache::get("dbi:$dbid"); #var_dump("get", $dbid, $result); return $result; } static function _state_put($dbid, $state, $shared = true) { #var_dump("put", $dbid, $state); $GLOBALS['it_dbi']->_state[$dbid] = $state; if ($shared) it_cache::put("dbi:$dbid", array('link' => null) + (array)$state); # link is not transferable } static function _state_purgeshared($dbid = null) { #var_dump("purgeshared", $dbid); it_cache::put("dbi:" . ($dbid ? $dbid : $GLOBALS['it_dbi']->_dbid), array()); # Nuke shared cache } # # Implement PHP 5 Iterator interface to make foreach work # Example: foreach (new T_User('firstname' => "foo") as $foouser) { ... } # function current() { return $this; } function key() { return isset($this->_key) ? $this->_key : $this->_iteratorkey++; } function next() { $this->iterate(); } function rewind() { if (!$this->_result) # Object without query used in foreach $this->select(); # Only rewind if not already at start and results present if (!$this->_nofetch && mysqli_num_rows($this->_result)) mysqli_data_seek($this->_result, 0); $this->_iteratorkey = 0; $this->iterate(); } function valid() { return (bool)$this->_data; } /** * Updates or inserts record */ function upsert($tags) { return $this->read($tags[$this->_p['keyfield']]) ? $this->update($tags) : $this->replace($tags); } /** * STATIC: create an object and read its record from DB. * @param $id ID of record to read * @return initialized object or null if ID not found */ static function get($id) { return isset($id) && ($rec = new static) && $rec->read($id) ? $rec : null; } } /* End class it_dbi */