亚洲激情专区-91九色丨porny丨老师-久久久久久久女国产乱让韩-国产精品午夜小视频观看

溫馨提示×

java接口限流如何實現

小億
153
2023-10-24 23:22:14
欄目: 編程語言

在Java中實現接口限流可以使用以下幾種方式:

  1. 計數器:維護一個計數器來統計每個接口的請求數量,當請求數量超過設定的閾值時,拒絕后續的請求。這可以使用并發容器如ConcurrentHashMap來實現,其中接口作為key,計數器作為value。
import java.util.concurrent.ConcurrentHashMap;

public class RateLimiter {
    private static ConcurrentHashMap<String, Integer> counters = new ConcurrentHashMap<>();
    private static final int MAX_REQUESTS = 100; // 設定的閾值

    public static boolean allowRequest(String interfaceName) {
        counters.putIfAbsent(interfaceName, 0);
        int count = counters.get(interfaceName);
        if (count >= MAX_REQUESTS) {
            return false;
        }
        counters.put(interfaceName, count + 1);
        return true;
    }

    public static void main(String[] args) {
        String interfaceName = "interface1";
        for (int i = 0; i < 110; i++) {
            if (allowRequest(interfaceName)) {
                System.out.println("Allow request for interface: " + interfaceName);
            } else {
                System.out.println("Reject request for interface: " + interfaceName);
            }
        }
    }
}
  1. 滑動窗口:使用一個固定長度的時間窗口,統計窗口內的請求數量。當請求數量超過設定的閾值時,拒絕后續的請求。這可以使用隊列或數組來保存請求的時間戳,并通過計算窗口內的請求數量來進行限流。
import java.util.ArrayDeque;
import java.util.Queue;

public class RateLimiter {
    private static Queue<Long> timestamps = new ArrayDeque<>();
    private static final int WINDOW_SIZE = 1000; // 窗口大小,單位為毫秒
    private static final int MAX_REQUESTS = 100; // 設定的閾值

    public static boolean allowRequest() {
        long now = System.currentTimeMillis();
        timestamps.offer(now);
        while (!timestamps.isEmpty() && now - timestamps.peek() > WINDOW_SIZE) {
            timestamps.poll();
        }
        return timestamps.size() <= MAX_REQUESTS;
    }

    public static void main(String[] args) {
        for (int i = 0; i < 110; i++) {
            if (allowRequest()) {
                System.out.println("Allow request");
            } else {
                System.out.println("Reject request");
            }
            try {
                Thread.sleep(100);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}
  1. 令牌桶:使用一個固定速率產生令牌,每個請求需要獲取一個令牌才能通過。當令牌數量不足時,拒絕后續的請求。這可以使用ScheduledExecutorService來定時產生令牌,并使用Semaphore來控制令牌的獲取。
import java.util.concurrent.Semaphore;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class RateLimiter {
    private static Semaphore tokens = new Semaphore(10); // 初始令牌數量
    private static final int RATE = 1; // 產生令牌的速率,單位為個/秒

    public static boolean allowRequest() {
        return tokens.tryAcquire();
    }

    public static void main(String[] args) {
        ScheduledThreadPoolExecutor executor = new ScheduledThreadPoolExecutor(1);
        executor.scheduleAtFixedRate(() -> {
            int availableTokens = tokens.availablePermits();
            if (availableTokens < RATE) {
                tokens.release(RATE - availableTokens);
            }
        }, 0, 1, TimeUnit.SECONDS);

        for (int i = 0; i < 20; i++) {
            if (allowRequest()) {
                System.out.println("Allow request");
            } else {
                System.out.println("Reject request");
            }
            try {
                Thread.sleep(100);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

        executor.shutdown();
    }
}

以上是幾種常見的Java接口限流的實現方式,可以根據實際需求選擇適合的方式。

0
赫章县| 库伦旗| 金平| 汶川县| 潼南县| 吉木萨尔县| 屯留县| 曲水县| 文登市| 山阴县| 米林县| 云霄县| 青冈县| 徐水县| 奉节县| 儋州市| 南汇区| 龙陵县| 阿克陶县| 茌平县| 武冈市| 顺昌县| 洛宁县| 松江区| 阿坝| 榆社县| 安徽省| 永康市| 兰溪市| 朔州市| 简阳市| 临沭县| 门源| 乐安县| 晋中市| 香格里拉县| 城固县| 钦州市| 兴山县| 北川| 泰来县|