快捷導(dǎo)航

SpringBoot?Aop實現(xiàn)接口請求次數(shù)統(tǒng)計

更新時間：2024年02月06日 14:44:20 作者：wx59bcc77095d22

我們通過Spring AOP在每次執(zhí)行方法前或執(zhí)行方法后進行切面的處理，進而統(tǒng)計方法訪問的次數(shù)等功能，本文主要介紹了SpringBoot?Aop實現(xiàn)接口請求次數(shù)統(tǒng)計

一、前言

我們通過Spring AOP在每次執(zhí)行方法前或執(zhí)行方法后進行切面的處理，進而統(tǒng)計方法訪問的次數(shù)等功能，可以持久化到數(shù)據(jù)庫或者后面進行接口并發(fā)的處理。

二、添加依賴

<dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.31</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>

三、項目配置

我們主要添加數(shù)據(jù)庫相關(guān)的配置。

spring.datasource.url=jdbc:mysql://127.0.0.1:3306/aop_db?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8&useSSL=false&allowPublicKeyRetrieval=true
spring.datasource.username=root
spring.datasource.password=123456
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver

spring.jpa.hibernate.ddl-auto=update
spring.jpa.show-sql=true

四、創(chuàng)建自定義計數(shù)注解

package com.example.aopdemo.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 訪問次數(shù)注解
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface VisitCount {

}

五、創(chuàng)建實體類和持久層

package com.example.aopdemo.entity;

import javax.persistence.*;

/**
 * @author qx
 * @date 2024/2/4
 * @des 訪問情況實體
 */
@Entity
@Table(name = "t_visit")
public class Visit {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    /**
     * 方法名稱
     */
    private String methodName;

    /**
     * 請求路徑
     */
    private String url;

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public String getMethodName() {
        return methodName;
    }

    public void setMethodName(String methodName) {
        this.methodName = methodName;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }
}

package com.example.aopdemo.repository;

import com.example.aopdemo.entity.Visit;
import org.springframework.data.jpa.repository.JpaRepository;

/**
 * @author qx
 * @date 2024/2/4
 * @des 訪問請求持久層
 */
public interface VisitRepository extends JpaRepository<Visit, Long> {
}

六、創(chuàng)建切面

package com.example.aopdemo.aspect;

import com.example.aopdemo.entity.Visit;
import com.example.aopdemo.repository.VisitRepository;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

/**
 * @author qx
 * @date 2024/2/4
 * @des 方法訪問切面
 */
@Aspect
@Component
public class VisitCountAspect {

    @Autowired
    private VisitRepository visitRepository;

    @Around("@annotation(com.example.aopdemo.annotation.VisitCount)")
    public Object VisitCountAround(ProceedingJoinPoint joinPoint) throws Throwable {
        // 獲取當(dāng)前請求的URL
        String url = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest().getRequestURL().toString();
        // 方法名
        String methodName = joinPoint.getSignature().getName();

        // 保存訪問數(shù)據(jù)
        Visit visit = new Visit();
        visit.setUrl(url);
        visit.setMethodName(methodName);
        visitRepository.save(visit);
        return joinPoint.proceed();
    }

}

七、創(chuàng)建控制層并添加注解

package com.example.aopdemo.controller;

import com.example.aopdemo.annotation.VisitCount;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * @author qx
 * @date 2024/2/4
 * @des
 */
@RestController
@RequestMapping("/index")
public class IndexController {

    @GetMapping("/one")
    @VisitCount
    public String showOne() {
        return "one";
    }

    @GetMapping("/two")
    @VisitCount
    public String showTwo() {
        return "two";
    }
}

八、測試

啟動程序，在瀏覽器上訪問請求地址。

SpringBoot Aop實現(xiàn)接口請求次數(shù)統(tǒng)計_spring

我們刷新數(shù)據(jù)庫，發(fā)現(xiàn)新增了數(shù)據(jù)。這樣我們就可以根據(jù)自己的需求獲取請求次數(shù)。

SpringBoot Aop實現(xiàn)接口請求次數(shù)統(tǒng)計_aop_02

到此這篇關(guān)于SpringBoot Aop實現(xiàn)接口請求次數(shù)統(tǒng)計的文章就介紹到這了,更多相關(guān)SpringBoot Aop接口請求次數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: