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/endpoint/ServiceEndpoint.java | 135 +++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 135 insertions(+), 0 deletions(-)
diff --git a/Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/endpoint/ServiceEndpoint.java b/Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/endpoint/ServiceEndpoint.java
new file mode 100644
index 0000000..690249d
--- /dev/null
+++ b/Source/BladeX-Tool/blade-starter-prometheus/src/main/java/org/springblade/core/prometheus/endpoint/ServiceEndpoint.java
@@ -0,0 +1,135 @@
+/*
+ * 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.endpoint;
+
+import lombok.RequiredArgsConstructor;
+import org.springblade.core.auto.annotation.AutoIgnore;
+import org.springblade.core.prometheus.data.Service;
+import org.springblade.core.prometheus.data.ServiceHealth;
+import org.springblade.core.prometheus.service.RegistrationService;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.util.Assert;
+import org.springframework.util.MultiValueMap;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import reactor.core.publisher.Mono;
+
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+import java.util.concurrent.ThreadLocalRandom;
+import java.util.concurrent.TimeUnit;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import static java.util.stream.Collectors.toList;
+
+/**
+ * consul catalog api
+ *
+ * @author L.cm
+ */
+@AutoIgnore
+@RestController
+@RequiredArgsConstructor
+public class ServiceEndpoint {
+ private static final String CONSUL_IDX_HEADER = "X-Consul-Index";
+ private static final String QUERY_PARAM_WAIT = "wait";
+ private static final String QUERY_PARAM_INDEX = "index";
+ private static final Pattern WAIT_PATTERN = Pattern.compile("(\\d*)(m|s|ms|h)");
+ private final RegistrationService registrationService;
+
+ @GetMapping(value = "/v1/catalog/services", produces = MediaType.APPLICATION_JSON_VALUE)
+ public Mono<ResponseEntity<Map<String, String[]>>> getServiceNames(
+ @RequestParam(name = QUERY_PARAM_WAIT, required = false) String wait,
+ @RequestParam(name = QUERY_PARAM_INDEX, required = false) Long index) {
+ return registrationService.getServiceNames(getWaitMillis(wait), index)
+ .map(item -> createResponseEntity(item.getItem(), item.getChangeIndex()));
+ }
+
+ @GetMapping(value = "/v1/catalog/service/{appName}", produces = MediaType.APPLICATION_JSON_VALUE)
+ public Mono<ResponseEntity<List<Service>>> getService(@PathVariable("appName") String appName,
+ @RequestParam(value = QUERY_PARAM_WAIT, required = false) String wait,
+ @RequestParam(value = QUERY_PARAM_INDEX, required = false) Long index) {
+ Objects.requireNonNull(appName, "service name can not be null");
+ return registrationService.getService(appName, getWaitMillis(wait), index)
+ .map(item -> createResponseEntity(item.getItem(), item.getChangeIndex()));
+ }
+
+ @GetMapping(value = "/v1/health/service/{appName}", produces = MediaType.APPLICATION_JSON_VALUE)
+ public Mono<ResponseEntity<List<ServiceHealth>>> getServiceHealth(@PathVariable("appName") String appName,
+ @RequestParam(value = QUERY_PARAM_WAIT, required = false) String wait,
+ @RequestParam(value = QUERY_PARAM_INDEX, required = false) Long index) {
+ Assert.isTrue(appName != null, "service name can not be null");
+ return registrationService.getService(appName, getWaitMillis(wait), index)
+ .map(item -> {
+ List<ServiceHealth> services = item.getItem().stream()
+ .map(registrationService::getServiceHealth).collect(toList());
+ return createResponseEntity(services, item.getChangeIndex());
+ });
+ }
+
+ private static MultiValueMap<String, String> createHeaders(long index) {
+ HttpHeaders headers = new HttpHeaders();
+ headers.add(CONSUL_IDX_HEADER, String.valueOf(index));
+ return headers;
+ }
+
+ private static <T> ResponseEntity<T> createResponseEntity(T body, long index) {
+ return new ResponseEntity<>(body, createHeaders(index), HttpStatus.OK);
+ }
+
+ /**
+ * Details to the wait behaviour can be found
+ * https://www.consul.io/api/index.html#blocking-queries
+ */
+ private static long getWaitMillis(String wait) {
+ // default from consul docu
+ long millis = TimeUnit.MINUTES.toMillis(5);
+ if (wait != null) {
+ Matcher matcher = WAIT_PATTERN.matcher(wait);
+ if (matcher.matches()) {
+ long value = Long.parseLong(matcher.group(1));
+ TimeUnit timeUnit = parseTimeUnit(matcher.group(2));
+ millis = timeUnit.toMillis(value);
+ } else {
+ throw new IllegalArgumentException("Invalid wait pattern");
+ }
+ }
+ return millis + ThreadLocalRandom.current().nextInt(((int) millis / 16) + 1);
+ }
+
+ private static TimeUnit parseTimeUnit(String unit) {
+ switch (unit) {
+ case "h":
+ return TimeUnit.HOURS;
+ case "m":
+ return TimeUnit.MINUTES;
+ case "s":
+ return TimeUnit.SECONDS;
+ case "ms":
+ return TimeUnit.MILLISECONDS;
+ default:
+ throw new IllegalArgumentException("No valid time unit");
+ }
+ }
+}
--
Gitblit v1.9.3