Java中的超时机制 - 执行一段有时间限制的任务

1. 场景描述
举个例子，有时候执行一段程序需要1 ~ 10分钟才能运行完，但你最多只能等5分钟。如果5分钟以内程序能够运行完，那么直接返回正确结果；如果过了5分钟程序还没运行完，那么就中止程序的运行，直接返回一个错误结果。

2. 一个工具类（用来执行有时间限制的任务）

package org.demo.task;

import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;

public class TimeoutTaskUtils {
    
	/**
	 * 执行一个有时间限制的任务
	 * @param task    待执行的任务
	 * @param seconds 超时时间(单位: 秒)
	 * @return
	 */
	public static Boolean execute(Callable<Boolean> task, int seconds)
	{
		Boolean result = Boolean.FALSE;
		ExecutorService threadPool = Executors.newCachedThreadPool();
		
		try 
		{
			Future<Boolean> future = threadPool.submit(task);
			result = future.get(seconds, TimeUnit.SECONDS);
		}
		catch (Exception e)
		{
			result = Boolean.FALSE;
			e.printStackTrace();
		}
		finally
		{
			threadPool.shutdownNow();
		}
		
		return result;
	}
}

3. 测试

package org.demo.task;

import java.util.concurrent.Callable;

public class Main {

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		
		boolean result = TimeoutTaskUtils.execute(new MyTimeoutTask(), 5);
		System.out.println("result = " + result);
		System.out.println("-- finished. --");
	}

}
class MyTimeoutTask implements Callable<Boolean> {
	
	@Override
	public Boolean call() throws Exception {
		for (int i=0; i<10; i++) {
		    System.out.println("i = " + i);
			Thread.sleep(1000);
		}
		return true;
	}
}

评论

1 楼 qiqitrue 2014-07-18  

为什么我执行了，结果报异常：

java.util.concurrent.TimeoutException执行状态：false

	at java.util.concurrent.FutureTask$Sync.innerGet(FutureTask.java:228)
	at java.util.concurrent.FutureTask.get(FutureTask.java:91)
	at com.cimstech.laiqi.common.demo.core.utils.TimeOutTask.execute(TimeOutTask.java:37)
	at com.cimstech.laiqi.common.demo.core.utils.TimeOutTask.main(TimeOutTask.java:53)