<?php
return array(
'passport' => array(
'write' => array(
'class' => 'CDbConnection',
'connectionString' => 'mysql:host=10.1.39.2;dbname=db1′,
'emulatePrepare' => true,
//'enableParamLogging' => true,
'enableProfiling' => true,
'username' => 'root',
'password' => '',
'charset' => 'utf8′,
'schemaCachingDuration'=>3600,
),
'read' => array(
array(
'class' => 'CDbConnection',
'connectionString' => 'mysql:host=10.1.39.3;dbname=db1,
'emulatePrepare' => true,
//'enableParamLogging' => true,
'enableProfiling' => true,
'username' => 'root',
'password' => '',
'charset' => 'utf8′,
'schemaCachingDuration'=>3600,
),
array(
'class' => 'CDbConnection',
'connectionString' => 'mysql:host=10.1.39.4;dbname=db3′,
'emulatePrepare' => true,
//'enableParamLogging' => true,
'enableProfiling' => true,
'username' => 'root',
'password' => '',
'charset' => 'utf8′,
'schemaCachingDuration'=>3600,
),
),
),
);

<?php
return array(
//key為數(shù)據(jù)庫(kù)名稱(chēng)，value為Model
'passport' => array('User','Post'),
'microblog' => array('…'),
);
?>

/**
* 基于CActiveRecord類(lèi)的封裝，實(shí)現(xiàn)多庫(kù)和主從讀寫(xiě)分離
* 所有Model都必須繼承些類(lèi).
*
*/
class ActiveRecord extends CActiveRecord
{
//model配置
public $modelConfig = '';
//數(shù)據(jù)庫(kù)配置
public $dbConfig = '';
//定義一個(gè)多數(shù)據(jù)庫(kù)集合
static $dataBase = array();
//當(dāng)前數(shù)據(jù)庫(kù)名稱(chēng)
public $dbName = '';
//定義庫(kù)類(lèi)型(讀或?qū)?
public $dbType = 'read'; //'read' or 'write'
/**
* 在原有基礎(chǔ)上添加了一個(gè)dbname參數(shù)
* @param string $scenario Model的應(yīng)用場(chǎng)景
* @param string $dbname 數(shù)據(jù)庫(kù)名稱(chēng)
*/
public function __construct($scenario='insert', $dbname = '')
{
if (!empty($dbname))
$this->dbName = $dbname;
parent::__construct($scenario);
}
/**
* 重寫(xiě)父類(lèi)的getDbConnection方法
* 多庫(kù)和主從都在這里切換
*/
public function getDbConnection()
{
//如果指定的數(shù)據(jù)庫(kù)對(duì)象存在則直接返回
if (self::$dataBase[$this->dbName]!==null)
return self::$dataBase[$this->dbName];
if ($this->dbName == 'db'){
self::$dataBase[$this->dbName] = Yii::app()->getDb();
}else{
$this->changeConn($this->dbType);
}
if(self::$dataBase[$this->dbName] instanceof CDbConnection){
self::$dataBase[$this->dbName]->setActive(true);
return self::$dataBase[$this->dbName];
} else
throw new CDbException(Yii::t('yii','Model requires a "db" CDbConnection application component.'));
}
/**
* 獲取配置文件
* @param unknown_type $type
* @param unknown_type $key
*/
private function getConfig($type="modelConfig",$key="){
$config = Yii::app()->params[$type];
if($key)
$config = $config[$key];
return $config;
}
/**
* 獲取數(shù)據(jù)庫(kù)名稱(chēng)
*/
private function getDbName(){
if($this->dbName)
return $this->dbName;
$modelName = get_class($this->model());
$this->modelConfig = $this->getConfig('modelConfig');
//獲取model所對(duì)應(yīng)的數(shù)據(jù)庫(kù)名
if($this->modelConfig)foreach($this->modelConfig as $key=>$val){
if(in_array($modelName,$val)){
$dbName = $key;
break;
}
}
return $dbName;
}
/**
* 切換數(shù)據(jù)庫(kù)連接
* @param unknown_type $dbtype
*/
protected function changeConn($dbtype = 'read'){
if($this->dbType == $dbtype && self::$dataBase[$this->dbName] !== null)
return self::$dataBase[$this->dbName];
$this->dbName = $this->getDbName();
if(Yii::app()->getComponent($this->dbName.'_'.$dbtype) !== null){
self::$dataBase[$this->dbName] = Yii::app()->getComponent($this->dbName.'_'.$dbtype);
return self::$dataBase[$this->dbName];
}
$this->dbConfig = $this->getConfig('dbConfig',$this->dbName);
//跟據(jù)類(lèi)型取對(duì)應(yīng)的配置(從庫(kù)是隨機(jī)值)
if($dbtype == 'write'){
$config = $this->dbConfig[$dbtype];
}else{
$slavekey = array_rand($this->dbConfig[$dbtype]);
$config = $this->dbConfig[$dbtype][$slavekey];
}
//將數(shù)據(jù)庫(kù)配置加到component中
if($dbComponent = Yii::createComponent($config)){
Yii::app()->setComponent($this->dbName.'_'.$dbtype,$dbComponent);
self::$dataBase[$this->dbName] = Yii::app()->getComponent($this->dbName.'_'.$dbtype);
$this->dbType = $dbtype;
return self::$dataBase[$this->dbName];
} else
throw new CDbException(Yii::t('yii','Model requires a "changeConn" CDbConnection application component.'));
}
/**
* 保存數(shù)據(jù)前選擇 主 數(shù)據(jù)庫(kù)
*/
protected function beforeSave(){
parent::beforeSave();
$this->changeConn('write');
return true;
}
/**
* 刪除數(shù)據(jù)前選擇 主 數(shù)據(jù)庫(kù)
*/
protected function beforeDelete(){
parent::beforeDelete();
$this->changeConn('write');
return true;
}
/**
* 讀取數(shù)據(jù)選擇 從 數(shù)據(jù)庫(kù)
*/
protected function beforeFind(){
parent::beforeFind();
$this->changeConn('read');
return true;
}
/**
* 獲取master庫(kù)對(duì)象
*/
public function dbWrite(){
return $this->changeConn('write');
}
/**
* 獲取slave庫(kù)對(duì)象
*/
public function dbRead(){
return $this->changeConn('read');
}
}