likeadmin_php_saas/server/app/common/model/BaseModel.php

<?php
// +----------------------------------------------------------------------
// | likeadmin快速开发前后端分离管理后台（PHP版）
// +----------------------------------------------------------------------
// | 欢迎阅读学习系统程序代码，建议反馈是我们前进的动力
// | 开源版本可自由商用，可去除界面版权logo
// | gitee下载：https://gitee.com/likeshop_gitee/likeadmin
// | github下载：https://github.com/likeshop-github/likeadmin
// | 访问官网：https://www.likeadmin.cn
// | likeadmin团队 版权所有 拥有最终解释权
// +----------------------------------------------------------------------
// | author: likeadminTeam
// +----------------------------------------------------------------------

namespace app\common\model;

use app\common\cache\TenantAdminTokenCache;
use app\common\enum\AdminTerminalEnum;
use app\common\model\tenant\Tenant;
use app\common\service\FileService;
use think\facade\Request;
use think\Model;

/**
 * 基础模型
 * Class BaseModel
 * @package app\common\model
 */
class BaseModel extends Model
{
    // 定义全局的查询范围
    protected $globalScope = ['tenantId'];

    private static bool $hasTenantId = false;

    // 不需要检查分表的范围
    private static $notCheckTables = ['tenant', 'recharge_order', 'refund_record', 'official_account_reply', 'config', 'hot_search', 'operation_log'];

    /**
     * @notes 公共处理图片，补全路径
     * @param $value
     * @return string
     * @author 张无忌
     * @date 2021/9/10 11:02
     */
    public function getImageAttr($value): string
    {
        return trim($value) ? FileService::getFileUrl($value) : '';
    }

    /**
     * @notes 公共图片处理,去除图片域名
     * @param $value
     * @return mixed|string
     * @author 张无忌
     * @date 2021/9/10 11:04
     */
    public function setImageAttr($value): mixed
    {
        return trim($value) ? FileService::setFileUrl($value) : '';
    }

    /**
     * @notes 增加全局
     * @param $query
     * @return void
     * @author yfdong
     * @date 2024/09/03 22:45
     */
    public static function scopeTenantId($query): void
    {
        $tenantId = self::checkTenant() ?: self::checkUser();
        // 获取当前查询模型表名
        $table = $query->getTable();
        // 获取对应表字段
        $fields = $query->getConnection()->getTableInfo($table, 'fields');
        // 判断是否存在 tenant_id 字段
        $hasTenantId = in_array('tenant_id', $fields);
        //去除表前缀
        $tableNoPrefix = str_replace(env('database.prefix', 'la_'), '', $table);
        // 来自租户端或者用户端，平台端查询租户就需要进行全局租户信息替换
        if ($tenantId || ($hasTenantId && isset(\request()->tenantId))) {
            if(!$tenantId){
                $tenantId = \request()->tenantId;
            }
            // 分表全局控制
            if (!in_array($tableNoPrefix, self::$notCheckTables)) {
                // 根据租户唯一标识获取对应表是否存在
                $tenantInfo = Tenant::query()->where('id', $tenantId)->find()->toArray();
                // 租户分表策略
                if ($tenantInfo['tactics'] == 1) {
                    $table = $table . '_' . $tenantInfo['sn'];
                    //判断对应表是否存在
                    $newQuery = $query->getConnection()->getTableInfo($table);
                    if ($newQuery) {
                        // 分表存在
                        $query->table($table);
                    } else {
                        // 表不存在
                        echo "表 " . $table . " 不存在";
                    }
                }
            }
            // 判断是否存在 tenant_id 字段
            if ($hasTenantId) {
                self::$hasTenantId = true;
                $query->where($table . '.tenant_id', $tenantId);
            }
        }
    }

    /**
     * @notes 插入语句前置处理
     * @param $model
     * @return void
     * @author JXDN
     * @date 2024/09/04 15:48
     */
    public static function onBeforeInsert($model): void
    {
        $tenantId = self::checkTenant() ?: self::checkUser();
        if ($tenantId) {
            if (self::$hasTenantId) {
                $model->tenant_id = $tenantId;
            }
        }
    }

    /**
     * @notes 更新语句前置处理
     * @param $model
     * @return void
     * @author JXDN
     * @date 2024/09/04 15:48
     */
    public static function onBeforeUpdate($model): void
    {
        $tenantId = self::checkTenant() ?: self::checkUser();
        if ($tenantId) {
            if (self::$hasTenantId) {
                $model->tenant_id = $tenantId;
            }
        }
    }

    /**
     * @notes 判断是否为租户端
     * @return false|mixed|void
     * @author JXDN
     * @date 2024/09/04 16:01
     */
    private static function checkTenant()
    {
        $token = Request::header('token');
        if (AdminTerminalEnum::isTenant() && $token !== "null") {
            $tenant_id = \request()->tenantId;
            if ($tenant_id) {
                return $tenant_id;
            } else {
                return false;
            }
        } else {
            return false;
        }
    }

    /**
     * @notes 判断是否为用户端
     * @return false|mixed|void
     * @author JXDN
     * @date 2024/09/04 16:01
     */
    private static function checkUser()
    {
        if (AdminTerminalEnum::isUser()) {
            $tenant_id = \request()->tenantId;
            if ($tenant_id) {
                return $tenant_id;
            } else {
                return false;
            }
        } else {
            return false;
        }
    }
}