Refactoring

Author: Amar3tto

sdks/java/io/debezium/src/main/java/org/apache/beam/io/debezium/DebeziumReadSchemaTransformProvider.java

@@ -151,13 +151,13 @@ public PCollectionRowTuple expand(PCollectionRowTuple input) {
                   .withMaxNumberOfRecords(testLimitRecords)
                   .withMaxTimeToRun(testLimitMilliseconds);
         } else {
-          if (configuration.getMaxNumberOfRecords() != null) {
-            readTransform =
-                readTransform.withMaxNumberOfRecords(configuration.getMaxNumberOfRecords());
+          Integer maxNumberOfRecords = configuration.getMaxNumberOfRecords();
+          if (maxNumberOfRecords != null) {
+            readTransform = readTransform.withMaxNumberOfRecords(maxNumberOfRecords);
           }
-
-          if (configuration.getMaxTimeToRun() != null) {
-            readTransform = readTransform.withMaxTimeToRun(configuration.getMaxTimeToRun());
+          Long maxTimeToRun = configuration.getMaxTimeToRun();
+          if (maxTimeToRun != null) {
+            readTransform = readTransform.withMaxTimeToRun(maxTimeToRun);
           }
         }
 
@@ -196,13 +196,12 @@ public PCollectionRowTuple expand(PCollectionRowTuple input) {
   @DefaultSchema(AutoValueSchema.class)
   @AutoValue
   public abstract static class DebeziumReadSchemaTransformConfiguration {
-    @Nullable
+
     @SchemaFieldDescription("Maximum number of records to read before stopping.")
-    public abstract Integer getMaxNumberOfRecords();
+    public abstract @Nullable Integer getMaxNumberOfRecords();
 
-    @Nullable
     @SchemaFieldDescription("Maximum time in milliseconds to run before stopping.")
-    public abstract Long getMaxTimeToRun();
+    public abstract @Nullable Long getMaxTimeToRun();
 
     @SchemaFieldDescription("Username used to connect to the source database.")
     public abstract String getUsername();