一步一步重写 CodeIgniter 框架 (10) —— 使用 CodeIgniter 类库（续）

上一节简单实现了 CI 的类库扩展模型，所以 _ci_load_class 和 _ci_init_class 写的不是很完备。根据上节课的分析，当 system/libraries 目录下存在 Email.php， 然后在 application/libraies 目录下存在 My_Email.php 时就可以实现扩展类库的功能。除了扩展之外，我们还需要：

1）直接覆盖原始类

2）完全自定义类

很简单，按照约定，当不存在MY_开头的类库文件，加载类库的情况必定属于以上两种，如下所示

// 直接加载代码
            $is_duplicate = FALSE;
            foreach ($this->_ci_library_paths as $path) {
                $filepath = $path.'libraries/'.$subdir.$class.'.php';

                if ( ! file_exists($filepath)) {
                    continue;
                }

                if (in_array($filepath, $this->_ci_loaded_files)) {

                    if ( ! is_null($object_name)) {
                        $CI =& get_instance();
                        if ( ! isset($CI->$object_name)) {
                            return $this->_ci_init_class($class, '', $params, $object_name);
                        }
                    }

                    $is_duplicate = TRUE;
                    log_message('debug', $class." class already loaded. Second attempt ignored.");
                    return;
                }

                include_once($filepath);
                $this->_ci_loaded_files[] = $filepath;
                return $this->_ci_init_class($class, '', $params, $object_name);
            }

其中 _ci_libraries_path 在初始构造函数中初始化如下：

$this->_ci_library_paths = array(APPPATH, BASEPATH);

注意顺序，先是 APPPATH， 然后再是 BASEPATH， 保证加载的顺序先是 application ，再是 system。

再来看一下 _ci_init_class 函数

public function _ci_init_class($class, $prefix = '', $config = FALSE, $object_name = NULL) {

        if ($prefix == '') {
            if (class_exists('CI_'.$class)) {
                $name = 'CI_'.$class;
            } elseif (class_exists(config_item('subclass_prefix').$class)) {
                $name = config_item('subclass_prefix');
            } else {
                $name = $class;
            }
        } else {
            $name = $prefix.$class;
        }

        if ( ! class_exists($name)) {
            log_message('error', "Non-existent class: ".$name);
            show_error("Non-existent class: ".$class);
        }

        $class = strtolower($class);

        // 这里对名字做了一个特殊的映射，针对 unit_test 就可以直接用 unit, 而非 unit_test
        if (is_null($object_name)) {
            $classvar = ( ! isset($this->_ci_varmap[$class])) ? $class : $this->_ci_varmap[$class];
        } else {
            $classvar = $object_name;
        }

        // 保存 class 名和对象名, 以供判断某个类库是否加载
        $this->_ci_classes[$class] = $classvar;

        $CI =& get_instance();
        if ($config !== NULL) {
            $CI->$classvar = new $name($config);
        } else {
            $CI->$classvar = new $name;
        }

分析以上代码可以发现，为了直接覆盖原始的类库文件，对 $prefix ==‘’ 的情况作了进一步处理，判断是否存在 CI 或自定义MY_前缀的类，从而保证获得正确的类名。

实例名也考虑到了一种特殊的情况，比如加载 unit_test 类的时候，直接使用 unit 作为变量，这也值得我们借鉴和考虑！！！也就是说，在考虑到一般的情况下，还可以做更灵活的处理，对类名较长的类直接通过配置的方式指定实例名，非常简洁~

 

总结： 整个过程加载类库的设计就完成了，特别灵活和方便。