Codeigniter 根据会话值自动检查数据库

Codeigniter autocheck db depending on session value

我试图强制我的应用在每次加载模型或控制器时根据我的会话值进行检查。

这实际上是 运行,但就在我扔掉这个模型的时候。

class News_model extends CI_Model {

public function __construct()
{
    parent::__construct();
    if($this->session->dbname=='db1'){
        $this->db=$this->load->database('db1', TRUE); 
    }       
    else{
        $this->db=$this->load->database('db2', TRUE); 
    }
}
public function get_news($slug = FALSE)
{
        if ($slug === FALSE)
        {
                $query = $this->db->get('news');
                return $query->result_array();
        }

        $query = $this->db->get_where('news', array('slug' => $slug));
        return $query->row_array();
}
}

但我不会war将 __construct 代码包含到我的所有模型或控制器中。

我已经尝试添加 autoload.php

$autoload['model'] = array('General');

我的通用代码是这样的。

class General extends CI_Model {

function __construct()
{
    parent::__construct();
    if($this->session->dbname=='db1'){
        $this->db=$this->load->database('db1', TRUE); 
    }       
    else{
        $this->db=$this->load->database('db2', TRUE); 
    }
}
}   

我该怎么做?

您可以通过创建一个基础模型来实现,该模型将由需要数据库检查的模型扩展。

我简化了检查和加载代码。一个简单的三元确定要使用的字符串并将其存储在变量 $dbname 中。该变量用于加载数据库,即 $this->load->database($dbname);

我认为您不需要 load::database() 的第二个参数,这意味着您不需要明确设置 $this->db。如果我错了,请使用

$this->db = $this->load->database($dbname, TRUE);

下面是 "base" 模型。文件名的前缀在 config.php 中通过设置 $config['subclass_prefix'] = 'MY_'; 确定。调整基本模型的文件和 class 名称以匹配您使用的 'subclass_prefix'。

/application/core/MY_Model.php

<?php
class MY_Model extends CI_Model
{
    public function __construct()
    {
        parent::__construct();
        $dbname = $this->session->dbname == 'db1' ? 'db1' : 'db2';
        $this->load->database($dbname);
    }

}

使用上面的方法创建其他模型...

class News_model extends MY_Model
{
    public function get_news($slug = FALSE)
    {
        if ($slug === FALSE)
        {
            $query = $this->db->get('news');
            return $query->result_array();
        }

        $query = $this->db->get_where('news', array('slug' => $slug));
        return $query->row_array();
    }
}