Expose functions to fetch record partitionColumn value (#3810)

apache · Oct 26, 2023 · be849ad · be849ad
1 parent 918716f
commit be849ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...e/src/main/java/org/apache/gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java b/...e/src/main/java/org/apache/gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java
@@ -85,7 +85,7 @@ public long getRecordTimestamp(GenericRecord record) {
   /**
    *  Check if the partition column value is present and is a Long object. Otherwise, use current system time.
    */
-  private long getRecordTimestamp(Optional<Object> writerPartitionColumnValue) {
+  protected long getRecordTimestamp(Optional<Object> writerPartitionColumnValue) {
 
     if (writerPartitionColumnValue.isPresent()) {
       Object val = writerPartitionColumnValue.get();
@@ -103,7 +103,7 @@ private long getRecordTimestamp(Optional<Object> writerPartitionColumnValue) {
   /**
    * Retrieve the value of the partition column field specified by this.partitionColumns
    */
-  private Optional<Object> getWriterPartitionColumnValue(GenericRecord record) {
+  protected Optional<Object> getWriterPartitionColumnValue(GenericRecord record) {
     if (!this.partitionColumns.isPresent()) {
       return Optional.absent();
     }