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