本篇內(nèi)容介紹了“guava的RateLimiter怎么使用”的有關(guān)知識(shí),在實(shí)際案例的操作過程中,不少人都會(huì)遇到這樣的困境,接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧!希望大家仔細(xì)閱讀,能夠?qū)W有所成!
站在用戶的角度思考問題,與客戶深入溝通,找到城口網(wǎng)站設(shè)計(jì)與城口網(wǎng)站推廣的解決方案,憑借多年的經(jīng)驗(yàn),讓設(shè)計(jì)與互聯(lián)網(wǎng)技術(shù)結(jié)合,創(chuàng)造個(gè)性化、用戶體驗(yàn)好的作品,建站類型包括:網(wǎng)站制作、成都做網(wǎng)站、企業(yè)官網(wǎng)、英文網(wǎng)站、手機(jī)端網(wǎng)站、網(wǎng)站推廣、域名注冊(cè)、雅安服務(wù)器托管、企業(yè)郵箱。業(yè)務(wù)覆蓋城口地區(qū)。
常用的限流算法有漏桶算法和令牌桶算法,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í)直接返回失敗;
創(chuàng)建一個(gè)maven工程,在pom中把guava的依賴添加進(jìn)來:
com.google.guava guava 18.0
把限流服務(wù)封裝到一個(gè)類中AccessLimitService,提供tryAcquire()方法,用來嘗試獲取令牌,返回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上即可,接下來我們寫一個(gè)類,十個(gè)線程并發(fā)訪問上面寫的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 { // 打開和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輸入流來讀取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塊來關(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)求沒有拿到令牌,我們可以根據(jù)實(shí)際業(yè)務(wù)來做區(qū)分處理。還有一點(diǎn)要注意,我們通過RateLimiter.create(5.0)配置的是每一秒5枚令牌,但是限流的時(shí)候發(fā)出的是6枚,改用其他值驗(yàn)證,也是實(shí)際的比配置的大1。
“guava的RateLimiter怎么使用”的內(nèi)容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注創(chuàng)新互聯(lián)網(wǎng)站,小編將為大家輸出更多高質(zhì)量的實(shí)用文章!