stopWhenComplete，增加动态修改完成时停止方法。 (#1169)

niuxiaozu · niuxiaozu · web-flow · commit 4d0cdb011fc4 · 2024-06-17T17:27:28.000+08:00
Co-authored-by: niuxiaozu &lt;niuxiaozu@yeah.net&gt;
diff --git a/webmagic-core/src/main/java/us/codecraft/webmagic/Spider.java b/webmagic-core/src/main/java/us/codecraft/webmagic/Spider.java
@@ -85,7 +85,7 @@ public class Spider implements Runnable, Task {
 
     protected AtomicInteger stat = new AtomicInteger(STAT_INIT);
 
-    protected boolean exitWhenComplete = true;
+    protected volatile boolean exitWhenComplete = true;
 
     protected final static int STAT_INIT = 0;
 
@@ -598,6 +598,13 @@ public void stop() {
         }
     }
 
+    /**
+     * Stop when all tasks in the queue are completed and all worker threads are also completed
+     */
+    public void stopWhenComplete(){
+        this.exitWhenComplete = true;
+    }
+
     /**
      * start with more than one threads
      *

Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ public class Spider implements Runnable, Task {`
`85`	`85`
`86`	`86`	`protected AtomicInteger stat = new AtomicInteger(STAT_INIT);`
`87`	`87`
`88`		`- protected boolean exitWhenComplete = true;`
	`88`	`+ protected volatile boolean exitWhenComplete = true;`
`89`	`89`
`90`	`90`	`protected final static int STAT_INIT = 0;`
`91`	`91`
`@@ -598,6 +598,13 @@ public void stop() {`
`598`	`598`	`}`
`599`	`599`	`}`
`600`	`600`
	`601`	`+ /**`
	`602`	`+ * Stop when all tasks in the queue are completed and all worker threads are also completed`
	`603`	`+ */`
	`604`	`+ public void stopWhenComplete(){`
	`605`	`+ this.exitWhenComplete = true;`
	`606`	`+ }`
	`607`	`+`
`601`	`608`	`/**`
`602`	`609`	`* start with more than one threads`
`603`	`610`	`*`