com.tomakehurst.crashlab.saboteur.SaboteurAdminClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crash-lab Show documentation
A Java resilience testing library
The newest version!
package com.tomakehurst.crashlab.saboteur;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.base.Function;
import com.google.common.collect.Lists;
import com.google.common.net.HostAndPort;
import com.ning.http.client.AsyncHttpClient;
import com.ning.http.client.ListenableFuture;
import com.ning.http.client.Response;

import java.io.IOException;
import java.util.List;

import static com.google.common.collect.Iterables.transform;
import static com.tomakehurst.crashlab.utils.Exceptions.throwUnchecked;
import static java.util.concurrent.TimeUnit.SECONDS;

public class SaboteurAdminClient {

    private final AsyncHttpClient client;
    private final List hosts;
    private final ObjectMapper objectMapper = new ObjectMapper();

    public SaboteurAdminClient(AsyncHttpClient client, List hostsAndPorts) {
        this.client = client;
        this.hosts = hostsAndPorts;
    }

    @SuppressWarnings("unchecked")
    public  void addFault(final T fault) {
        executeRequestForAllHosts(new HttpClientAction("Add fault") {
            public ListenableFuture doWithClient(AsyncHttpClient client, String host, int port) throws IOException {
                StringBuilder url = new StringBuilder().append("http://").append(host).append(":").append(port).append("/");
                return client.preparePost(url.toString())
                    .setBody(objectMapper.writeValueAsBytes(fault))
                    .execute();
            }
        });
    }

    public void reset() {
        executeRequestForAllHosts(new HttpClientAction("Reset") {
            public ListenableFuture doWithClient(AsyncHttpClient client, String host, int port) throws IOException {
                StringBuilder url = new StringBuilder().append("http://").append(host).append(":").append(port).append("/");
                return client.prepareDelete(url.toString()).execute();
            }
        });
    }

    private void executeRequestForAllHosts(final HttpClientAction httpClientAction) {
        Iterable> responseFutures = transform(hosts, new Function>() {
            public ListenableFuture apply(HostAndPort host) {
                try {
                    return httpClientAction.doWithClient(client, host.getHostText(), host.getPort());
                } catch (IOException e) {
                    return throwUnchecked(e, ListenableFuture.class);
                }
            }
        });

        for (ListenableFuture responseFuture: responseFutures) {
            try {
                Response response = responseFuture.get(10, SECONDS);
                System.out.println(httpClientAction.name + ": " + response.getStatusText());
            } catch (Exception e) {
                e.printStackTrace();
                throwUnchecked(e);
            }
        }
    }

    private static abstract class HttpClientAction {

        public final String name;

        protected HttpClientAction(String name) {
            this.name = name;
        }

        public abstract ListenableFuture doWithClient(AsyncHttpClient client, String host, int port) throws IOException;
    }
}