PHP CodeIgniter框架實現讀寫分離

時間  2019-11-19
標籤 php codeigniter 框架 實現 讀寫 分離 欄目 PHP 简体版
原文   原文鏈接

1、目標

當前服務器只作了主從，未配置讀寫分離，讀寫分離的功能就只有交給程序來實現，本文主要談談Codeigniter怎麼實現讀寫分離，而且須要知足如下兩點：php

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

網上有方案經過手動load多個DB來實現讀寫分離，這樣的分離跟業務關聯太緊，增長了開發難度也不利於維護，咱們要作的是默認讀重庫，寫則寫主庫，讀寫分離對開發者透明mysql

二、配置簡單。sql

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

2、實現思路

  • 一、要實現讀寫分離最簡單的思路就是在最終執行查詢的地方根據查詢語句判斷是插入主庫仍是讀取從庫，因此須要找到該函數。
  • 二、應該只鏈接一次數據庫，下次操做該連接應當可複用。也就是連一次重庫後全部的讀操做均可用，不需再次鏈接，主庫同理。因此咱們能夠將連接放在CI超級對象中。
  • 三、主從的判斷是根據最終執行的SQL語句來判斷的，因此數據庫配置中的自動連接autoinit參數就不用設置爲true了，若是默認鏈接了而又不須要操做該庫就浪費資源了。
  • 四、模型中可使用$this->db來直接操做查詢，不須要其餘調整。
  • 五、不直接修改system下的文件

3、實現讀寫分離

CI的DB類固定爲讀取system下的文件，咱們能夠經過適當的重寫來實現。首先是Loader.php，其中的database方法用來加載數據庫對象，固定引用了system/database/DB.php文件，咱們判斷下是否存在自定義DB.php文件，存在則引入。數組

3.1 重寫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方法，用來讀取配置文件並進行初始化工做。一樣有兩處地方須要重寫下：服務器

3.2 重寫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語句的方法，咱們能夠在這裏進行數據庫連接的判斷。app

3.3 重寫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方法，能夠考慮下是否能夠在該方法中關閉？這裏認爲是不行的。負載均衡

3.4 關閉數據庫連接

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便可。

3.5 模型調整

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的實現方式。

3.6 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中的分配方式。

 

文章轉自：http://itopic.org/codeigniter-mysql-proxy.html 

關注微信公衆號：lovephp

相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程