From 352887870c15a1601288508a1840c925576f1961 Mon Sep 17 00:00:00 2001
From: "yihua.huang" <code4crafter@gmail.com>
Date: Thu, 31 Oct 2013 22:22:14 +0800
Subject: [PATCH] remove shutdown call

---
 .../java/us/codecraft/webmagic/Spider.java    | 31 +++++++++++++------
 1 file changed, 21 insertions(+), 10 deletions(-)

diff --git a/webmagic-core/src/main/java/us/codecraft/webmagic/Spider.java b/webmagic-core/src/main/java/us/codecraft/webmagic/Spider.java
index 149f0a8..65ee7af 100644
--- a/webmagic-core/src/main/java/us/codecraft/webmagic/Spider.java
+++ b/webmagic-core/src/main/java/us/codecraft/webmagic/Spider.java
@@ -74,6 +74,8 @@ public class Spider implements Runnable, Task {
 
     protected AtomicInteger stat = new AtomicInteger(STAT_INIT);
 
+    protected boolean exitWhenComplete = false;
+
     protected final static int STAT_INIT = 0;
 
     protected final static int STAT_RUNNING = 1;
@@ -240,7 +242,7 @@ public class Spider implements Runnable, Task {
         while (!Thread.currentThread().isInterrupted() && stat.get() == STAT_RUNNING) {
             Request request = scheduler.poll(this);
             if (request == null) {
-                if (threadAlive.get() == 0) {
+                if (threadAlive.get() == 0 && exitWhenComplete) {
                     break;
                 }
                 // when no request found but some thread is alive, sleep a
@@ -258,7 +260,7 @@ public class Spider implements Runnable, Task {
                         try {
                             processRequest(requestFinal);
                         } catch (Exception e) {
-                            logger.error("download "+requestFinal+" error",e);
+                            logger.error("download " + requestFinal + " error", e);
                         } finally {
                             threadAlive.decrementAndGet();
                         }
@@ -372,20 +374,12 @@ public class Spider implements Runnable, Task {
 
     public void stop() {
         if (stat.compareAndSet(STAT_RUNNING, STAT_STOPPED)) {
-            if (executorService != null) {
-                executorService.shutdown();
-            }
             logger.info("Spider " + getUUID() + " stop success!");
         } else {
             logger.info("Spider " + getUUID() + " stop fail!");
         }
     }
 
-    public void stopAndDestroy() {
-        stop();
-        destroy();
-    }
-
     /**
      * start with more than one threads
      *
@@ -413,6 +407,23 @@ public class Spider implements Runnable, Task {
         EnvironmentUtil.setUseXsoup(false);
     }
 
+    public boolean isExitWhenComplete() {
+        return exitWhenComplete;
+    }
+
+    /**
+     * Exit when complete. <br/>
+     * True: exit when all url of the site is downloaded. <br/>
+     * False: not exit until call stop manually.<br/>
+     *
+     * @param exitWhenComplete
+     * @return
+     */
+    public Spider setExitWhenComplete(boolean exitWhenComplete) {
+        this.exitWhenComplete = exitWhenComplete;
+        return this;
+    }
+
     @Override
     public String getUUID() {
         if (uuid != null) {