Merge pull request #2 from twolight/master

Add finish callback method in task

demo/src/main/java/cn/zhaiyifan/demo/DemoApplication.java

@@ -31,6 +31,10 @@ protected void start() {
  public boolean runOnProcess(String processName) {
  return processName.equals("cn.zhaiyifan.init");
  }
+
+ @Override
+ protected void onResult() {
+ }
  };
 
  Task task2 = new Task("task2", false, 5) {

init/src/main/java/cn/zhaiyifan/init/Init.java

@@ -18,6 +18,7 @@ public class Init {
  private static Context sContext;
  private static int mThreadPoolSize = DEFAULT_THREAD_POOL_SIZE;
 
+ private static ExecutorService mExecutorService;
  /**
  * Init with context.
  *
@@ -87,7 +88,10 @@ public static Context getContext() {
  * @param size thread pool size, value less or equal than 0 will produce a cached thread pool.
  */
  public static void setThreadPoolSize(int size) {
- mThreadPoolSize = size;
+ if(mThreadPoolSize != size){
+ mThreadPoolSize = size;
+ createThreadPool();
+ }
  }
 
  public static Flow getFlow(String flowName) {
@@ -157,10 +161,20 @@ public static int getFlowStatus(String flowName) {
  * @return thread pool
  */
  public static ExecutorService getThreadPool() {
+ if(mExecutorService == null){
+ createThreadPool();
+ }
+ return mExecutorService;
+ }
+
+ /**
+ * create a thread pool.
+ */
+ public static void createThreadPool(){
  if (mThreadPoolSize <= 0) {
- return Executors.newCachedThreadPool();
+ mExecutorService =  Executors.newCachedThreadPool();
  } else {
- return Executors.newFixedThreadPool(mThreadPoolSize);
+ mExecutorService = Executors.newFixedThreadPool(mThreadPoolSize);
  }
  }
 }

init/src/main/java/cn/zhaiyifan/init/Task.java

@@ -1,5 +1,8 @@
 package cn.zhaiyifan.init;
 
+import android.os.Handler;
+import android.os.Looper;
+import android.os.Message;
 import java.util.concurrent.CountDownLatch;
 
 /**
@@ -15,7 +18,8 @@ public abstract class Task implements Runnable {
  private boolean mIsBlocked = true;
  private long mDelay = 0;
  private int mStatus = Status.STATUS_PENDING_START;
-
+ private static InternalHandler sHandler;
+ private static final int MESSAGE_POST_RUSULT = 0x1;
  // for asynchronous task chain
  private Task mParentTask;
  private Task mChildTask;
@@ -92,6 +96,7 @@ public void run() {
  long startTime = System.currentTimeMillis();
 
  start();
+ getHandler().obtainMessage(MESSAGE_POST_RUSULT, this).sendToTarget();
 
  long endTime = System.currentTimeMillis();
  LogImpl.i(TAG, getName() + " runs " + (endTime - startTime));
@@ -107,6 +112,10 @@ public void run() {
  mStatus = Status.STATUS_DONE;
  }
 
+ /**
+ * This method is invoked by ui thread when the task finish.
+ */
+ protected void onResult(){};
  /**
  * Run task.
  */
@@ -176,4 +185,34 @@ public boolean runOnProcess(String processName) {
  public String getName() {
  return mTaskName;
  }
+
+ /**
+ * Provide a public handler for all the task;
+ * @return a handler object
+ */
+ private static Handler getHandler() {
+ synchronized (Task.class) {
+ if (sHandler == null) {
+ sHandler = new InternalHandler();
+ }
+ return sHandler;
+ }
+ }
+
+ private static class InternalHandler extends Handler {
+ public InternalHandler() {
+ super(Looper.getMainLooper());
+ }
+ @Override
+ public void handleMessage(Message msg) {
+ switch (msg.what) {
+ case MESSAGE_POST_RUSULT:
+ Task task = (Task) msg.obj;
+ task.onResult();
+ break;
+ default:
+ break;
+ }
+ }
+ }
 }