定制mapreduce输出

最新推荐文章于 2024-01-30 21:55:32 发布

hjxhjh

最新推荐文章于 2024-01-30 21:55:32 发布

阅读量858

点赞数 1

这里以redis数据库为例。

这里的例子是，我想统计日志文件中的某天各个小时的访问量，日志格式为：

?

1	`2014` `-` `02` `-` `10` `04` `:` `52` `:` `34` `127.0` `.` `0.1` `xxx`

我们知道在写mapreduce job时，要配置输入输出，然后编写mapper和reducer类，hadoop默认输出是到hdfs的文件中，例如：

?

1	`job.setOutputFormatClass(FileOutputFormat.` `class` `);`

现在我们想要将任务计算结果输出到数据库(redis)中，怎么做呢？可以继承FileOutputFormat类，定制自己的类，看代码：

?

publicclass LoginLogOutputFormat<K, V> extends FileOutputFormat<K, V> {
    /**
     * 重点也是定制一个RecordWriter类，每一条reduce处理后的记录，我们便可将该记录输出到数据库中
     */
    protectedstatic class RedisRecordWriter<K, V> extends RecordWriter<K, V>{
        privateJedis jedis; //redis的client实例
         
        publicRedisRecordWriter(Jedis jedis){
            this.jedis = jedis;
        }
         
        @Override
        publicvoid write(K key, V value) throws IOException,
                InterruptedException {
             
            booleannullKey = key == null;
            booleannullValue = value == null;
            if(nullKey || nullValue) return;
             
            String[] sKey = key.toString().split("-");
            String outKey = sKey[0]+"-"+sKey[1]+"-"+sKey[2]+"_login_stat";//zset key为yyyy-MM-dd_login_stat
            jedis.zadd(outKey.getBytes("UTF-8"), -1,
                        (sKey[3]+":"+value).getBytes("UTF-8"));//zadd, 其值格式为: 时刻:访问量
        }
 
        @Override
        publicvoid close(TaskAttemptContext context) throws IOException,
                InterruptedException {
            if(jedis != null) jedis.disconnect(); //关闭链接
        }
    }
     
    @Override
    publicRecordWriter<K, V> getRecordWriter(TaskAttemptContext job)
            throwsIOException, InterruptedException {
        Jedis jedis = RedisClient.newJedis(); //构建一个redis，这里你可以自己根据实际情况来构建数据库连接对象
        //System.out.println("构建RedisRecordWriter");
        returnnew RedisRecordWriter<K, V>(jedis);
    }
}

下面就是整个job实现：

?

publicclass LoginLogStatTask extends Configured implements Tool {
    publicstatic class MyMapper extends Mapper<LongWritable, Text, Text, IntWritable>{
        @Override
        protectedvoid map(LongWritable key, Text value, Context context)
                throwsIOException, InterruptedException {
            if(value == null || "".equals(value))return;
            // 解析value,如: 2014-02-10 04:52:34 127.0.0.1 xxx
            String[] fields = value.toString().split(" ");
            String date = fields[0];
            String time = fields[1];
            String hour = time.split(":")[0];
            String outKey = date+"-"+hour;
            context.write(newText(outKey), new IntWritable(1));
        }
    }
     
    publicstatic class MyReducer extends Reducer<Text, IntWritable, Text, IntWritable>{
        @Override
        protectedvoid reduce(Text key, Iterable<IntWritable> values,
                Context context)
                throwsIOException, InterruptedException {
            intcount = 0;
            while(values.iterator().hasNext()){ //统计数量
                count ++;
                values.iterator().next();
            }
            context.write(key,new IntWritable(count));
        }
    }
 
    @Override
    publicint run(String[] args) throws Exception {
        Configuration conf = getConf();
        List<Path> inputs = new ArrayList<>();
        String inputPath = args[0];
        if(inputPath.endsWith("/")){//如果是目录
            inputs.addAll(HdfsUtil.listFiles(inputPath, conf));
        }else{//如果是文件
            inputs.add(newPath(inputPath));
        }
        longts = System.currentTimeMillis();
        String jobName = "login_logs_stat_job_" + ts;
        Job job = Job.getInstance(conf, jobName);
        job.setJarByClass(LoginLogStatTask.class);
        //添加输入文件路径
        for(Path p : inputs){
            FileInputFormat.addInputPath(job, p);
        }
        //设置输出路径
        Path out = new Path(jobName + ".out");//以jobName.out作为输出
        FileOutputFormat.setOutputPath(job, out);
        //设置mapper
        job.setMapperClass(MyMapper.class);
        //设置reducer
        job.setReducerClass(MyReducer.class);
         
        //设置输入格式
        job.setInputFormatClass(TextInputFormat.class);
        //设置输出格式
        job.setOutputFormatClass(LoginLogOutputFormat.class);
        //设置输出key类型
        job.setOutputKeyClass(Text.class);
        //设置输出value类型
        job.setOutputValueClass(IntWritable.class);
        job.waitForCompletion(true);
        returnjob.isSuccessful()?0:1;
    }
      
    publicstatic void main(String[] args) throws Exception {
        Configuration conf = new Configuration();
        intres = ToolRunner.run(conf, new LoginLogStatTask(), args);
        System.exit(res);
    }