Spring Boot实现数据访问计数器方案详解

在一个Web应用中，我们经常需要统计某些数据的访问次数，用于后续的分析或优化。Spring Boot提供了丰富的支持来实现这个计数器方案。

步骤一：定义计数器服务

首先我们需要定义一个计数器服务，用于记录各种数据的访问次数。这个服务可以定义为一个Spring Bean，并用注解标记为@Service：

@Service
public class CounterService {
    private Map<String, AtomicInteger> counters = new HashMap<>();

    public int increment(String key) {
        AtomicInteger counter = counters.computeIfAbsent(key, k -> new AtomicInteger());
        return counter.incrementAndGet();
    }

    public int getCount(String key) {
        AtomicInteger counter = counters.get(key);
        return counter != null ? counter.get() : 0;
    }
}

这个计数器服务中包含了一个并发安全的计数器AtomicInteger的集合counters，以及两个方法increment和getCount用于增加和获取计数器的值。

步骤二：定义计数器拦截器

接下来我们需要定义一个计数器拦截器，用于拦截指定URL的请求，并调用计数器服务来增加计数器的值。这个拦截器可以定义为一个Spring Bean，并用注解标记为@Component：

@Component
public class CounterInterceptor implements HandlerInterceptor {
    private CounterService counterService;

    public CounterInterceptor(CounterService counterService) {
        this.counterService = counterService;
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String path = request.getRequestURI();
        counterService.increment(path);
        return true;
    }
}

这个计数器拦截器中通过构造函数注入了计数器服务CounterService，并在preHandle方法中拦截了指定URL的请求，并调用计数器服务的increment方法来增加计数器的值。

步骤三：配置计数器拦截器

最后我们需要在Spring Boot的配置文件中配置计数器拦截器，使其生效。这个配置可以通过实现WebMvcConfigurer接口的方式来完成：

@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
    private CounterInterceptor counterInterceptor;

    public WebMvcConfig(CounterInterceptor counterInterceptor) {
        this.counterInterceptor = counterInterceptor;
    }

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(counterInterceptor).addPathPatterns("/api/**");
    }
}

这个配置中通过构造函数注入了计数器拦截器CounterInterceptor，并在addInterceptors方法中将其加入InterceptorRegistry中，并指定拦截的URL路径为/api/**。

示例一：统计API接口访问次数

假设我们有一个API接口，其URL为/api/books，用于获取书籍列表。我们希望统计该接口的访问次数。实现方式如下：

@RestController
@RequestMapping("/api")
public class ApiRestController {
    @GetMapping("/books")
    public List<Book> getBooks() {
        // ...
    }
}

示例二：统计网站页面访问次数

假设我们有一个网站，其页面URL为/pages/home，用于显示首页。我们希望统计该页面的访问次数。实现方式如下：

@Controller
public class PageController {
    @GetMapping("/pages/home")
    public String home(Model model) {
        // ...
        return "home";
    }
}

总结

通过以上步骤，我们就可以实现一个数据访问计数器，用于统计Web应用的各种数据的访问次数。在实际应用中，我们可以根据需要拦截不同的URL，并对不同的URL设置不同的计数器服务，以便对不同的数据进行统计和分析。