Examples with ServerErrorCode com.github.ambry.commons.ServerErrorCode used on opensource projects

Example 11 with ServerErrorCode

use of com.github.ambry.commons.ServerErrorCode in project ambry by linkedin.

the class AmbryRequests method handleBlobStoreControlRequest.

/**
 * Handles {@link com.github.ambry.protocol.AdminRequestOrResponseType#BlobStoreControl}.
 * @param requestStream the serialized bytes of the request.
 * @param adminRequest the {@link AdminRequest} received.
 * @return the {@link AdminResponse} to the request.
 * @throws IOException if there is any I/O error reading from the {@code requestStream}.
 */
private AdminResponse handleBlobStoreControlRequest(DataInputStream requestStream, AdminRequest adminRequest) throws IOException {
    ServerErrorCode error;
    BlobStoreControlAdminRequest blobStoreControlAdminRequest = BlobStoreControlAdminRequest.readFrom(requestStream, adminRequest);
    if (blobStoreControlAdminRequest.getNumReplicasCaughtUpPerPartition() > 0) {
        PartitionId partitionId = blobStoreControlAdminRequest.getPartitionId();
        if (partitionId != null) {
            error = validateRequest(partitionId, RequestOrResponseType.AdminRequest);
            if (error.equals(ServerErrorCode.No_Error)) {
                if (blobStoreControlAdminRequest.shouldEnable()) {
                    // TODO: start BlobStore properly
                    error = ServerErrorCode.Temporarily_Disabled;
                } else {
                    Collection<PartitionId> partitionIds = Collections.singletonList(partitionId);
                    if (storageManager.disableCompactionForBlobStore(partitionId)) {
                        controlRequestForPartitions(RequestOrResponseType.PutRequest, partitionIds, false);
                        controlRequestForPartitions(RequestOrResponseType.DeleteRequest, partitionIds, false);
                        if (replicationManager.controlReplicationForPartitions(partitionIds, Collections.<String>emptyList(), false)) {
                            if (isRemoteLagLesserOrEqual(partitionIds, 0, blobStoreControlAdminRequest.getNumReplicasCaughtUpPerPartition())) {
                                controlRequestForPartitions(RequestOrResponseType.GetRequest, partitionIds, false);
                                controlRequestForPartitions(RequestOrResponseType.ReplicaMetadataRequest, partitionIds, false);
                                // Shutdown the BlobStore completely
                                if (storageManager.shutdownBlobStore(partitionId)) {
                                    error = ServerErrorCode.No_Error;
                                    logger.info("store shutdown for partition: {}", partitionId);
                                } else {
                                    error = ServerErrorCode.Unknown_Error;
                                    logger.error("Shutting down BlobStore fails on {}", partitionId);
                                }
                            } else {
                                error = ServerErrorCode.Retry_After_Backoff;
                                logger.debug("Catchup not done on {}", partitionIds);
                            }
                        } else {
                            error = ServerErrorCode.Unknown_Error;
                            logger.error("Could not disable replication on {}", partitionIds);
                        }
                    } else {
                        error = ServerErrorCode.Unknown_Error;
                        logger.error("Disable compaction on given BlobStore failed for {}", partitionId);
                    }
                }
            } else {
                logger.debug("Validate request fails for {} with error code {}", partitionId, error);
            }
        } else {
            error = ServerErrorCode.Partition_Unknown;
            logger.debug("The partition Id should not be null.");
        }
    } else {
        error = ServerErrorCode.Bad_Request;
        logger.debug("The number of replicas to catch up should not be less or equal to zero {}", blobStoreControlAdminRequest.getNumReplicasCaughtUpPerPartition());
    }
    return new AdminResponse(adminRequest.getCorrelationId(), adminRequest.getClientId(), error);
}