首页 > 代码库 > java使用默认线程池踩过的坑(三)

java使用默认线程池踩过的坑(三)

                              云智慧(北京)科技有限公司  陈鑫

重新启动线程池
TaskManager
public class TaskManager implements Runnable {
…..
public TaskManager (Setrunners) {
super();
this.runners = runners;
executeTasks(runners);
}

private voidexecuteTasks(Set<FileTask> runners) {
    for (FileTask task : runners){
       pool.execute(task);
       System.out.println(task.getClass().getSimpleName() + " has beenstarted");
    }
}

@Override
public void run() {
    while(!Thread.currentThread().isInterrupted()) {
        try {
           long current = System.currentTimeMillis();
           for (FileTask wrapper : runners) {
               if (wrapper.getLastExecTime() != 0 && current -wrapper.getLastExecTime() > wrapper.getInterval() * 5 * 1000) {   // 開始忘了乘以1000
                   wrapper.interrupt();
                   if (wrapper.getFiles() != null){
                       for (File file : wrapper.getFiles()){
                           file.delete();
                       }
                   }
                   System.out.println("Going to shutdown thethread pool");
                   List<Runnable> shutdownNow = pool.shutdownNow();    // 不等当前pool里的任务运行完。直接关闭线程池
                   for (Runnable run : shutdownNow) {
                       System.out.println(run + " goingto be shutdown");
                   }
                  while (pool.awaitTermination(1, TimeUnit.SECONDS)) { 
                       System.out.println("The threadpool has been shutdown " + new Date());
                       executeTasks(runners);//又一次运行
                       Thread.sleep(200);
                   }
               }
            }
        } catch(Exception e1) {
           e1.printStackTrace();
        }
        try {
           Thread.sleep(500);
        } catch(InterruptedException e) {
        }
    }
}
public static void main(String[] args) {
    Set<FileTask> tasks =new HashSet<FileTask>();

    FileTask task = newFileTask();
    task.setInterval(1);
    task.setName("task-1");
    tasks.add(task);


    FileTask task1 = newFileTask();
    task1.setInterval(2);
   task.setName("task-2");
    tasks.add(task1);

    TaskManager  codeManager = new TaskManager (tasks);
    newThread(codeManager).start();
}

}

成功。把整个的ThreadPoolExector里所有的worker所有停止,之后再向其队列里又一次增加要运行的两个task(注意这里并没有清空,仅仅是停止而已)。这样做尽管能够及时处理task,可是一个非常致命的缺点在于。假设不能明白的知道ThreadPoolExecutor要运行的task。就没有办法又一次运行这些任务。


定制线程池
好吧!停止钻研别人的东西!

我们全然能够自己写一个自己的ThreadPoolExecutor。仅仅要把worker暴露出来就能够了。这里是不是回忆起前面的start问题来了,没错。我们即便能够直接针对Thread进行interrupt, 可是不能再次start它了。那么clone一个相同的Thread行不行呢?
Thread
@Override
protectedObject clone() throws CloneNotSupportedException{
throw newCloneNotSupportedException();
}
答案显而易见。线程是不支持clone 的。

我们须要又一次new 一个Thread来又一次运行。事实上我们仅仅须要将原来的Worker里的Runnable换成我们自己的task,然后将訪问权限适当放开就能够了。

还有,就是让我们的CustomThreadPoolExecutor继承Thread,由于它须要定时监控自己的所有的worker里Thread的运行状态。


CustomThreadPoolExecutor
public class CustomThreadPoolExecutor extendsThreadPoolExecutor implements Runnable {
public voidexecute(Testask command) {
….//将运行接口改为接收我们的业务类
}


private final class Worker
extends AbstractQueuedSynchronizer
implements Runnable
{

Testask firstTask; //将Runnable改为我们的业务类,方便查看状态

Worker(Testask firstTask) {
…//相同将初始化參数改为我们的业务类
}

}
public staticvoid main(String[] args) {
CustomThreadPoolExecutor pool = new CustomThreadPoolExecutor(0, Integer.MAX_VALUE,
60L, TimeUnit.SECONDS,
newSynchronousQueue());

    Testasktask = new Testask();
   task.setInterval(1);
   pool.execute(task);

    Testasktask1 = new Testask();
   task1.setInterval(2);
   pool.execute(task1);

    newThread(pool).start();
}

@Override
public voidrun() {
    while(!Thread.currentThread().isInterrupted()) {
        try {
           long current = System.currentTimeMillis();
           Set<Testask> toReExecute = new HashSet<Testask>();
           System.out.println("\t number is " + number);
            for(Worker wrapper : workers) {
                Testask tt = wrapper.firstTask;
               if (tt != null) {
                   if (current - tt.getLastExecTime() > tt.getInterval() * 5 * 1000) {

wrapper.interruptIfStarted();
remove(tt);
if (tt.getFiles() != null) {
for (File file: tt.getFiles()) {
file.delete();
}
}
System.out.println(“THread is timeout : ” + tt + ” “+ new Date());
toReExecute.add(tt);
}
}
}
if(toReExecute.size() > 0) {
mainLock.lock();
try {
for (Testask tt :toReExecute) {
execute(tt); // execute this task again
}
} finally {
mainLock.unlock();
}
}
} catch(Exception e1) {
System.out.println(“Error happens when we trying to interrupt andrestart a code task “);
}
try {
Thread.sleep(500);
} catch(InterruptedException e) {
}
}
}
}
Testask
class Testask implements Runnable {
…..

@Override
public voidrun() {
    while(!Thread.currentThread().isInterrupted()) {
       lastExecTime = System.currentTimeMillis();
       System.out.println(Thread.currentThread().getName() + " is running-> " + new Date());
        try {
           CustomThreadPoolExecutor.number++;
           Thread.sleep(getInterval() * 6 * 1000);
           System.out.println(Thread.currentThread().getName() + " aftersleep");
        } catch(InterruptedException e) {

Thread.currentThread().interrupt();
System.out.println(“InterruptedException happens”);
}
}
System.out.println(“Going to die”);
}
}
终于方案
综上,最稳妥的就是使用JDK自带的ThreadPoolExecutor,假设须要对池里的task进行随意时间的控制,能够考虑全面更新,全方面,360度无死角的定制自己的线程池当然是最好的方案。可是一定要注意对于共享对象的处理,适当的处理好并发訪问共享对象的方法。
鉴于我们的场景。由于时间紧,并且须要了解的task并不多。临时选用所有又一次更新的策略。

上线后。抽时间把自己定制的ThreadPoolExecutor搞定,然后更新上去!

<script type="text/javascript"> $(function () { $(‘pre.prettyprint code‘).each(function () { var lines = $(this).text().split(‘\n‘).length; var $numbering = $(‘
    ‘).addClass(‘pre-numbering‘).hide(); $(this).addClass(‘has-numbering‘).parent().append($numbering); for (i = 1; i <= lines; i++) { $numbering.append($(‘
  • ‘).text(i)); }; $numbering.fadeIn(1700); }); }); </script>

java使用默认线程池踩过的坑(三)