快捷導(dǎo)航

Java 實現(xiàn)限流器處理Rest接口請求詳解流程

更新時間：2021年11月02日 11:37:20 作者：劍客阿良_ALiang

在工作中是否會碰到這樣的場景，高并發(fā)的請求但是無法全部執(zhí)行，需要一定的限流。如果你是使用的微服務(wù)框架，比如SpringCloud，可以使用Gateway增加限流策略來解決。本篇文章是在沒有框架的情況實現(xiàn)限流器

Maven依賴

        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>31.0.1-jre</version>
        </dependency>

代碼

上代碼，不廢話。

首先是限流器代碼。

package com.huyi.csdn.tools.rate;
 
import com.google.common.util.concurrent.Monitor;
import com.google.common.util.concurrent.RateLimiter;
 
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.function.Consumer;
 
/**
 * @Program: csdn @ClassName: RequestRateLimiter @Author: huyi @Date: 2021-10-30 22:16 @Description:
 * 請求限流器 @Version: V1.0
 */
public class RequestRateLimiter {
  // 請求隊列
  private final ConcurrentLinkedQueue<Request> bucket = new ConcurrentLinkedQueue<>();
  // 隊列上限
  private static final int BUCKET_CAPACITY = 100;
  // 漏桶下沿水流速度
  private final RateLimiter rateLimiter = RateLimiter.create(10.0D);
  // 請求監(jiān)視器
  private final Monitor requestMoniter = new Monitor();
  // 處理監(jiān)視器
  private final Monitor handleMoniter = new Monitor();
 
  /** 請求實體 */
  public static class Request {
    private int data;
 
    public Request(int data) {
      this.data = data;
    }
 
    @Override
    public String toString() {
      return "Request{" + "data=" + data + '}';
    }
  }
 
  public void submitRequest(int data) {
    this.submitRequest(new Request(data));
  }
 
  public void submitRequest(Request request) {
    // 請求監(jiān)視器，創(chuàng)建監(jiān)視向?qū)?，隊列?shù)據(jù)量小于上限
    if (requestMoniter.enterIf(requestMoniter.newGuard(() -> bucket.size() < BUCKET_CAPACITY))) {
      try {
        boolean result = bucket.offer(request);
        if (result) {
          System.out.println("成功向隊列加入新的請求！" + Thread.currentThread() + " request:" + request);
        } else {
          System.out.println("加入新請求失?。?);
        }
      } finally {
        requestMoniter.leave();
      }
    } else {
      // 隊列已滿
      // System.out.println("請求隊列已經(jīng)上限，請稍后重試！");
    }
  }
 
  // 處理請求方法
  public void handleRequest(Consumer<Request> consumer) {
    if (handleMoniter.enterIf(handleMoniter.newGuard(() -> !bucket.isEmpty()))) {
      try {
        // 勻速處理
        rateLimiter.acquire();
        consumer.accept(bucket.poll());
      } finally {
        handleMoniter.leave();
      }
    }
  }
}

代碼說明

1、有個長度100的任務(wù)隊列，增加了監(jiān)視器。

2、添加了限流器，限流為10。

驗證代碼

package com.huyi.csdn.tools.rate;
 
import java.time.LocalTime;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.IntStream;
 
/**
 * @Program: csdn @ClassName: TestRateLimiter @Author: huyi @Date: 2021-10-30
 * 22:35 @Description: @Version: V1.0
 */
public class TestRateLimiter {
  private static final AtomicInteger DATA = new AtomicInteger(0);
  private static final RequestRateLimiter HANDLE = new RequestRateLimiter();
 
  public static void main(String[] args) {
    IntStream.range(0, 10)
        .forEach(
            (x) ->
                new Thread(
                        () -> {
                          while (true) {
                            HANDLE.submitRequest(DATA.getAndIncrement());
                            try {
                              TimeUnit.MILLISECONDS.sleep(100);
                            } catch (InterruptedException e) {
                              e.printStackTrace();
                            }
                          }
                        })
                    .start());
 
    IntStream.range(0, 20)
        .forEach(
            (x) ->
                new Thread(
                        () -> {
                          while (true) {
                            HANDLE.handleRequest(
                                y ->
                                    System.out.println(
                                        LocalTime.now() + ":處理數(shù)據(jù) -> " + y.toString()));
                          }
                        })
                    .start());
  }
}

驗證執(zhí)行結(jié)果