CodeIgniter讀寫分離實現方法詳解_php執行個體

本文執行個體講述了CodeIgniter讀寫分離實現方法。分享給大家供大家參考，具體如下：

當前伺服器只做了主從，未配置讀寫分離，讀寫分離的功能就只有交給程式來實現，這裡主要談談Codeigniter怎麼實現讀寫分離，並且需要滿足以下兩點：

1、讀寫分離對開發應該透明。

網上有方案通過手動load多個DB來實現讀寫分離，這樣的分離跟業務關聯太緊，增加了開發難度也不利於維護，我們要做的是預設讀重庫，寫則寫主庫，讀寫分離對開發人員透明

2、配置簡單。

保留現有的配置方式，通過增加一個數組來配置讀寫分離，不影響原有使用方式。

思路

1、要實現讀寫分離最簡單的思路就是在最終執行查詢的地方根據查詢語句判斷是插入主庫還是讀取從庫，所以需要找到該函數。

2、應該只串連一次資料庫，下次操作該連結應當可複用。也就是連一次重庫後所有的讀操作都可用，不需再次串連，主庫同理。所以我們可以將連結放在CI超級對象中。

3、主從的判斷是根據最終執行的SQL語句來判斷的，所以資料庫配置中的自動連結autoinit參數就不用設定為true了，如果預設串連了而又不需要操作該庫就浪費資源了。

4、模型中可以使用$this->db來直接操作查詢，不需要其他調整。

5、不直接修改system下的檔案

實現讀寫分離

CI的DB類固定為讀取system下的檔案，我們可以通過適當的重寫來實現。首先是Loader.php，其中的database方法用來載入資料庫物件，固定引用了system/database/DB.php檔案，我們判斷下是否存在自訂DB.php檔案，存在則引入。

重寫Loader.php

public function database($params = '', $return = FALSE, $active_record = NULL){  $CI =& get_instance();  if (class_exists('CI_DB') AND $return == FALSE AND $active_record == NULL AND isset($CI->db) AND is_object($CI->db)) {    return FALSE;  }  if(file_exists(APPPATH.'core/database/DB.php')) {    require_once(APPPATH.'core/database/DB.php');  } else {    require_once(BASEPATH.'database/DB.php');  }  if ($return === TRUE) {    return DB($params, $active_record);  }  $CI->db = '';  $CI->db =& DB($params, $active_record);}/* End of file MY_Loader.php *//* Location: ./application/core/MY_Loader.php */

接著我們在application/core下建立database/DB.php，該檔案只有一個DB方法，用來讀取設定檔並進行初始化工作。同樣有兩處地方需要重寫下：

重寫DB.php

//DB_driver.php為所有驅動方式的父類，最終執行查詢的方法在該檔案中//第一處修改為判斷自訂的DB_driver.php是否存在，存在則引入if(file_exists(APPPATH.'core/database/DB_driver.php')) {  require_once(APPPATH.'core/database/DB_driver.php');} else {  require_once(BASEPATH.'database/DB_driver.php');}//第二處 $params['dbdriver'].'_driver.php' 該檔案可不調整，實際未修改該檔案，為了方便調試也加了//mysql驅動對應system/database/drivers/mysql/mysql_driver.php，mysql的最後執行方法在這裡，//包括資料庫開啟和關閉、查詢等,可以該檔案增加相應日誌查看讀寫分離是否有效if(file_exists(APPPATH.'core/database/drivers/'.$params['dbdriver'].'/'.$params['dbdriver'].'_driver.php')) {  require_once(APPPATH.'core/database/drivers/'.$params['dbdriver'].'/'.$params['dbdriver'].'_driver.php');} else {  require_once(BASEPATH.'database/drivers/'.$params['dbdriver'].'/'.$params['dbdriver'].'_driver.php');}//將當前group name賦值給param，方便判斷$params['group_name'] = $active_group; /* End of file DB.php *//* Location: ./application/core/database/DB.php */

整個DB.php調整的也基本上是檔案的引入，group name的引入是為了方便後面的判斷， 不引入則可以通過主機、資料庫名稱這些來配置。如果想強制關閉autoint，可以在DB.php中刪掉下面這段：

if ($DB->autoinit == TRUE){  $DB->initialize();}

接下來就是最核心的地方。根據查詢語句實現讀寫分離。
DB_driver.php中的simple_query方法可以理解為最後執行SQL語句的方法，我們可以在這裡進行資料庫連結的判斷。

重寫DB_driver.php

//增加屬性，表示當前組var $active_group;//增加屬性，使用強制使用主庫var $db_force_master;//該方法為執行查詢的必經之地，我們可以在這雷根據類型判斷使用哪個連結。function simple_query($sql){  //load_db_proxy_setting方法這裡寫在helper中，也可以直接寫在該類中，寫在helper中則需要在自動載入中載入該helper  //該方法的作用是根據當前連結group name 和sql讀寫類型，以及是否強制使用主庫判斷使用哪個連結。使用主庫 OR 重庫？  //主重庫的負載平衡，單點故障都可以在這裡考慮。也就是根據3個參數返回一個可用的配置數組。  $proxy_setting = load_db_proxy_setting($this->group_name, $this->is_write_type($sql), $this->db_force_master);  if(is_array($proxy_setting) && ! empty($proxy_setting)) {    $proxy_setting_key = key($proxy_setting);    $this->group_name = $proxy_setting_key;    //將當前配置重新賦值給類的屬性，如果database.php配置的是DSN字串，則需要在load_db_proxy_setting中做處理    foreach($proxy_setting[$proxy_setting_key] as $key => $val) {      $this->$key = $val;    }    //定義連結ID為conn_首碼    $proxy_conn_id = 'conn_'.$proxy_setting_key;    $CI = & get_instance();    //賦值給CI超級對象或者直接從CI超級對象中讀取    if(isset($CI->$proxy_conn_id) && is_resource($CI->$proxy_conn_id)) {      $this->conn_id = $CI->$proxy_conn_id;    } else {      $this->conn_id = false;      $this->initialize();      $CI->$proxy_conn_id = $this->conn_id;    }    //強制只一次有效，下次查詢失效，防止一直強制主庫    $this->reset_force_master();  }  if ( ! $this->conn_id)  {    $this->initialize();  }  return $this->_execute($sql);}//某些情況會強制使用主庫，先執行該方法即可public function force_master(){  $this->db_force_master = TRUE;}public function reset_force_master(){  $this->db_force_master = FALSE;}/* End of file DB_driver.php *//* Location: ./application/core/database/DB_driver.php */

到這裡讀寫分離即基本實現了，但做事情得善始善終，連結的資料庫物件需要關閉，可以在公用控制器中執行完畢後關掉串連。

DB_driver.php中也有close方法，可以考慮下是否可以在該方法中關閉？這裡認為是不行的。

關閉資料庫連結

class MY_Controller extends CI_Controller {  public function __construct()   {    parent::__construct();    $this->load->service('common/helper_service', NULL, 'helper');    //下面這段為關閉CI超級對象中的資料庫物件和資料庫連結，db的對象Codeigniter.php中會關閉    register_shutdown_function(function(){      foreach(get_object_vars($this) as $key => $val) {        if(substr($key, 0, 3) == 'db_' && is_object($this->{$key}) && method_exists($this->{$key}, 'close')) {          $this->{$key}->close();        }        if(substr($key, 0, 5) == 'conn_' && is_resource($this->{$key})) {          $this->db->_close($val);          unset($this->{$key});        }      }    });  }}/* End of file MY_Controller.php *//* Location: ./application/core/MY_Controller.php */

模型中的使用，為了使每個model中都可使用$this->db，以及不多次串連資料庫，這裡也是將連結放在CI超級對象中。這裡就算不讀寫分離也可以這麼處理，可以很方便的串連多個DB，具體的model要使用其他庫只需要在建構函式中傳入group name即可。

模型調整

public function __construct($group_name = ''){  parent::__construct();  $this->initDb($group_name);}private function initDb($group_name = ''){  $db_conn_name = $this->getDbName($group_name);  $CI = & get_instance();  if(isset($CI->{$db_conn_name}) && is_object($CI->{$db_conn_name})) {    $this->db = $CI->{$db_conn_name};  } else {    $CI->{$db_conn_name} = $this->db = $this->load->database($group_name, TRUE);  }}private function getDbName($group_name = ''){  if($group_name == '') {    $db_conn_name = 'db';  } else {    $db_conn_name = 'db_'.$group_name;  }  return $db_conn_name;}/* End of file MY_Model.php *//* Location: ./application/core/MY_Model.php */

最後的資料庫配置方式，只需要在原有的基礎上配置一個數組即可。是使用雙主還是一主多從就看這裡的配置方式。最開始想到直接在原配置上加鍵名來處理，但主與從的對應關係還是沒有這樣子明了，這裡的定義方式決定了load_db_proxy_setting的實現方式。

database.php配置

$_master_slave_relation = array(  'default_master' => array('default_slave1', 'default_slave2', 'default_slave3'),);/* End of file database.php *//* Location: ./application/config/database.php */

最開始的資料庫連結並未放到CI超級對象中，發現load多個模型時每次都會開啟連結，所以完成讀寫分離之後一定要測試，可以在資料庫連結開啟和關閉的地方查看是否按預期執行(方法對應application/core/database/drivers/mysql/mysql_driver.php中的db_connect和_close)。整個調整過程最重要的兩點就是simple_query方法以及建構函式中關閉資料庫連結。模型中的調整是為了更方便的連結多個庫，未實現讀寫分離時也是這麼調整的，常用的方法獨立成一個檔案，MY_Model去繼承。

實現MYSQL讀寫分離的中介軟體挺多，在沒有用到這些時可以通過程式上的控制來實現讀寫分離。當然這裡只是實現了讀寫分離，可以強制使用主庫。如果想要更好的分配方式，可以好好想想load_db_proxy_setting中的分配方式。

更多關於CodeIgniter架構相關內容感興趣的讀者可查看本站專題：《codeigniter入門教程》和《CI(CodeIgniter)架構進階教程》。

希望本文所述對大家基於CodeIgniter架構的PHP程式設計有所協助。

