ThinkPHP6 业务分表之五：优化

<?php
declare (strict_types = 1);

namespace app\util;

use think\facade\Db;

class Subtable {
    /**
     * 支持分表的列表
     * 格式按照 表前缀 -> 分表数，生成会按照 `_` 来拼接
     *
     * @var array
     */
    private static $tables = [
        'users' => 16,
        'order' => 32,
    ];

    /*
     * 获取指定表的 Db 实例
     *
     * @param  string  $name 名称
     * @param  int  $uid UID
     * @return \think\facade\Db
     */
    public static function db(string $name, ?int $uid = null): Db
    {
        return Db::table(self::name($name, $uid));
    }

    /*
     * 获取指定表的名称
     *
     * @param  string  $name 名称
     * @param  int  $uid UID
     * @return string
     */
    public static function name(string $name, ?int $uid = null): string
    {
        $index = self::index($name, $uid);

        return sprintf('%s_%s', $name, $index);
    }

    /*
     * 获取指定表的 ID（十六进制）
     *
     * @param  string  $name 名称
     * @param  int  $uid UID
     * @return string
     */
    public static function index(string $name, ?int $uid = null) :string
    {
        if(is_null($uid)) {
            // 这里需要替换成获取 UID 的方法
            throw new \Exception('UID 值异常');
        }

        if(!isset(self::$tables[$name])) {
            throw new \Exception(sprintf('不支持当前 %s 表', $name));
        }

        if((int) $uid === 0) {
            throw new \Exception('UID 值异常');
        }

        return sprintf('%x', $uid % self::$tables[$name]);
    }

    /*
     * 获取指定表的分表数
     *
     * @param  string  $name 名称
     * @return int
     */
    public static function number(string $name) :int
    {
        if(!isset(self::$tables[$name])) {
            throw new \Exception(sprintf('不支持当前 %s 表', $name));
        }

        return (int) self::$tables[$name];
    }

    /*
     * UNION 生成
     *
     * @param  string  $name 名称
     * @param  string  $field 查询字段
     * @param  string  $other 其他条件，比如 where
     * @param  bool  $has_all 是否使用 UNION ALL
     * @return string
     */
    public static function union(string $name, string $field = '*', string $other = '', ?bool $has_all = true) :string
    {
        $number = self::number($name);

        $sql = '';

        for($i = 0; $i < $number; $i++) {
            if($i !== 0) {
                $sql .= $has_all ? ' UNION ALL ' : ' UNION ';
            }

            $sql .= sprintf('SELECT %s FROM %s_%x %s', $field, $name, $i, $other);
        }

        return $sql;
    }
}