1、說明
java已經為我們提供了解決方案。 jdk1.5帶來的並發庫Future類別可以滿足此需求。 Future類別中重要的方法有get()和cancel()。 get()取得資料對象,如果資料沒有加載,則在取得資料之前堵塞,cancel()取消資料加載。另一個get(timeout)操作表明,如果timeout時間內沒有得到,就會失敗回來,不會堵塞。
利用泛型和函數式介面編寫一個工具類,可以讓超時處理更方便,而不用到處寫程式碼。
2、實例
/** * TimeoutUtil <br> * * @author lys * @date 2021/2/25 */ @Slf4j @Component @NoArgsConstructor public class TimeoutUtil { private ExecutorService executorService; public TimeoutUtil(ExecutorService executorService) { this.executorService = executorService; } /** * 有超时限制的方法 * * @param bizSupplier 业务函数 * @param timeout 超时时间,ms * @return 返回值 */ public <R> Result<R> doWithTimeLimit(Supplier<R> bizSupplier, int timeout) { return doWithTimeLimit(bizSupplier, null, timeout); } /** * 有超时限制的方法 * * @param bizSupplier 业务函数 * @param defaultResult 默认值 * @param timeout 超时时间,ms * @return 返回值 */ public <R> Result<R> doWithTimeLimit(Supplier<R> bizSupplier, R defaultResult, int timeout) { R result; String errMsg = "Null value"; FutureTask<R> futureTask = new FutureTask<>(bizSupplier::get); executorService.execute(futureTask); try { result = futureTask.get(timeout, TimeUnit.MILLISECONDS); } catch (InterruptedException | ExecutionException | TimeoutException e) { errMsg = String.format("doWithTimeLimit执行超过%d毫秒,强制结束", timeout); log.error(errMsg, e); futureTask.cancel(true); result = defaultResult; } return of(result, errMsg); } /** * 随机耗时的测试方法 */ private String randomSpentTime() { Random random = new Random(); int time = (random.nextInt(10) + 1) * 1000; log.info("预计randomSpentTime方法执行将耗时: " + time + "毫秒"); try { Thread.sleep(time); } catch (Exception e) { } return "randomSpentTime --> " + time; } public static void main(String[] args) throws Exception { ExecutorService executorService = new ThreadPoolExecutor(1, 1, 0L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<Runnable>(), runnable -> { Thread thread = new Thread(runnable); // 以守护线程方式启动 thread.setDaemon(true); return thread; }); TimeoutUtil timeoutUtil = new TimeoutUtil(executorService); for (int i = 1; i <= 10; i++) { log.info("\n=============第{}次超时测试=============", i); Thread.sleep(6000); long start = System.currentTimeMillis(); String result = timeoutUtil.doWithTimeLimit(() -> timeoutUtil.randomSpentTime(), 5000).getOrElse("默认"); log.info("doWithTimeLimit方法实际耗时{}毫秒,结果:{}", System.currentTimeMillis() - start, result); } } }
以上是怎麼使用Java編寫超時工具類的詳細內容。更多資訊請關注PHP中文網其他相關文章!