From 7c9b1000cacc760a022fba54404b0f10b9a91037 Mon Sep 17 00:00:00 2001
From: v-kkhuang <62878639+v-kkhuang@users.noreply.github.com>
Date: Tue, 29 Aug 2023 21:48:11 +0800
Subject: [PATCH] eureka instance list (#257)

Co-authored-by: huangKai-2323 <62878639+huangKai-2323@users.noreply.github.com>
---
 .../label/restful/InstanceRestful.java        | 40 ++++++++++++++-----
 1 file changed, 30 insertions(+), 10 deletions(-)

diff --git a/linkis-public-enhancements/linkis-instance-label/linkis-instance-label-server/src/main/java/org/apache/linkis/instance/label/restful/InstanceRestful.java b/linkis-public-enhancements/linkis-instance-label/linkis-instance-label-server/src/main/java/org/apache/linkis/instance/label/restful/InstanceRestful.java
index 5b333cd6f6..68f81548ee 100644
--- a/linkis-public-enhancements/linkis-instance-label/linkis-instance-label-server/src/main/java/org/apache/linkis/instance/label/restful/InstanceRestful.java
+++ b/linkis-public-enhancements/linkis-instance-label/linkis-instance-label-server/src/main/java/org/apache/linkis/instance/label/restful/InstanceRestful.java
@@ -37,19 +37,14 @@
 import org.apache.commons.logging.LogFactory;
 
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestMethod;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.cloud.client.discovery.DiscoveryClient;
+import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletRequest;
 
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.HashSet;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 import com.fasterxml.jackson.databind.JsonNode;
 import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
@@ -74,6 +69,9 @@ public class InstanceRestful {
 
   @Autowired private DefaultInsLabelService insLabelService;
 
+  @Autowired
+  private DiscoveryClient discoveryClient;
+
   @ApiOperation(
       value = "listAllInstanceWithLabel",
       notes = "list all instance with label",
@@ -172,4 +170,26 @@ public Message getServiceRegistryURL(HttpServletRequest request) throws Exceptio
     ModuleUserUtils.getOperationUser(request, "getServiceRegistryURL");
     return Message.ok().data("url", serviceRegistryURL);
   }
+
+  @ApiOperation(value = "getServiceInstances", response = Message.class)
+  @ApiImplicitParams({
+          @ApiImplicitParam(name = "serviceName", required = false, dataType = "String"),
+          @ApiImplicitParam(name = "ip", required = false, dataType = "ip")
+  })
+  @RequestMapping(path = "/serviceInstances", method = RequestMethod.GET)
+  public Message getServiceInstance(HttpServletRequest request,
+                                    @RequestParam(value = "serviceName", required = false) String serviceName,
+                                    @RequestParam(value = "ip", required = false) String ip){
+    Stream<String> serviceStream = discoveryClient.getServices().stream();
+    serviceStream = serviceStream.filter(s -> s.toUpperCase().contains("LINKIS"));
+    if (StringUtils.isNotBlank(serviceName)) {
+      serviceStream = serviceStream.filter(s -> s.toUpperCase().contains(serviceName.toUpperCase()));
+    }
+    List<org.springframework.cloud.client.ServiceInstance> instanceList =
+            serviceStream.flatMap(serviceId -> discoveryClient.getInstances(serviceId).stream()).collect(Collectors.toList());
+    if (StringUtils.isNotBlank(ip)) {
+      instanceList = instanceList.stream().filter(s -> s.getHost().equals(ip)).collect(Collectors.toList());
+    }
+    return Message.ok().data("list", instanceList);
+  }
 }