mapreduce统计多文件的单词个数(map方式+combiner)

public class InvertIndex2 {

    static {
        System.setProperty("hadoop.home.dir", "D:\\software\\hadoop-2.9.2");
    }

    public static void main(String[] args) throws IOException, ClassNotFoundException, InterruptedException {
       //0.初始化一个作业
        Configuration conf = new Configuration();
        Job job = Job.getInstance(conf, "invert-index-2");
        job.setJarByClass(InvertIndex2.class);
        //1.输入文件
        FileInputFormat.addInputPaths(job, args[0]);
        //2.map处理
        job.setMapperClass(MyMapper.class);
        job.setMapOutputKeyClass(Text.class);
        job.setMapOutputValueClass(Text.class);
        //3.shuffle处理
        job.setCombinerClass(MyCombiner.class);
        //4.reduce处理
        job.setReducerClass(MyReducer.class);
        job.setOutputKeyClass(Text.class);
        job.setOutputValueClass(Text.class);
        //5.输出文件
        FileOutputFormat.setOutputPath(job, new Path(args[1]));
        //6.启动
        boolean result = job.waitForCompletion(true);
        System.exit(result ? 1 : 0);
    }

    public static class MyMapper extends Mapper<LongWritable, Text, Text, Text> {
        @Override
        protected void map(LongWritable key, Text value, Context context) throws IOException, InterruptedException {
            String line = value.toString();
            String[] arr = line.split("\t");

            String fileName = ((FileSplit) context.getInputSplit()).getPath().getName();

            for (String word : arr) {
                //<welcome_a.txt,1>
                //<welcome_b.txt,1>
                context.write(new Text(word + "_" + fileName), new Text("1"));
            }
        }
    }

    public static class MyCombiner extends Reducer<Text, Text, Text, Text> {
        @Override
        protected void reduce(Text key, Iterable<Text> values, Context context) throws IOException, InterruptedException {
            Integer sum = 0;
            for (Text value : values) {
                sum += Integer.parseInt(value.toString());
            }

            String[] arr = key.toString().split("_");
            //<welcome,"a.txt->2">
            //<welcome,"b.txt->2">
            context.write(new Text(arr[0]), new Text(arr[1] + "->" + sum));
        }
    }

    public static class MyReducer extends Reducer<Text, Text, Text, Text> {
        @Override
        protected void reduce(Text key, Iterable<Text> values, Context context) throws IOException, InterruptedException {
            Map<String, Integer> map = new HashMap<>();
            for (Text value : values) {
                String[] arr = value.toString().split("->");
            if (map.get(arr[0]) == null) {
                map.put(arr[0], Integer.parseInt(arr[1]));
            } else {
                map.put(arr[0], map.get(arr[0]) + Integer.parseInt(arr[1]));
            }
        }

        StringBuffer sb = new StringBuffer();
            for (Map.Entry<String, Integer> entry : map.entrySet()) {
            sb.append(entry.getKey()).append("->").append(entry.getValue());
            sb.append(",");
        }
            context.write(key, new Text(sb.substring(0, sb.length() - 1)));
        }
    }

}
 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值