com.azure.cosmos.implementation.changefeed.implementation.PartitionCheckpointerImpl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of azure-cosmos Show documentation
Show all versions of azure-cosmos Show documentation
This Package contains Microsoft Azure Cosmos SDK (with Reactive Extension Reactor support) for Azure Cosmos DB SQL API
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos.implementation.changefeed.implementation;
import com.azure.cosmos.implementation.changefeed.Lease;
import com.azure.cosmos.implementation.changefeed.LeaseCheckpointer;
import com.azure.cosmos.implementation.changefeed.PartitionCheckpointer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import reactor.core.publisher.Mono;
import static com.azure.cosmos.implementation.guava25.base.Preconditions.checkArgument;
import static com.azure.cosmos.implementation.guava25.base.Preconditions.checkNotNull;
/**
* Checkpoint the given partition up to the given continuation token.
*/
class PartitionCheckpointerImpl implements PartitionCheckpointer {
private final Logger logger = LoggerFactory.getLogger(PartitionCheckpointerImpl.class);
private final LeaseCheckpointer leaseCheckpointer;
private Lease lease;
public PartitionCheckpointerImpl(LeaseCheckpointer leaseCheckpointer, Lease lease) {
this.leaseCheckpointer = leaseCheckpointer;
this.lease = lease;
}
@Override
public Mono checkpointPartition(ChangeFeedState continuationState) {
checkNotNull(continuationState, "Argument 'continuationSttae' must not be null.");
checkArgument(
continuationState.getContinuation().getContinuationTokenCount() == 1,
"For ChangeFeedProcessor the continuation state should always have one range/continuation");
return this.leaseCheckpointer.checkpoint(
this.lease,
continuationState.getContinuation().getCurrentContinuationToken().getToken())
.map(lease1 -> {
this.lease = lease1;
logger.info("Checkpoint: partition {}, new continuation {}", this.lease.getLeaseToken(), this.lease.getContinuationToken());
return lease1;
});
}
}