Merge pull request #54 from MadEgg/binary-content

yasserg · yasserg · commit bdbdc3e503ec · 2015-07-18T20:07:19.000-07:00
Added configuration parameter processBinaryContentInCrawling
diff --git a/src/main/java/edu/uci/ics/crawler4j/crawler/CrawlConfig.java b/src/main/java/edu/uci/ics/crawler4j/crawler/CrawlConfig.java
@@ -79,6 +79,11 @@ public class CrawlConfig {
    * Should we fetch binary content such as images, audio, ...?
    */
   private boolean includeBinaryContentInCrawling = false;
+  
+  /**
+   * Should we process binary content such as image, audio, ... using TIKA?
+   */
+  private boolean processBinaryContentInCrawling = false;
 
   /**
    * Maximum Connections per host
@@ -306,6 +311,17 @@ public boolean isIncludeBinaryContentInCrawling() {
   public void setIncludeBinaryContentInCrawling(boolean includeBinaryContentInCrawling) {
     this.includeBinaryContentInCrawling = includeBinaryContentInCrawling;
   }
+  
+  public boolean isProcessBinaryContentInCrawling() {
+    return processBinaryContentInCrawling;
+  }
+
+  /**
+   * Should we process binary content such as images, audio, ... using TIKA?
+   */
+  public void setProcessBinaryContentInCrawling(boolean processBinaryContentInCrawling) {
+    this.processBinaryContentInCrawling = processBinaryContentInCrawling;
+  }  
 
   public int getMaxConnectionsPerHost() {
     return maxConnectionsPerHost;
diff --git a/src/main/java/edu/uci/ics/crawler4j/parser/Parser.java b/src/main/java/edu/uci/ics/crawler4j/parser/Parser.java
@@ -60,7 +60,11 @@ public void parse(Page page, String contextURL) throws NotAllowedContentExceptio
     if (Util.hasBinaryContent(page.getContentType())) { // BINARY
       BinaryParseData parseData = new BinaryParseData();
       if (config.isIncludeBinaryContentInCrawling()) {
-        parseData.setBinaryContent(page.getContentData());
+        if (config.isProcessBinaryContentInCrawling()) {
+          parseData.setBinaryContent(page.getContentData());
+        } else {
+          parseData.setHtml("<html></html>");
+        }
         page.setParseData(parseData);
         if (parseData.getHtml() == null) {
           throw new ParseException();
