Spring/spring-tools
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add extension APIs to get live data

Browse Source 
Signed-off-by: Yan Zhang <yanzh@microsoft.com>
This commit is contained in:
Yan Zhang
2022-04-02 18:38:13 +08:00
parent e66303e28e
commit f332ae4e69
6 changed files with 139 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/java/livehover/v2/SpringProcessCommandHandler.java
View File

@@ -38,6 +38,7 @@ public class SpringProcessCommandHandler {
private static final String COMMAND_CONNECT = "sts/livedata/connect";
private static final String COMMAND_REFRESH = "sts/livedata/refresh";
private static final String COMMAND_DISCONNECT = "sts/livedata/disconnect";
private static final String COMMAND_GET = "sts/livedata/get";
private final SpringProcessConnectorService connectorService;
private final SpringProcessConnectorLocal localProcessConnector;
@@ -68,6 +69,11 @@ public class SpringProcessCommandHandler {
return disconnect(params);
});
log.info("Registered command handler: {}",COMMAND_DISCONNECT);
server.onCommand(COMMAND_GET, (params) -> {
return get(params);
});
log.info("Registered command handler: {}",COMMAND_GET);
}
private CompletableFuture<Object> connect(ExecuteCommandParams params) {
@@ -174,16 +180,20 @@ public class SpringProcessCommandHandler {
}
private String getProcessKey(ExecuteCommandParams params) {
return getArgumentByKey(params, "processKey");
}
private String getArgumentByKey(ExecuteCommandParams params, String name) {
List<Object> arguments = params.getArguments();
for (Object arg : arguments) {
if (arg instanceof Map<?, ?>) {
Object value = ((Map<?, ?>) arg).get("processKey");
Object value = ((Map<?, ?>) arg).get(name);
if (value != null) {
return value.toString();
}
}
else if (arg instanceof JsonObject) {
JsonElement element = ((JsonObject) arg).get("processKey");
JsonElement element = ((JsonObject) arg).get(name);
if (element != null) {
return element.getAsString();
}
@@ -192,5 +202,42 @@ public class SpringProcessCommandHandler {
return null;
}
private CompletableFuture<Object> get(ExecuteCommandParams params) {
String processKey = getProcessKey(params);
String dataKey = getArgumentByKey(params, "dataKey");
if (processKey != null) {
SpringProcessLiveData data = connectorService.getLiveData(processKey);
switch(dataKey) {
case "properties": {
return CompletableFuture.completedFuture(data.getLiveProperties());
}
case "beans": {
String beanName = getArgumentByKey(params, "beanName");
if (beanName != null) {
return CompletableFuture.completedFuture(data.getBeans().getBeansOfName(beanName));
}
String dependingOn = getArgumentByKey(params, "dependingOn");
if (dependingOn != null) {
return CompletableFuture.completedFuture(data.getBeans().getBeansDependingOn(dependingOn));
}
return CompletableFuture.completedFuture(data.getBeans().getBeanNames());
}
case "mappings": {
return CompletableFuture.completedFuture(data.getRequestMappings());
}
case "contextPath": {
return CompletableFuture.completedFuture(data.getContextPath());
}
case "port": {
return CompletableFuture.completedFuture(data.getPort());
}
default: {}
}
}
return CompletableFuture.completedFuture(null);
}
}

4
headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/java/livehover/v2/SpringProcessConnectorService.java
View File

@@ -110,6 +110,10 @@ public class SpringProcessConnectorService {
}
}
public SpringProcessLiveData getLiveData(String processKey) {
return this.liveDataProvider.getCurrent(processKey);
}
public void disconnectProcess(String processKey) {
log.info("disconnect from process: " + processKey);