THINKPHP框架动态连接不同配置数据库

最近做的一个项目,需要应用程序根据不同的区域编码连接不同的区域数据库,比如请求地址https://test.com/11/login 连接a数据库 , 请求https://test.com/21/login 需要连接b数据库 ,这样的设计可以让应用程序满足按一定规则分数据库的场景,虽然可以使用tp框架的多应用模式实现,但是如果每一个应用都是一样的功能,就相当于复制了一份代码,只是修改了数据库配置,为了提高代码的复用,减少代码维护工作,采用单应用模式,动态根据区域编码连接不同区域的数据库;

下面我们研究一下如何解决这个问题;

使用xdebug跟踪tp框架的入口,会发现tp框架通过Config实例读取Config下配置文件,实例化该类。下面是config类的代码:


// +----------------------------------------------------------------------
declare(strict_types=1);

namespace think;

/**
 * 配置管理类
 * @package think
 */
class Config
{
    /**
     * 配置参数
     * @var array
     */
    protected $config = [];

    /**
     * 构造方法
     * @access public
     */
    public function __construct(protected string $path = '', protected string $ext = '.php')
    {
    }

    public static function __make(App $app)
    {
        $path = $app->getConfigPath();
        $ext  = $app->getConfigExt();

        return new static($path, $ext);
    }

    /**
     * 加载配置文件(多种格式)
     * @access public
     * @param  string $file 配置文件名
     * @param  string $name 一级配置名
     * @return array
     */
    public function load(string $file, string $name = ''): array
    {
        if (is_file($file)) {
            $filename = $file;
        } elseif (is_file($this->path . $file . $this->ext)) {
            $filename = $this->path . $file . $this->ext;
        }

        if (isset($filename)) {
            return $this->parse($filename, $name);
        }

        return $this->config;
    }

    /**
     * 解析配置文件
     * @access public
     * @param  string $file 配置文件名
     * @param  string $name 一级配置名
     * @return array
     */
    protected function parse(string $file, string $name): array
    {
        $type   = pathinfo($file, PATHINFO_EXTENSION);
        $config = [];
        $config = match ($type) {
            'php'           =>  include $file,
            'yml','yaml'    =>  function_exists('yaml_parse_file') ? yaml_parse_file($file) : [],
            'ini'           =>  parse_ini_file($file, true, INI_SCANNER_TYPED) ?: [],
            'json'          =>  json_decode(file_get_contents($file), true),
            default         =>  [],
        };

        return is_array($config) ? $this->set($config, strtolower($name)) : [];
    }

    /**
     * 检测配置是否存在
     * @access public
     * @param  string $name 配置参数名(支持多级配置 .号分割)
     * @return bool
     */
    public function has(string $name): bool
    {
        if (!str_contains($name, '.') && !isset($this->config[strtolower($name)])) {
            return false;
        }

        return !is_null($this->get($name));
    }

    /**
     * 获取一级配置
     * @access protected
     * @param  string $name 一级配置名
     * @return array
     */
    protected function pull(string $name): array
    {
        $name = strtolower($name);

        return $this->config[$name] ?? [];
    }

    /**
     * 获取配置参数 为空则获取所有配置
     * @access public
     * @param  string $name    配置参数名(支持多级配置 .号分割)
     * @param  mixed  $default 默认值
     * @return mixed
     */
    public function get(string $name = null, $default = null)
    {
        // 无参数时获取所有
        if (empty($name)) {
            return $this->config;
        }

        if (!str_contains($name, '.')) {
            return $this->pull($name);
        }

        $name    = explode('.', $name);
        $name[0] = strtolower($name[0]);
        $config  = $this->config;

        // 按.拆分成多维数组进行判断
        foreach ($name as $val) {
            if (isset($config[$val])) {
                $config = $config[$val];
            } else {
                return $default;
            }
        }

        return $config;
    }

    /**
     * 设置配置参数 name为数组则为批量设置
     * @access public
     * @param  array  $config 配置参数
     * @param  string $name 配置名
     * @return array
     */
    public function set(array $config, string $name = null): array
    {
        if (empty($name)) {
            $this->config = array_merge($this->config, array_change_key_case($config));
            return $this->config;
        }

        if (isset($this->config[$name])) {
            $result = array_merge($this->config[$name], $config);
        } else {
            $result = $config;
        }

        $this->config[$name] = $result;

        return $result;
    }
}

通过阅读代码,配置类又提供了set()方法,我们正好可以使用它改变配置参数。

接下来的问题,就是我们在哪里修改配置比较合适呢,那肯定在业务层操作数据库之前,在http请求之前修改比较好,tp框架提供的中间件功能,可以在请求之前添加前置的过滤器处理,我们就使用它来修改配置吧。

首先在middleware 目录下添加一个前置处理类:

class RequestDataBaseConfig
{
    /**
     * @name:handle
     * @datetime:2023/11/23
     * @author:
     * @param $request
     * @param \Closure $next
     * @return mixed|\think\response\Json
     */
    public function handle($request, \Closure $next)
    {

        // 添加中间件执行代码
        $uri = $request->server("REQUEST_URI");
        $parts = explode('/', $uri);
        $area = reset($parts);
        
        setDataBaseConfig($area)

        return $next($request);
    }



    public function setDataBaseConfig($area)
    {
        $provinceConfig = Config('mutidatabase.'.$area); //读取 配置类下配置参数
        $newConfig = new Config();   // 配置类实例
        $newConfig->set($provinceConfig,"database"); // 数据库配置名称
        Db::setConfig($newConfig);  // 设置新的数据库配置

    }
}

然后 在 middleware.php 配置文件中 配置 中间件 :

ok,万事具备了,我们在config目录下添加mutidatabase.php 配置文件,配置不同的数据库参数就好了。

你可能感兴趣的:(数据库)