前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >多线程执行超时处理

多线程执行超时处理

作者头像
程裕强
发布2022-05-06 17:14:32
9720
发布2022-05-06 17:14:32
举报
文章被收录于专栏:大数据学习笔记

多线程执行超时处理:

代码语言:javascript
复制
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

代码语言:javascript
复制
/**
 * 开启监控程序,远程脚本类似守护进程,不停止不返回
 * @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;
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2017-01-22,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档