Changeset 31194 for other-projects/hathitrust/wcsa

Timestamp:

2016-12-12T09:51:07+13:00 (7 years ago)

Author:

davidb

Message:

Serialize in and out methods added

File:

• other-projects/hathitrust/wcsa/extracted-features-solr/trunk/solr-ingest/src/main/java/org/hathitrust/extractedfeatures/WhitelistBloomFilter.java (modified) (2 diffs)

other-projects/hathitrust/wcsa/extracted-features-solr/trunk/solr-ingest/src/main/java/org/hathitrust/extractedfeatures/WhitelistBloomFilter.java

@@ -2 +2 @@
 
 import java.io.BufferedInputStream;
+import java.io.BufferedOutputStream;
 import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.OutputStream;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
@@ -90 +94 @@
         return _bloomFilter.mightContain(key);
     }
+    
+    public void serializeOut(String filename)
+    {
+        try {
+            FileOutputStream fos = new FileOutputStream(filename);
+
+            BufferedOutputStream bfos = new BufferedOutputStream(fos);
+
+            _bloomFilter.writeTo(bfos);
+            
+            bfos.close();
+        }
+        catch (FileNotFoundException e) {
+            System.err.println("Unable to open Bloom file:" + filename);
+            e.printStackTrace();
+        } catch (IOException e) {
+            System.err.println("Error reading in Bloom file:" + filename);
+            e.printStackTrace();
+        }
+    }
+    
+    public void serializeIn(String filename)
+    {
+        try {
+            FileInputStream fis = new FileInputStream(filename);
+
+            BufferedInputStream bfis = new BufferedInputStream(fis);
+
+            Funnel<CharSequence> string_funnel = Funnels.stringFunnel(StandardCharsets.UTF_8);
+            _bloomFilter = BloomFilter.readFrom(bfis,string_funnel);
+            
+            bfis.close();
+        }
+        catch (FileNotFoundException e) {
+            System.err.println("Unable to open Bloom file:" + filename);
+            e.printStackTrace();
+        } catch (IOException e) {
+            System.err.println("Error writing out Bloom file:" + filename);
+            e.printStackTrace();
+        }
+    }
+    
+    
 }