com.palantir.atlasdb.services.test.TestSweeperModule Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of atlasdb-dagger Show documentation
Palantir open source project
There is a newer version: 0.1152.0
/*
 * (c) Copyright 2018 Palantir Technologies Inc. All rights reserved.
 *
 * Licensed 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 com.palantir.atlasdb.services.test;

import com.google.common.collect.ImmutableList;
import com.palantir.atlasdb.cleaner.Follower;
import com.palantir.atlasdb.keyvalue.api.KeyValueService;
import com.palantir.atlasdb.services.ServicesConfig;
import com.palantir.atlasdb.sweep.CellsSweeper;
import com.palantir.atlasdb.sweep.SweepTaskRunner;
import com.palantir.atlasdb.transaction.impl.SerializableTransactionManager;
import com.palantir.atlasdb.transaction.impl.SweepStrategyManager;
import com.palantir.atlasdb.transaction.service.TransactionService;
import dagger.Module;
import dagger.Provides;
import java.util.Optional;
import java.util.function.LongSupplier;
import javax.inject.Named;
import javax.inject.Singleton;

@Module
public class TestSweeperModule {

    private final Optional unreadableTs;
    private final Optional immutableTs;

    public static TestSweeperModule create(LongSupplier ts) {
        return new TestSweeperModule(Optional.of(ts), Optional.of(ts));
    }

    public static TestSweeperModule create(LongSupplier unreadableTs, LongSupplier immutableTs) {
        return new TestSweeperModule(Optional.of(unreadableTs), Optional.of(immutableTs));
    }

    public TestSweeperModule() {
        this.unreadableTs = Optional.empty();
        this.immutableTs = Optional.empty();
    }

    private TestSweeperModule(Optional unreadableTs, Optional immutableTs) {
        this.unreadableTs = unreadableTs;
        this.immutableTs = immutableTs;
    }

    @Provides
    @Singleton
    @SuppressWarnings("OptionalOrElseMethodInvocation")
    public SweepTaskRunner provideSweepTaskRunner(
            SerializableTransactionManager txm,
            @Named("kvs") KeyValueService kvs,
            TransactionService transactionService,
            SweepStrategyManager sweepStrategyManager,
            Follower follower,
            ServicesConfig config) {
        LongSupplier unreadable = unreadableTs.orElse(txm::getUnreadableTimestamp);
        LongSupplier immutable = immutableTs.orElse(txm::getImmutableTimestamp);
        return new SweepTaskRunner(
                kvs, unreadable, immutable, transactionService, new CellsSweeper(txm, kvs, ImmutableList.of(follower)));
    }
}