<?php
/**************************************************
*  Created:  2010-06-08
*  LastModify: 2010-10-23
*  MysqlDB类
*
*  @Xweibo (C)1996-2099 SINA Inc.
*  @Author xuzhenquan <zhenquan@staff.sina.com.cn>
*
***************************************************/

if (!class_exists('interface_db')) {
	require('interface_db.php');
}

class mysql_db extends interface_db 
{
	var $table = ''; // default table name
	var $last_sql = ''; // last query sql
	var $querys = array(); // query history
	var $autoFree = false; // switch of auto free result 
	var $ignore_insert = false; // 设置是否为忽略插入
	//var $last_query_id = false; // query id
	var $prefix = ''; // 表名前缀
	var $params = array(); // db params
	var $debug = false;
	//var $last_query_connect = null;
	var $state = array();
	/**
	 * contructor
	 * @param $table default table name
	 * @return void
	 */
	function mysql_db() {
		
	}

	function setIgnoreInsert($ignore = true) {
		$this->ignore_insert = (boolean)$ignore;
	}
	/**
	 * 设置调试模式(注意：调试模式将打印出Mysql帐号信息)
	 */
	function setDebug($flag = true) {
		$this->debug = (boolean)$flag;
	}

	/**
	 * 得到表前缀
	 */
	function getPrefix() {
		return $this->prefix;
	}

	/**
	 * 框架初始化类时装配置传给该方法
	 */
	function adp_init($params) {
		$this->params = $params;
		$this->prefix = $params['tbpre'];
	}
	
	/**
	 * 设置主控表名
	 * @param $table string 表名
	 *
	 */
	function setTable($table) {
		$this->table = $table;
	}
	
	/**
	 * 取得主控表名称
	 * @param $table_name string 表名称 
	 * @param $tb_prefix sting 表前缀，如不设置，则使用getPrefix()设置的前缀
	 * @return string
	 */
	function getTable($table_name='', $tb_prefix='') {
		if (empty($tb_prefix)) {
			$tb_prefix = $this->getPrefix();
		}
		$table_name = trim($table_name);
		$table_name = $table_name == ''? $this->table : $table_name;
		return $tb_prefix . $table_name;
	}
	
	/**
	 * 记录调用的SQL
	 * @param $sql string SQL语句
	 */
	function pushSql($sql) {
		array_push($this->querys, $sql);
		return $this->last_sql = $sql;
	}

	/**
	 * 设置是否每次查询完成后，是否自动释放内存
	 * @param $v boolean true表示自动释放，false关闭自动释放
	 */
	function setAutoFree($v) {
		$this->autoFree = (bool)$v;
	}

	/**
	 * 过滤字串
	 * @param $str string 要过滤的字串
	 * @return string
	 */
	function escape($str) {
		return mysql_escape_string($str);
	}

	/**
	 * 得到最后插入记录的自增ID
	 * @return int
	 */
	function getInsertId() {
		return $this->state['last_insert_id'] ? $this->state['last_insert_id']: false;
	}

	/**
	 * 由指定的值,取得一条记录
	 * @param $id int|string 要查询的ID或关键词,查询的字段由$id_name指定，默认查询名为id的字段
	 * @param $table string 表名称
	 * @param $id_name string 提供查询的字段名
	 * @return array
	 */
	function get($id, $table='', $id_name = 'id') {
		if (!($table = $this->getTable($table))) {
			return false;
		}

		$sql = 'SELECT * FROM ' . $table . ' WHERE ' .$id_name . ' =\'' . $this->escape($id) . '\'';

		return $this->getRow($sql);
		
	}
	
	/**
	 * 根据关键字删除记录,  只用作全等比较
	 * @param $id mixed 要查询的关键字
	 * @param $table string 表名称
	 * @param $id_name string 要查询的字段名，默认为id
	 * @return int|false 如果成功则返回影响行业,失败返回false
	 */
	function delete($id, $table = '', $id_name = 'id') {
		if (!$table = $this->getTable($table)) {
			return false;
		}
		if (!is_array($id)) {
			$id = (array)$id;
		}
		$sql = 'DELETE FROM ' . $table . ' WHERE ' . $id_name . ' IN(' . implode(',', $id). ')';
		if (!$this->execute($sql)) {
			return false;
		}
		return $this->getAffectedRows();
	}
	
	/**
	 * 添加或插入数据
	 * @param $data 键/值对应的数组，其中键为字段名，值为要插入的内容
	 * @param $id int 更新时使用的关键字，如果指定该项，则执行更新操作
	 * @param $table string 表单名，如果不指定，则使用setTable()设置的默认表名
	 * @param $id_name string 字段名，更新时用于查询的字段，默认查询名为'id'的字段
	 * @return int|false 返回最后插入的记录ID或更新记录的ID,失败返回false
	 */
	function save($data, $id = '', $table = '', $id_name = 'id') {

		if ($id == '') {
			$type = 'insert';
		} else {
			$type = 'update';
		}
		$table = $this->getTable($table);
		if ($type == 'insert') {
			$keys = array();
			$values = array();
			foreach ($data as $key => $value) {
				$keys[] = '`' . $this->escape($key). '`';
				$values[] = '"' . $this->escape($value) . '"';
			}
			if (sizeof($keys) != sizeof($values)) {
				return false;
			}
			$ignore = '';
			if ($this->ignore_insert) {
				$ignore = ' IGNORE ';
			}

			$sql = 'INSERT ' . $ignore . ' INTO ' . $table . '(' .implode(',', $keys). ') VALUES('. implode(',', $values) .')';
			if (!$this->execute($sql)) {
				return false;
			}

			return $this->getInsertId();
		}
		$values = array();
		foreach ($data as $key=>$value) {
			$values[] = '`' .trim($key) . '`="' . $this->escape($value) . '"';
		}
		if (!sizeof($values)) {
			return false;
		}
		$sql = 'UPDATE ' . $table . ' SET ' . implode(',', $values) . ' WHERE ' . $id_name . '=' . $id;
		
		if (!$this->execute($sql)) {
			return false;
		}
		return $this->getAffectedRows();
	}
	
	
	/**
	 * 添加或插入数据
	 * @param $data 键/值对应的数组，其中键为字段名，值为要插入的内容
	 * @param $id int 更新时使用的关键字，如果指定该项，则执行更新操作
	 * @param $table string 表单名，如果不指定，则使用setTable()设置的默认表名
	 * @param $id_name string 字段名，更新时用于查询的字段，默认查询名为'id'的字段
	 * @return int|false 返回最后插入的记录ID或更新是否成,失败返回false
	 */
	function boolSave($data, $id = '', $table = '', $id_name = 'id') 
	{
		$type  = ($id=='') ? 'insert' : 'update';
		$table = $this->getTable($table);
		
		// insert
		if ($type == 'insert') 
		{
			$keys   = array();
			$values = array();
			foreach ($data as $key => $value) {
				$keys[]   = '`' . $this->escape($key) . '`';
				$values[] = '"' . $this->escape($value). '"';
			}
			if (sizeof($keys) != sizeof($values)) {
				return false;
			}
			
			$ignore = '';
			if ($this->ignore_insert) {
				$ignore = ' IGNORE ';
			}

			$sql = 'INSERT ' . $ignore . ' INTO ' . $table . '(' .implode(',', $keys). ') VALUES('. implode(',', $values) .')';
			if ($this->execute($sql)) {
				return $this->getInsertId();
			}
			return false;
		}
		
		// update
		$values = array();
		foreach ($data as $key=>$value) {
			$values[] = '`' .trim($key) . '`="' . $this->escape($value) . '"';
		}
		if (!sizeof($values)) {
			return false;
		}
		
		$sql = 'UPDATE ' . $table . ' SET ' . implode(',', $values) . ' WHERE ' . $id_name . '=' . $id;
		return $this->execute($sql);
	}
	
	/**
	 * 得到读数据库连接(使用主从数据库)时，如果不是主从架构，则返回唯一的连接(主从设置可以站点根目录中的config.php中设置)
	 * @param $mode 'read'|'write' 使用读服务器还是写服务器
	 * @param $index int 指定要使用的读服务器配置项,如果不设置则随机选择配置项中指定的服务器，以达到读服务器均衡效果
	 * @param $reconnect boolean 指定是否强制重新连接,true为强制重连
	 * @return resource 返回mysql读数据库连接
	 */
	function getConnect($mode = 'write', $index = null, $reconnect = false) {
		static $connect = null;
		static $count_reconnect = 0;// 重连次数
		static $error_connect = 0;// 连接错误的服务器数

		$mode = in_array($mode, array('write', 'read')) ? strtolower($mode) : 'write';

		// 如果第一次连接
		if (!isset($connect[$mode]) || $reconnect !== false) {
			$count_reconnect = 0; // 重新计算重连次数
			// 如果设置时使用读服务器，但没有相关配置项，则尝试使用写服务器
			if ($mode == 'read' && isset($this->params['slaves']) && !empty($this->params['slaves'])) {
				// 配置的服务器数量
				$count = count($this->params['slaves']);
				if ($index === null || !is_int($index) || $index < 0 || $index>=$count) {
					$index = rand(0, $count - 1);
				} 
				$p = $this->params['slaves'][$index];
			} else {
				$p = $this->params;
			}

			// 检查配置项完整
			if (!(isset($p['host']) && isset($p['user']) && isset($p['pwd']) )) {
				$this->debug && print('Mysql config error');
				exit;
				//return $this->getWriteConnect();
			}
			if (!isset($p['port'])) {
				$p['port'] = 3306;
			}

			$this->debug && print('Create Mysql (' . $mode . ')connection:Host:' .$p['host'] . ':' . $p['port'] . ' , User:' . $p['user'] . ' , ' . $p['pwd'] . "<br />");
			
			$connect[$mode] = @mysql_connect($p['host'] . ':' . $p['port'], $p['user'], $p['pwd']);

			// 如果连接失败，则尝试连接下一台读服务器
			if (!$connect[$mode]) {
				$this->log('连接失败.Error:'.mysql_error());
				F('error', mysql_error());
				if ($mode == 'read') {
					++ $error_connect;
					//如查所有读服务器都连接失败,尝试连接写服务器
					if ($error_connect >= $count) {
						$this->debug && print('Try to connect next (write)server<br />'."\n");
						return $this->getConnect();
					}
					if (++$index > $count) {
						$index = 0;
					}
					$this->debug && print('Try to connect next (read)server<br />'."\n");
					return $this->getConnect('read', $index);
				} else {
					exit('Connect Mysql server error.<br />' . "\n");
				}
			}
			// 默认使用UTF8编码
			$p['charset'] = isset($p['charset']) ? $p['charset'] : 'UTF8';
			$this->_setCharset($p['charset'], $connect[$mode]);
			//mysql_query('SET NAMES ' . $p['charset'], $connect[$mode]);
			// 默认使用和写数据库相同的库名
			if (!isset($p['db'])) {
				$p['db'] = $this->params['db'];
			}
			$this->debug && print('Using db:' . $p['db'] . "<br />\n");
			mysql_select_db($p['db'], $connect[$mode]);
		}
		// 如果出现长时间没mysql动作而引起超时，则尝试重连，重连次数为3
		if (!mysql_ping($connect[$mode])) {
			if ($count_reconnect < 3) {
				$count_reconnect ++;
				mysql_close($connect[$mode]);
				$this->debug && print('Try reconnect<br />' . "\n");
				return $this->getConnect('read', $index, true);
			} else {
				$this->debug && print("Reconnect MySQL read server error <br />\n");
				return false;
			}
		}
		//$this->last_query_connect = $connect;
		return $connect[$mode];
	}
	
	/**
	 * 设置db charset
	 * @param string $charset
	 * @param resource $link_identifier
	 */
	function _setCharset($charset, $link_identifier){
		$version = mysql_get_server_info($link_identifier);
		$charset = strtolower($charset);
		if($version > '4.1'){
			$sql = $charset ? "character_set_connection={$charset}, character_set_results={$charset}, character_set_client=binary" : '';
			$sql .= $version > '5.0.1' ? ((empty($sql) ? '' : ', ')."sql_mode=''") : '';
			$sql && mysql_query("SET {$sql}", $link_identifier);
		}
	}
	
	/**
	 * 根据SQL语句返回相应的数据库连接
	 * @param $sql string SQL语句
	 * @return resource mysql连接
	 */
	function autoConnect($sql = null) {
		$write_command = array('insert', 'update', 'delete', 'replace', 'alter', 'create', 'drop', 'rename', 'truncate');
		if ($sql !== null) {
			$sql = explode(' ', trim((string)$sql));
		}
		if ($sql === null || !in_array(strtolower($sql[0]), $write_command)) {
			return $this->getConnect('read');
		}
		return $this->getConnect();
	}

	/**
	 * 执行一SQL语句
	 * @param $sql string SQL语句
	 * @return int|false 失败则返回false,成功则返回mysql resource
	 */
	function execute($sql) {
		$sql = $this->pushSql($sql);
		$conn = $this->autoConnect($sql);
		if (!$conn) return false;
		$rs= mysql_query($sql, $conn);
		if (!$rs) {
			$this->log('query error:'. mysql_error($conn) . '. SQL:' . $sql );
		}
		// 查询后的状态
		$this->state = array(
					'sql' => $sql,
					'result' => $rs,
					'last_insert_id' => mysql_insert_id($conn),
					'affected_rows' => mysql_affected_rows($conn),
					'error' => mysql_error($conn),
					'errno' => mysql_errno($conn)
					);

		$this->debug && print_r($this->state);
		return $rs ? $rs : false;
	}
	
	/**
	 * 得到执行一SQL写操作后的影响记录行数
	 * @return int 返回受影响的记录行数
	 */
	function getAffectedRows() {
		return $this->state['affected_rows'];
	}
	
	/**
	 * 查询并返回查询结果
	 * @param $sql string SQL查询语句
	 * @return array|false 成功则返回二维数组
	 */
	function query($sql, $fetch_mode = MYSQL_ASSOC) {
		if (!$rs = $this->execute($sql)) {
			return false;
		}
		$data = array();
		while ( $row = mysql_fetch_array($rs, $fetch_mode) ) {
			$data[] = $row;
		}

		if ($this->autoFree) {
			$this->free($rs);
		}
		return $data;
	}

	/**
	 * 根据SQL查询，返回第一条记录中的一项
	 * @param $sql string SQL查询语句
	 * @param $index int 指定要返回第$index个字段的内容
	 * @return mixed|false 成功则返回指字段内容,如果找不到记录则返回一空数组
	 */
	function getOne($sql, $index=0) {
		$rs = $this->getRow($sql);
		if ($rs === false) {
			return false;
		}
		return isset($rs[$index]) ? $rs[$index] : array_shift($rs);
	}

	/**
	 * 得到一条记录
	 * @param $sql string SQL查询语句
	 * @return array|false 成功则返回第一条记录，找不到记录则返回空数组，失败返回false
	 */
	function getRow($sql) {
		if (!$rs = $this->execute($sql)){
			return false;
		}
		$rst = mysql_fetch_array($rs, MYSQL_ASSOC);
		if (!$rst) {
			return array();
		}
		if ($this->autoFree) {
			$this->free($rs);
		}
		return $rst;
	}
	
	/**
	 * 得到最后一次执行的SQL语句
	 * @return string 最后一次执行的SQL语句
	 */
	function getLastQuery() {
		return $this->last_sql;
	}

	/**
	 * @return 得到mysql错误信息
	 */
	function getError() {
		return $this->state['error'];
	}

	/**
	 * 得到mysql错误号
	 * @return int 得到mysql错误号
	 */
	function getErrorNo() {
		return $this->state['errno'];
	}
	
	/**
	 * 释放mysql查询结果
	 * @param int query handle
	 */
	function free($query_id=""){
		if ($query_id == '') {
			return;
		}
		mysql_free_result($query_id);
		$this->last_query_id = '';
    	}

	/**
	 * 清空表
	 */
	function truncate($table) {
		$sql = 'TRUNCATE ' . $this->getTable($table);
		return $this->execute($sql);
	}

	/**
	 * 关闭mysql连接
	 */
	function close() {
		$conn = $this->getConnect('read');
		$conn && mysql_close($conn);
		$conn = $this->getConnect('write');
		$conn && mysql_close($conn);
	}
	
	/**
	 * 以数组的方式返回所有执行过的的mysql语句
	 * @return array
	 */
	function getHistory() {
		return $this->querys;
	}
}

