@Component
@Slf4j
public class CompanyContextInterceptor implements HandlerInterceptor {
    @Resource
    private RedisTemplate<String, Object> redisTemplate;
    private static final String REDIS_TOKEN_KEY_PREFIX = "login:";
    // 請求開始前：解析 Header，設(shè)置公司編碼到上下文
    @Override
    public boolean preHandle(HttpServletRequest request, @NotNull HttpServletResponse response, @NotNull Object handler) throws Exception {
        // 1. 從請求頭獲取Token（Bearer Token格式）
        String authHeader = request.getHeader("Authorization");
        if (StringUtils.isBlank(authHeader) || !authHeader.startsWith("Bearer ")) {
            // 未攜帶Token，直接通過（或返回401，根據(jù)業(yè)務(wù)需求）
            return true;
        }
        String token = authHeader.substring(7); // 截取Bearer后的Token
        // 2. 從Redis查詢登錄用戶信息
        String redisKey = REDIS_TOKEN_KEY_PREFIX + token;
        LoginController.LoginWxUser loginWxUser = (LoginController.LoginWxUser) redisTemplate.opsForValue().get(redisKey);
        if (loginWxUser == null || StringUtils.isBlank(loginWxUser.getPhone())) {
            // Token無效/過期，直接通過（或返回401）
            return true;
        }
        // 3. 核心：將loginWxUser存入request，key必須是「loginWxUser」（和接口一致?。?
        request.setAttribute("loginWxUser", loginWxUser);
        // 可選：Token續(xù)期
        redisTemplate.expire(redisKey, 7, TimeUnit.DAYS);
        // 獲取請求路徑
        String requestURI = request.getRequestURI();
        // 檢查是否為需要忽略的登錄接口
        if (requestURI.equals("/workaffairs/api/user/loginByPhone") ||
                requestURI.startsWith("/workaffairs/api/user/loginByPhone")) {
            return true; // 直接放行，不設(shè)置公司編碼
        }
        // 從 Header 獲取公司編碼
        String companyCode = request.getHeader("x-Company-Code");
        if (companyCode != null && !companyCode.isEmpty()) {
            CompanyContextHolder.setCompanyCode(companyCode);
            log.info("獲取到公司編碼----" + companyCode);
        }
        return true; // 放行請求
    }
    // 請求結(jié)束后：清理上下文（必須執(zhí)行，避免內(nèi)存泄漏）
    @Override
    public void afterCompletion(@NotNull HttpServletRequest request, @NotNull HttpServletResponse response, @NotNull Object handler, Exception ex) throws Exception {
        CompanyContextHolder.clear();
    }
}

/**
 * 攔截器配置類：注冊攔截器并配置攔截規(guī)則
 */
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
    // 注入自定義攔截器
    @Resource
    private CompanyContextInterceptor companyContextInterceptor;
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(companyContextInterceptor)
                // 攔截所有需要切換數(shù)據(jù)源的業(yè)務(wù)接口（根據(jù)你的實(shí)際路徑調(diào)整）
                .addPathPatterns("/purchase/**", "/delivery/**","/api/user/getSupplierByCompany")
                // 排除不需要攔截的接口（比如登錄接口）
                .excludePathPatterns("/api/user/loginByPhone");
    }
}

@Aspect // 聲明為切面類
@Component // 交給Spring容器管理
@Slf4j
public class DbOperationLogAspect {
    /**
     * 定義切點(diǎn)：指定切面作用的目標(biāo)方法
     * 表達(dá)式規(guī)則：execution(訪問修飾符 返回值 包名.類名.方法名(參數(shù)類型))
     * 示例：匹配com.tgerp.workaffairs.mapper包下所有類的所有方法
     */
    @Pointcut("execution(* com.tgerp.workaffairs.mapper..*(..))")
    public void dbOperationPointCut() {
    }
    @Around("dbOperationPointCut()")
    public Object aroundDbOperation(ProceedingJoinPoint joinPoint) throws Throwable {
        // 1. 此時(shí)controller已經(jīng)執(zhí)行了setDbName，能拿到正確的數(shù)據(jù)庫名
        String currentDbName = DynamicDataSourceContextHolder.peek(); // 從數(shù)據(jù)源上下文拿
//        }
        String className = joinPoint.getTarget().getClass().getName();
        String methodName = joinPoint.getSignature().getName();
        // 打印日志
        log.info("[workaffairs-數(shù)據(jù)庫: {}] 執(zhí)行數(shù)據(jù)庫操作 - 類：{}，方法：{}", currentDbName, className, methodName);
        try {
            return joinPoint.proceed();
        } catch (Exception e) {
            log.error("[workaffairs-數(shù)據(jù)庫: {}] 數(shù)據(jù)庫操作失敗 - 類：{}，方法：{}，異常信息：{}",
                    currentDbName, className, methodName, e.getMessage(), e);
            throw e;
        }
    }
}