feat: <java> add field type support

phodal · Feb 9, 2020 · e585d33 · e585d33
1 parent 8c08bb8
commit e585d33
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/chapi-ast-java/src/main/kotlin/chapi/ast/javaast/JavaFullIdentListener.kt b/chapi-ast-java/src/main/kotlin/chapi/ast/javaast/JavaFullIdentListener.kt
@@ -323,10 +323,13 @@ open class JavaFullIdentListener(fileName: String) : JavaAstListener() {
     private fun parseTargetType(target: String?): String? {
         var targetType = target
 
+        val fieldType = fieldsMap[targetType]
         val formalType = formalParameters[targetType]
         val localVarType = localVars[targetType]
 
-        if (formalType != null && formalType != "") {
+        if (fieldType != null && fieldType != "") {
+            targetType = fieldType
+        } else if (formalType != null && formalType != "") {
             targetType = formalType
         } else if (localVarType != null && localVarType != "") {
             targetType = localVarType