/*
|
* Copyright (c) 2018-2028, DreamLu All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the dreamlu.net developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: DreamLu 卢春梦 (596392912@qq.com)
|
*/
|
package org.springblade.core.prometheus.service;
|
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springblade.core.prometheus.data.ChangeItem;
|
import org.springblade.core.prometheus.data.Service;
|
import org.springblade.core.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 L.cm
|
*/
|
@Slf4j
|
@RequiredArgsConstructor
|
public class RegistrationService {
|
private static final String[] NO_SERVICE_TAGS = new String[0];
|
private final DiscoveryClient discoveryClient;
|
|
public Mono<ChangeItem<Map<String, String[]>>> getServiceNames(long waitMillis, Long index) {
|
return returnDeferred(waitMillis, index, () -> {
|
List<String> services = discoveryClient.getServices();
|
Set<String> set = new HashSet<>(services);
|
Map<String, String[]> result = new HashMap<>();
|
for (String item : set) {
|
result.put(item, NO_SERVICE_TAGS);
|
}
|
return result;
|
});
|
}
|
|
public Mono<ChangeItem<List<Service>>> getService(String appName, long waitMillis, Long index) {
|
return returnDeferred(waitMillis, index, () -> {
|
List<ServiceInstance> instances = discoveryClient.getInstances(appName);
|
List<Service> list = new ArrayList<>();
|
if (instances == null || instances.isEmpty()) {
|
return Collections.emptyList();
|
}
|
Set<ServiceInstance> 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 <T> Mono<ChangeItem<T>> returnDeferred(long waitMillis, Long index, Supplier<T> fn) {
|
return Mono.just(new ChangeItem<>(fn.get(), System.currentTimeMillis()));
|
}
|
}
|
