org.macrocloud.kernel.prometheus.service.RegistrationService Maven / Gradle / Ivy

Go to download
package org.macrocloud.kernel.prometheus.service;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

import org.macrocloud.kernel.prometheus.data.ChangeItem;
import org.macrocloud.kernel.prometheus.data.Service;
import org.macrocloud.kernel.prometheus.data.ServiceHealth;
import org.springframework.cloud.client.ServiceInstance;
import org.springframework.cloud.client.discovery.DiscoveryClient;
import reactor.core.publisher.Mono;

import java.util.*;
import java.util.function.Supplier;

/**
 * Returns Services and List of Service with its last changed
 *
 * @author macro
 */
@Slf4j
@RequiredArgsConstructor
public class RegistrationService {
	private static final String[] NO_SERVICE_TAGS = new String[0];
	private final DiscoveryClient discoveryClient;

	public Mono>> getServiceNames(long waitMillis, Long index) {
		return returnDeferred(waitMillis, index, () -> {
			List services = discoveryClient.getServices();
			Set set = new HashSet<>(services);
			Map result = new HashMap<>();
			for (String item : set) {
				result.put(item, NO_SERVICE_TAGS);
			}
			return result;
		});
	}

	public Mono>> getService(String appName, long waitMillis, Long index) {
		return returnDeferred(waitMillis, index, () -> {
			List instances = discoveryClient.getInstances(appName);
			List list = new ArrayList<>();
			if (instances == null || instances.isEmpty()) {
				return Collections.emptyList();
			}
			Set instSet = new HashSet<>(instances);
			for (ServiceInstance instance : instSet) {
				Service service = Service.builder().address(instance.getHost()).node(instance.getServiceId())
						.serviceAddress(instance.getHost()).servicePort(instance.getPort())
						.serviceName(instance.getServiceId()).serviceId(instance.getHost() + ":" + instance.getPort())
						.nodeMeta(Collections.emptyMap()).serviceMeta(instance.getMetadata())
						.serviceTags(Collections.emptyList()).build();
				list.add(service);
			}
			return list;
		});
	}

	public ServiceHealth getServiceHealth(Service instanceInfo) {
		String address = instanceInfo.getAddress();
		ServiceHealth.Node node = ServiceHealth.Node.builder().name(instanceInfo.getServiceName()).address(address)
				.meta(Collections.emptyMap()).build();
		ServiceHealth.Service service = ServiceHealth.Service.builder().id(instanceInfo.getServiceId())
				.name(instanceInfo.getServiceName()).tags(Collections.emptyList()).address(address)
				.meta(instanceInfo.getServiceMeta()).port(instanceInfo.getServicePort()).build();
		ServiceHealth.Check check = ServiceHealth.Check.builder().node(instanceInfo.getServiceName())
				.checkId("service:" + instanceInfo.getServiceId())
				.name("Service '" + instanceInfo.getServiceId() + "' check")
				// nacos 实时性很高，可认定为健康
				.status("UP").build();
		return ServiceHealth.builder().node(node).service(service).checks(Collections.singletonList(check)).build();
	}

	private static  Mono> returnDeferred(long waitMillis, Long index, Supplier fn) {
		return Mono.just(new ChangeItem<>(fn.get(), System.currentTimeMillis()));
	}
}