fix NPE in IndexGeneratorJob (apache#4371)

* fix NPE in IndexGeneratorJob * address review comment * review comments
caobaohe · Jun 7, 2017 · 4ace65a · 4ace65a
1 parent 2d15215
commit 4ace65a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/indexing-hadoop/src/main/java/io/druid/indexer/IndexGeneratorJob.java b/indexing-hadoop/src/main/java/io/druid/indexer/IndexGeneratorJob.java
@@ -454,7 +454,7 @@ public int getPartition(BytesWritable bytesWritable, Writable value, int numPart
       final ByteBuffer bytes = ByteBuffer.wrap(bytesWritable.getBytes());
       bytes.position(4); // Skip length added by SortableBytes
       int shardNum = bytes.getInt();
-      if (config.get("mapred.job.tracker").equals("local")) {
+      if ("local".equals(config.get("mapreduce.jobtracker.address")) || "local".equals(config.get("mapred.job.tracker"))) {
         return shardNum % numPartitions;
       } else {
         if (shardNum >= numPartitions) {