/nova/core/database/mysql.class.php
PHP | 393 lines | 195 code | 18 blank | 180 comment | 41 complexity | 4ce5555f11ff0bed6c59dc4415b265ac MD5 | raw file
Possible License(s): AGPL-1.0
- <?php
- class Mysql extends Database {
- /**
- +----------------------------------------------------------
- * ???? ?????????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @param array $config ???????
- +----------------------------------------------------------
- */
- public function __construct($config=''){
- if ( !extension_loaded('mysql') ) {
- throw_exception(L('_NOT_SUPPERT_').':mysql');
- }
- if(!empty($config)) {
- $this->config = $config;
- }
- }
- /**
- +----------------------------------------------------------
- * ???????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function connect($config='',$linkNum=0) {
- if ( !isset($this->linkID[$linkNum]) ) {
- if(empty($config)) $config = $this->config;
- // ????????socket????
- $host = $config['hostname'].($config['hostport']?":{$config['hostport']}":'');
- if($this->pconnect) {
- $this->linkID[$linkNum] = mysql_pconnect( $host, $config['username'], $config['password'],CLIENT_MULTI_RESULTS);
- }else{
- $this->linkID[$linkNum] = mysql_connect( $host, $config['username'], $config['password'],true,CLIENT_MULTI_RESULTS);
- }
- if ( !$this->linkID[$linkNum] || (!empty($config['database']) && !mysql_select_db($config['database'], $this->linkID[$linkNum])) ) {
- throw_exception(mysql_error());
- }
- $dbVersion = mysql_get_server_info($this->linkID[$linkNum]);
- if ($dbVersion >= "4.1") {
- //??UTF8????? ??mysql 4.1.0????
- mysql_query("SET NAMES '".C('DB_CHARSET')."'", $this->linkID[$linkNum]);
- }
- //?? sql_model
- if($dbVersion >'5.0.1'){
- mysql_query("SET sql_mode=''",$this->linkID[$linkNum]);
- }
- // ??????
- $this->connected = true;
- // ???????????
- if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
- }
- return $this->linkID[$linkNum];
- }
- /**
- +----------------------------------------------------------
- * ??????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- */
- public function free() {
- @mysql_free_result($this->queryID);
- $this->queryID = 0;
- }
- /**
- +----------------------------------------------------------
- * ???? ?????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @param string $str sql??
- +----------------------------------------------------------
- * @return mixed
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function query($str) {
- $this->initConnect(false);
- if ( !$this->_linkID ) return false;
- $this->queryStr = $str;
- //?????????
- if ( $this->queryID ) { $this->free(); }
- $this->Q(1);
- $this->queryID = mysql_query($str, $this->_linkID);
- $this->debug();
- if ( false === $this->queryID ) {
- $this->error();
- return false;
- } else {
- $this->numRows = mysql_num_rows($this->queryID);
- return $this->getAll();
- }
- }
- /**
- +----------------------------------------------------------
- * ????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @param string $str sql??
- +----------------------------------------------------------
- * @return integer
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function execute($str) {
- $this->initConnect(true);
- if ( !$this->_linkID ) return false;
- $this->queryStr = $str;
- //?????????
- if ( $this->queryID ) { $this->free(); }
- $this->W(1);
- $result = mysql_query($str, $this->_linkID) ;
- $this->debug();
- if ( false === $result) {
- $this->error();
- return false;
- } else {
- $this->numRows = mysql_affected_rows($this->_linkID);
- $this->lastInsID = mysql_insert_id($this->_linkID);
- return $this->numRows;
- }
- }
- /**
- +----------------------------------------------------------
- * ????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @return void
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function startTrans() {
- $this->initConnect(true);
- if ( !$this->_linkID ) return false;
- //??rollback ??
- if ($this->transTimes == 0) {
- mysql_query('START TRANSACTION', $this->_linkID);
- }
- $this->transTimes++;
- return ;
- }
- /**
- +----------------------------------------------------------
- * ????????????????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @return boolen
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function commit()
- {
- if ($this->transTimes > 0) {
- $result = mysql_query('COMMIT', $this->_linkID);
- $this->transTimes = 0;
- if(!$result){
- throw_exception($this->error());
- }
- }
- return true;
- }
- /**
- +----------------------------------------------------------
- * ????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @return boolen
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function rollback()
- {
- if ($this->transTimes > 0) {
- $result = mysql_query('ROLLBACK', $this->_linkID);
- $this->transTimes = 0;
- if(!$result){
- throw_exception($this->error());
- }
- }
- return true;
- }
- /**
- +----------------------------------------------------------
- * ?????????
- +----------------------------------------------------------
- * @access private
- +----------------------------------------------------------
- * @return array
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- private function getAll() {
- //?????
- $result = array();
- if($this->numRows >0) {
- while($row = mysql_fetch_assoc($this->queryID)){
- $result[] = $row;
- }
- mysql_data_seek($this->queryID,0);
- }
- return $result;
- }
- /**
- +----------------------------------------------------------
- * ??????????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- */
- public function getFields($tableName) {
- $result = $this->query('SHOW COLUMNS FROM '.$tableName);
- $info = array();
- if($result) {
- foreach ($result as $key => $val) {
- $info[$val['Field']] = array(
- 'name' => $val['Field'],
- 'type' => $val['Type'],
- 'notnull' => (bool) ($val['Null'] === ''), // not null is empty, null is yes
- 'default' => $val['Default'],
- 'primary' => (strtolower($val['Key']) == 'pri'),
- 'autoinc' => (strtolower($val['Extra']) == 'auto_increment'),
- );
- }
- }
- return $info;
- }
- /**
- +----------------------------------------------------------
- * ?????????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- */
- public function getTables($dbName='') {
- if(!empty($dbName)) {
- $sql = 'SHOW TABLES FROM '.$dbName;
- }else{
- $sql = 'SHOW TABLES ';
- }
- $result = $this->query($sql);
- $info = array();
- foreach ($result as $key => $val) {
- $info[$key] = current($val);
- }
- return $info;
- }
- /**
- +----------------------------------------------------------
- * ????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @param mixed $data ??
- * @param array $options ?????
- +----------------------------------------------------------
- * @return false | integer
- +----------------------------------------------------------
- */
- public function replace($data,$options=array()) {
- foreach ($data as $key=>$val){
- $value = $this->parseValue($val);
- if(is_scalar($value)) { // ???????
- $values[] = $value;
- $fields[] = $this->addSpecialChar($key);
- }
- }
- $sql = 'REPLACE INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')';
- return $this->execute($sql);
- }
- /**
- +----------------------------------------------------------
- * ????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @param mixed $datas ??
- * @param array $options ?????
- +----------------------------------------------------------
- * @return false | integer
- +----------------------------------------------------------
- */
- public function insertAll($datas,$options=array()) {
- if(!is_array($datas[0])) return false;
- $fields = array_keys($datas[0]);
- array_walk($fields, array($this, 'addSpecialChar'));
- $values = array();
- foreach ($datas as $data){
- $value = array();
- foreach ($data as $key=>$val){
- $val = $this->parseValue($val);
- if(is_scalar($val)) { // ???????
- $value[] = $val;
- }
- }
- $values[] = '('.implode(',', $value).')';
- }
- $sql = 'INSERT INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES '.implode(',',$values);
- return $this->execute($sql);
- }
- /**
- +----------------------------------------------------------
- * ?????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @throws ThinkExecption
- +----------------------------------------------------------
- */
- public function close() {
- if (!empty($this->queryID))
- mysql_free_result($this->queryID);
- if ($this->_linkID && !mysql_close($this->_linkID)){
- throw_exception($this->error());
- }
- $this->_linkID = 0;
- }
- /**
- +----------------------------------------------------------
- * ???????
- * ??????SQL??
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @return string
- +----------------------------------------------------------
- */
- public function error() {
- $this->error = mysql_error($this->_linkID);
- if($this->debug && '' != $this->queryStr){
- $this->error .= "\n [ SQL?? ] : ".$this->queryStr;
- }
- return $this->error;
- }
- /**
- +----------------------------------------------------------
- * SQL??????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- * @param string $str SQL???
- +----------------------------------------------------------
- * @return string
- +----------------------------------------------------------
- */
- public function escape_string($str) {
- return mysql_escape_string($str);
- }
- /**
- +----------------------------------------------------------
- * ????
- +----------------------------------------------------------
- * @access public
- +----------------------------------------------------------
- */
- public function __destruct()
- {
- // ????
- $this->close();
- }
- }//?????
- ?>