11package com .hsjfans .github .parser ;
22
33import com .github .javaparser .ast .CompilationUnit ;
4+ import com .github .javaparser .ast .body .TypeDeclaration ;
45import com .google .common .collect .Maps ;
56
67import java .util .Map ;
@@ -18,15 +19,15 @@ public class ClassCache {
1819
1920 private static final Map <String ,Class <?>> classCache ;
2021
21- private static final Map <String , CompilationUnit > compilationUnitCache ;
22+ private static final Map <String , TypeDeclaration <?> > compilationUnitCache ;
2223
2324 static {
2425 classCache = Maps .newHashMap ();
2526 compilationUnitCache = Maps .newHashMap ();
2627 }
2728
2829
29- public static void putCompilationUnit (String filePath ,CompilationUnit compilationUnit ){
30+ public static void putCompilationUnit (String filePath ,TypeDeclaration compilationUnit ){
3031 boolean contain ;
3132 compilationUnitReadWriteLock .readLock ().lock ();
3233 contain = compilationUnitCache .containsKey (filePath );
@@ -39,8 +40,8 @@ public static void putCompilationUnit(String filePath,CompilationUnit compilatio
3940 compilationUnitReadWriteLock .writeLock ().unlock ();
4041 }
4142
42- public static CompilationUnit getCompilationUnit (String filePath ){
43- CompilationUnit compilationUnit ;
43+ public static TypeDeclaration <?> getCompilationUnit (String filePath ){
44+ TypeDeclaration compilationUnit ;
4445 compilationUnitReadWriteLock .readLock ().lock ();
4546 compilationUnit = compilationUnitCache .get (filePath );
4647 compilationUnitReadWriteLock .readLock ().unlock ();
@@ -75,7 +76,7 @@ public static Map<String, Class<?>> getClassCache() {
7576 return classCache ;
7677 }
7778
78- public static Map <String , CompilationUnit > getCompilationUnitCache () {
79- return compilationUnitCache ;
79+ public static Map <String , TypeDeclaration <?> > getCompilationUnitCache () {
80+ return getCompilationUnitCache () ;
8081 }
8182}
0 commit comments