SpringBoot项目限制带参数接口配置使用数量实现

最新推荐文章于 2025-10-24 17:24:34 发布

原创最新推荐文章于 2025-10-24 17:24:34 发布 · 316 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#spring boot #后端 #java

要求系统在使用过程中对某类带参数的接口做限制，也就只能配置几个接口供用户使用。带参数接口如：http://localhost:8088/sys/list?d=a001&c=b001,类似d=a001&c=b001这样的参数只能配置限定使用数量。超出使用数量，系统报出超过最大配置限制提醒。

1、过滤器

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;

/**
 * @Description:TODO
 * @Author:sclbq@sina.com
 * @date:2025年07月2815:35
 */
@Component
public class ApiLimitFilter implements Filter {
    // 允许的最大唯一组合数
    private static final int MAX_CONFIGS = 2;
    // 使用线程安全的Set存储唯一组合
    private final Set<String> configSet = new CopyOnWriteArraySet<>();
    private static final Logger logger = LoggerFactory.getLogger(ApiLimitFilter.class);

    @Override
    public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain)
            throws IOException, ServletException {

        HttpServletRequest request = (HttpServletRequest) req;
        HttpServletResponse response = (HttpServletResponse) res;

        String clinicCode = request.getParameter("clinicCode");
        String departCode = request.getParameter("departCode");

        logger.info("请求参数: clinicCode={}, departCode={}, 请求路径={}", clinicCode, departCode, request.getRequestURI());

        if (clinicCode == null || departCode == null) {
            logger.warn("缺少必要参数，clinicCode或departCode未提供");
            response.sendError(HttpServletResponse.SC_BAD_REQUEST, "缺少必要参数");
            return;
        }

        // 创建唯一标识符
        String configKey = clinicCode + ":" + departCode;

        // 添加新组合前检查数量
        if(!configSet.contains(configKey)) {
            if(configSet.size() >= MAX_CONFIGS) {
                logger.warn("超过最大配置限制: {}", configKey);
                response.sendError(403, "超过最大配置限制");
                return;
            }
            configSet.add(configKey);
            logger.info("新增配置组合: {}，当前总量: {}", configKey, configSet.size());
        }

        chain.doFilter(req, res);
    }

    @Override
    public void destroy() {
        configSet.clear();
    }
    /**
     * 供外部调用的清理方法
     */
    public void clearConfigSet() {
        configSet.clear();
        logger.info("configSet已手动清理");
    }
}

2、配置文件

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @Description:TODO
 * @Author:sclbq@sina.com
 * @date:2025年07月2815:32
 */
@Configuration
@Slf4j
public class ApiLimitConfig {
    @Autowired
    private ApiLimitFilter apiLimitFilter;
    @Bean
    public FilterRegistrationBean<ApiLimitFilter> customApiLimitFilter() {
        FilterRegistrationBean<ApiLimitFilter> registration = new FilterRegistrationBean<>();
        registration.setFilter(apiLimitFilter);
        // 添加多个URL匹配模式
//        registration.addUrlPatterns("/jhtqs/queuing/qspatienapi/*");
        registration.addUrlPatterns("/queuing/qspatienapi/*");
        registration.setOrder(1);
        log.info("对指定接口进行过滤时的初始化...");
        return registration;
    }
}