jack-moseley commented on code in PR #3670: URL: https://github.com/apache/gobblin/pull/3670#discussion_r1163325383
########## gobblin-data-management/src/main/java/org/apache/gobblin/data/management/version/finder/LookbackDateTimeDatasetVersionFinder.java: ########## @@ -0,0 +1,98 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.gobblin.data.management.version.finder; + +import java.io.IOException; +import java.util.Collection; +import java.util.HashSet; +import java.util.Set; + +import org.apache.hadoop.fs.FileSystem; +import org.apache.hadoop.fs.Path; +import org.joda.time.DateTime; +import org.joda.time.Duration; +import org.joda.time.Instant; +import org.joda.time.Period; +import org.joda.time.format.PeriodFormatter; +import org.joda.time.format.PeriodFormatterBuilder; + +import com.google.common.annotations.VisibleForTesting; +import com.google.common.base.Preconditions; +import com.typesafe.config.Config; + +import org.apache.gobblin.data.management.version.FileSystemDatasetVersion; +import org.apache.gobblin.data.management.version.TimestampedDatasetVersion; +import org.apache.gobblin.dataset.Dataset; +import org.apache.gobblin.dataset.FileSystemDataset; +import org.apache.gobblin.util.ConfigUtils; + + +/** + * {@link DatasetVersionFinder} that constructs {@link TimestampedDatasetVersion}s without actually checking for existence + * of the version path. The version path is constructed by appending the version partition pattern to the dataset root. + * The versions are found by looking back a specific period of time and finding unique date partitions between that + * time and the current time. + */ +public class LookbackDateTimeDatasetVersionFinder extends DateTimeDatasetVersionFinder { + public static final String VERSION_PATH_PREFIX = "version.path.prefix"; + public static final String VERSION_LOOKBACK_PERIOD = "version.lookback.period"; + + private final Duration stepDuration; + private final Period lookbackPeriod; + private final String pathPrefix; + private final Instant endTime; + + public LookbackDateTimeDatasetVersionFinder(FileSystem fs, Config config) { + this(fs, config, Instant.now()); + } + + @VisibleForTesting + public LookbackDateTimeDatasetVersionFinder(FileSystem fs, Config config, Instant endTime) { + super(fs, config); + Preconditions.checkArgument(config.hasPath(VERSION_LOOKBACK_PERIOD) , "Missing required property " + VERSION_LOOKBACK_PERIOD); + PeriodFormatter periodFormatter = + new PeriodFormatterBuilder().appendYears().appendSuffix("y").appendMonths().appendSuffix("M").appendDays() + .appendSuffix("d").appendHours().appendSuffix("h").appendMinutes().appendSuffix("m").toFormatter(); + this.stepDuration = Duration.standardMinutes(1); + this.pathPrefix = ConfigUtils.getString(config, VERSION_PATH_PREFIX, ""); + this.lookbackPeriod = periodFormatter.parsePeriod(config.getString(VERSION_LOOKBACK_PERIOD)); + this.endTime = endTime; + } + + @Override + public Class<? extends FileSystemDatasetVersion> versionClass() { + return TimestampedDatasetVersion.class; + } + + @Override + public Collection<TimestampedDatasetVersion> findDatasetVersions(Dataset dataset) throws IOException { Review Comment: If we set the iceberg retention lookback to one day longer than the file retention (which is what my plan is), it means each drop_files GMCE will contain a list of 24 previous partitions instead of about 3. I expect there should be little to no performance impact due to that. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
