Merge remote-tracking branch 'origin/mikhail.golubev/py-attribute-inference' into...
[idea/community.git] / python / src / com / jetbrains / python / psi / resolve / PythonPathCache.java
1 /*
2  * Copyright 2000-2014 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.jetbrains.python.psi.resolve;
17
18 import com.intellij.openapi.vfs.*;
19 import com.intellij.psi.PsiElement;
20 import com.intellij.psi.util.QualifiedName;
21 import com.intellij.util.containers.ConcurrentHashMap;
22 import org.jetbrains.annotations.NotNull;
23
24 import java.util.List;
25 import java.util.Map;
26
27 /**
28  * @author yole
29  */
30 public abstract class PythonPathCache {
31   private final Map<QualifiedName, List<PsiElement>> myCache = new ConcurrentHashMap<QualifiedName, List<PsiElement>>();
32   private final Map<VirtualFile, List<QualifiedName>> myQNameCache = new ConcurrentHashMap<VirtualFile, List<QualifiedName>>();
33
34   public void clearCache() {
35     myCache.clear();
36     myQNameCache.clear();
37   }
38
39   public List<PsiElement> get(QualifiedName qualifiedName) {
40     return myCache.get(qualifiedName);
41   }
42
43   public void put(QualifiedName qualifiedName, List<PsiElement> results) {
44     myCache.put(qualifiedName, results);
45   }
46
47   public List<QualifiedName> getNames(VirtualFile vFile) {
48     return myQNameCache.get(vFile);
49   }
50   
51   public void putNames(VirtualFile vFile, List<QualifiedName> qNames) {
52     myQNameCache.put(vFile, qNames);
53   }
54
55   protected class MyVirtualFileAdapter extends VirtualFileAdapter {
56     @Override
57     public void fileCreated(@NotNull VirtualFileEvent event) {
58       clearCache();
59     }
60
61     @Override
62     public void fileDeleted(@NotNull VirtualFileEvent event) {
63       clearCache();
64     }
65
66     @Override
67     public void fileMoved(@NotNull VirtualFileMoveEvent event) {
68       clearCache();
69     }
70
71     @Override
72     public void fileCopied(@NotNull VirtualFileCopyEvent event) {
73       clearCache();
74     }
75
76     @Override
77     public void propertyChanged(@NotNull VirtualFilePropertyEvent event) {
78       if (event.getPropertyName().equals(VirtualFile.PROP_NAME)) {
79         clearCache();
80       }
81     }
82   }
83 }