diff --git a/src/main/java/com/cloudera/hadoop/HadoopFile.java b/src/main/java/com/cloudera/hadoop/HadoopFile.java index fbf1a88..a616931 100644 --- a/src/main/java/com/cloudera/hadoop/HadoopFile.java +++ b/src/main/java/com/cloudera/hadoop/HadoopFile.java @@ -36,7 +36,7 @@ public HadoopFile(String file, Configuration configuration) { this.configuration = configuration; } - public void inspect() throws IOException { + public String inspect() throws IOException { FileSystem fileSystem = FileSystem.get(configuration); Path filePath = new Path(file); FSDataInputStream is = fileSystem.open(filePath); @@ -46,12 +46,11 @@ public void inspect() throws IOException { for (Detector detector : REGISTERED_DETECTORS) { if (detector.detect(header, read)) { PathData pd = new PathData(file, configuration); - System.out.println(file + ": " + detector.analyze(pd, "")); - return; + return(file + ": " + detector.analyze(pd, "")); } } - System.out.println(file + ": octet-stream"); + return (file + ": octet-stream"); } finally { is.close(); @@ -61,7 +60,7 @@ public void inspect() throws IOException { @Override public int run(String[] strings) throws Exception { - inspect(); + System.out.println(inspect()); return 0; } @@ -75,4 +74,4 @@ public static void main(String[] args) throws Exception { hadoopFile.run(args); } -} \ No newline at end of file +} diff --git a/src/test/java/com/cloudera/hadoop/HadoopFileTest.java b/src/test/java/com/cloudera/hadoop/HadoopFileTest.java index 77ea7a5..3a65f62 100644 --- a/src/test/java/com/cloudera/hadoop/HadoopFileTest.java +++ b/src/test/java/com/cloudera/hadoop/HadoopFileTest.java @@ -25,31 +25,31 @@ public void testTextFile() { @Test public void testParquetFile() throws IOException { HadoopFile hadoopFile = new HadoopFile("src/test/resources/testparq.parq", configuration); - hadoopFile.inspect(); + String result = hadoopFile.inspect(); } @Test public void testAvroFile() throws IOException { HadoopFile hadoopFile = new HadoopFile("src/test/resources/testavro.avro", configuration); - hadoopFile.inspect(); + String result = hadoopFile.inspect(); } @Test public void testSeqFile() throws IOException { HadoopFile hadoopFile = new HadoopFile("src/test/resources/testseq.seq", configuration); - hadoopFile.inspect(); + String result = hadoopFile.inspect(); } @Test public void testRCFile() throws IOException { HadoopFile hadoopFile = new HadoopFile("src/test/resources/testrcfile.rcf", configuration); - hadoopFile.inspect(); + String result = hadoopFile.inspect(); } @Test public void testORCFile() throws IOException { HadoopFile hadoopFile = new HadoopFile("src/test/resources/testorc.orc", configuration); - hadoopFile.inspect(); + String result = hadoopFile.inspect(); } }