DB类

阅读更多

/**
 * 数据库类
 *
 * Copyright(c) 2005-2008 by 陈毅鑫(深空). All rights reserved
 *
 * To contact the author write to {@link mailto:[email protected]}
 *
 * @author 陈毅鑫(深空)
 * @version $Id: DB.class.php 2018 2008-07-28 03:37:35Z skchen $
 * @package common
 */

defined('FW') || header('HTTP/1.0 400 Bad Request');

/**
 * DB 抽象类
 * 其中主要是创建了一个静态变量$db，所有集成类的对象实例化到$db中方便调用
 * 该抽象类初始化时候根据配置文件存入$db变量，并调用子类进行DB实例化，使用DB::init()进行调用
 * 本类只实现了一个静态方法，并规定了其子类必须实现的一些方法。
 *
 */
abstract class DB {
    const DB_FETCH_ASSOC    = 1;
    const DB_FETCH_ARRAY    = 3;
    const DB_FETCH_ROW      = 2;
    const DB_FETCH_DEFAULT  = self::DB_FETCH_ASSOC;

    public static $db;
    protected static $db_type = array('mysqli' => 'MySQLi', 'oracle' => 'Oracle');

    protected $u_conn;
    protected $q_conn;
    protected $dsn;
    protected $db_key;
    protected $fecth_mode;
    protected $sql;
    protected $sqls;
    protected $qrs;
    protected $urs;
    protected $u_sqls;
    protected $q_sqls;
    protected $query_num;
    protected $update_num;

    protected function __construct() {
    }

    /**
     * DB初始化
     *
     * @param array $dsn 配置文件中的DB信息
     * @param string $db_key 配置中的数据库KEY名
     * @param const $fetch_mode 返回数据的KEY类型
     * @return array|DB DB对象
     */
    public static function &init(& $dsn, $db_key, $fetch_mode = self::DB_FETCH_ASSOC) {
        $key = explode('.', $db_key);
        $key = "['" . implode("']['" , $key) . "']";
        eval('$flag = isset(self::$db' . $key . ');');
        eval("\$db_info = \$dsn" . $key . ";");
        if (!$flag) {
            $class_name = 'DB_' . self::$db_type[strtolower($db_info['db_type'])];
            $obj = new $class_name($db_info, $db_key, $fetch_mode);
            eval('self::$db' . $key . ' =& $obj;');
            unset($obj);
        }
        return self::$db;
    }

    public abstract function connect($type = "slave");
    public abstract function close();
    public abstract function query($sql, $limit = null, $quick = false);
    public abstract function update($sql);
    public abstract function getOne($sql);
    public abstract function getCol($sql, $limit = null);
    public abstract function getRow($sql, $fetch_mode = self::DB_FETCH_DEFAULT);
    public abstract function getAll($sql, $limit = null, $fetch_mode = self::DB_FETCH_DEFAULT);
}

class DB_MySQLi extends DB {
    /**
     * MySQLi构造函数
     *
     * @param array $db_info 数据库配置信息
     * @param string $db_key db的key
     * @param 返回的数据格式 $fetch_mode
     */
    public function __construct(& $db_info, $db_key, $fetch_mode) {
        $this->db_key = $db_key;
        $this->dsn =& $db_info;
        $this->fecth_mode = $fetch_mode;
    }

    /**
     * 连接数据库
     *
     * 连接数据库之前可能需要改变DSN，一般不建议使用此方法
     *
     * @param string $type 选择连接主服务器或者从服务器
     * @return boolean
     */
    public function connect($type = "slave") {
        if ($type == "master" || !isset($this->dsn["slave"])) {
            $db_host = isset($this->dsn["master"]) ? $this->dsn["master"]["db_host"] : $this->dsn["db_host"];
            $db_name = isset($this->dsn["master"]) ? $this->dsn["master"]["db_name"] : $this->dsn["db_name"];
            $db_user = isset($this->dsn["master"]) ? $this->dsn["master"]["db_user"] : $this->dsn["db_user"];
            $db_pass = isset($this->dsn["master"]) ? $this->dsn["master"]["db_pass"] : $this->dsn["db_pass"];
            $this->u_conn = mysqli_connect($db_host, $db_user, $db_pass);
            if (!$this->u_conn) {
                throw new DB_Exception('更新数据库连接失败');
            }
            if (!mysqli_select_db($this->u_conn, $db_name)) {
                throw new DB_Exception('更新数据库选择失败');
            }
            if (!isset($this->dsn["slave"])) {
                $this->q_conn =& $this->u_conn;
            }
        } else {
            if (empty($this->dsn["slave"])) {
                $this->connect('master');
                return $this->q_conn =& $this->u_conn;
            }
            if (empty($_COOKIE[COOKIE_PREFIX . $this->db_key . '_db_no'])) {
                $db_no = array_rand($this->dsn["slave"]);
                setcookie(COOKIE_PREFIX . $this->db_key . '_db_no', $db_no, null, COOKIE_PATH, COOKIE_DOMAIN);
            } else {
                $db_no = $_COOKIE[COOKIE_PREFIX . $this->db_key . '_db_no'];
            }
            $db_info = $this->dsn["slave"][$db_no];
            $db_host = $db_info["db_host"];
            $db_name = $db_info["db_name"];
            $db_user = $db_info["db_user"];
            $db_pass = $db_info["db_pass"];
            $this->q_conn = mysqli_connect($db_host, $db_user, $db_pass);

            if (!$this->q_conn) {
                if (!$this->u_conn) {
                    $this->connect('slave');
                }
                $this->q_conn =& $this->u_conn;
                if (!$this->q_conn) {
                    throw new DB_Exception('查询数据库连接失败');
                }
            } else {
                if (!mysqli_select_db($this->q_conn, $db_name)) {
                    throw new DB_Exception('查询数据库选择失败');
                }
            }
        }
        return true;
    }

    /**
     * 关闭数据库连接
     *
     * 一般不需要调用此方法
     */
    public function close() {
        if ($this->u_conn === $this->q_conn) {
            if (is_object($this->u_conn)) {
                mysqli_close($this->u_conn);
            }
        } else {
            if (is_object($this->u_conn)) {
                mysqli_close($this->u_conn);
            }
            if (is_object($this->q_conn)) {
                mysqli_close($this->q_conn);
            }
        }
    }

    /**
     * 执行一个SQL查询
     *
     * 本函数仅限于执行SELECT类型的SQL语句
     *
     * @param string $sql SQL查询语句
     * @param mixed $limit 整型或者字符串类型，如10|10,10
     * @param boolean $quick 是否快速查询
     * @return resource 返回查询结果资源句柄
     */
    public function query($sql, $limit = null, $quick = false) {
        if ($limit != null) {
            $sql = $sql . " LIMIT " . $limit;
        }
        $this->sqls[] = $sql;
        $this->q_sqls[] = $sql;
        $this->sql = $sql;

        if (!$this->q_conn) {
            $this->connect("slave");
        }

        $this->qrs = mysqli_query($this->q_conn, $sql, $quick ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT);
        if (!$this->qrs) {
            throw new DB_Exception('查询失败:' . mysqli_error($this->q_conn));
        } else {
            $this->query_num++;
            return $this->qrs;
        }
    }

    /**
     * 获取结果集
     *
     * @param resource $rs 查询结果资源句柄
     * @param const $fetch_mode 返回的数据格式
     * @return array 返回数据集每一行，并将$rs指针下移
     */
    public function fetch($rs, $fetch_mode = self::DB_FETCH_DEFAULT) {
        switch ($fetch_mode) {
            case 1:
                $fetch_mode = self::DB_FETCH_ASSOC;
                break;
            case 2:
                $fetch_mode = self::DB_FETCH_ROW;
                break;
            case 3:
                $fetch_mode = self::DB_FETCH_ARRAY;
                break;
            default:
                $fetch_mode = self::DB_FETCH_DEFAULT;
                break;
        }
        return mysqli_fetch_array($rs, $fetch_mode);
    }

    /**
     * 执行一个SQL更新
     *
     * 本方法仅限数据库UPDATE操作
     *
     * @param string $sql 数据库更新SQL语句
     * @return boolean
     */
    public function update($sql) {
        $this->sql = $sql;
        $this->sqls[] = $this->sql;
        $this->u_sqls[] = $this->sql;
        if (!$this->u_conn) {
            $this->connect("master");
        }

        $this->urs = mysqli_query($this->u_conn, $sql);

        if (!$this->urs) {
            throw new DB_Exception('更新失败:' . mysqli_error($this->u_conn));
        } else {
            $this->update_num++;
            return $this->urs;
        }
    }

    /**
     * 返回SQL语句执行结果集中的第一行第一列数据
     *
     * @param string $sql 需要执行的SQL语句
     * @return mixed 查询结果
     */
    public function getOne($sql) {
        if (!$rs = $this->query($sql, 1, true)) {
            return false;
        }
        $row = $this->fetch($rs, self::DB_FETCH_ROW);
        $this->free($rs);
        return $row[0];
    }

    /**
     * 返回SQL语句执行结果集中的第一列数据
     *
     * @param string $sql 需要执行的SQL语句
     * @param mixed $limit 整型或者字符串类型，如10|10,10
     * @return array 结果集数组
     */
    public function getCol($sql, $limit = null) {
        if (!$rs = $this->query($sql, $limit, true)) {
            return false;
        }
        $result = array();
        while ($rows = $this->fetch($rs, self::DB_FETCH_ROW)) {
            $result[] = $rows[0];
        }
        $this->free($rs);
        return $result;
    }

    /**
     * 返回SQL语句执行结果中的第一行数据
     *
     * @param string $sql 需要执行的SQL语句
     * @param const $fetch_mode 返回