Main.java Source code

Introduction

Here is the source code for Main.java

Source

//package com.java2s;
import java.util.concurrent.CountDownLatch;

public class Main {
    /**
     * 
     * @param threadCount
     * @param task
     * @return costTime
     * @throws InterruptedException 
     */
    public static long executeAndWait(int threadCount, final Runnable task) throws InterruptedException {
        CountDownLatch doneSignal = execute(threadCount, task);
        long startTime = System.currentTimeMillis();
        doneSignal.await();
        return System.currentTimeMillis() - startTime;
    }

    public static CountDownLatch execute(int threadCount, final Runnable task) {
        final CountDownLatch startSignal = new CountDownLatch(1);
        final CountDownLatch startedSignal = new CountDownLatch(threadCount);
        final CountDownLatch doneSignal = new CountDownLatch(threadCount);
        for (int i = 0; i < threadCount; i++) {
            Thread t = new Thread() {
                public void run() {
                    startedSignal.countDown();
                    try {
                        startSignal.await();
                    } catch (InterruptedException e) {
                        //ignore
                    }

                    try {
                        task.run();
                    } finally {
                        doneSignal.countDown();
                    }

                }
            };

            t.start();
        }

        try {
            startedSignal.await();
        } catch (InterruptedException e) {
            //ignore
        }
        startSignal.countDown();
        return doneSignal;
    }
}