Avoid nullpointer exception when optional extras are left out

rejasupotaro · Jun 28, 2015 · c778b7e · c778b7e
1 parent 64bc38f
commit c778b7e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler/src/main/java/se/emilsjolander/intentbuilder/Processor.java b/compiler/src/main/java/se/emilsjolander/intentbuilder/Processor.java
@@ -153,7 +153,9 @@ private TypeSpec getBuilderSpec(Element annotatedElement) {
                 .addStatement("$T extras = intent.getExtras()", Bundle.class);
         for (Element e : all) {
             String paramName = e.getSimpleName().toString();
+            injectMethod.beginControlFlow("if (extras.containsKey($S))", paramName);
             injectMethod.addStatement("activity.$N = ($T) extras.get($S)", paramName, e.asType(), paramName);
+            injectMethod.endControlFlow();
         }
         builder.addMethod(injectMethod.build());