io.camunda.zeebe.shared.management.BackupEndpoint 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.shared.management;
import io.camunda.zeebe.broker.client.api.BrokerClient;
import io.camunda.zeebe.broker.client.api.BrokerErrorException;
import io.camunda.zeebe.gateway.admin.IncompleteTopologyException;
import io.camunda.zeebe.gateway.admin.backup.BackupAlreadyExistException;
import io.camunda.zeebe.gateway.admin.backup.BackupApi;
import io.camunda.zeebe.gateway.admin.backup.BackupRequestHandler;
import io.camunda.zeebe.gateway.admin.backup.BackupStatus;
import io.camunda.zeebe.gateway.admin.backup.PartitionBackupStatus;
import io.camunda.zeebe.gateway.admin.backup.State;
import io.camunda.zeebe.management.backups.BackupInfo;
import io.camunda.zeebe.management.backups.Error;
import io.camunda.zeebe.management.backups.PartitionBackupInfo;
import io.camunda.zeebe.management.backups.StateCode;
import io.camunda.zeebe.management.backups.TakeBackupResponse;
import io.camunda.zeebe.protocol.management.BackupStatusCode;
import io.netty.channel.ConnectTimeoutException;
import java.net.ConnectException;
import java.time.Instant;
import java.time.OffsetDateTime;
import java.time.ZoneId;
import java.util.concurrent.CompletionException;
import java.util.concurrent.TimeoutException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.actuate.endpoint.annotation.DeleteOperation;
import org.springframework.boot.actuate.endpoint.annotation.ReadOperation;
import org.springframework.boot.actuate.endpoint.annotation.Selector;
import org.springframework.boot.actuate.endpoint.annotation.WriteOperation;
import org.springframework.boot.actuate.endpoint.web.WebEndpointResponse;
import org.springframework.boot.actuate.endpoint.web.annotation.WebEndpoint;
import org.springframework.lang.NonNull;
import org.springframework.stereotype.Component;
@Component
@WebEndpoint(id = "backups")
public final class BackupEndpoint {
private final BackupApi api;
@SuppressWarnings("unused") // used by Spring
@Autowired
public BackupEndpoint(final BrokerClient client) {
this(new BackupRequestHandler(client));
}
BackupEndpoint(final BackupApi api) {
this.api = api;
}
@WriteOperation
public WebEndpointResponse take(final long backupId) {
try {
if (backupId <= 0) {
return new WebEndpointResponse<>(
new Error().message("A backupId must be provided and it must be > 0"),
WebEndpointResponse.STATUS_BAD_REQUEST);
}
api.takeBackup(backupId).toCompletableFuture().join();
return new WebEndpointResponse<>(
new TakeBackupResponse()
.message(
"A backup with id %d has been scheduled. Use GET actuator/backups/%d to monitor the status."
.formatted(backupId, backupId)),
202);
} catch (final Exception e) {
return mapErrorResponse(e);
}
}
@ReadOperation
public WebEndpointResponse status(@Selector @NonNull final long id) {
try {
final BackupStatus status = api.getStatus(id).toCompletableFuture().join();
if (status.status() == State.DOES_NOT_EXIST) {
return doestNotExistResponse(status.backupId());
}
final BackupInfo backupInfo = getBackupInfoFromBackupStatus(status);
return new WebEndpointResponse<>(backupInfo);
} catch (final Exception e) {
return mapErrorResponse(e);
}
}
@ReadOperation
public WebEndpointResponse list() {
try {
final var backups = api.listBackups().toCompletableFuture().join();
final var response = backups.stream().map(this::getBackupInfoFromBackupStatus).toList();
return new WebEndpointResponse<>(response);
} catch (final Exception e) {
return mapErrorResponse(e);
}
}
@DeleteOperation
public WebEndpointResponse delete(@Selector @NonNull final long id) {
try {
api.deleteBackup(id).toCompletableFuture().join();
return new WebEndpointResponse<>(WebEndpointResponse.STATUS_NO_CONTENT);
} catch (final Exception e) {
return mapErrorResponse(e);
}
}
private BackupInfo getBackupInfoFromBackupStatus(final BackupStatus status) {
final BackupInfo backupInfo =
new BackupInfo().backupId(status.backupId()).state(getBackupStateCode(status.status()));
status.failureReason().ifPresent(backupInfo::setFailureReason);
final var details = status.partitions().stream().map(this::toPartitionBackupInfo).toList();
backupInfo.setDetails(details);
return backupInfo;
}
private static WebEndpointResponse doestNotExistResponse(final long backupId) {
return new WebEndpointResponse<>(
new Error().message("Backup with id %d does not exist".formatted(backupId)),
WebEndpointResponse.STATUS_NOT_FOUND);
}
private PartitionBackupInfo toPartitionBackupInfo(final PartitionBackupStatus partitionStatus) {
final var partitionBackupInfo =
new PartitionBackupInfo()
.partitionId(partitionStatus.partitionId())
.state(getPartitionBackupStateCode(partitionStatus.status()));
partitionStatus.failureReason().ifPresent(partitionBackupInfo::setFailureReason);
partitionStatus
.createdAt()
.ifPresent(
time -> {
final var i = Instant.parse(time);
partitionBackupInfo.createdAt(OffsetDateTime.ofInstant(i, ZoneId.of("UTC")));
});
partitionStatus
.lastUpdatedAt()
.ifPresent(
time -> {
final var i = Instant.parse(time);
partitionBackupInfo.lastUpdatedAt(OffsetDateTime.ofInstant(i, ZoneId.of("UTC")));
});
partitionStatus.brokerId().ifPresent(partitionBackupInfo::setBrokerId);
partitionStatus.brokerVersion().ifPresent(partitionBackupInfo::setBrokerVersion);
partitionStatus.snapshotId().ifPresent(partitionBackupInfo::setSnapshotId);
partitionStatus.checkpointPosition().ifPresent(partitionBackupInfo::setCheckpointPosition);
return partitionBackupInfo;
}
private StateCode getBackupStateCode(final State state) {
return switch (state) {
case IN_PROGRESS -> StateCode.IN_PROGRESS;
case COMPLETED -> StateCode.COMPLETED;
case FAILED -> StateCode.FAILED;
case DOES_NOT_EXIST -> StateCode.DOES_NOT_EXIST;
case INCOMPLETE -> StateCode.INCOMPLETE;
};
}
private StateCode getPartitionBackupStateCode(final BackupStatusCode status) {
return switch (status) {
case IN_PROGRESS -> StateCode.IN_PROGRESS;
case COMPLETED -> StateCode.COMPLETED;
case FAILED -> StateCode.FAILED;
case DOES_NOT_EXIST -> StateCode.DOES_NOT_EXIST;
default -> throw new IllegalStateException("Unknown BackupState %s".formatted(status));
};
}
private WebEndpointResponse mapErrorResponse(final Throwable exception) {
final int errorCode;
final String message;
if (exception instanceof CompletionException) {
final var error = exception.getCause();
if (error instanceof BackupAlreadyExistException) {
errorCode = 409;
message = error.getMessage();
} else if (error instanceof IncompleteTopologyException) {
errorCode = 502;
message = error.getMessage();
} else if (error instanceof TimeoutException || error instanceof ConnectTimeoutException) {
errorCode = 504;
message = "Request from gateway to broker timed out. " + error.getMessage();
} else if (error instanceof ConnectException) {
errorCode = 502;
message = "Failed to send request from gateway to broker." + error.getMessage();
} else if (error instanceof final BrokerErrorException brokerError) {
final var rootError = brokerError.getError();
errorCode =
switch (rootError.getCode()) {
case PARTITION_LEADER_MISMATCH -> 502;
case RESOURCE_EXHAUSTED -> WebEndpointResponse.STATUS_SERVICE_UNAVAILABLE;
case UNSUPPORTED_MESSAGE -> WebEndpointResponse.STATUS_BAD_REQUEST;
default -> WebEndpointResponse.STATUS_INTERNAL_SERVER_ERROR;
};
message = rootError.getMessage();
} else {
errorCode = WebEndpointResponse.STATUS_INTERNAL_SERVER_ERROR;
message = error.getMessage();
}
} else {
errorCode = WebEndpointResponse.STATUS_INTERNAL_SERVER_ERROR;
message = exception.getMessage();
}
return new WebEndpointResponse<>(new Error().message(message), errorCode);
}
}