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中文網其他相關文章！
