94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
"""gvsdsdk.db_router — 开箱即用的 SAAS 数据库路由器
|
||
|
||
子服务器只需在 settings.py 中配置：
|
||
DATABASE_ROUTERS = ['gvsdsdk.db_router.SAASDatabaseRouter']
|
||
|
||
即可自动将 gvsdsdk 的所有模型路由到 SAAS 主库。
|
||
|
||
配置项（settings.py）：
|
||
GVSDSDK_SAAS_DB_ALIAS = 'saas' # SAAS 数据库别名，默认 'saas'
|
||
GVSDSDK_SAAS_READ_ONLY = True # 子服务器是否只读，默认 True
|
||
"""
|
||
|
||
|
||
class SAASDatabaseRouter:
|
||
"""SAAS 模型数据库路由器——将 gvsdsdk 模型路由到 SAAS 主库
|
||
|
||
用法：
|
||
# settings.py
|
||
DATABASES = {
|
||
'default': { ... }, # 本地数据库
|
||
'saas': { # SAAS 主库
|
||
'ENGINE': 'django.db.backends.mysql',
|
||
'NAME': 'Users',
|
||
...
|
||
},
|
||
}
|
||
DATABASE_ROUTERS = ['gvsdsdk.db_router.SAASDatabaseRouter']
|
||
|
||
# 可选配置
|
||
GVSDSDK_SAAS_DB_ALIAS = 'saas' # 默认 'saas'
|
||
GVSDSDK_SAAS_READ_ONLY = True # 默认 True（子服务器只读）
|
||
"""
|
||
|
||
# gvsdsdk 内所有模块前缀
|
||
_GVSDSDK_MODULES = frozenset({
|
||
'gvsdsdk.models',
|
||
'gvsdsdk.commission.models',
|
||
'gvsdsdk.payment.models',
|
||
})
|
||
|
||
# gvsdsdk 内所有 app_label
|
||
_GVSDSDK_APP_LABELS = frozenset({
|
||
'gvsdsdk',
|
||
})
|
||
|
||
def _get_saas_db(self):
|
||
from django.conf import settings
|
||
return getattr(settings, 'GVSDSDK_SAAS_DB_ALIAS', 'saas')
|
||
|
||
def _is_read_only(self):
|
||
from django.conf import settings
|
||
return getattr(settings, 'GVSDSDK_SAAS_READ_ONLY', True)
|
||
|
||
def _is_gvsdsdk_model(self, model):
|
||
"""判断模型是否属于 gvsdsdk"""
|
||
# 按模块路径判断
|
||
module = getattr(model, '__module__', '')
|
||
if module in self._GVSDSDK_MODULES or module.startswith('gvsdsdk.'):
|
||
return True
|
||
# 按 app_label 判断
|
||
app_label = getattr(model._meta, 'app_label', '')
|
||
if app_label in self._GVSDSDK_APP_LABELS:
|
||
return True
|
||
return False
|
||
|
||
def db_for_read(self, model, **hints):
|
||
if self._is_gvsdsdk_model(model):
|
||
return self._get_saas_db()
|
||
return None
|
||
|
||
def db_for_write(self, model, **hints):
|
||
if self._is_gvsdsdk_model(model):
|
||
if self._is_read_only():
|
||
raise RuntimeError(
|
||
"gvsdsdk SAAS database is read-only for sub-servers. "
|
||
"Set GVSDSDK_SAAS_READ_ONLY=False to allow writes."
|
||
)
|
||
return self._get_saas_db()
|
||
return None
|
||
|
||
def allow_relation(self, obj1, obj2, **hints):
|
||
# 允许同一数据库内的关联
|
||
db1 = self.db_for_read(obj1.__class__)
|
||
db2 = self.db_for_read(obj2.__class__)
|
||
if db1 and db2 and db1 == db2:
|
||
return True
|
||
return None
|
||
|
||
def allow_migrate(self, db, app_label, model_name=None, **hints):
|
||
# gvsdsdk 模型不迁移（managed=False 或由主服务器管理）
|
||
if app_label in self._GVSDSDK_APP_LABELS:
|
||
return db != self._get_saas_db()
|
||
return None
|