最近做的一个项目,需要应用程序根据不同的区域编码连接不同的区域数据库,比如请求地址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 配置文件,配置不同的数据库参数就好了。