[gobblin] branch master updated: Expose functions to fetch record partitionColumn value (#3810)

[zi...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zihanli58 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/gobblin.git


The following commit(s) were added to refs/heads/master by this push:
     new be849adf3 Expose functions to fetch record partitionColumn value (#3810)
be849adf3 is described below

commit be849adf3e0b1b8e1e1498ab9014b22b8d8556dc
Author: Andy Jiang <an...@outlook.com>
AuthorDate: Thu Oct 26 16:42:04 2023 -0700

    Expose functions to fetch record partitionColumn value (#3810)
---
 .../gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gobblin-core/src/main/java/org/apache/gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java b/gobblin-core/src/main/java/org/apache/gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java
index 5ee1975c3..28e0f7bad 100644
--- a/gobblin-core/src/main/java/org/apache/gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java
+++ b/gobblin-core/src/main/java/org/apache/gobblin/writer/partitioner/TimeBasedAvroWriterPartitioner.java
@@ -85,7 +85,7 @@ public class TimeBasedAvroWriterPartitioner extends TimeBasedWriterPartitioner<G
   /**
    *  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 @@ public class TimeBasedAvroWriterPartitioner extends TimeBasedWriterPartitioner<G
   /**
    * 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();
     }