SpringBoot接口限流器

package com.ciih.iot.common.util;

import com.google.common.util.concurrent.RateLimiter;

import java.util.concurrent.ConcurrentHashMap;


/**
 * 接口限流器
 *
 * @author sunziwen
 * @date 2023/7/17
 */
public class LimitingUtil {

    public final static ConcurrentHashMap<String, RateLimiter> HASHMAP = new ConcurrentHashMap<>();


    /**
     * 限流策略
     *
     * @param QPS      每秒吞吐量
     * @param uniqueId 唯一标识，此ID不可重复，否则会出现限流穿透。
     * @return boolean 是否放行
     */
    public static boolean limit(double QPS, String uniqueId) {
        RateLimiter rateLimiter = HASHMAP.get(uniqueId);
        if (rateLimiter == null) {
            // 初始化限流器
            rateLimiter = RateLimiter.create(QPS);
            HASHMAP.put(uniqueId, rateLimiter);
        }
        return rateLimiter.tryAcquire();
    }

    public static void main(String[] args) throws InterruptedException {
        for (int i = 0; i < 50; i++) {
            // 每秒放置0.8个通行牌。
            if (LimitingUtil.limit(0.8, "iotDeviceInnerCarBarrierChannel/remoteDoorClosing")) {
                System.out.println("放行");
            } else {
                System.out.println("操作频繁");
            }
            Thread.sleep(500);
        }
    }
}