实际工作中哪里用到了自定义注解的方法

2024年 1月 12日 75.6k 0

自定义注解可以标记在方法上或类上,用于在编译期或运行期进行特定的业务功能处理。

在Java中,自定义注解使用 @interface 关键字来定义,它可以实现如:日志记录、性能监控、权限校验等功能。

在 Spring Boot 中实现一个自定义注解,可以通过 AOP(面向切面编程)或拦截器(Interceptor)来实现。

图片[1]-实际工作中哪里用到了自定义注解的方法-不念博客

1.实现自定义注解

下面我们先使用 AOP 的方式来实现一个打印日志的自定义注解,它的实现步骤如下:

  • 添加 Spring AOP 依赖。
  • 创建自定义注解。
  • 编写 AOP 拦截(自定义注解)的逻辑代码。
  • 使用自定义注解。
  • 具体实现如下。

    ① 添加 Spring AOP 依赖

    在 pom.xml 中添加如下依赖:

    <dependencies>
      <!-- Spring AOP dependency -->
      <dependency>
        <groupIdorg.springframework.boot</groupId>
          <artifactIdspring-boot-starter-aop</artifactId>
          </dependency>
    </dependencies>

    ② 创建自定义注解

    创建一个新的 Java 注解类,通过 @interface 关键字来定义,并可以添加元注解以及属性。

    import java.lang.annotation.*;
    @Target(ElementType.METHOD)
    @Retention(RetentionPolicy.RUNTIME)
    public @interface CustomLogAnnotation {
        String value() default "";
        boolean enable() default true;
    }

    在上面的例子中,我们定义了一个名为 CustomLogAnnotation 的注解,它有两个属性:value 和 enable,分别设置了默认值。

    • @Target(ElementType.METHOD) 指定了该注解只能应用于方法级别。
    • @Retention(RetentionPolicy.RUNTIME) 表示这个注解在运行时是可见的,这样 AOP 代理才能在运行时读取到这个注解。

    ③ 编写 AOP 拦截(自定义注解)的逻辑代码

    使用 Spring AOP 来拦截带有自定义注解的方法,并在其前后执行相应的逻辑。

    import org.aspectj.lang.ProceedingJoinPoint;
    import org.aspectj.lang.annotation.Around;
    import org.aspectj.lang.annotation.Aspect;
    import org.springframework.stereotype.Component;
    @Aspect
    @Component
    public class CustomLogAspect {
        @Around("@annotation(customLog)")
        public Object logAround(ProceedingJoinPoint joinPoint, CustomLogAnnotation customLog) throws Throwable {
            if (customLog.enable()) {
                // 方法执行前的处理
                System.out.println("Before method execution: " + joinPoint.getSignature().getName());
                long start = System.currentTimeMillis();
                // 执行目标方法
                Object result = joinPoint.proceed();
                // 方法执行后的处理
                long elapsedTime = System.currentTimeMillis() - start;
                System.out.println("After method execution (" + elapsedTime + 
                                   "ms): " + customLog.value());
                return result;
            } else {
                return joinPoint.proceed();
            }
        }
    }

    ④ 使用自定义注解

    将自定义注解应用于需要进行日志记录的方法上,如下代码所示:

    @RestController
    public class MyController {
        @CustomLogAnnotation(value = "This is a test method", enable = true)
        @GetMapping("/test")
        public String testMethod() {
            // 业务逻辑代码
            return "Hello from the annotated method!";
        }
    }

    2.实际工作中的自定义注解

    实际工作中我们通常会使用自定义注解来实现如权限验证,或者是幂等性判断等功能。

    幂等性判断是指在分布式系统或并发环境中,对于同一操作的多次重复请求,系统的响应结果应该是一致的。简而言之,无论接收到多少次相同的请求,系统的行为和结果都应该是相同的。”

    3.如何实现自定义幂等性注解?

    下面我们使用拦截器 + Redis 的方式来实现一下自定义幂等性注解,它的实现步骤如下:

  • 创建自定义幂等性注解。
  • 创建拦截器,实现幂等性逻辑判断。
  • 配置拦截规则。
  • 使用自定义幂等性注解。
  • 具体实现如下。

    ① 创建自定义幂等性注解

    @Retention(RetentionPolicy.RUNTIME) // 程序运行时有效
    @Target(ElementType.METHOD) // 方法注解
    public @interface Idempotent {
        /**
         * 请求标识符的参数名称,默认为"requestId"
         */
        String requestId() default "requestId";
        /**
         * 幂等有效时长(单位:秒)
         */
        int expireTime() default 60;
    }

    ② 创建拦截器

    @Component
    public class IdempotentInterceptor extends HandlerInterceptorAdapter {
        @Autowired
        private RedisTemplate<String, Object> redisTemplate;
        @Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
            Method method = ((HandlerMethod) handler).getMethod();
            Idempotent idempotent = method.getAnnotation(Idempotent.class);
            if (idempotent != null) {
                // 获取请求中的唯一标识符
                String requestId = obtainRequestId(request, idempotent.requestId());
                // 判断该请求是否已经处理过
                if (redisTemplate.opsForValue().get(idempotentKey(requestId)) != null) {
                    // 已经处理过,返回幂等响应
                    response.getWriter().write("重复请求");
                    return false;
                } else {
                    // 将请求标识符存入Redis,并设置过期时间
                    redisTemplate.opsForValue().set(idempotentKey(requestId), "processed", idempotent.expireTime(), TimeUnit.SECONDS);
                    return true; // 继续执行业务逻辑
                }
            }
            return super.preHandle(request, response, handler);
        }
    
        private String idempotentKey(String requestId) {
            return "idempotent:" + requestId;
        }
    
        private String obtainRequestId(HttpServletRequest request, String paramName) {
            // 实现从请求中获取唯一标识符的方法
            return request.getParameter(paramName);
        }
    }

    ③ 配置拦截器

    在 Spring Boot 配置文件类中,添加拦截器配置:

    @Configuration
    public class WebConfig implements WebMvcConfigurer {
        @Autowired
        private IdempotentInterceptor idempotentInterceptor;
        @Override
        public void addInterceptors(InterceptorRegistry registry) {
            registry.addInterceptor(idempotentInterceptor)
             .addPathPatterns("/**"); // 拦截所有接口
        }
    }

    ④ 使用自定义注解

    最后,在需要进行幂等控制的 Controller 方法上使用 @Idempotent 注解:

    Java
    @RestController
    public class TestController {
        @PostMapping("/order")
        @Idempotent(requestId = "orderId") // 假设orderId是从客户端传来的唯一标识订单请求的参数
        public String placeOrder(@RequestParam("orderId") String orderId, ...) {
            // 业务处理逻辑
        }
    }

    这样,当有相同的请求 ID 在指定的有效期内再次发起请求时,会被拦截器识别并阻止其重复执行业务逻辑。

    总结

    自定义注解被广泛应用于日常开发中,像日志记录、性能监控、权限判断和幂等性判断等功能的实现,使用自定义注解来实现是非常方便的。

    在 Spring Boot 中,使用 @interface 关键字来定义自定义注解,之后再使用 AOP 或拦截器的方式实现自定义注解,就可以方便的使用自定义注解了。

    相关文章

    JavaScript2024新功能:Object.groupBy、正则表达式v标志
    PHP trim 函数对多字节字符的使用和限制
    新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
    使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
    为React 19做准备:WordPress 6.6用户指南
    如何删除WordPress中的所有评论

    发布评论