/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.jclouds.rackspace.cloudbigdata.v1.predicates;
import static com.google.common.base.Preconditions.checkNotNull;
import static java.util.concurrent.TimeUnit.SECONDS;
import static org.jclouds.util.Predicates2.retry;
import org.jclouds.rackspace.cloudbigdata.v1.domain.Cluster;
import org.jclouds.rackspace.cloudbigdata.v1.domain.Cluster.Status;
import org.jclouds.rackspace.cloudbigdata.v1.features.ClusterApi;
import com.google.common.base.Predicate;
/**
* Tests to see if Cluster has reached status. This class is most useful when paired with a RetryablePredicate as
* in the code below. This class can be used to block execution until the Cluster status has reached a desired state.
* This is useful when your Cluster needs to be 100% ready before you can continue with execution.
*
* <pre>
* {@code
* Cluster Cluster = ClusterApi.create(100);
*
* RetryablePredicate<String> awaitAvailable = RetryablePredicate.create(
* ClusterPredicates.available(ClusterApi), 600, 10, 10, TimeUnit.SECONDS);
*
* if (!awaitAvailable.apply(Cluster.getId())) {
* throw new TimeoutException("Timeout on Cluster: " + Cluster);
* }
* }
* </pre>
*
* You can also use the static convenience methods as follows.
*
* <pre>
* {@code
* Cluster Cluster = ClusterApi.create(100);
*
* if (!ClusterPredicates.awaitAvailable(ClusterApi).apply(Cluster.getId())) {
* throw new TimeoutException("Timeout on Cluster: " + Cluster);
* }
* }
* </pre>
*/
public class ClusterPredicates {
/**
* Wait until an Cluster is Available.
*
* @param clusterApi The ClusterApi in the zone where your Cluster resides.
* @return RetryablePredicate That will check the status every 5 seconds for a maxiumum of 10 minutes.
*/
public static Predicate<Cluster> awaitAvailable(ClusterApi clusterApi) {
StatusUpdatedPredicate statusPredicate = new StatusUpdatedPredicate(clusterApi, Cluster.Status.ACTIVE);
return retry(statusPredicate, 600, 5, 5, SECONDS);
}
/**
* Wait until an Cluster no longer exists.
*
* @param clusterApi The ClusterApi in the zone where your Cluster resides.
* @return RetryablePredicate That will check whether the Cluster exists.
* every 5 seconds for a maximum of 10 minutes.
*/
public static Predicate<Cluster> awaitDeleted(ClusterApi clusterApi) {
DeletedPredicate deletedPredicate = new DeletedPredicate(clusterApi);
return retry(deletedPredicate, 600, 5, 5, SECONDS);
}
/**
* Wait until Cluster is in the status specified.
*
* @param clusterApi The ClusterApi in the zone where your Cluster resides.
* @param status Wait until Cluster in in this status.
* @param maxWaitInSec Maximum time to wait.
* @param periodInSec Interval between retries.
* @return RetryablePredicate That will check whether the Cluster exists.
*/
public static Predicate<Cluster> awaitStatus(
ClusterApi clusterApi, Cluster.Status status, long maxWaitInSec, long periodInSec) {
StatusUpdatedPredicate statusPredicate = new StatusUpdatedPredicate(clusterApi, status);
return retry(statusPredicate, maxWaitInSec, periodInSec, periodInSec, SECONDS);
}
private static class StatusUpdatedPredicate implements Predicate<Cluster> {
private ClusterApi clusterApi;
private Status status;
public StatusUpdatedPredicate(ClusterApi clusterApi, Cluster.Status status) {
this.clusterApi = checkNotNull(clusterApi, "ClusterApi must be defined");
this.status = checkNotNull(status, "status must be defined");
}
/**
* @return boolean Return true when the Cluster reaches status, false otherwise.
*/
@Override
public boolean apply(Cluster cluster) {
checkNotNull(cluster, "Cluster must be defined");
if (status.equals(cluster.getStatus())) {
return true;
}
else {
Cluster ClusterUpdated = clusterApi.get(cluster.getId());
checkNotNull(ClusterUpdated, "Cluster %s not found.", cluster.getId());
return status.equals(ClusterUpdated.getStatus());
}
}
}
private static class DeletedPredicate implements Predicate<Cluster> {
private ClusterApi clusterApi;
public DeletedPredicate(ClusterApi clusterApi) {
this.clusterApi = checkNotNull(clusterApi, "ClusterApi must be defined");
}
/**
* @return boolean Return true when the snapshot is deleted, false otherwise.
*/
@Override
public boolean apply(Cluster cluster) {
checkNotNull(cluster, "Cluster must be defined");
return clusterApi.get(cluster.getId()) == null;
}
}
}