From 80b6cbfc9c861469146318d0b3dd5f8b8b525b8a Mon Sep 17 00:00:00 2001
From: xiejun <xiejun@vci-tech.com>
Date: 星期五, 01 十一月 2024 15:11:19 +0800
Subject: [PATCH] Revert "集成获取mdm分发通用数据格式接口集成"

---
 Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/service/RegistrationService.java |  112 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 112 insertions(+), 0 deletions(-)

diff --git a/Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/service/RegistrationService.java b/Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/service/RegistrationService.java
new file mode 100644
index 0000000..1f0d0ee
--- /dev/null
+++ b/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()));
+	}
+}

--
Gitblit v1.9.3