多线程执行超时处理

多线程执行超时处理：

package util;
import java.util.concurrent.Callable;  
import java.util.concurrent.ExecutionException;  
import java.util.concurrent.ExecutorService;  
import java.util.concurrent.Executors;  
import java.util.concurrent.Future;
import java.util.concurrent.FutureTask;
import java.util.concurrent.TimeUnit;  
import java.util.concurrent.TimeoutException;
/** 
 * 在某些情况下需要控制方法的运行时间，通过Thread+Callable+FutureTask完成
 * Thread用于新开线程运行指定方法，Callable和FutrueTask用于监控方法运行时间。
 * 启动一个任务，然后等待任务的计算结果，如果等待时间超出预设定的超时时间，则中止任务。 
 *  
 * @author Cheng 
 */  
public class TimeoutUtil{
  
    public static boolean checkTask(int timeout,Callable<Boolean> task) {  
       ExecutorService exec = Executors.newCachedThreadPool();
       Boolean taskResult = false;  
        String failReason = null;
        Future<Boolean> future = exec.submit(task);  
        try {  
            // 等待计算结果，最长等待timeout秒(在第timeout秒取结果)，timeout秒后中止任务  
            taskResult = future.get(timeout, TimeUnit.SECONDS);
        } catch (InterruptedException e) { 
            failReason = "主线程在等待计算结果时被中断！";  
        } catch (ExecutionException e) {  
            failReason = "主线程等待计算结果，但计算抛出异常！";  
        } catch (TimeoutException e) {
            failReason = "主线程等待计算结果超时，因此中断任务线程！";  
            exec.shutdownNow();  
        }finally{
           exec.shutdown();
        }  
        return taskResult;
    }  
    //测试
   public static void main(String[] args) {  
        Boolean flag=null;
        System.out.println("Start ..."); 
        flag=TimeoutUtil.checkTask(15,new MyTask()); // 任务成功结束后等待计算结果，不需要等到15秒   
        System.out.println("\n执行结果："+flag);  
        System.out.println("\nStart ..."); 
        flag=TimeoutUtil.checkTask(5,new MyTask()); // 只等待5秒，任务还没结束，所以将任务中止 
        System.out.println("\n执行结果："+flag);  
    }
}

class MyTask implements Callable<Boolean> {
    @Override
    public Boolean call() throws Exception {
       // 总计耗时约10秒
        for (int i = 0; i < 100L; i++) {
            Thread.sleep(100); // 睡眠0.1秒
            System.out.print('-');
            if (Thread.interrupted()){ //很重要
                return false;
            }
        }
        return Boolean.TRUE;
    }
}

输出信息：

Start ...
----------------------------------------------------------------------------------------------------
执行结果：true

Start ...
-------------------------------------------------
执行结果：false

Process finished with exit code 0

/**
 * 开启监控程序，远程脚本类似守护进程，不停止不返回
 * @param req
 * @param resp
 * @return
 * @throws Exception
 */
@RequestMapping(value = "monitor.do",method=RequestMethod.GET)
@ResponseBody
public Map<String, Object> monitor(HttpServletRequest req, HttpServletResponse resp) throws Exception {
    System.out.println("monitor.do.............");
    ScpClient scp=new ScpClient("192.168.1.110",22,"root","123456");
    //匿名类方式
    /*Boolean flag=TimeoutUtil.checkTask(15, new Callable<Boolean>() {
        @Override
        public Boolean call() throws Exception {
            System.out.println("开始执行远程节点脚本....");
            String rs=scp.exe("sh /root/Storm_shell/flume.sh");
            System.out.println("远程节点脚本执行结束！");
            return Boolean.TRUE;
        }
    });*/
    //lambda表达式
    Boolean flag=TimeoutUtil.checkTask(15, ()-> {
            System.out.println("开始执行远程节点脚本....");
            String rs=scp.exe("sh /root/Storm_shell/flume.sh");
            System.out.println("远程节点脚本执行结束！");
            return Boolean.TRUE;
    });
    System.out.println("15秒时间到！");
    if(flag){
        System.out.println("true");
    }else{
        System.out.println("false");
    }
    scp.close();
    Map<String, Object> map=new HashMap<String, Object>();
    map.put("ok","程序正在执行...");
    return map;
}