CakePHP Shell脚本环境设置

时间:2011-11-02 02:39:04

标签: cakephp cakephp-2.0

我是Cake的shell开发新手。我面临的问题是在脚本本身设置数据源。我的database.php是;

function __construct() 
{ 
    if(getenv('ENVIRONMENT') == 'staging') {
        $this->default = $this->staging;
    } else {
        $this->default = $this->production;
    }
} 

所以,我正在根据Web服务器的环境设置设置数据库。当然,php-cli无法访问此变量。我最终做的是创建一个cakephp shell任务。

class SelectEnvTask extends Shell {

    public function execute() 
    {
        App::Import('ConnectionManager', 'Model');
        $configs = ConnectionManager::enumConnectionObjects();

        if (!is_array($configs) || empty($configs)) {
            $this->out('Error! No database configuration has been found.');
        }

        $connections = array_keys($configs);

        if(!isset($this->args[0])) {
            $this->out('Error! Please enter one of the environment settings as an argument: ' . implode('/', $connections) . 
                "\n\n" . 'eg. ./Console/cake *script_name* *environment*', 2);
            exit(1);
        }

        if(!in_array($this->args[0], $connections)) {
            $this->out($this->args[0] . ' environment could not be found!', 2);
            exit(1);
        }

        //hacky solution until finding a better one
        $models = App::objects('Model');
        foreach($models as $model) {
            ClassRegistry::init($model)->setDataSource($this->args[0]);
        }
    }
}

这样可以正常工作,但是正如您在任务的下方所看到的,我获得了所有模型名称并更改了它们的数据库连接,这不是一个好习惯。我也不想在数据库类中设置更多变量,并希望在shell / tasks中处理这些变量。

有没有更优雅的方法来实现这个目标?

谢谢,

1 个答案:

答案 0 :(得分:0)

这是一个更优雅的解决方案。将以下方法添加到shell或任务中,然后在需要时执行它以便动态更改数据配置文件(在app / Config / database.php中列出):

function change_database_profile($database = 'default') {
    $connected = ConnectionManager::getDataSource($database);
    if($connected->isConnected()) { 
        return true; 
    }   
    return false;
}