com.facebook.airlift.discovery.client.ServiceSelectorManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of discovery Show documentation

There is a newer version: 0.215

package com.facebook.airlift.discovery.client;

import com.google.common.annotations.Beta;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;

import javax.inject.Inject;

import java.util.List;
import java.util.Set;
import java.util.concurrent.ExecutionException;

import static java.util.Objects.requireNonNull;

@Beta
public class ServiceSelectorManager
{
    private final Set serviceSelectors;

    @Inject
    public ServiceSelectorManager(Set serviceSelectors)
    {
        requireNonNull(serviceSelectors, "serviceSelectors is null");
        this.serviceSelectors = ImmutableSet.copyOf(serviceSelectors);
    }

    public Set getServiceSelectors()
    {
        return serviceSelectors;
    }

    public void forceRefresh()
    {
        for (ListenableFuture future : initiateRefresh()) {
            Futures.getUnchecked(future);
        }
    }

    public void attemptRefresh()
    {
        try {
            Futures.successfulAsList(initiateRefresh()).get();
        }
        catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
        catch (ExecutionException ignored) {
        }
    }

    private List> initiateRefresh()
    {
        ImmutableList.Builder> futures = ImmutableList.builder();
        for (ServiceSelector selector : serviceSelectors) {
            futures.add(selector.refresh());
        }
        return futures.build();
    }
}