io.camunda.zeebe.broker.partitioning.startup.steps.RaftJoinStep Maven / Gradle / Ivy
/*
* Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH under
* one or more contributor license agreements. See the NOTICE file distributed
* with this work for additional information regarding copyright ownership.
* Licensed under the Camunda License 1.0. You may not use this file
* except in compliance with the Camunda License 1.0.
*/
package io.camunda.zeebe.broker.partitioning.startup.steps;
import io.camunda.zeebe.broker.partitioning.startup.PartitionStartupContext;
import io.camunda.zeebe.scheduler.future.ActorFuture;
import io.camunda.zeebe.scheduler.startup.StartupStep;
public class RaftJoinStep implements StartupStep {
@Override
public String getName() {
return "Joining Raft Partition";
}
@Override
public ActorFuture startup(final PartitionStartupContext context) {
final var result = context.concurrencyControl().createFuture();
final var partition =
context
.raftPartitionFactory()
.createRaftPartition(context.partitionMetadata(), context.partitionDirectory());
// Immediately save the partition to the context, so that it can be closed in case of an error.
context.raftPartition(partition);
partition
.join(context.partitionManagementService(), context.snapshotStore())
.whenComplete(
(raftPartition, throwable) -> {
if (throwable == null) {
result.complete(context);
} else {
result.completeExceptionally(throwable);
}
});
return result;
}
@Override
public ActorFuture shutdown(final PartitionStartupContext context) {
final var result = context.concurrencyControl().createFuture();
final var raftPartition = context.raftPartition();
if (raftPartition == null) {
result.complete(context);
return result;
}
raftPartition
.close()
.whenComplete(
(ignored, throwable) -> {
if (throwable == null) {
result.complete(context.raftPartition(null));
} else {
result.completeExceptionally(throwable);
}
});
return result;
}
}