MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-04-19 04:51:24 +07:00
Code Issues Packages Projects Releases Wiki Activity

PY-56541 Introduce TypeEvalContext.getProcessingContext()

Browse Source 
GitOrigin-RevId: 869b5905d05fb3abfb81233e03ab001b5301a3b6
This commit is contained in:
Mikhail Golubev
2023-02-24 10:39:07 +02:00
committed by intellij-monorepo-bot
parent f6d7b71285
commit 0f85b19994
2 changed files with 26 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
python/python-psi-api/src/com/jetbrains/python/psi/types/TypeEvalContext.java
View File

@@ -7,8 +7,11 @@ import com.intellij.openapi.util.RecursionManager;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.util.ProcessingContext;
import com.jetbrains.python.psi.PyCallable;
import com.jetbrains.python.psi.PyTypedElement;
import com.jetbrains.python.psi.impl.PyTypeProvider;
import org.jetbrains.annotations.ApiStatus;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
@@ -33,6 +36,8 @@ public final class TypeEvalContext {
private List<String> myTrace;
private String myTraceIndent = "";
private final ThreadLocal<ProcessingContext> myProcessingContext = ThreadLocal.withInitial(ProcessingContext::new);
private final Map<PyTypedElement, PyType> myEvaluated = new HashMap<>();
private final Map<PyCallable, PyType> myEvaluatedReturn = new HashMap<>();
@@ -206,6 +211,20 @@ public final class TypeEvalContext {
);
}
/**
* Normally, each {@link PyTypeProvider} is supposed to perform all the necessary analysis independently
* and hence should completely isolate its state. However, on rare occasions when several type providers have to
* recursively call each other, it might be necessary to preserve some state for subsequent calls to the same provider with
* the same instance of {@link TypeEvalContext}. Each {@link TypeEvalContext} instance contains an associated thread-local
* {@link ProcessingContext} that can be used for such caching. Should be used with discretion.
*
* @return a thread-local instance of {@link ProcessingContext} bound to this {@link TypeEvalContext} instance
*/
@ApiStatus.Experimental
public @NotNull ProcessingContext getProcessingContext() {
return myProcessingContext.get();
}
private static void assertValid(@Nullable PyType result, @NotNull PyTypedElement element) {
if (result != null) {
result.assertValid(element.toString());