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'].'https://www.jb51.net/'.$params['dbdriver'].'_driver.php')) { require_once(APPPATH.'core/database/drivers/'.$params['dbdriver'].'https://www.jb51.net/'.$params['dbdriver'].'_driver.php'); } else { require_once(BASEPATH.'database/drivers/'.$params['dbdriver'].'https://www.jb51.net/'.$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方法,可以考虑下是否可以在该方法中关闭?这里认为是不行的。

关闭数据库链接

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:http://www.heiqu.com/f7a39fb269b76c49edf33f139f2bb741.html