This repository has been archived by the owner on Jan 31, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 4
Cancel initialize job when removing workspace #60
Closed
Closed
Changes from all commits
Commits
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
290 changes: 290 additions & 0 deletions
290
org.elastic.jdt.ls.core/src/org/elastic/jdt/ls/core/internal/ExtendedInitHandler.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,290 @@ | ||
| package org.elastic.jdt.ls.core.internal; | ||
|
|
||
| import static org.elastic.jdt.ls.core.internal.ElasticJavaLanguageServerPlugin.logInfo; | ||
|
|
||
| import java.util.ArrayList; | ||
| import java.util.Arrays; | ||
| import java.util.Collection; | ||
| import java.util.Collections; | ||
| import java.util.Map; | ||
| import java.util.Set; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| import org.eclipse.core.resources.ResourcesPlugin; | ||
| import org.eclipse.core.resources.WorkspaceJob; | ||
| import org.eclipse.core.runtime.CoreException; | ||
| import org.eclipse.core.runtime.IPath; | ||
| import org.eclipse.core.runtime.IProgressMonitor; | ||
| import org.eclipse.core.runtime.IStatus; | ||
| import org.eclipse.core.runtime.OperationCanceledException; | ||
| import org.eclipse.core.runtime.Status; | ||
| import org.eclipse.core.runtime.SubMonitor; | ||
| import org.eclipse.core.runtime.jobs.Job; | ||
| import org.eclipse.jdt.ls.core.internal.JSONUtility; | ||
| import org.eclipse.jdt.ls.core.internal.JavaClientConnection; | ||
| import org.eclipse.jdt.ls.core.internal.JavaLanguageServerPlugin; | ||
| import org.eclipse.jdt.ls.core.internal.ResourceUtils; | ||
| import org.eclipse.jdt.ls.core.internal.ServiceStatus; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.BundleUtils; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.CompletionHandler; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.InitHandler; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.RenameHandler; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.SignatureHelpHandler; | ||
| import org.eclipse.jdt.ls.core.internal.handlers.WorkspaceExecuteCommandHandler; | ||
| import org.eclipse.jdt.ls.core.internal.highlighting.SemanticHighlightingService; | ||
| import org.eclipse.jdt.ls.core.internal.managers.ProjectsManager; | ||
| import org.eclipse.jdt.ls.core.internal.preferences.PreferenceManager; | ||
| import org.eclipse.jdt.ls.core.internal.preferences.Preferences; | ||
| import org.eclipse.lsp4j.ClientCapabilities; | ||
| import org.eclipse.lsp4j.CodeLensOptions; | ||
| import org.eclipse.lsp4j.DocumentOnTypeFormattingOptions; | ||
| import org.eclipse.lsp4j.ExecuteCommandOptions; | ||
| import org.eclipse.lsp4j.InitializeParams; | ||
| import org.eclipse.lsp4j.InitializeResult; | ||
| import org.eclipse.lsp4j.SaveOptions; | ||
| import org.eclipse.lsp4j.SemanticHighlightingServerCapabilities; | ||
| import org.eclipse.lsp4j.ServerCapabilities; | ||
| import org.eclipse.lsp4j.TextDocumentSyncKind; | ||
| import org.eclipse.lsp4j.TextDocumentSyncOptions; | ||
| import org.eclipse.lsp4j.WorkspaceFoldersOptions; | ||
| import org.eclipse.lsp4j.WorkspaceServerCapabilities; | ||
|
|
||
| /** | ||
| * Handler for the VS Code extension initialization | ||
| */ | ||
| final public class ExtendedInitHandler { | ||
|
|
||
| public static final String JAVA_LS_INITIALIZATION_JOBS = "java-ls-initialization-jobs"; | ||
| private static final String BUNDLES_KEY = "bundles"; | ||
| public static final String SETTINGS_KEY = "settings"; | ||
|
|
||
| private ProjectsManager projectsManager; | ||
| private PreferenceManager preferenceManager; | ||
|
|
||
| public ExtendedInitHandler(ProjectsManager manager, PreferenceManager preferenceManager) { | ||
| this.projectsManager = manager; | ||
| this.preferenceManager = preferenceManager; | ||
| } | ||
|
|
||
| @SuppressWarnings("unchecked") | ||
| public InitializeResult initialize(InitializeParams param) { | ||
| logInfo("Initializing Java Language Server " + JavaLanguageServerPlugin.getVersion()); | ||
| Map<?, ?> initializationOptions = this.getInitializationOptions(param); | ||
| Map<String, Object> extendedClientCapabilities = getInitializationOption(initializationOptions, "extendedClientCapabilities", Map.class); | ||
| if (param.getCapabilities() == null) { | ||
| preferenceManager.updateClientPrefences(new ClientCapabilities(), extendedClientCapabilities); | ||
| } else { | ||
| preferenceManager.updateClientPrefences(param.getCapabilities(), extendedClientCapabilities); | ||
| } | ||
|
|
||
|
|
||
| Collection<IPath> rootPaths = new ArrayList<>(); | ||
| Collection<String> workspaceFolders = getInitializationOption(initializationOptions, "workspaceFolders", Collection.class); | ||
| if (workspaceFolders != null && !workspaceFolders.isEmpty()) { | ||
| for (String uri : workspaceFolders) { | ||
| IPath filePath = ResourceUtils.canonicalFilePathFromURI(uri); | ||
| if (filePath != null) { | ||
| rootPaths.add(filePath); | ||
| } | ||
| } | ||
| } else { | ||
| String rootPath = param.getRootUri(); | ||
| if (rootPath == null) { | ||
| rootPath = param.getRootPath(); | ||
| if (rootPath != null) { | ||
| logInfo("In LSP 3.0, InitializeParams.rootPath is deprecated in favour of InitializeParams.rootUri!"); | ||
| } | ||
| } | ||
| if (rootPath != null) { | ||
| IPath filePath = ResourceUtils.canonicalFilePathFromURI(rootPath); | ||
| if (filePath != null) { | ||
| rootPaths.add(filePath); | ||
| } | ||
| } | ||
| } | ||
| if (rootPaths.isEmpty()) { | ||
| IPath workspaceLocation = ResourcesPlugin.getWorkspace().getRoot().getLocation(); | ||
| logInfo("No workspace folders or root uri was defined. Falling back on " + workspaceLocation); | ||
| rootPaths.add(workspaceLocation); | ||
| } | ||
| if (initializationOptions.get(SETTINGS_KEY) instanceof Map) { | ||
| Object settings = initializationOptions.get(SETTINGS_KEY); | ||
| @SuppressWarnings("unchecked") | ||
| Preferences prefs = Preferences.createFrom((Map<String, Object>) settings); | ||
| preferenceManager.update(prefs); | ||
| } | ||
| preferenceManager.getPreferences().setRootPaths(rootPaths); | ||
|
|
||
| Collection<IPath> triggerPaths = new ArrayList<>(); | ||
| Collection<String> triggerFiles = getInitializationOption(initializationOptions, "triggerFiles", Collection.class); | ||
| if (triggerFiles != null) { | ||
| for (String uri : triggerFiles) { | ||
| IPath filePath = ResourceUtils.canonicalFilePathFromURI(uri); | ||
| if (filePath != null) { | ||
| triggerPaths.add(filePath); | ||
| } | ||
| } | ||
| } | ||
| preferenceManager.getPreferences().setTriggerFiles(triggerPaths); | ||
|
|
||
| triggerInitialization(rootPaths); | ||
| Integer processId = param.getProcessId(); | ||
| if (processId != null) { | ||
| JavaLanguageServerPlugin.getLanguageServer().setParentProcessId(processId.longValue()); | ||
| } | ||
| try { | ||
| Collection<String> bundleList = getInitializationOption(initializationOptions, BUNDLES_KEY, Collection.class); | ||
| BundleUtils.loadBundles(bundleList); | ||
| } catch (CoreException e) { | ||
| // The additional plug-ins should not affect the main language server loading. | ||
| JavaLanguageServerPlugin.logException("Failed to load extension bundles ", e); | ||
| } | ||
| InitializeResult result = new InitializeResult(); | ||
| ServerCapabilities capabilities = new ServerCapabilities(); | ||
| if (!preferenceManager.getClientPreferences().isCompletionDynamicRegistered()) { | ||
| capabilities.setCompletionProvider(CompletionHandler.DEFAULT_COMPLETION_OPTIONS); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isFormattingDynamicRegistrationSupported()) { | ||
| capabilities.setDocumentFormattingProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isRangeFormattingDynamicRegistrationSupported()) { | ||
| capabilities.setDocumentRangeFormattingProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isOnTypeFormattingDynamicRegistrationSupported()) { | ||
| capabilities.setDocumentOnTypeFormattingProvider(new DocumentOnTypeFormattingOptions(";", Arrays.asList("\n", "}"))); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isCodeLensDynamicRegistrationSupported()) { | ||
| capabilities.setCodeLensProvider(new CodeLensOptions(true)); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isSignatureHelpDynamicRegistrationSupported()) { | ||
| capabilities.setSignatureHelpProvider(SignatureHelpHandler.createOptions()); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isRenameDynamicRegistrationSupported()) { | ||
| capabilities.setRenameProvider(RenameHandler.createOptions()); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isCodeActionDynamicRegistered()) { | ||
| capabilities.setCodeActionProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isExecuteCommandDynamicRegistrationSupported()) { | ||
| Set<String> commands = WorkspaceExecuteCommandHandler.getCommands(); | ||
| capabilities.setExecuteCommandProvider(new ExecuteCommandOptions(new ArrayList<>(commands))); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isWorkspaceSymbolDynamicRegistered()) { | ||
| capabilities.setWorkspaceSymbolProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isDocumentSymbolDynamicRegistered()) { | ||
| capabilities.setDocumentSymbolProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isDefinitionDynamicRegistered()) { | ||
| capabilities.setDefinitionProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isTypeDefinitionDynamicRegistered()) { | ||
| capabilities.setTypeDefinitionProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isHoverDynamicRegistered()) { | ||
| capabilities.setHoverProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isReferencesDynamicRegistered()) { | ||
| capabilities.setReferencesProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isDocumentHighlightDynamicRegistered()) { | ||
| capabilities.setDocumentHighlightProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isFoldgingRangeDynamicRegistered()) { | ||
| capabilities.setFoldingRangeProvider(Boolean.TRUE); | ||
| } | ||
| if (!preferenceManager.getClientPreferences().isImplementationDynamicRegistered()) { | ||
| capabilities.setImplementationProvider(Boolean.TRUE); | ||
| } | ||
| TextDocumentSyncOptions textDocumentSyncOptions = new TextDocumentSyncOptions(); | ||
| textDocumentSyncOptions.setOpenClose(Boolean.TRUE); | ||
| textDocumentSyncOptions.setSave(new SaveOptions(Boolean.TRUE)); | ||
| textDocumentSyncOptions.setChange(TextDocumentSyncKind.Incremental); | ||
| if (preferenceManager.getClientPreferences().isWillSaveRegistered()) { | ||
| textDocumentSyncOptions.setWillSave(Boolean.TRUE); | ||
| } | ||
|
|
||
| if (preferenceManager.getClientPreferences().isWillSaveWaitUntilRegistered()) { | ||
| textDocumentSyncOptions.setWillSaveWaitUntil(Boolean.TRUE); | ||
| } | ||
| capabilities.setTextDocumentSync(textDocumentSyncOptions); | ||
|
|
||
| if (preferenceManager.getClientPreferences().isSemanticHighlightingSupported()) { | ||
| SemanticHighlightingServerCapabilities semanticHighlightingCapabilities = new SemanticHighlightingServerCapabilities(); | ||
| semanticHighlightingCapabilities.setScopes(SemanticHighlightingService.getAllScopes()); | ||
| capabilities.setSemanticHighlighting(semanticHighlightingCapabilities); | ||
| } | ||
|
|
||
| WorkspaceServerCapabilities wsCapabilities = new WorkspaceServerCapabilities(); | ||
| WorkspaceFoldersOptions wsFoldersOptions = new WorkspaceFoldersOptions(); | ||
| wsFoldersOptions.setSupported(Boolean.TRUE); | ||
| wsFoldersOptions.setChangeNotifications(Boolean.TRUE); | ||
| wsCapabilities.setWorkspaceFolders(wsFoldersOptions); | ||
| capabilities.setWorkspace(wsCapabilities); | ||
|
|
||
| result.setCapabilities(capabilities); | ||
| return result; | ||
| } | ||
|
|
||
| public static void cancelInitJobFromURI(String uri) { | ||
| IPath rootName = ResourceUtils.canonicalFilePathFromURI(uri); | ||
| Job[] jobs = Job.getJobManager().find(rootName); | ||
| for (Job job: jobs) { | ||
| JavaLanguageServerPlugin.logInfo("Cancel init job: " + job.toString()); | ||
| job.cancel(); | ||
| } | ||
| } | ||
|
|
||
| private void triggerInitialization(Collection<IPath> roots) { | ||
| Job job = new WorkspaceJob("Initialize Workspace") { | ||
| @Override | ||
| public IStatus runInWorkspace(IProgressMonitor monitor) { | ||
| long start = System.currentTimeMillis(); | ||
| SubMonitor subMonitor = SubMonitor.convert(monitor, 100); | ||
| try { | ||
| JavaLanguageServerPlugin.logInfo("Start initialization"); | ||
| projectsManager.setAutoBuilding(false); | ||
| projectsManager.initializeProjects(roots, subMonitor); | ||
| projectsManager.setAutoBuilding(preferenceManager.getPreferences().isAutobuildEnabled()); | ||
| JavaLanguageServerPlugin.logInfo("Workspace initialized in " + (System.currentTimeMillis() - start) + "ms"); | ||
| } catch (OperationCanceledException e) { | ||
| } catch (Exception e) { | ||
| JavaLanguageServerPlugin.logException("Initialization failed ", e); | ||
| } | ||
| return Status.OK_STATUS; | ||
| } | ||
|
|
||
| /* (non-Javadoc) | ||
| * @see org.eclipse.core.runtime.jobs.Job#belongsTo(java.lang.Object) | ||
| */ | ||
| @Override | ||
| public boolean belongsTo(Object family) { | ||
| boolean belongsToAnyRoot = false; | ||
| for (IPath root: roots) { | ||
| belongsToAnyRoot |= root.equals(family); | ||
| } | ||
| return JAVA_LS_INITIALIZATION_JOBS.equals(family) || belongsToAnyRoot; | ||
| } | ||
|
|
||
| }; | ||
| job.setPriority(Job.BUILD); | ||
| job.setRule(ResourcesPlugin.getWorkspace().getRoot()); | ||
| job.schedule(); | ||
| } | ||
|
|
||
| private Map<?, ?> getInitializationOptions(InitializeParams params) { | ||
| Map<?, ?> initOptions = JSONUtility.toModel(params.getInitializationOptions(), Map.class); | ||
| return initOptions == null ? Collections.emptyMap() : initOptions; | ||
| } | ||
|
|
||
| private <T> T getInitializationOption(Map<?, ?> initializationOptions, String key, Class<T> clazz) { | ||
| if (initializationOptions != null) { | ||
| Object bundleObject = initializationOptions.get(key); | ||
| if (clazz.isInstance(bundleObject)) { | ||
| return clazz.cast(bundleObject); | ||
| } | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| } | ||
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please leave a new line at the end of the file. It's optional.