wangli/axzo-framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加context info Handler

Browse Source
This commit is contained in:
TanJ 2023-03-29 11:27:24 +08:00
parent 519f0c9e13
commit fed63c5607
4 changed files with 38 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
axzo-auth-spring-boot-starter/src/main/java/cn/axzo/framework/auth/annotation/PreBuildContext.java
View File

@ -7,10 +7,13 @@ import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import cn.axzo.framework.auth.domain.ContextInfo;
import cn.axzo.framework.auth.domain.contextvalidate.ContextInfoHandler;
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface PreBuildContext {
Class<? extends ContextInfo> value() default ContextInfo.class;
Class<? extends ContextInfoHandler>[] filter() default {};
}

12
axzo-auth-spring-boot-starter/src/main/java/cn/axzo/framework/auth/domain/contextvalidate/ContextInfoHandler.java Normal file
View File

@ -0,0 +1,12 @@
package cn.axzo.framework.auth.domain.contextvalidate;
import cn.axzo.framework.auth.domain.ContextInfo;
/**
* @author tanjie@axzo.cn
* @date 2023/3/29 10:59
*/
public interface ContextInfoHandler {
void handler(ContextInfo contextInfo);
}

18
axzo-auth-spring-boot-starter/src/main/java/cn/axzo/framework/auth/domain/contextvalidate/VerifyContext.java Normal file
View File

@ -0,0 +1,18 @@
package cn.axzo.framework.auth.domain.contextvalidate;
import cn.axzo.framework.auth.AuthException;
import cn.axzo.framework.auth.domain.ContextInfo;
/**
* @author tanjie@axzo.cn
* @date 2023/3/29 11:12
*/
public class VerifyContext implements ContextInfoHandler{
@Override
public void handler(ContextInfo contextInfo) {
Integer verifiedStatus = contextInfo.getUserInfo().getVerifiedStatus();
if (2 != verifiedStatus) {
throw new AuthException("该功能需要实名认证之后才可使用");
}
}
}

6
axzo-auth-spring-boot-starter/src/main/java/cn/axzo/framework/auth/service/ContextInfoBuilderAspect.java
View File

@ -11,6 +11,7 @@ import cn.axzo.framework.auth.domain.SystemAndDeviceInfo;
import cn.axzo.framework.auth.domain.TerminalInfo;
import cn.axzo.framework.auth.domain.UserInfo;
import cn.axzo.framework.auth.domain.UserInfoMap;
import cn.axzo.framework.auth.domain.contextvalidate.ContextInfoHandler;
import cn.axzo.framework.auth.enums.EnvEnum;
import cn.azxo.framework.common.logger.logback.PodNamespacePropertyDefiner;
import cn.hutool.core.bean.BeanUtil;
@ -124,7 +125,10 @@ public class ContextInfoBuilderAspect {
// 定制一些信息的处理
contextInfo.buildCustomInfo(httpRequest, contextInfo.getUserInfoMap());
//针对contextInfo拦截。或修改参数等
for (Class<? extends ContextInfoHandler> filter : preBuildContext.filter()) {
filter.newInstance().handler(contextInfo);
}
// 把ContextInfo放到ThreadLocal中
ContextInfoHolder.set(contextInfo);
Object[] args = parseContextInfoAndReturnArgs(httpRequest, pjp, contextInfo);