www.gusucode.com > baigo SSO单点登录系统 v3.0源码程序 > baigosso_v3.0/core/class/database.class.php

    <?php
/*-----------------------------------------------------------------
！！！！警告！！！！
以下为系统文件，请勿修改
-----------------------------------------------------------------*/

//不能非法包含或直接执行
if (!defined('IN_BAIGO')) {
    exit('Access Denied');
}

/*-------------数据库类-------------*/
class CLASS_DATABASE {

    public $db_host     = 'localhost';
    public $db_port     = '3306';
    public $db_user     = 'root';
    public $db_pass     = '';
    public $db_name     = '';
    public $db_charset  = 'utf8';
    public $db_debug    = false;
    public $obj_mysqli;
    public $obj_result;

    function __construct($cfg_host) {
        if (isset($cfg_host['host'])) {
            $this->db_host      = $cfg_host['host'];
        }

        if (isset($cfg_host['port'])) {
            $this->db_port      = $cfg_host['port'];
        }

        if (isset($cfg_host['user'])) {
            $this->db_user      = $cfg_host['user'];
        }

        if (isset($cfg_host['pass'])) {
            $this->db_pass      = $cfg_host['pass'];
        }

        if (isset($cfg_host['name'])) {
            $this->db_name      = $cfg_host['name'];
        }

        if (isset($cfg_host['charset'])) {
            $this->db_charset   = $cfg_host['charset'];
        }

        if (isset($cfg_host['debug'])) {
            $this->db_debug     = $cfg_host['debug'];
        }

        $this->connect();
        $this->select_db();
    }

    function connect() {
        $this->obj_mysqli = new mysqli($this->db_host, $this->db_user, $this->db_pass, $this->db_name, $this->db_port);
        if ($this->obj_mysqli->connect_errno > 0) {
            if ($this->db_debug) {
                $_str_msg   = $this->obj_mysqli->connect_error;
                $_str_msg  .= ', Error No: ' . $this->obj_mysqli->connect_errno;
            } else {
                $_str_msg   = 'Can not connect to database';
            }
            exit('{"rcode":"x","msg":"Fatal Error: ' . $_str_msg . '!"}');
        } else {
            $this->obj_mysqli->query('SET NAMES ' . $this->db_charset);
        }
    }

    function select_db() {
        if (!$this->obj_mysqli->select_db($this->db_name)) {
            if ($this->db_debug) {
                $_str_msg   = $this->obj_mysqli->error;
                $_str_msg  .= ', Error No: ' . $this->obj_mysqli->errno;
            } else {
                $_str_msg   = 'Can not select database';
            }
            exit('{"rcode":"x","msg":"Fatal Error: ' . $_str_msg . '!"}');
        }
    }

    function query($sql) {
        $query = $this->obj_mysqli->query($sql);
        if ($query) {
            $this->obj_result = $query;
            return $query;
        } else {
            return false;
        }
    }

    function fetch_row($rs = '') {
        $rs = $rs ? $rs : $this->obj_result;
        return $rs->fetch_row();
    }

    function fetch_assoc($rs = '') {
        $rs = $rs ? $rs : $this->obj_result;
        return $rs->fetch_assoc();
    }

    function affected_rows() {
        return $this->obj_mysqli->affected_rows;
    }

    function insert_id() {
        $_obj_temp    = $this->query('SELECT LAST_INSERT_ID()');
        $this->obj_result  = $this->fetch_row($_obj_temp);
        return $this->obj_result[0];
    }

    function close() {
        $this->obj_mysqli->close();
    }

    function create_table($table, $data, $primary, $comment = '', $engine = 'InnoDB') {
        $sql      = 'CREATE TABLE IF NOT EXISTS `' . $table . '` (';
        $values   = array();
        foreach ($data as $key => $value) {
            $values[] = '`' . $key . '` ' . $value;
        }
        $sql         .= implode(',', $values);
        $sql         .= ', PRIMARY KEY (`' . $primary . '`)) ENGINE=' . $engine . '  DEFAULT CHARSET=' . $this->db_charset . ' COMMENT=\'' . $comment . '\' AUTO_INCREMENT=1';
        $this->obj_result  = $this->query($sql);
        return $this->obj_result;
    }

    function create_index($index, $table, $data, $type = 'BTREE', $drop_overlap = false) {
        if ($drop_overlap) {
            $sql = 'DROP INDEX `' . $index . '` ON `' . $table . '`';
            $this->query($sql);
        }

        $sql      = 'CREATE INDEX `' . $index . '` ON `' . $table . '` (';
        $values   = array();
        foreach ($data as $key=>$value) {
            $values[] = '`' . $value . '` ';
        }
        $sql         .= implode(',', $values);
        $sql         .= ') USING ' . $type;
        $this->obj_result  = $this->query($sql);
        return $this->obj_result;
    }

    function create_view($view, $data, $table, $join) {
        $sql      = 'CREATE OR REPLACE SQL SECURITY INVOKER VIEW `' . $view . '` AS SELECT ';
        $values   = array();
        foreach ($data as $key=>$value) {
            $_str_view = '`' . $value[1] . '`.`' . $value[0] . '`';
            if (isset($value[2])) {
                $_str_view .= ' AS `' . $value[2] . '`';
            } else {
                $_str_view .= ' AS `' . $value[0] . '`';
            }
            $values[] = $_str_view;
        }
        $sql         .= implode(',', $values);
        $sql         .= ' FROM `' . $table . '` ' . $join;
        //print_r($sql);
        $this->obj_result  = $this->query($sql);
        return $this->obj_result;
    }

    function copy_table($table, $table_src, $data, $primary, $comment = '', $engine = 'InnoDB') {
        $sql  = 'CREATE TABLE IF NOT EXISTS `' . $table . '` (';

        $values   = array();
        foreach ($data as $key => $value) {
            $values[] = '`' . $key . '` ' . $value;
        }
        $sql     .= implode(',', $values);

        $sql     .= ', PRIMARY KEY (`' . $primary . '`)) ENGINE=' . $engine . ' DEFAULT CHARSET=' . $this->db_charset . ' COMMENT=\'' . $comment . '\' ';
        $sql     .= ' SELECT ';

        $values   = array();
        foreach ($data as $key => $value) {
            $values[] = '`' . $key . '`';
        }
        $sql .= implode(',', $values);

        $sql .= ' FROM `' . $table_src . '` WHERE 1';

        $this->obj_result  = $this->query($sql);
        return $this->obj_result;
    }

    function alter_table($table, $data = false, $rename = false) {
        $sql = 'ALTER TABLE `' . $table . '` ';
        if (!fn_isEmpty($rename)) {
            $sql .= ' RENAME TO `' . $rename . '`';
        }
        if (!fn_isEmpty($data)) {
            $values  = array();
            foreach ($data as $key => $value) {
                switch ($value[0]) {
                    case 'ADD':
                        $values[] = $value[0] . ' COLUMN `' . $key . '` ' . $value[1];
                    break;
                    case 'DROP':
                        $values[] = $value[0] . ' COLUMN `' . $key . '`';
                    break;
                    case 'DROP PRIMARY KEY':
                        $values[] = $value[0];
                    break;
                    case 'ADD PRIMARY KEY':
                        $values[] = $value[0] . ' (`' . $value[1] . '`)';
                    break;
                    case 'CHANGE':
                        $values[] = $value[0] . ' COLUMN `' . $key . '` `' . $value[2] . '` ' . $value[1];
                    break;
                }
            }
            $sql .= implode(',', $values);
        }
        $this->obj_result = $this->query($sql);
        return $this->obj_result;
    }

    function duplicate($table_dst, $data_dst, $table_src, $data_src, $where = '') {
        $sql    = 'INSERT INTO `' . $table_dst . '` ';
        $sql   .= '(`' . implode('`,`', $data_dst) . '`) SELECT ';
        $sql   .= '`' . implode('`,`', $data_src) . '` FROM `' . $table_src . '`';
        if (!fn_isEmpty($where)) {
            $sql .= ' WHERE ' . $where;
        }

        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        return $this->insert_id();
    }

    function insert($table, $data, $field = false) {
        $sql      = 'INSERT INTO `' . $table . '` SET ';
        $values   = array();
        foreach ($data as $key => $value) {
            if ($field) {
                $values[] = '`' . $key . '`=' . $value;
            } else {
                $values[] = '`' . $key . '`=\'' . $value . '\'';
            }
        }
        $sql         .= implode(',', $values);
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        return $this->insert_id();
    }

    function update($table, $data, $where = '', $field = false) {
        $sql      = 'UPDATE `' . $table . '` SET ';
        $values   = array();
        foreach ($data as $key => $value) {
            if ($field) {
                $values[] = '`' . $key . '`=' . $value;
            } else {
                $values[] = '`' . $key . '`=\'' . $value . '\'';
            }
        }
        $sql .= implode(',', $values);
        if (!fn_isEmpty($where)) {
            $sql .= ' WHERE ' . $where;
        }
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        return $this->affected_rows();
    }

    function delete($table, $where) {
        $sql          = 'DELETE FROM `' . $table . '` WHERE ' . $where;
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        return $this->affected_rows();
    }

    function count($table, $where = '', $distinct = '') {
        $sql = 'SELECT';
        if (fn_isEmpty($distinct)) {
            $sql .= ' COUNT(*)';
        } else {
            $sql .= ' COUNT(DISTINCT `' . implode('`,`', $distinct) . '`)';
        }
        $sql .= ' FROM `' . $table . '`';
        if (!fn_isEmpty($where)) {
            $sql .= ' WHERE ' . $where;
        }
        //print_r($sql);
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        $obj_temp     = $this->fetch_row($this->obj_result);
        return $obj_temp[0];
    }

    function select($table, $data = '', $where = '', $group = '', $order = '', $length = 0, $start = 0, $field = false) {
        $sql = 'SELECT';
        if (fn_isEmpty($data)) {
            $sql .= ' *';
        } else {
            if ($field) {
                $sql .= ' ' . implode(',', $data);
            } else {
                $sql .= ' `' . implode('`,`', $data) . '`';
            }
        }
        $sql .= ' FROM `' . $table . '`';
        if (!fn_isEmpty($where)) {
            $sql .= ' WHERE ' . $where;
        }
        if (!fn_isEmpty($group)) {
            $sql .= ' GROUP BY `' . implode('`,`', $group) . '`';
        }
        if (!fn_isEmpty($order)) {
            $sql_order = ' ORDER BY ';
            foreach ($order as $_key=>$_value) {
                $sql_order .= '`' . $_value[0] . '` ' . $_value[1] . ',';
            }
            $sql_order = rtrim($sql_order, ',');
            $sql .= $sql_order;
        }
        if ($length > 0) {
            $sql .= ' LIMIT ' . $start . ', ' . $length;
        }

        //print_r($sql);
        /*if ($field) {
        print_r($sql . '\n');
        }*/

        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        $obj = array();

        while ($obj_temp = $this->fetch_assoc($this->obj_result)) {
            $obj[] = $obj_temp;
            unset($obj_temp);
        }
        return $obj;
    }

    function show_columns($table) {
        $sql = 'SHOW COLUMNS FROM `' . $table . '`';
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        $obj = array();

        while ($obj_temp = $this->fetch_assoc($this->obj_result)) {
            $obj[] = $obj_temp;
            unset($obj_temp);
        }
        return $obj;
    }


    function show_indexs($table) {
        $sql = 'SHOW INDEX FROM `' . $table . '`';
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        $obj          = array();

        while ($obj_temp = $this->fetch_assoc($this->obj_result)) {
            $obj[] = $obj_temp;
            unset($obj_temp);
        }
        return $obj;
    }


    function show_tables() {
        $sql = 'SHOW TABLES FROM `' . $this->db_name . '`';
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        $obj          = array();

        while ($obj_temp = $this->fetch_assoc($this->obj_result)) {
            $obj[] = $obj_temp;
            unset($obj_temp);
        }
        return $obj;
    }


    function show_databases() {
        $sql = 'SHOW DATABASES';
        $this->obj_result  = $this->query($sql);
        if (!$this->obj_result) {
            return false;
        }
        $obj          = array();

        while ($obj_temp = $this->fetch_assoc($this->obj_result)) {
            $obj[] = $obj_temp;
            unset($obj_temp);
        }
        return $obj;
    }


    function __destruct() {
        if ($this->obj_mysqli) {
            //$this->close();
            //unset($this->obj_mysqli);
        }
    }
}