guava的RateLimiter怎么使用

本篇內(nèi)容介紹了“guava的RateLimiter怎么使用”的有關(guān)知識(shí)，在實(shí)際案例的操作過(guò)程中，不少人都會(huì)遇到這樣的困境，接下來(lái)就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

關(guān)于限流

常用的限流算法有漏桶算法和令牌桶算法，guava的RateLimiter使用的是令牌桶算法，也就是以固定的頻率向桶中放入令牌，例如一秒鐘10枚令牌，實(shí)際業(yè)務(wù)在每次響應(yīng)請(qǐng)求之前都從桶中獲取令牌，只有取到令牌的請(qǐng)求才會(huì)被成功響應(yīng)，獲取的方式有兩種：阻塞等待令牌或者取不到立即返回失敗，下圖：

本次實(shí)戰(zhàn)，我們用的是guava的RateLimiter，場(chǎng)景是spring mvc在處理請(qǐng)求時(shí)候，從桶中申請(qǐng)令牌，申請(qǐng)到了就成功響應(yīng)，申請(qǐng)不到時(shí)直接返回失??；

實(shí)戰(zhàn)開(kāi)發(fā)

創(chuàng)建一個(gè)maven工程，在pom中把guava的依賴(lài)添加進(jìn)來(lái)：

<dependency>
      <groupId>com.google.guava</groupId>
      <artifactId>guava</artifactId>
      <version>18.0</version>
    </dependency>

把限流服務(wù)封裝到一個(gè)類(lèi)中AccessLimitService，提供tryAcquire()方法，用來(lái)嘗試獲取令牌，返回true表示獲取到，如下所示：

@Service
public class AccessLimitService {

    //每秒只發(fā)出5個(gè)令牌
    RateLimiter rateLimiter = RateLimiter.create(5.0);

    /**
     * 嘗試獲取令牌
     * @return
     */
    public boolean tryAcquire(){
        return rateLimiter.tryAcquire();
    }
}

調(diào)用方是個(gè)普通的controller，每次收到請(qǐng)求的時(shí)候都嘗試去獲取令牌，獲取成功和失敗打印不同的信息，如下：

@Controller
public class HelloController {

	private static SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

	@Autowired
	private AccessLimitService accessLimitService;

	@RequestMapping("/access")
	@ResponseBody
	public String access(){
		//嘗試獲取令牌
		if(accessLimitService.tryAcquire()){
			//模擬業(yè)務(wù)執(zhí)行500毫秒
			try {
				Thread.sleep(500);
			}catch (InterruptedException e){
				e.printStackTrace();
			}
			return "aceess success [" + sdf.format(new Date()) + "]";
		}else{
			return "aceess limit [" + sdf.format(new Date()) + "]";
		}
	}
}

以上就是服務(wù)端的代碼了，打包部署在tomcat上即可，接下來(lái)我們寫(xiě)一個(gè)類(lèi)，十個(gè)線程并發(fā)訪問(wèn)上面寫(xiě)的controller：

public class AccessClient {
    ExecutorService fixedThreadPool = Executors.newFixedThreadPool(10);

    /**
     * get請(qǐng)求
     * @param realUrl
     * @return
     */
    public static String sendGet(URL realUrl) {
        String result = "";
        BufferedReader in = null;
        try {
            // 打開(kāi)和URL之間的連接
            URLConnection connection = realUrl.openConnection();
            // 設(shè)置通用的請(qǐng)求屬性
            connection.setRequestProperty("accept", "*/*");
            connection.setRequestProperty("connection", "Keep-Alive");
            connection.setRequestProperty("user-agent",
                    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            // 建立實(shí)際的連接
            connection.connect();

            // 定義 BufferedReader輸入流來(lái)讀取URL的響應(yīng)
            in = new BufferedReader(new InputStreamReader(
                    connection.getInputStream()));
            String line;
            while ((line = in.readLine()) != null) {
                result += line;
            }
        } catch (Exception e) {
            System.out.println("發(fā)送GET請(qǐng)求出現(xiàn)異常！" + e);
            e.printStackTrace();
        }
        // 使用finally塊來(lái)關(guān)閉輸入流
        finally {
            try {
                if (in != null) {
                    in.close();
                }
            } catch (Exception e2) {
                e2.printStackTrace();
            }
        }
        return result;
    }



    public void access() throws Exception{
        final URL url = new URL("http://localhost:8080/guavalimitdemo/access");

        for(int i=0;i<10;i++) {
            fixedThreadPool.submit(new Runnable() {
                public void run() {
                    System.out.println(sendGet(url));
                }
            });
        }

        fixedThreadPool.shutdown();
        fixedThreadPool.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
    }

    public static void main(String[] args) throws Exception{
        AccessClient accessClient = new AccessClient();
        accessClient.access();
    }
}

直接執(zhí)行AccessClient的main方法，可以看到結(jié)果如下：

部分請(qǐng)求由于獲取的令牌可以成功執(zhí)行，其余請(qǐng)求沒(méi)有拿到令牌，我們可以根據(jù)實(shí)際業(yè)務(wù)來(lái)做區(qū)分處理。還有一點(diǎn)要注意，我們通過(guò)RateLimiter.create(5.0)配置的是每一秒5枚令牌，但是限流的時(shí)候發(fā)出的是6枚，改用其他值驗(yàn)證，也是實(shí)際的比配置的大1。

“guava的RateLimiter怎么使用”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！