From 9b4433fddf5b401edb0aace8a404ac733b122702 Mon Sep 17 00:00:00 2001 From: 田源 <tianyuan@vci-tech.com> Date: 星期四, 03 四月 2025 14:35:02 +0800 Subject: [PATCH] 添加非密字段显示 --- Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/dynamic/TenantDataSourceAnnotationInterceptor.java | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 50 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/dynamic/TenantDataSourceAnnotationInterceptor.java b/Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/dynamic/TenantDataSourceAnnotationInterceptor.java new file mode 100644 index 0000000..54ee169 --- /dev/null +++ b/Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/dynamic/TenantDataSourceAnnotationInterceptor.java @@ -0,0 +1,50 @@ +/* + * Copyright (c) 2018-2028, Chill Zhuang All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * Neither the name of the dreamlu.net developer nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * Author: Chill 搴勯獮 (smallchill@163.com) + */ +package org.springblade.core.tenant.dynamic; + +import com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationInterceptor; +import com.baomidou.dynamic.datasource.processor.DsProcessor; +import lombok.Setter; +import org.aopalliance.intercept.MethodInvocation; +import org.springblade.core.secure.utils.AuthUtil; +import org.springblade.core.tenant.exception.TenantDataSourceException; + +/** + * 绉熸埛鏁版嵁婧愬垏鎹㈡嫤鎴櫒 + * + * @author Chill + */ +public class TenantDataSourceAnnotationInterceptor extends DynamicDataSourceAnnotationInterceptor { + + @Setter + private TenantDataSourceHolder holder; + + public TenantDataSourceAnnotationInterceptor(Boolean allowedPublicOnly, DsProcessor dsProcessor) { + super(allowedPublicOnly, dsProcessor); + } + + @Override + public Object invoke(MethodInvocation invocation) throws Throwable { + try { + holder.handleDataSource(AuthUtil.getTenantId()); + return super.invoke(invocation); + } catch (Exception exception) { + throw new TenantDataSourceException(exception.getMessage()); + } + } + +} -- Gitblit v1.9.3