CheckStyle自定义检查 - 检索所有参数名称

时间:2012-05-13 22:26:59

标签: java static-analysis checkstyle

我正在尝试使用CheckStyle检索给定源文件中找到的所有方法中的所有参数名称。以下是相关代码:


public int[] getDefaultTokens()
{
   return new int[] { TokenTypes.METHOD_DEF};
}


public void visitToken(DetailAST aDetailAST)
{
String returnType;        // The return type of the method.
int numberOfParameters;   // The number of parameters in the method's parameter list... not    returned in log.
String [] parameterNames; // The names of all method parameters.
int openingBraceLine;     // The line number of the opening method bracket.

  returnType = aDetailAST.findFirstToken(TokenTypes.TYPE).getFirstChild().getText(); // get the return type.

  numberOfParameters = aDetailAST.findFirstToken(TokenTypes.PARAMETERS).getChildCount(TokenTypes.PARAMETER_DEF); // get num of parameters.
  parameterNames = new String[numberOfParameters]; // create array to store the parameter names.
  if (numberOfParameters > 0) // only bother if parameters existed.
  {
     List <DetailAST> parameters = DetailASTUtil.getDetailASTsForTypeInBranch                       // Get all PARAMETER_DEF nodes.
                                             (aDetailAST.findFirstToken(TokenTypes.PARAMETERS)
                                                                  , TokenTypes.PARAMETER_DEF);

     int i = 0;

     for (DetailAST currentParameter: parameters) // iterate through all parameters.
     {
        parameterNames[i] = currentParameter.findFirstToken(TokenTypes.IDENT).getText();
        // Get the parameter name, store it in the array.
        i++; // iterate to next parameter name array storage index.
     }
  }

  // parameterNames now contains all parameter names in the parameter list. Format it for log message.

  String formattedParameterNames = "";


  if (numberOfParameters > 1) // if more than one parameter was present, then create comma list.
  {
     for (int i = 0; i < parameterNames.length-1; i++) // put all names in comma-separated string except for last.
     {
        formattedParameterNames += parameterNames[i] + ", ";
     }

     formattedParameterNames += parameterNames[numberOfParameters-1]; // add the last element of the comma list.
  }
  else if (numberOfParameters == 1) // only one parameter -- don't comma-delimit.
  {
     formattedParameterNames = parameterNames[0];
  }

  if (numberOfParameters == 2) // debug to see if string formatting is messing up the parameter names or if tree traversal is bad.
  {
     formattedParameterNames = "Param 1: " + parameterNames[0] + " Param 2: " + parameterNames[1];
  }

  log(aDetailAST.getLineNo(), "[" + returnType + "]" + ", [" + formattedParameterNames + "], ");
  // will be further parsed in actual applet since I don't think there's a way to get individual lines of code via CheckStyle... I would like if a getTextForLineofCode(lineNumber) func existed with CheckStyle, but I don't think it does.
}

public static List<DetailAST> getDetailASTsForTypeInBranch(DetailAST expr,
        int tokenType) {
    return getDetailASTsForTypeInBranch(expr, tokenType, null);
}

private static List<DetailAST> getDetailASTsForTypeInBranch(DetailAST expr,
        int tokenType, List<DetailAST> list) {
    if (list == null)
        list = new ArrayList<DetailAST>();
    DetailAST child = (DetailAST) expr.getFirstChild();
    while (child != null) {
        if (child.getType() == tokenType) {
            list.add(child);
        } else {
            list = getDetailASTsForTypeInBranch(child, tokenType, list);
        }
        child = (DetailAST) child.getNextSibling();
    }
    return list;
}

当我在我的主applet中检索此日志消息时,没有/单个参数列表的函数看起来很好,但双参数函数要么根本没有注册,要么返回消息“secondParmeterNameHere]”,其中secondParameterNameHere是特定功能的第二个参数名称。

关于我的算法获取所有参数名称有什么问题的任何想法?感谢。

2 个答案:

答案 0 :(得分:0)

我的两分钱来优化您的代码。

使用for循环

在您的递归辅助方法中,您可以替换

    DetailAST child = (DetailAST) expr.getFirstChild();
while (child != null) {
    if (child.getType() == tokenType) {
        list.add(child);
    } else {
        list = getDetailASTsForTypeInBranch(child, tokenType, list);
    }
    child = (DetailAST) child.getNextSibling();
}

通过

for(DetailAST child = (DetailAST) expr.getFirstChild(); child != null; child = (DetailAST) child.getNextSibling()) {
    if (child.getType() == tokenType) {
        list.add(child);
    } else {
        list = getDetailASTsForTypeInBranch(child, tokenType, list);
    }
}

决定分配“list”的一种方式

这是某种“分配重复:

list = getDetailASTsForTypeInBranch(child, tokenType, list);

您通过引用传递“列表”。在递归过程中,您可以使用此引用(list.add(..))。但是你仍然可以稍后返回此列表作为返回值。在递归中,您将此返回值重新分配给原始输入变量,尽管在递归过程中已经修改过这个值?!

您应该使用返回值并且不要将列表作为输入参数传递。或者您可以传递列表,但是您的方法不应该有返回值。

减少格式化参数名称的代码

变量 parameterNames 已过时,因为您只使用它来再次迭代它并连接一个String。您的字符串连接也是重复且低效的。我建议采用以下解决方案:

        StringBuilder formattedParameterNames = new StringBuilder();
    for (Iterator<DetailAST> iterator = parameters.iterator(); iterator.hasNext();) {
        DetailAST detailAST = iterator.next();
        formattedParameterNames.append(detailAST.findFirstToken(TokenTypes.IDENT).getText());
        if(iterator.hasNext()) {
            formattedParameterNames.append(", ");
        }
    }

您可以在StringBuilder上调用toString()来获取日志记录语句。

答案 1 :(得分:0)

希望以下代码段对您有所帮助

[ERROR] path\Test.java:8:9: Method Name : add [ListMethodParameters]
[ERROR] path\Test.java:8:9: No Of Parameters : 0 [ListMethodParameters]
[ERROR] path\Test.java:8:9: Return Type : void [ListMethodParameters]
[ERROR] path\Test.java:12:9: Method Name : sub [ListMethodParameters]
[ERROR] path\Test.java:12:9: No Of Parameters : 3 [ListMethodParameters]
[ERROR] path\Test.java:12:9: Parameter 1 (int a) [ListMethodParameters]
[ERROR] path\Test.java:12:9: Parameter 2 (int b) [ListMethodParameters]
[ERROR] path\Test.java:12:9: Parameter 3 (int c) [ListMethodParameters]
[ERROR] path\Test.java:12:9: Return Type : int [ListMethodParameters]

输出将类似于

public class Test {
    public void add(){
        name.length();
    }
    public int sub(int a,int b,int c){
        return a+b+c;
    }
}

Java文件用于测试以上代码

<base href="/">