利用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怎么實現一個讀寫分離功能的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！