GIRAPH-525: Add PartitionClass to the ConfigurationUtils
Project: http://git-wip-us.apache.org/repos/asf/giraph/repo Commit: http://git-wip-us.apache.org/repos/asf/giraph/commit/c0ee023a Tree: http://git-wip-us.apache.org/repos/asf/giraph/tree/c0ee023a Diff: http://git-wip-us.apache.org/repos/asf/giraph/diff/c0ee023a Branch: refs/heads/trunk Commit: c0ee023aaa3c7712c7f61c8a797460290c225d14 Parents: 5d00532 Author: Claudio Martella <[email protected]> Authored: Mon Feb 18 19:58:00 2013 +0100 Committer: Claudio Martella <[email protected]> Committed: Mon Feb 18 19:58:00 2013 +0100 ---------------------------------------------------------------------- .../apache/giraph/utils/ConfigurationUtils.java | 7 +++++++ 1 files changed, 7 insertions(+), 0 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/giraph/blob/c0ee023a/giraph-core/src/main/java/org/apache/giraph/utils/ConfigurationUtils.java ---------------------------------------------------------------------- diff --git a/giraph-core/src/main/java/org/apache/giraph/utils/ConfigurationUtils.java b/giraph-core/src/main/java/org/apache/giraph/utils/ConfigurationUtils.java index 5e744a6..bd2a9c4 100644 --- a/giraph-core/src/main/java/org/apache/giraph/utils/ConfigurationUtils.java +++ b/giraph-core/src/main/java/org/apache/giraph/utils/ConfigurationUtils.java @@ -36,6 +36,7 @@ import org.apache.giraph.io.EdgeInputFormat; import org.apache.giraph.io.VertexInputFormat; import org.apache.giraph.io.VertexOutputFormat; import org.apache.giraph.master.MasterCompute; +import org.apache.giraph.partition.Partition; import org.apache.giraph.vertex.Vertex; import org.apache.giraph.worker.WorkerContext; import org.apache.hadoop.fs.Path; @@ -76,6 +77,7 @@ public final class ConfigurationUtils { OPTIONS.addOption("aw", "aggregatorWriter", true, "AggregatorWriter class"); OPTIONS.addOption("mc", "masterCompute", true, "MasterCompute class"); OPTIONS.addOption("cf", "cacheFile", true, "Files for distributed cache"); + OPTIONS.addOption("pc", "partitionClass", true, "Partition class"); OPTIONS.addOption("ca", "customArguments", true, "provide custom" + " arguments for the job configuration in the form:" + " -ca <param1>=<value1>,<param2>=<value2> -ca <param3>=<value3> etc." + @@ -236,6 +238,11 @@ public final class ConfigurationUtils { "does not require one."); } } + if (cmd.hasOption("pc")) { + giraphConfiguration.setPartitionClass( + (Class<? extends Partition>) + Class.forName(cmd.getOptionValue("pc"))); + } if (cmd.hasOption("ca")) { for (String caOptionValue : cmd.getOptionValues("ca")) { for (String paramValue :
