Java中线程的终止条件

时间:2011-08-25 22:47:41

标签: java multithreading concurrency

我编写了一个多线程Java应用程序,它从目录中读取一堆.jar文件。此应用程序生成多个线程,每个线程读取一堆jar文件。我无法确定此应用程序的停止条件。我怎样才能确定所有文件都已被阅读?

以下是从每个线程的run()方法调用的片段函数。

    import java.io.*;
import java.util.Enumeration;
import java.util.jar.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipException;
import java.io.FilenameFilter;

public class ArchiveFileTest implements Runnable {
     private static boolean stopAll = false;
     private static int threadNumber = 0;
     private int myNumber = 0;

     public ArchiveFileTest () {
            myNumber = threadNumber;
            threadNumber++;
     }

     public static boolean setStopAll () {
            return setStopAll(true);
     }

     public static boolean setStopAll (boolean b) {
            stopAll = b;
            return stopAll;
     }

     public static String[] listFiles (File parentDir,final String ext1,final String ext2,final String ext3,final String ext4) {
            String allFiles[] = parentDir.list(new FilenameFilter() { 
                 public boolean accept(File pDir, String fName) {
                        if (fName.endsWith("."+ext1) || fName.endsWith("."+ext2) || fName.endsWith("."+ext3) || fName.endsWith("."+ext4)) return true;
                        else return false;
                 }
            });
            for (int i=0; i<allFiles.length; i++)
                 allFiles[i] = parentDir.getAbsolutePath() + File.separator +  allFiles[i];
            return allFiles;
     }

     public ZipFile getMyZipFile (File parentDir) {
            String fn[] = listFiles(parentDir, "jar", "zip", "war", "rar");
            int fileNum = myNumber % fn.length;
            ZipFile zFile = null;
            for (int i=0; i<fn.length; i++) {
                 String jFile = fn[(fileNum + i)%fn.length];
                 try {
                        zFile = new ZipFile(jFile);
                        break;
                 } catch  (IOException e) {
                     setStopAll();
                 }
            }
            return zFile;
     }

     public void doStuff() throws Exception {
            File dName = new File("/home/sqatest/chander/sample-files");
          final int N_TIMES = 15;
            final int N_FILES = 500;
          int counter = 0;
            int fCount  = 0;
            if (!dName.isDirectory() || !dName.exists()) {
                 System.err.println("The parent directory given should point to an existing directory...");
                 setStopAll();
                 return;
            }
          while (counter < N_TIMES) { 
                 ZipFile zipFile = getMyZipFile(dName);
                 if (zipFile == null) {
                        System.err.println("No zip file entry for the Thread-" + myNumber);
                        break;
                 }
                 try {
                        Enumeration <? extends ZipEntry> zipEntries = zipFile.entries(); 
                        fCount = 0;
                        ZipEntry ze = null;
                        while (zipEntries.hasMoreElements()) {
                             ze = zipEntries.nextElement();

                             if (ze.isDirectory()) continue; // if it is a directory go to next entry

                             InputStream is = zipFile.getInputStream(ze); 
                             fCount++;
                             int readCount = 0;
                             try {
                                    while(is.read((new byte[50])) != -1 && readCount != 200) readCount++;
                                    System.out.println("Successfully Read " + zipFile.toString());

                                    //is.close();
                             } catch (IOException e) {
                                    e.printStackTrace();
                             }
                             if (fCount == N_FILES) break;  // read maximum of N_FILES
                        }
                        if (stopAll) break;
                 } catch (Exception e) {
                     e.printStackTrace();
                 } finally {
                        counter++; 
                 }
          } 
     }

     public void run () {
            try {
                 doStuff();
            } catch (IOException e) {
                 e.printStackTrace();
                 setStopAll();
            } catch (Exception e) {
                 e.printStackTrace();
            }
     }

     public static void main (String[] args) throws Exception { 
          final int MAX_THREADS = 500;
          final int MAX_HOLDING_THREADS = 5;
            int loopCount = 0;
            Thread mainThread = Thread.currentThread();
            for (int m=0; ; m++) {
                 Thread t[] = new Thread[MAX_HOLDING_THREADS];
                 for (int n=0; n<t.length; n++) {
                        t[n] = new Thread(new ArchiveFileTest());
                        t[n].start();
                        if ((m+1)*(n+1)==MAX_THREADS) {
                             System.out.println("\n" + MAX_THREADS + " reached... \nMain Sleeping for some mins..."); 
                             loopCount++;
                             try {
                                    t[n].join();
                                    System.out.println("\nMain is back... (" + loopCount + ")");
                             } catch (InterruptedException e) {
                                    e.printStackTrace();
                                    setStopAll();
                             }
                             m = 0;
                        }
                 }
            }
     }
}

3 个答案:

答案 0 :(得分:1)

  1. 在启动线程的类中,为运行的线程设置一个volatile计数器。
  2. 在您的线程构造函数中传递对启动类的引用。
  3. 使用synchronized方法让线程通知启动类它们已完成。
  4. 实例化并启动线程后,等待计数器变为0;

    while(getRunningThreads() > 0) // getRunningThreads must be synchronized too
        Thread.sleep(500); // Check every half second.
    

答案 1 :(得分:1)

如果停止意味着终止,则当所有非守护进程特殊情况的线程完成时,应用程序将停止。

答案 2 :(得分:1)

我认为您的申请不会停止。你在main方法中有一个无限循环:

for (int m=0; ; m++) {
    ....
}

注意,在正文中设置m=0不会破坏循环,所以即使你没有文件我也认为你永远不会结束。然后它会不断读取目录中的所有zip / jar / war / rar文件(选择基于旋转计数器myNumber的文件不是很容易维护),但永远不会退出循环。

如果您要求使用多个线程读取ZIP文件,那么我会采用不同的方式。

  1. 创建一组您想要查看的文件。
  2. 创建ThreadPoolExecutor以创建5个线程的固定池
  3. 迭代文件集并创建一个新的Runnable,它执行Zip提取(虽然我不太清楚为什么你读取ZIP条目的前10000字节,然后不做任何事情它),并调用execute方法。这将使用线程池一次处理5个文件。
  4. 提交所有runnables之后使用shutdown方法,它将等待所有提交的任务完成,并关闭线程池。