|
@@ -68,8 +68,8 @@ public class WordCount {
|
|
public static void main(String[] args) throws Exception {
|
|
public static void main(String[] args) throws Exception {
|
|
Configuration conf = new Configuration();
|
|
Configuration conf = new Configuration();
|
|
String[] otherArgs = new GenericOptionsParser(conf, args).getRemainingArgs();
|
|
String[] otherArgs = new GenericOptionsParser(conf, args).getRemainingArgs();
|
|
- if (otherArgs.length != 2) {
|
|
|
|
- System.err.println("Usage: wordcount <in> <out>");
|
|
|
|
|
|
+ if (otherArgs.length < 2) {
|
|
|
|
+ System.err.println("Usage: wordcount <in> [<in>...] <out>");
|
|
System.exit(2);
|
|
System.exit(2);
|
|
}
|
|
}
|
|
Job job = new Job(conf, "word count");
|
|
Job job = new Job(conf, "word count");
|
|
@@ -79,8 +79,11 @@ public class WordCount {
|
|
job.setReducerClass(IntSumReducer.class);
|
|
job.setReducerClass(IntSumReducer.class);
|
|
job.setOutputKeyClass(Text.class);
|
|
job.setOutputKeyClass(Text.class);
|
|
job.setOutputValueClass(IntWritable.class);
|
|
job.setOutputValueClass(IntWritable.class);
|
|
- FileInputFormat.addInputPath(job, new Path(otherArgs[0]));
|
|
|
|
- FileOutputFormat.setOutputPath(job, new Path(otherArgs[1]));
|
|
|
|
|
|
+ for (int i = 0; i < otherArgs.length - 1; ++i) {
|
|
|
|
+ FileInputFormat.addInputPath(job, new Path(otherArgs[i]));
|
|
|
|
+ }
|
|
|
|
+ FileOutputFormat.setOutputPath(job,
|
|
|
|
+ new Path(otherArgs[otherArgs.length - 1]));
|
|
System.exit(job.waitForCompletion(true) ? 0 : 1);
|
|
System.exit(job.waitForCompletion(true) ? 0 : 1);
|
|
}
|
|
}
|
|
}
|
|
}
|