mirror of
https://github.com/torrentpier/torrentpier
synced 2025-08-21 13:54:02 -07:00
Transfer announce to the php7-optimized database layer
This commit is contained in:
Yuriy Pikhtarev
parent
195fe90fc8
commit
54693c00b8
4 changed files with 184 additions and 463 deletions
|
|
@ -27,9 +27,11 @@ if (!defined('IN_TRACKER')) {
|
|||
die(basename(__FILE__));
|
||||
}
|
||||
|
||||
global $tr_cfg;
|
||||
|
||||
// Exit if tracker is disabled
|
||||
if ($tr_cfg['off']) {
|
||||
tr_die($tr_cfg['off_reason']);
|
||||
msg_die($tr_cfg['off_reason']);
|
||||
}
|
||||
|
||||
//
|
||||
|
|
@ -77,419 +79,3 @@ function error_exit($msg = '')
|
|||
|
||||
tracker_exit();
|
||||
}
|
||||
|
||||
// Database
|
||||
class sql_db
|
||||
{
|
||||
public $cfg = array();
|
||||
public $cfg_keys = array('dbhost', 'dbname', 'dbuser', 'dbpasswd', 'charset', 'persist');
|
||||
public $link = null;
|
||||
public $result = null;
|
||||
public $db_server = '';
|
||||
public $selected_db = null;
|
||||
|
||||
public $locked = false;
|
||||
|
||||
public $num_queries = 0;
|
||||
public $sql_starttime = 0;
|
||||
public $sql_inittime = 0;
|
||||
public $sql_timetotal = 0;
|
||||
public $sql_last_time = 0;
|
||||
public $slow_time = 0;
|
||||
|
||||
public $dbg = array();
|
||||
public $dbg_id = 0;
|
||||
public $dbg_enabled = false;
|
||||
public $cur_query = null;
|
||||
|
||||
public $DBS = array();
|
||||
|
||||
/**
|
||||
* Constructor
|
||||
*/
|
||||
public function __construct($cfg_values)
|
||||
{
|
||||
global $DBS;
|
||||
|
||||
$this->cfg = array_combine($this->cfg_keys, $cfg_values);
|
||||
$this->dbg_enabled = sql_dbg_enabled();
|
||||
$this->slow_time = SQL_SLOW_QUERY_TIME;
|
||||
|
||||
$this->DBS['num_queries'] =& $DBS->num_queries;
|
||||
$this->DBS['sql_inittime'] =& $DBS->sql_inittime;
|
||||
$this->DBS['sql_timetotal'] =& $DBS->sql_timetotal;
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize connection
|
||||
*/
|
||||
public function init()
|
||||
{
|
||||
// Connect to server
|
||||
$this->link = $this->connect();
|
||||
|
||||
// Select database
|
||||
$this->selected_db = $this->select_db();
|
||||
|
||||
// Set charset
|
||||
if ($this->cfg['charset'] && !@mysql_set_charset($this->cfg['charset'], $this->link)) {
|
||||
if (!$this->sql_query("SET NAMES {$this->cfg['charset']}")) {
|
||||
error_exit("Could not set charset {$this->cfg['charset']}");
|
||||
}
|
||||
}
|
||||
|
||||
$this->num_queries = 0;
|
||||
$this->sql_inittime = $this->sql_timetotal;
|
||||
$this->DBS['sql_inittime'] += $this->sql_inittime;
|
||||
}
|
||||
|
||||
/**
|
||||
* Open connection
|
||||
*/
|
||||
public function connect()
|
||||
{
|
||||
$this->cur_query = 'connect';
|
||||
$this->debug('start');
|
||||
|
||||
$connect_type = ($this->cfg['persist']) ? 'mysql_pconnect' : 'mysql_connect';
|
||||
|
||||
if (!$link = $connect_type($this->cfg['dbhost'], $this->cfg['dbuser'], $this->cfg['dbpasswd'])) {
|
||||
$this->log_error();
|
||||
}
|
||||
|
||||
register_shutdown_function(array(&$this, 'close'));
|
||||
|
||||
$this->debug('end');
|
||||
$this->cur_query = null;
|
||||
|
||||
# if (DBG_LOG) dbg_log(' ', 'DB-connect'. ($link ? '' : '-FAIL'));
|
||||
|
||||
if (!$link) {
|
||||
if (function_exists('dummy_exit')) {
|
||||
dummy_exit(mt_rand(1200, 2400));
|
||||
} else {
|
||||
die;
|
||||
}
|
||||
}
|
||||
|
||||
return $link;
|
||||
}
|
||||
|
||||
/**
|
||||
* Select database
|
||||
*/
|
||||
public function select_db()
|
||||
{
|
||||
$this->cur_query = 'select db';
|
||||
$this->debug('start');
|
||||
|
||||
if (!mysql_select_db($this->cfg['dbname'], $this->link)) {
|
||||
$this->log_error();
|
||||
error_exit("Could not select database '{$this->cfg['dbname']}'");
|
||||
}
|
||||
|
||||
$this->debug('end');
|
||||
$this->cur_query = null;
|
||||
|
||||
return $this->cfg['dbname'];
|
||||
}
|
||||
|
||||
/**
|
||||
* Base query method
|
||||
*/
|
||||
public function sql_query($query)
|
||||
{
|
||||
if (!is_resource($this->link)) {
|
||||
$this->init();
|
||||
}
|
||||
$this->cur_query = $query;
|
||||
$this->debug('start');
|
||||
|
||||
if (!$this->result = mysql_query($query, $this->link)) {
|
||||
$this->log_error();
|
||||
}
|
||||
|
||||
$this->debug('end');
|
||||
$this->cur_query = null;
|
||||
|
||||
$this->num_queries++;
|
||||
$this->DBS['num_queries']++;
|
||||
|
||||
return $this->result;
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute query WRAPPER (with error handling)
|
||||
*/
|
||||
public function query($query)
|
||||
{
|
||||
if (!$result = $this->sql_query($query)) {
|
||||
$this->trigger_error();
|
||||
}
|
||||
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return number of rows
|
||||
*/
|
||||
public function num_rows($result = false)
|
||||
{
|
||||
$num_rows = false;
|
||||
|
||||
if ($result or $result = $this->result) {
|
||||
$num_rows = is_resource($result) ? mysql_num_rows($result) : false;
|
||||
}
|
||||
|
||||
return $num_rows;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return number of affected rows
|
||||
*/
|
||||
public function affected_rows()
|
||||
{
|
||||
return is_resource($this->link) ? mysql_affected_rows($this->link) : -1;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch current row
|
||||
*/
|
||||
public function sql_fetchrow($result)
|
||||
{
|
||||
return is_resource($result) ? mysql_fetch_assoc($result) : false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Alias of sql_fetchrow()
|
||||
*/
|
||||
public function fetch_next($result)
|
||||
{
|
||||
return $this->sql_fetchrow($result);
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch row WRAPPER (with error handling)
|
||||
*/
|
||||
public function fetch_row($query)
|
||||
{
|
||||
if (!$result = $this->sql_query($query)) {
|
||||
$this->trigger_error();
|
||||
}
|
||||
|
||||
return $this->sql_fetchrow($result);
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch all rows
|
||||
*/
|
||||
public function sql_fetchrowset($result)
|
||||
{
|
||||
$rowset = array();
|
||||
|
||||
while ($row = mysql_fetch_assoc($result)) {
|
||||
$rowset[] = $row;
|
||||
}
|
||||
|
||||
return $rowset;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch all rows WRAPPER (with error handling)
|
||||
*/
|
||||
public function fetch_rowset($query)
|
||||
{
|
||||
if (!$result = $this->sql_query($query)) {
|
||||
$this->trigger_error();
|
||||
}
|
||||
|
||||
return $this->sql_fetchrowset($result);
|
||||
}
|
||||
|
||||
/**
|
||||
* Escape string used in sql query
|
||||
*/
|
||||
public function escape($v, $check_type = false)
|
||||
{
|
||||
if (!is_resource($this->link)) {
|
||||
$this->init();
|
||||
}
|
||||
if (!$check_type) {
|
||||
return mysql_real_escape_string($v);
|
||||
}
|
||||
|
||||
switch (true) {
|
||||
case is_string($v):
|
||||
return "'" . mysql_real_escape_string($v) . "'";
|
||||
case is_int($v):
|
||||
return "$v";
|
||||
case is_bool($v):
|
||||
return ($v) ? '1' : '0';
|
||||
case is_float($v):
|
||||
return "'$v'";
|
||||
case is_null($v):
|
||||
return 'NULL';
|
||||
}
|
||||
// if $v has unsuitable type
|
||||
$this->trigger_error(__FUNCTION__ . ' - wrong params');
|
||||
}
|
||||
|
||||
/**
|
||||
* Return sql error array
|
||||
*/
|
||||
public function sql_error()
|
||||
{
|
||||
$return_ary = array(
|
||||
'code' => '',
|
||||
'message' => 'not connected',
|
||||
);
|
||||
|
||||
if (is_resource($this->link)) {
|
||||
$return_ary = array(
|
||||
'code' => mysql_errno($this->link),
|
||||
'message' => mysql_error($this->link),
|
||||
);
|
||||
}
|
||||
|
||||
return $return_ary;
|
||||
}
|
||||
|
||||
/**
|
||||
* Close sql connection
|
||||
*/
|
||||
public function close()
|
||||
{
|
||||
if (is_resource($this->link)) {
|
||||
mysql_close($this->link);
|
||||
}
|
||||
|
||||
$this->link = $this->selected_db = null;
|
||||
|
||||
if (DBG_LOG) {
|
||||
dbg_log(str_repeat(' ', $this->num_queries), 'DB-num_queries-' . php_sapi_name());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get info about last query
|
||||
*/
|
||||
public function query_info()
|
||||
{
|
||||
$info = array();
|
||||
|
||||
if ($num = $this->num_rows($this->result)) {
|
||||
$info[] = "$num rows";
|
||||
}
|
||||
|
||||
if (is_resource($this->link) and $ext = mysql_info($this->link)) {
|
||||
$info[] = "$ext";
|
||||
} elseif (!$num && ($aff = $this->affected_rows($this->result) and $aff != -1)) {
|
||||
$info[] = "$aff rows";
|
||||
}
|
||||
|
||||
return join(', ', $info);
|
||||
}
|
||||
|
||||
/**
|
||||
* Store debug info
|
||||
*/
|
||||
public function debug($mode)
|
||||
{
|
||||
if (!SQL_DEBUG) {
|
||||
return;
|
||||
}
|
||||
|
||||
if ($mode == 'start') {
|
||||
if (SQL_CALC_QUERY_TIME || DBG_LOG || SQL_LOG_SLOW_QUERIES) {
|
||||
$this->sql_starttime = utime();
|
||||
$this->sql_last_time = 0;
|
||||
}
|
||||
} elseif ($mode == 'end') {
|
||||
if (SQL_CALC_QUERY_TIME || DBG_LOG || SQL_LOG_SLOW_QUERIES) {
|
||||
$this->sql_last_time = utime() - $this->sql_starttime;
|
||||
$this->sql_timetotal += $this->sql_last_time;
|
||||
$this->DBS['sql_timetotal'] += $this->sql_last_time;
|
||||
|
||||
if (SQL_LOG_SLOW_QUERIES && $this->sql_last_time > $this->slow_time) {
|
||||
$msg = date('m-d H:i:s') . LOG_SEPR;
|
||||
$msg .= sprintf('%03d', round($this->sql_last_time));
|
||||
$msg .= LOG_SEPR . sprintf('%.1f', sys('la'));
|
||||
$msg .= LOG_SEPR . str_compact($this->cur_query);
|
||||
$msg .= LOG_SEPR . ' # ' . $this->query_info();
|
||||
$msg .= LOG_SEPR . $this->debug_find_source();
|
||||
bb_log($msg . LOG_LF, 'sql_slow_tr');
|
||||
}
|
||||
}
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
/**
|
||||
* Trigger error
|
||||
*/
|
||||
public function trigger_error($msg = '')
|
||||
{
|
||||
if (error_reporting()) {
|
||||
if (!$msg) {
|
||||
$msg = 'DB Error';
|
||||
}
|
||||
|
||||
if (DBG_TRACKER === true) {
|
||||
$err = $this->sql_error();
|
||||
$msg .= trim(sprintf(' #%06d %s', $err['code'], $err['message']));
|
||||
} else {
|
||||
$msg .= " [" . $this->debug_find_source() . "]";
|
||||
}
|
||||
|
||||
error_exit($msg);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Find caller source
|
||||
*/
|
||||
public function debug_find_source()
|
||||
{
|
||||
$source = '';
|
||||
$backtrace = debug_backtrace();
|
||||
|
||||
foreach ($backtrace as $trace) {
|
||||
if ($trace['file'] !== __FILE__) {
|
||||
$source = str_replace(BB_PATH, '', $trace['file']) . '(' . $trace['line'] . ')';
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return $source;
|
||||
}
|
||||
|
||||
/**
|
||||
* Log error
|
||||
*/
|
||||
public function log_error()
|
||||
{
|
||||
if (!SQL_LOG_ERRORS) {
|
||||
return;
|
||||
}
|
||||
if (!error_reporting()) {
|
||||
return;
|
||||
}
|
||||
|
||||
$msg = array();
|
||||
$err = $this->sql_error();
|
||||
$msg[] = str_compact(sprintf('#%06d %s', $err['code'], $err['message']));
|
||||
$msg[] = '';
|
||||
$msg[] = str_compact($this->cur_query);
|
||||
$msg[] = '';
|
||||
$msg[] = 'Source : ' . $this->debug_find_source();
|
||||
$msg[] = 'IP : ' . @$_SERVER['REMOTE_ADDR'];
|
||||
$msg[] = 'Date : ' . date('Y-m-d H:i:s');
|
||||
$msg[] = 'Agent : ' . @$_SERVER['HTTP_USER_AGENT'];
|
||||
$msg[] = 'Req_URI : ' . @$_SERVER['REQUEST_URI'];
|
||||
$msg[] = 'Referer : ' . @$_SERVER['HTTP_REFERER'];
|
||||
$msg[] = 'Method : ' . @$_SERVER['REQUEST_METHOD'];
|
||||
$msg[] = 'Request : ' . trim(print_r($_REQUEST, true)) . str_repeat('_', 78) . LOG_LF;
|
||||
$msg[] = '';
|
||||
bb_log($msg, 'sql_error_tr');
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue