利用CodeIgniter怎么實(shí)現(xiàn)一個(gè)讀寫分離功能

利用CodeIgniter怎么實(shí)現(xiàn)一個(gè)讀寫分離功能？相信很多沒有經(jīng)驗(yàn)的人對(duì)此束手無策，為此本文總結(jié)了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個(gè)問題。

1、讀寫分離對(duì)開發(fā)應(yīng)該透明。

網(wǎng)上有方案通過手動(dòng)load多個(gè)DB來實(shí)現(xiàn)讀寫分離，這樣的分離跟業(yè)務(wù)關(guān)聯(lián)太緊，增加了開發(fā)難度也不利于維護(hù)，我們要做的是默認(rèn)讀重庫，寫則寫主庫，讀寫分離對(duì)開發(fā)者透明

2、配置簡(jiǎn)單。

保留現(xiàn)有的配置方式，通過增加一個(gè)數(shù)組來配置讀寫分離，不影響原有使用方式。

思路

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

2、應(yīng)該只連接一次數(shù)據(jù)庫，下次操作該鏈接應(yīng)當(dāng)可復(fù)用。也就是連一次重庫后所有的讀操作都可用，不需再次連接，主庫同理。所以我們可以將鏈接放在CI超級(jí)對(duì)象中。

3、主從的判斷是根據(jù)最終執(zhí)行的SQL語句來判斷的，所以數(shù)據(jù)庫配置中的自動(dòng)鏈接autoinit參數(shù)就不用設(shè)置為true了，如果默認(rèn)連接了而又不需要操作該庫就浪費(fèi)資源了。

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

5、不直接修改system下的文件

實(shí)現(xiàn)讀寫分離

CI的DB類固定為讀取system下的文件，我們可以通過適當(dāng)?shù)闹貙憗韺?shí)現(xiàn)。首先是Loader.php，其中的database方法用來加載數(shù)據(jù)庫對(duì)象，固定引用了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 */

接著我們?cè)赼pplication/core下創(chuàng)建database/DB.php，該文件只有一個(gè)DB方法，用來讀取配置文件并進(jìn)行初始化工作。同樣有兩處地方需要重寫下：

重寫DB.php

//DB_driver.php為所有驅(qū)動(dòng)方式的父類，最終執(zhí)行查詢的方法在該文件中
//第一處修改為判斷自定義的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' 該文件可不調(diào)整，實(shí)際未修改該文件，為了方便調(diào)試也加了
//mysql驅(qū)動(dòng)對(duì)應(yīng)system/database/drivers/mysql/mysql_driver.php，mysql的最后執(zhí)行方法在這里，
//包括數(shù)據(jù)庫打開和關(guān)閉、查詢等,可以該文件增加相應(yīng)日志查看讀寫分離是否有效
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');
}
//將當(dāng)前group name賦值給param，方便判斷
$params['group_name'] = $active_group;
 
/* End of file DB.php */
/* Location: ./application/core/database/DB.php */

整個(gè)DB.php調(diào)整的也基本上是文件的引入，group name的引入是為了方便后面的判斷， 不引入則可以通過主機(jī)、數(shù)據(jù)庫名稱這些來配置。如果想強(qiáng)制關(guān)閉autoint，可以在DB.php中刪掉下面這段：

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

接下來就是最核心的地方。根據(jù)查詢語句實(shí)現(xiàn)讀寫分離。
DB_driver.php中的simple_query方法可以理解為最后執(zhí)行SQL語句的方法，我們可以在這里進(jìn)行數(shù)據(jù)庫鏈接的判斷。

重寫DB_driver.php

//增加屬性，表示當(dāng)前組
var $active_group;
//增加屬性，使用強(qiáng)制使用主庫
var $db_force_master;
//該方法為執(zhí)行查詢的必經(jīng)之地，我們可以在這里根據(jù)類型判斷使用哪個(gè)鏈接。
function simple_query($sql)
{
  //load_db_proxy_setting方法這里寫在helper中，也可以直接寫在該類中，寫在helper中則需要在自動(dòng)加載中加載該helper
  //該方法的作用是根據(jù)當(dāng)前鏈接group name 和sql讀寫類型，以及是否強(qiáng)制使用主庫判斷使用哪個(gè)鏈接。使用主庫 OR 重庫？
  //主重庫的負(fù)載均衡，單點(diǎn)故障都可以在這里考慮。也就是根據(jù)3個(gè)參數(shù)返回一個(gè)可用的配置數(shù)組。
  $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;
    //將當(dāng)前配置重新賦值給類的屬性，如果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超級(jí)對(duì)象或者直接從CI超級(jí)對(duì)象中讀取
    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;
    }
    //強(qiáng)制只一次有效，下次查詢失效，防止一直強(qiáng)制主庫
    $this->reset_force_master();
  }
  if ( ! $this->conn_id)
  {
    $this->initialize();
  }
  return $this->_execute($sql);
}
//某些情況會(huì)強(qiáng)制使用主庫，先執(zhí)行該方法即可
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 */

到這里讀寫分離即基本實(shí)現(xiàn)了，但做事情得善始善終，鏈接的數(shù)據(jù)庫對(duì)象需要關(guān)閉，可以在公用控制器中執(zhí)行完畢后關(guān)掉連接。

DB_driver.php中也有close方法，可以考慮下是否可以在該方法中關(guān)閉？這里認(rèn)為是不行的。

關(guān)閉數(shù)據(jù)庫鏈接

class MY_Controller extends CI_Controller 
{
  public function __construct() 
  {
    parent::__construct();
    $this->load->service('common/helper_service', NULL, 'helper');
    //下面這段為關(guān)閉CI超級(jí)對(duì)象中的數(shù)據(jù)庫對(duì)象和數(shù)據(jù)庫鏈接，db的對(duì)象Codeigniter.php中會(huì)關(guān)閉
    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 */

模型中的使用，為了使每個(gè)model中都可使用$this->db，以及不多次連接數(shù)據(jù)庫，這里也是將鏈接放在CI超級(jí)對(duì)象中。這里就算不讀寫分離也可以這么處理，可以很方便的連接多個(gè)DB，具體的model要使用其他庫只需要在構(gòu)造函數(shù)中傳入group name即可。

模型調(diào)整

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 */

最后的數(shù)據(jù)庫配置方式，只需要在原有的基礎(chǔ)上配置一個(gè)數(shù)組即可。是使用雙主還是一主多從就看這里的配置方式。最開始想到直接在原配置上加鍵名來處理，但主與從的對(duì)應(yīng)關(guān)系還是沒有這樣子明了，這里的定義方式?jīng)Q定了load_db_proxy_setting的實(shí)現(xiàn)方式。

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 */

最開始的數(shù)據(jù)庫鏈接并未放到CI超級(jí)對(duì)象中，發(fā)現(xiàn)load多個(gè)模型時(shí)每次都會(huì)打開鏈接，所以完成讀寫分離之后一定要測(cè)試，可以在數(shù)據(jù)庫鏈接打開和關(guān)閉的地方查看是否按預(yù)期執(zhí)行(方法對(duì)應(yīng)application/core/database/drivers/mysql/mysql_driver.php中的db_connect和_close)。整個(gè)調(diào)整過程最重要的兩點(diǎn)就是simple_query方法以及構(gòu)造函數(shù)中關(guān)閉數(shù)據(jù)庫鏈接。模型中的調(diào)整是為了更方便的鏈接多個(gè)庫，未實(shí)現(xiàn)讀寫分離時(shí)也是這么調(diào)整的，常用的方法獨(dú)立成一個(gè)文件，MY_Model去繼承。

實(shí)現(xiàn)MYSQL讀寫分離的中間件挺多，在沒有用到這些時(shí)可以通過程序上的控制來實(shí)現(xiàn)讀寫分離。當(dāng)然這里只是實(shí)現(xiàn)了讀寫分離，可以強(qiáng)制使用主庫。如果想要更好的分配方式，可以好好想想load_db_proxy_setting中的分配方式。

看完上述內(nèi)容，你們掌握利用CodeIgniter怎么實(shí)現(xiàn)一個(gè)讀寫分離功能的方法了嗎？如果還想學(xué)到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！