org.smartfrog.services.hadoop.benchmark.citerank.InitializeRanks.java Source code

Introduction

Here is the source code for org.smartfrog.services.hadoop.benchmark.citerank.InitializeRanks.java

Source

/** (C) Copyright 2009 Hewlett-Packard Development Company, LP
    
 This library is free software; you can redistribute it and/or
 modify it under the terms of the GNU Lesser General Public
 License as published by the Free Software Foundation; either
 version 2.1 of the License, or (at your option) any later version.
    
 This library is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 Lesser General Public License for more details.
    
 You should have received a copy of the GNU Lesser General Public
 License along with this library; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
    
 For more information: www.smartfrog.org
    
 */
package org.smartfrog.services.hadoop.benchmark.citerank;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapred.JobConf;
import org.apache.hadoop.util.ToolRunner;

public class InitializeRanks extends CiteRankTool {
    @Override
    protected String getName() {
        return "InitializeRanks";
    }

    @Override
    public int run(String[] args) throws Exception {
        if (args.length != 3) {
            return usage("<input path> <output path> <number of pages>");
        }

        JobConf conf = createInputOutputConfiguration(args);

        conf.set(CiteRankTool.RANK_COUNT, args[2]);

        conf.setMapperClass(InitializeRanksMapper.class);

        conf.setMapOutputKeyClass(Text.class);
        conf.setMapOutputValueClass(Text.class);

        conf.setNumMapTasks(CiteRankTool.NUM_MAP_TASKS);
        conf.setNumReduceTasks(CiteRank.NUM_REDUCE_TASKS);

        return runJob(conf);
    }

    public static void main(String[] args) throws Exception {
        int exitCode = ToolRunner.run(new Configuration(), new InitializeRanks(), args);
        System.exit(exitCode);
    }

}