feat(tenant): 实现多租户功能支持

2025-12-10 21:14:37 +08:00
parent f16c1e6227
commit 329b3551f7
17 changed files with 1787 additions and 0 deletions
--- a/src/main/java/com/youlai/boot/system/controller/TenantController.java
+++ b/src/main/java/com/youlai/boot/system/controller/TenantController.java
@@ -0,0 +1,110 @@
+package com.youlai.boot.system.controller;
+
+import com.youlai.boot.common.tenant.TenantContextHolder;
+import com.youlai.boot.core.web.Result;
+import com.youlai.boot.security.util.SecurityUtils;
+import com.youlai.boot.system.model.vo.TenantVO;
+import com.youlai.boot.system.service.TenantService;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
+
+/**
+ * 租户管理控制器
+ * <p>
+ * 提供租户切换、查询等功能
+ * </p>
+ *
+ * @author Ray.Hao
+ * @since 3.0.0
+ */
+@Tag(name = "租户管理接口")
+@RestController
+@RequestMapping("/api/v1/tenant")
+@RequiredArgsConstructor
+@Slf4j
+@ConditionalOnProperty(prefix = "youlai.tenant", name = "enabled", havingValue = "true", matchIfMissing = false)
+public class TenantController {
+
+    private final TenantService tenantService;
+
+    /**
+     * 获取当前用户的租户列表
+     * <p>
+     * 根据当前登录用户查询其所属的所有租户
+     * </p>
+     *
+     * @return 租户列表
+     */
+    @Operation(summary = "获取当前用户的租户列表")
+    @GetMapping("/list")
+    public Result<List<TenantVO>> getTenantList() {
+        Long userId = SecurityUtils.getUserId();
+        List<TenantVO> tenantList = tenantService.getTenantListByUserId(userId);
+        log.info("获取用户 {} 的租户列表，共 {} 个租户", userId, tenantList.size());
+        return Result.success(tenantList);
+    }
+
+    /**
+     * 获取当前租户信息
+     *
+     * @return 当前租户信息
+     */
+    @Operation(summary = "获取当前租户信息")
+    @GetMapping("/current")
+    public Result<TenantVO> getCurrentTenant() {
+        Long tenantId = TenantContextHolder.getTenantId();
+        if (tenantId == null) {
+            return Result.success(null);
+        }
+        TenantVO tenant = tenantService.getTenantById(tenantId);
+        return Result.success(tenant);
+    }
+
+    /**
+     * 切换租户
+     * <p>
+     * 切换当前用户的租户上下文，需要验证用户是否有权限访问该租户
+     * </p>
+     *
+     * @param tenantId 目标租户ID
+     * @return 切换结果
+     */
+    @Operation(summary = "切换租户")
+    @PostMapping("/switch/{tenantId}")
+    public Result<TenantVO> switchTenant(
+            @Parameter(description = "租户ID") @PathVariable Long tenantId
+    ) {
+        Long userId = SecurityUtils.getUserId();
+        log.info("用户 {} 请求切换租户到 {}", userId, tenantId);
+
+        // 验证用户是否有权限访问该租户
+        boolean hasPermission = tenantService.hasTenantPermission(userId, tenantId);
+        if (!hasPermission) {
+            log.warn("用户 {} 无权限访问租户 {}", userId, tenantId);
+            return Result.failed("无权限访问该租户");
+        }
+
+        // 验证租户是否存在且正常
+        TenantVO tenant = tenantService.getTenantById(tenantId);
+        if (tenant == null) {
+            return Result.failed("租户不存在");
+        }
+        if (tenant.getStatus() == null || tenant.getStatus() != 1) {
+            return Result.failed("租户已禁用");
+        }
+
+        // 设置新的租户上下文
+        TenantContextHolder.setTenantId(tenantId);
+        log.info("用户 {} 成功切换租户到 {}", userId, tenantId);
+
+        return Result.success(tenant);
+    }
+}
+