Example #1
0
 /**
  * 连接数据库,返回连接上的PDO对象
  * @param int $DBName    数据库名称
  * @param string $master_or_slave   master;主库|slave:从库
  * @return master db handle;
  */
 public static function connectDB($DBName, $master_or_slave = 'slave', $DBConfig = array(), $reConnect = false)
 {
     $master_or_slave === 'master' || ($master_or_slave = 'slave');
     if ($master_or_slave === 'master' && !empty($_SERVER['HTTP_HOST']) && isset($_SERVER['REQUEST_METHOD']) && strtoupper($_SERVER['REQUEST_METHOD']) !== 'POST' && BaseModelSwitch::check(BaseModelSwitch::SWITCH_MASTERDB_POST_ONLY) === true) {
         throw new BaseModelDBException('请求方法不允许,主库操作必须为post', 90300);
     }
     $DBType = DAGGER_ENV;
     in_array($DBType, array('dev', 'test', 'product'), true) || ($DBType = 'product');
     empty($DBName) && ($DBName = DAGGER_DB_DEFAULT);
     if (empty(DBConfig::$config['mysql'][$DBName][$DBType]['master']['host'])) {
         $DBType = 'product';
     }
     $username = empty($DBConfig[$master_or_slave]['user']) ? DBConfig::$config['mysql'][$DBName][$DBType][$master_or_slave]['user'] : $DBConfig[$master_or_slave]['user'];
     $password = empty($DBConfig[$master_or_slave]['pass']) ? DBConfig::$config['mysql'][$DBName][$DBType][$master_or_slave]['pass'] : $DBConfig[$master_or_slave]['pass'];
     $hostspec = empty($DBConfig[$master_or_slave]['host']) ? DBConfig::$config['mysql'][$DBName][$DBType][$master_or_slave]['host'] : $DBConfig[$master_or_slave]['host'];
     $port = !isset($DBConfig[$master_or_slave]['port']) || !is_numeric($DBConfig[$master_or_slave]['port']) ? DBConfig::$config['mysql'][$DBName][$DBType][$master_or_slave]['port'] : $DBConfig[$master_or_slave]['port'];
     $database = empty($DBConfig[$master_or_slave]['database']) ? DBConfig::$config['mysql'][$DBName][$DBType][$master_or_slave]['database'] : $DBConfig[$master_or_slave]['database'];
     $charset = empty($DBConfig['charset']) ? DBConfig::$config['mysql'][$DBName][$DBType][$master_or_slave]['charset'] : strtolower($DBConfig['charset']);
     $db_key = md5(implode('-', array($hostspec, $port, $username, $database, $charset)));
     self::$linkConfig = array('host' => $hostspec, 'port' => $port, 'db' => $database, 'charset' => $charset);
     if (isset(self::$links[$db_key]) && !$reConnect) {
         return self::$links[$db_key];
     }
     // self::$links[$db_key] = new PDO($dsn, $username, $password);
     $dsn = "mysql:dbname={$database};port={$port};host={$hostspec}";
     $connectType = $reConnect ? 'db_reconnect' : 'db_connect';
     defined('DAGGER_DEBUG') && BaseModelCommon::debug($dsn . "|username:{$username}|pw:***", $connectType);
     $mysqli = mysqli_init();
     $mysqli->options(MYSQLI_OPT_CONNECT_TIMEOUT, 4);
     if ($mysqli->real_connect($hostspec, $username, $password, $database, $port)) {
         $mysqli->set_charset($charset);
         defined('DAGGER_DEBUG') && BaseModelCommon::debug($charset, "db_set_charset");
         self::$links[$db_key] = $mysqli;
         return self::$links[$db_key];
     } else {
         return false;
     }
 }