Merge branch 'mbalassi-getters' containing getters for unexposed Wind…

…owFn fields
chinglinli · May 12, 2015 · 5b5e631 · 5b5e631
2 parents 6fbb53e + a96562f
commit 5b5e631
Show file tree

Hide file tree

Showing 4 changed files with 77 additions and 0 deletions.
diff --git a/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/CalendarWindows.java b/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/CalendarWindows.java
@@ -142,6 +142,19 @@ public boolean isCompatible(WindowFn other) {
           && startDate == that.startDate
           && timeZone == that.timeZone;
     }
+
+    public int getNumber() {
+      return number;
+    }
+
+    public DateTime getStartDate() {
+      return startDate;
+    }
+
+    public DateTimeZone getTimeZone() {
+      return timeZone;
+    }
+
   }
 
   /**
@@ -218,6 +231,23 @@ public boolean isCompatible(WindowFn other) {
           && startDate == that.startDate
           && timeZone == that.timeZone;
     }
+
+    public int getNumber() {
+      return number;
+    }
+
+    public int getDayOfMonth() {
+      return dayOfMonth;
+    }
+
+    public DateTime getStartDate() {
+      return startDate;
+    }
+
+    public DateTimeZone getTimeZone() {
+      return timeZone;
+    }
+
   }
 
   /**
@@ -299,5 +329,26 @@ public boolean isCompatible(WindowFn other) {
           && startDate == that.startDate
           && timeZone == that.timeZone;
     }
+
+    public DateTimeZone getTimeZone() {
+      return timeZone;
+    }
+
+    public DateTime getStartDate() {
+      return startDate;
+    }
+
+    public int getDayOfMonth() {
+      return dayOfMonth;
+    }
+
+    public int getMonthOfYear() {
+      return monthOfYear;
+    }
+
+    public int getNumber() {
+      return number;
+    }
+
   }
 }
diff --git a/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/FixedWindows.java b/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/FixedWindows.java
@@ -91,4 +91,13 @@ public boolean isCompatible(WindowFn<?, ?> other) {
         && (size.equals(((FixedWindows) other).size))
         && (offset.equals(((FixedWindows) other).offset));
   }
+
+  public Duration getSize() {
+    return size;
+  }
+
+  public Duration getOffset() {
+    return offset;
+  }
+
 }
diff --git a/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/Sessions.java b/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/Sessions.java
@@ -84,4 +84,8 @@ public boolean isCompatible(WindowFn<?, ?> other) {
   public IntervalWindow getSideInputWindow(BoundedWindow window) {
     throw new UnsupportedOperationException("Sessions is not allowed in side inputs");
   }
+
+  public Duration getGapDuration() {
+    return gapDuration;
+  }
 }
diff --git a/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/SlidingWindows.java b/sdk/src/main/java/com/google/cloud/dataflow/sdk/transforms/windowing/SlidingWindows.java
@@ -163,4 +163,17 @@ static Duration getDefaultPeriod(Duration size) {
     }
     return Duration.millis(1);
   }
+
+  public Duration getPeriod() {
+    return period;
+  }
+
+  public Duration getSize() {
+    return size;
+  }
+
+  public Duration getOffset() {
+    return offset;
+  }
+
 }