svn commit: r1175809 - /incubator/opennlp/sandbox/corpus-server-tools/src/main/java/org/apache/opennlp/corpus_server/tools/CASImporter.java

[jo...@apache.org]

Author: joern
Date: Mon Sep 26 11:46:21 2011
New Revision: 1175809

URL: http://svn.apache.org/viewvc?rev=1175809&view=rev
Log:
OPENNLP-300 Added ability to import a single xmi file or a folder of xmi files.

Modified:
    incubator/opennlp/sandbox/corpus-server-tools/src/main/java/org/apache/opennlp/corpus_server/tools/CASImporter.java

Modified: incubator/opennlp/sandbox/corpus-server-tools/src/main/java/org/apache/opennlp/corpus_server/tools/CASImporter.java
URL: http://svn.apache.org/viewvc/incubator/opennlp/sandbox/corpus-server-tools/src/main/java/org/apache/opennlp/corpus_server/tools/CASImporter.java?rev=1175809&r1=1175808&r2=1175809&view=diff
==============================================================================
--- incubator/opennlp/sandbox/corpus-server-tools/src/main/java/org/apache/opennlp/corpus_server/tools/CASImporter.java (original)
+++ incubator/opennlp/sandbox/corpus-server-tools/src/main/java/org/apache/opennlp/corpus_server/tools/CASImporter.java Mon Sep 26 11:46:21 2011
@@ -18,6 +18,7 @@
 package org.apache.opennlp.corpus_server.tools;
 
 import java.io.File;
+import java.io.FilenameFilter;
 
 import javax.ws.rs.core.MediaType;
 
@@ -30,7 +31,7 @@ public class CASImporter {
 	public static void main(String[] args) throws Exception {
 		
 		if (args.length != 2) {
-			System.out.println("WikinewsImporter address xmiFile");
+			System.out.println("WikinewsImporter address xmiFileOrFolder");
 			System.exit(-1);
 		}
 		
@@ -38,16 +39,33 @@ public class CASImporter {
 		
 		WebResource r = c.resource(args[0]);
 		
-		File xmiFile = new File(args[1]);
-		byte xmiBytes[] = FileUtil.fileToBytes(xmiFile);
+		File xmiFileOrFolder = new File(args[1]);
 		
-		ClientResponse response = r
-				.path(xmiFile.getName())
-				.accept(MediaType.TEXT_XML)
-				// TODO: How to fix this? Shouldn't accept do it?
-				.header("Content-Type", MediaType.TEXT_XML)
-				.post(ClientResponse.class, xmiBytes);
+		File xmiFiles[];
 		
-		System.out.println(xmiFile.getName() + " " + response.getStatus());
+		if (xmiFileOrFolder.isFile()) {
+		  xmiFiles = new File[]{xmiFileOrFolder};
+		}
+		else {
+		  xmiFiles = xmiFileOrFolder.listFiles(new FilenameFilter() {
+            @Override
+            public boolean accept(File dir, String name) {
+              return name.toLowerCase().endsWith(".xmi");
+            }
+          });
+		}
+		
+		for (File xmiFile : xmiFiles) {
+    		byte xmiBytes[] = FileUtil.fileToBytes(xmiFile);
+    		
+    		ClientResponse response = r
+    				.path(xmiFile.getName())
+    				.accept(MediaType.TEXT_XML)
+    				// TODO: How to fix this? Shouldn't accept do it?
+    				.header("Content-Type", MediaType.TEXT_XML)
+    				.post(ClientResponse.class, xmiBytes);
+    		
+    		System.out.println(xmiFile.getName() + " " + response.getStatus());
+		}
 	}
 }