from sqlalchemy.orm import Session from typing import List, Optional import uuid from app.core.logging_config import get_business_logger from app.repositories.tenant_repository import TenantRepository from app.repositories.user_repository import UserRepository from app.repositories.workspace_repository import WorkspaceRepository from app.schemas.tenant_schema import ( TenantCreate, TenantUpdate, Tenant, TenantQuery, TenantList ) from app.schemas.user_schema import User from app.schemas.workspace_schema import WorkspaceCreate from app.models.tenant_model import Tenants from app.models.user_model import User as UserModel from app.core.exceptions import BusinessException from app.core.error_codes import BizCode # 获取业务逻辑专用日志器 business_logger = get_business_logger() class TenantService: """租户业务逻辑层""" def __init__(self, db: Session): self.db = db self.tenant_repo = TenantRepository(db) self.user_repo = UserRepository(db) self.workspace_repo = WorkspaceRepository(db) def create_tenant(self, tenant_data: TenantCreate) -> Tenants: """创建租户""" # 检查租户名称是否已存在 existing_tenant = self.tenant_repo.get_tenant_by_name(tenant_data.name) if existing_tenant: raise BusinessException(f"租户名称 '{tenant_data.name}' 已存在", code=BizCode.DUPLICATE_NAME) try: tenant = self.tenant_repo.create_tenant(tenant_data) business_logger.info(f"创建租户成功: {tenant.name} (ID: {tenant.id})") return tenant except Exception as e: business_logger.error(f"创建租户失败: {str(e)}") raise BusinessException(f"创建租户失败: {str(e)}", code=BizCode.DB_ERROR) def create_tenant_and_assign_user(self, tenant_data: TenantCreate, user_id: uuid.UUID) -> Tenants: """创建租户并分配用户""" try: # 创建租户 tenant = self.create_tenant(tenant_data) # 将用户分配给租户 success = self.user_repo.assign_user_to_tenant(user_id, tenant.id) if not success: raise BusinessException("分配用户到租户失败", code=BizCode.STATE_CONFLICT) business_logger.info(f"创建租户并分配用户成功: {tenant.name}") return tenant except Exception as e: business_logger.error(f"创建租户和分配用户失败: {str(e)}") self.db.rollback() raise BusinessException(f"创建租户失败: {str(e)}", code=BizCode.DB_ERROR) def get_tenant(self, tenant_id: uuid.UUID) -> Optional[Tenants]: """获取租户""" return self.tenant_repo.get_tenant_by_id(tenant_id) def get_tenant_by_name(self, name: str) -> Optional[Tenants]: """根据名称获取租户""" return self.tenant_repo.get_tenant_by_name(name) def get_tenants(self, query: TenantQuery) -> TenantList: """获取租户列表""" skip = (query.page - 1) * query.size tenants = self.tenant_repo.get_tenants( skip=skip, limit=query.size, is_active=query.is_active, search=query.search ) total = self.tenant_repo.count_tenants( is_active=query.is_active, search=query.search ) pages = (total + query.size - 1) // query.size return TenantList( items=[Tenant.model_validate(tenant) for tenant in tenants], total=total, page=query.page, size=query.size, pages=pages ) def update_tenant(self, tenant_id: uuid.UUID, tenant_data: TenantUpdate) -> Optional[Tenants]: """更新租户""" # 如果更新名称，检查是否重复 if tenant_data.name: existing_tenant = self.tenant_repo.get_tenant_by_name(tenant_data.name) if existing_tenant and existing_tenant.id != tenant_id: raise BusinessException(f"租户名称 '{tenant_data.name}' 已存在", code=BizCode.DUPLICATE_NAME) try: tenant = self.tenant_repo.update_tenant(tenant_id, tenant_data) if tenant: business_logger.info(f"更新租户成功: {tenant.name} (ID: {tenant.id})") return tenant except Exception as e: business_logger.error(f"更新租户失败: {str(e)}") raise BusinessException(f"更新租户失败: {str(e)}", code=BizCode.DB_ERROR) def delete_tenant(self, tenant_id: uuid.UUID) -> bool: """删除租户""" try: # 检查租户是否存在 tenant = self.tenant_repo.get_tenant_by_id(tenant_id) if not tenant: return False # 检查是否有关联的用户 users = self.tenant_repo.get_tenant_users(tenant_id) if users: raise BusinessException("无法删除租户，存在关联的用户", code=BizCode.STATE_CONFLICT) # 检查是否有关联的工作空间 workspaces = self.workspace_repo.get_workspaces_by_tenant(tenant_id) if workspaces: raise BusinessException("无法删除租户，存在关联的工作空间", code=BizCode.STATE_CONFLICT) success = self.tenant_repo.delete_tenant(tenant_id) if success: business_logger.info(f"删除租户成功: {tenant.name} (ID: {tenant.id})") return success except Exception as e: business_logger.error(f"删除租户失败: {str(e)}") raise BusinessException(f"删除租户失败: {str(e)}", code=BizCode.DB_ERROR) # 租户用户管理 def get_tenant_users( self, tenant_id: uuid.UUID, skip: int = 0, limit: int = 100, is_active: Optional[bool] = None, search: Optional[str] = None ) -> List[UserModel]: """获取租户下的用户列表""" return self.user_repo.get_users_by_tenant( tenant_id=tenant_id, skip=skip, limit=limit, is_active=is_active, search=search ) def count_tenant_users( self, tenant_id: uuid.UUID, is_active: Optional[bool] = None, search: Optional[str] = None ) -> int: """统计租户下的用户数量""" return self.user_repo.count_users_by_tenant( tenant_id=tenant_id, is_active=is_active, search=search ) def assign_user_to_tenant(self, user_id: uuid.UUID, tenant_id: uuid.UUID) -> bool: """将用户分配给租户""" # 检查租户是否存在 tenant = self.tenant_repo.get_tenant_by_id(tenant_id) if not tenant: raise BusinessException("租户不存在", code=BizCode.TENANT_NOT_FOUND) try: success = self.user_repo.assign_user_to_tenant(user_id, tenant_id) if success: business_logger.info(f"分配用户到租户成功: 用户ID {user_id}, 租户ID {tenant_id}") return success except Exception as e: business_logger.error(f"分配用户到租户失败: {str(e)}") raise BusinessException(f"分配用户到租户失败: {str(e)}", code=BizCode.DB_ERROR) def get_user_tenant(self, user_id: uuid.UUID) -> Optional[Tenants]: """获取用户所属的租户""" return self.tenant_repo.get_user_tenant(user_id) def remove_user_from_tenant(self, user_id: uuid.UUID) -> bool: """将用户从租户中移除（设置tenant_id为None）""" try: user = self.user_repo.get_user_by_id(user_id) if not user: return False success = self.user_repo.assign_user_to_tenant(user_id, None) if success: business_logger.info(f"移除用户租户关联成功: 用户ID {user_id}") return success except Exception as e: business_logger.error(f"移除用户租户关联失败: {str(e)}") raise BusinessException(f"移除用户租户关联失败: {str(e)}", code=BizCode.DB_ERROR) def get_users_without_tenant( self, skip: int = 0, limit: int = 100, is_active: Optional[bool] = None ) -> List[UserModel]: """获取没有租户的用户列表""" return self.user_repo.get_users_without_tenant( skip=skip, limit=limit, is_active=is_active )