版本說明：
- thinkphp3.2.3
新增自定義行為類
- 文件位置：Application/Common/Behaviors/SwitchDbBehavior.class.php
- 文件內(nèi)容：
namespace Common\Behaviors;
class SwitchDbBehavior
{
//私有庫id,如何連接公有庫則設(shè)置為share字符串
private $_privateId = '';
/*
* 行為擴展的執(zhí)行入口必須是run
* $param 為引用傳值忘巧，所以實參必須是變量
*/
public function run(&$params)
{
//獲取私有庫id
$this->_privateId = empty($params) && session('?privateId') ? (int)session('privateId') : trim($params);
//echo 'curent database is '.$this->_privateId;
//默認(rèn)連接公有庫
if(empty($this->_privateId)) $this->_privateId = 'share';
$this->_checkDb();
//連接公有庫
if('share' == trim($this->_privateId))
{
$share = $this->_connectShare();
}else
//連接私有庫
{
$share = $this->_connectPrivate($this->_privateId);
}
//循環(huán)修改數(shù)據(jù)庫配置信息
foreach($share as $dbKey=>$dbVal)
{
C($dbKey,$dbVal);
}
}
/*
* 判斷數(shù)據(jù)庫是否存在
*/
private function _checkDb()
{
//C('PRIVATEIDS') 在配置文件Appliation/Common/Conf/dbname.php中定義
if(!in_array($this->_privateId,C('PRIVATEIDS')))
{
exit(__CLASS__.'->'.__FUNCTION__.': dbName error!');
}
}
/*
* 返回連接私有庫配置
*/
private function _connectPrivate($privateId)
{
return array(
'DB_TYPE' => 'mysql', // 數(shù)據(jù)庫類型
'DB_HOST' => DB_WECHAT_HOST, // 服務(wù)器地址
'DB_NAME' => DB_WECHAT_NAME.(int)$privateId, // 數(shù)據(jù)庫名
'DB_USER' => DB_WECHAT_USER, // 用戶名
'DB_PWD' => DB_WECHAT_PASS, // 密碼
'DB_PORT' => DB_WECHAT_PORT, // 端口
'DB_PARAMS' => array(), // 數(shù)據(jù)庫連接參數(shù)
'DB_PREFIX' => '', // 數(shù)據(jù)庫表前綴
'DB_CHARSET'=> 'utf8', // 字符集
'DB_DEBUG' => DB_WECHAT_DEBUG, // 數(shù)據(jù)庫調(diào)試模式 開啟后可以記錄SQL日志
);
}
/*
* 返回連接公有庫配置
*/
private function _connectShare()
{
return array(
'DB_TYPE' => 'mysql', // 數(shù)據(jù)庫類型
'DB_HOST' => DB_WECHAT_HOST, // 服務(wù)器地址
'DB_NAME' => DB_WECHAT_SHARE_NAME, // 數(shù)據(jù)庫名
'DB_USER' => DB_WECHAT_USER, // 用戶名
'DB_PWD' => DB_WECHAT_PASS, // 密碼
'DB_PORT' => DB_WECHAT_PORT, // 端口
'DB_PARAMS' => array(), // 數(shù)據(jù)庫連接參數(shù)
'DB_PREFIX' => '', // 數(shù)據(jù)庫表前綴
'DB_CHARSET'=> 'utf8', // 字符集
'DB_DEBUG' => DB_WECHAT_DEBUG, // 數(shù)據(jù)庫調(diào)試模式 開啟后可以記錄SQL日志
);
}
}
- 以上文件中用到的常量在我們的配置文件Application/Common/Conf/constant.php中． 內(nèi)容如下：
/*數(shù)據(jù)庫配置*/
define('DB_WECHAT_HOST','127.0.0.1'); //主機host
define('DB_WECHAT_USER','common'); //用戶名
define('DB_WECHAT_PASS','common'); //密碼
define('DB_WECHAT_NAME','wechat_'); //私有庫前綴
define('DB_WECHAT_SHARE_NAME','wechat_share'); //共有庫名
define('DB_WECHAT_PORT','3306'); //端口
define('DB_WECHAT_DEBUG',TRUE); //數(shù)據(jù)庫調(diào)試模式 開啟后可以記錄SQL日志
- 圖中代碼驗證數(shù)據(jù)庫存在不存在的C('PRIVATEIDS')在文件Application/Common/Conf/dbname.php中配置內(nèi)容如下：
<?php
/*
* 本配置文件主要存儲數(shù)據(jù)庫后綴名障般，
* 前綴為wechat_,
* 在Application/Common/Behaviors/TestBehavior.class.php中驗證
* 注意：在新增數(shù)據(jù)庫的時候，注意修改該文件
*
* @author: liangxifeng
* @date: 2016-08-13
*/
return array(
'PRIVATEIDS'=>array('share',1,2,3,4)
);
- constant.php和dbname.php則是在配置文件Application/Common/Conf/config.php中使用擴展配置選項配置加載；
'LOAD_EXT_CONFIG' => 'constants,systemConfig,dbname',//加載常量配置
- 在使用自定義行為類的時候要在Application/Common/Conf/中新增tags.php
內(nèi)容如下：
/*
* 擴展行為類配置文件
* @author:liangxifeng
* @date:2016-08-13
*/
return array(
//應(yīng)用開始標(biāo)簽位添加切換數(shù)據(jù)庫行為
'action_begin'=>array('Common\\Behaviors\\SwitchDbBehavior'),
);
使用方法
在注冊session后或手動加載使用，比如在控制器中：
public function index()
{
//注冊session切換數(shù)據(jù)庫
session('privateId',2);
//手動切換數(shù)據(jù)庫為私有庫
tag('action_begin',$params='share');
$wechat = D('Wechat');
//查詢數(shù)據(jù)庫
$res = $wechat->where('wechat_id=1')->find();
echo "<pre>";
var_dump($res);
exit;
｝
