springboot 自定义注解 入参赋值

2022-11-14 20:14:47 浏览数 (1)

第一步:自定义注解

代码语言:javascript复制
@Target({ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface TeacherInfoAnnotation {
}

注解声明方式

        通过关键字 @interface 声明为注解

注解的元素类型

        主要有@Target,@Retention,@Document,@Inherited 用来修饰注解。

   @Target 表明该注解可以应用的java元素类型。

Target类型

描述

ElementType.TYPE

应用于类、接口(包括注解类型)、枚举

ElementType.FIELD

应用于属性(包括枚举中的常量)

ElementType.METHOD

应用于方法

ElementType.PARAMETER

应用于方法的形参

ElementType.CONSTRUCTOR

应用于构造函数

ElementType.LOCAL_VARIABLE

应用于局部变量

ElementType.ANNOTATION_TYPE

应用于注解类型

ElementType.PACKAGE

应用于包

ElementType.TYPE_PARAMETER

1.8版本新增,应用于类型变量)

ElementType.TYPE_USE

1.8版本新增,应用于任何使用类型的语句中(例如声明语句、泛型和强制转换语句中的类型)

@Retention:表明该注解的生命周期

生命周期类型

描述

RetentionPolicy.SOURCE

编译时被丢弃,不包含在类文件中

RetentionPolicy.CLASS

JVM加载时被丢弃,包含在类文件中,默认值

RetentionPolicy.RUNTIME

由JVM 加载,包含在类文件中,在运行时可以被获取到

@Documented  表明该注解标记的元素可以被Javadoc 或类似的工具文档化。

@Inherited  表明使用了@Inherited注解的注解,所标记的类的子类也会拥有这个注解

第二步:创建一个处理方法参数的解析器

        这个类主要是针对于注解标记的参数进行处理

        第一个方法如果返回true时,会进入第二个方法

代码语言:javascript复制
@Component
public class TeacherInfoResover implements HandlerMethodArgumentResolver {
    @Autowired
    private TokenManager tokenManager;
/**
     * 此方法确定当前参数是否需要处理
     * @param methodParameter
     * @return
     */

    @Override
    public boolean supportsParameter(MethodParameter methodParameter) {
        if (methodParameter.hasParameterAnnotation(TeacherInfoAnnotation.class)) {
            return true;
        } else {
            return false;
        }
    }
/**
     * 此方法是对参数的处理结果,返回值将被赋值给handler method的对应参数
     * @param methodParameter
     * @param modelAndViewContainer
     * @param nativeWebRequest
     * @param webDataBinderFactory
     * @return
     * @throws Exception
     */

    @Override
    public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer, NativeWebRequest nativeWebRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
        String token = nativeWebRequest.getHeader("token");
        String subject = tokenManager.getTokenClaim(token).getSubject();
        TeacherInfoDTO teacherInfo = JSONUtil.readValue(subject, TeacherInfoDTO.class);
        return teacherInfo;
    }
}

第三步:将我们第二步创建好的参数解析器,添加到spring中

代码语言:javascript复制
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {

    @Bean
    public TeacherInfoResover teacherInfoResover() {
        return new TeacherInfoResover();
    }

    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
        resolvers.add(teacherInfoResover());
    }


}
jvm

0 人点赞