|
12 | 12 |
|
13 | 13 | import java.util.ArrayList;
|
14 | 14 | import java.util.Collection;
|
| 15 | +import java.util.HashMap; |
15 | 16 | import java.util.HashSet;
|
16 | 17 | import java.util.List;
|
17 | 18 | import java.util.Map;
|
@@ -43,6 +44,8 @@ public class SpringProcessCommandHandler {
|
43 | 44 | private static final String COMMAND_LIST_CONNECTED = "sts/livedata/listConnected";
|
44 | 45 | private static final String COMMAND_GET_METRICS = "sts/livedata/get/metrics";
|
45 | 46 | private static final String COMMAND_GET_REFRESH_METRICS = "sts/livedata/refresh/metrics";
|
| 47 | + private static final String COMMAND_GET_LOGGERS = "sts/livedata/getLoggers"; |
| 48 | + private static final String COMMAND_CONFIGURE_LOGLEVEL = "sts/livedata/configure/logLevel"; |
46 | 49 |
|
47 | 50 | private final SpringProcessConnectorService connectorService;
|
48 | 51 | private final SpringProcessConnectorLocal localProcessConnector;
|
@@ -88,6 +91,16 @@ public SpringProcessCommandHandler(SimpleLanguageServer server, SpringProcessCon
|
88 | 91 | return refreshMetrics(params);
|
89 | 92 | });
|
90 | 93 | log.info("Registered command handler: {}",COMMAND_GET_METRICS);
|
| 94 | + |
| 95 | + server.onCommand(COMMAND_GET_LOGGERS, (params) -> { |
| 96 | + return getLoggers(params); |
| 97 | + }); |
| 98 | + log.info("Registered command handler: {}",COMMAND_GET_LOGGERS); |
| 99 | + |
| 100 | + server.onCommand(COMMAND_CONFIGURE_LOGLEVEL, (params) -> { |
| 101 | + return configureLogLevel(params); |
| 102 | + }); |
| 103 | + log.info("Registered command handler: {}",COMMAND_CONFIGURE_LOGLEVEL); |
91 | 104 |
|
92 | 105 | server.onCommand(COMMAND_LIST_CONNECTED, (params) -> {
|
93 | 106 | List<LiveProcessSummary> result = new ArrayList<>();
|
@@ -320,4 +333,34 @@ private CompletableFuture<Object> handleLiveMetricsProcessRequest(ExecuteCommand
|
320 | 333 | return CompletableFuture.completedFuture(null);
|
321 | 334 | }
|
322 | 335 |
|
| 336 | + |
| 337 | + private CompletableFuture<Object> getLoggers(ExecuteCommandParams params) { |
| 338 | + SpringProcessLoggersData loggersData = null; |
| 339 | + SpringProcessParams springProcessParams = new SpringProcessParams(); |
| 340 | + springProcessParams.setProcessKey(getProcessKey(params)); |
| 341 | + springProcessParams.setEndpoint(getArgumentByKey(params, "endpoint")); |
| 342 | + if (springProcessParams.getProcessKey() != null) { |
| 343 | + loggersData = connectorService.getLoggers(springProcessParams); |
| 344 | + return CompletableFuture.completedFuture(loggersData); |
| 345 | + } |
| 346 | + |
| 347 | + return CompletableFuture.completedFuture(loggersData); |
| 348 | + } |
| 349 | + |
| 350 | + private CompletableFuture<Object> configureLogLevel(ExecuteCommandParams params) { |
| 351 | + Map<String, String> args = new HashMap<>(); |
| 352 | + args.put("packageName", getArgumentByKey(params, "packageName")); |
| 353 | + args.put("configuredLevel", getArgumentByKey(params, "configuredLevel")); |
| 354 | + args.put("effectiveLevel", getArgumentByKey(params, "effectiveLevel")); |
| 355 | + SpringProcessParams springProcessParams = new SpringProcessParams(); |
| 356 | + springProcessParams.setProcessKey(getProcessKey(params)); |
| 357 | + springProcessParams.setArgs(args); |
| 358 | + |
| 359 | + if (springProcessParams.getProcessKey() != null) { |
| 360 | + connectorService.configureLogLevel(springProcessParams); |
| 361 | + } |
| 362 | + |
| 363 | + return CompletableFuture.completedFuture(null); |
| 364 | + } |
| 365 | + |
323 | 366 | }
|
0 commit comments