Commit 20422f1b authored by Yihua Huang's avatar Yihua Huang

Merge pull request #161 from zhugw/patch-4

Update FileCacheQueueScheduler.java
parents 3a9c1d30 eb3c78b9
...@@ -49,6 +49,7 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement ...@@ -49,6 +49,7 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement
filePath += "/"; filePath += "/";
} }
this.filePath = filePath; this.filePath = filePath;
initDuplicateRemover();
} }
private void flush() { private void flush() {
...@@ -74,6 +75,9 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement ...@@ -74,6 +75,9 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement
new DuplicateRemover() { new DuplicateRemover() {
@Override @Override
public boolean isDuplicate(Request request, Task task) { public boolean isDuplicate(Request request, Task task) {
if (!inited.get()) {
init(task);
}
return !urls.add(request.getUrl()); return !urls.add(request.getUrl());
} }
...@@ -113,7 +117,7 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement ...@@ -113,7 +117,7 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement
urls = new LinkedHashSet<String>(); urls = new LinkedHashSet<String>();
readCursorFile(); readCursorFile();
readUrlFile(); readUrlFile();
initDuplicateRemover(); // initDuplicateRemover();
} catch (FileNotFoundException e) { } catch (FileNotFoundException e) {
//init //init
logger.info("init cache file " + getFileName(fileUrlAllName)); logger.info("init cache file " + getFileName(fileUrlAllName));
...@@ -164,9 +168,9 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement ...@@ -164,9 +168,9 @@ public class FileCacheQueueScheduler extends DuplicateRemovedScheduler implement
@Override @Override
protected void pushWhenNoDuplicate(Request request, Task task) { protected void pushWhenNoDuplicate(Request request, Task task) {
if (!inited.get()) { /* if (!inited.get()) {
init(task); init(task);
} }*/
queue.add(request); queue.add(request);
fileUrlWriter.println(request.getUrl()); fileUrlWriter.println(request.getUrl());
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment