SpringBoot統(tǒng)計一個Bean中方法的調(diào)用次數(shù)的實現(xiàn)步驟

更新時間：2024年01月28日 09:03:46 作者：蒾酒

這篇文章主要給大家介紹了SpringBoot統(tǒng)計一個Bean中方法的調(diào)用次數(shù)的實現(xiàn)步驟,文中通過代碼示例和圖文結(jié)合的方式給大家講解的非常詳細,對大家的學習具有一定的幫助,需要的朋友可以參考下

實現(xiàn)思路

通過AOP即可實現(xiàn)，通過AOP對Bean進行代理，在每次執(zhí)行方法前或者后進行幾次計數(shù)統(tǒng)計。這個主要就是考慮好如何避免并發(fā)情況下不準，以及如何使用AOP實現(xiàn)代理。

前置條件

首先搭建一個spring boot工程，我這里用的是3x版本

搭建步驟：

新版idea創(chuàng)建springboot項目

新版idea創(chuàng)建spring boot項目的詳細教程_java_腳本之家 (jb51.net)

導入依賴：

pom.xml:

       <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>

實現(xiàn)步驟

首先我們先自定義一個注解

有了這個注解之后，我們可以在想要統(tǒng)計的方法上加上這個注解

名稱隨便起但要見名知意

代碼如下：

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
 
/**
 * @author mijiupro
 */
 
 
@Retention(RetentionPolicy.RUNTIME)// 指定注解的保留策略為RUNTIME
@Target(ElementType.METHOD)// 指定該注解可以用于方法
public @interface MethodCallCount {
}

接下來定義一個切面

通過該切面來對這個注解進行增強處理

代碼如下：

import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;
 
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
 
/**
 * @author mijiupro
 */
 
@Aspect// 聲明這是一個切面
@Component// 聲明這是一個Spring Bean(交給Spring管理)
@Slf4j
public class MethodCallCountAspect {
    // 用于存儲方法調(diào)用次數(shù)的Map，使用ConcurrentMap保證線程安全
    private final Map<String, AtomicInteger> counterMap = new ConcurrentHashMap<>();
    @Around("@annotation(com.mijiu.commom.aop.annotation.MethodCallCount)")
    public Object methodCallCountAspect(ProceedingJoinPoint joinPoint) {
 
        String methodName = joinPoint.getSignature().toShortString();
        try{
            return joinPoint.proceed();
        }catch (Throwable ignored){
        //異常處理
            return null;
        }finally {
            AtomicInteger counter = counterMap.computeIfAbsent(methodName,k -> new AtomicInteger(0));
            counter.incrementAndGet();
            log.info("方法 {} 調(diào)用次數(shù)：{}", methodName, counter.get());
        }
 
    }
 
    // 提供一個方法，用于獲取方法調(diào)用次數(shù)的Map
    public Map<String, AtomicInteger> getCounterMap() {
        return new ConcurrentHashMap<>(counterMap);
    }
}