Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/service/RegistrationService.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,112 @@
/*
 *      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()));
   }
}