HealthCheckServiceImpl.java

package org.kairosdb.core.health;

import com.google.inject.Binding;
import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Key;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import static com.google.common.base.Preconditions.checkNotNull;

public class HealthCheckServiceImpl implements HealthCheckService
{
	private final List<HealthStatus> checks = new ArrayList<HealthStatus>();

	@Inject
	public HealthCheckServiceImpl(Injector injector)
	{
		checkNotNull(injector);

		Map<Key<?>, Binding<?>> bindings = injector.getAllBindings();

		for (Key<?> key : bindings.keySet())
		{
			Class<?> bindingClass = key.getTypeLiteral().getRawType();
			if (HealthStatus.class.isAssignableFrom(bindingClass))
			{
				checks.add((HealthStatus) injector.getInstance(bindingClass));
			}
		}
	}

	@Override
	public List<HealthStatus> getChecks()
	{
		return Collections.unmodifiableList(checks);
	}
}