JCodeModel - 如何链接调用方法

Question

我试图找出如何使用循环来构建一个JExpression ，我可以在使用.assign()调用。 我知道您可以使用JExpr.invoke("methodA").invoke("methodB")执行JExpr.invoke("methodA").invoke("methodB")的操作，但我希望能够根据方法名称列表添加每个.invoke()方法。 有谁知道如何做到这一点？

我的生成器代码示例[假设我已经修改了 ArrayList.add() 方法以返回 ArrayList]：

JDefinedClass newClass = jCodeModel._class("com.me.GeneratedClass");
JClass refObject = jCodeModel.ref(Object.class);
JClass refArrayList = jCodeModel.ref(ArrayList.class);

JFieldVar objectList = newClass.field(refArrayList, "objects");

JBlock methodBody = newClass.method(JMod.PUBLIC, refArrayList, "getNewObjectList");

String[] methodsToInvoke = {"add", "add", "add"};

JExpression rhsAssignmentExpression = JExpr._new(refArrayList).narrow(refObject);
for(String methodName : methodsToInvoke) {
    rhsAssignmentExpression.invoke(methodName).arg(JExpr._new(refObject));
}

methodBody.assign(objectList, rhsAssignmentExpression);
methodBody._return(objectList);

我试图生成的示例：

class GeneratedClass {

   ArrayList<Object> objects;

   public ArrayList<Object> getNewObjectList()
   {
      objects = new ArrayList<Object>().add(new Object()).add(new Object()).add(new Object());
      return objects;
   }
}

我的问题是此方法不会链接调用方法，而是替换它们，因此生成的代码如下所示：

class GeneratedClass {

   ArrayList<Object> objects;

   public ArrayList<Object> getNewObjectList()
   {
      objects = new ArrayList<Object>().add(new Object());
      return objects;
   }
}

Answer 1

这也可以通过让 JCodeModel 打印原始字符串而不是声明 JCode 对象来完成。 对于这个特定的问题，不需要生成 java 导入——因此这可以通过 directStatement() 来完成。

StringBuilder expression = new StringBuilder();
expression.append (“objects = new ArrayList<Object>()”);

if ((methodsToInvoke != null) && (methodsToInvoke.length > 0))
{
   for(String methodName : methodsToInvoke)
   {
      expression.append(“.add(new Object())”);
   }
}

expression.append(“;”);

methodBoby.directStatement(expression.toString());

Answer 2

您需要做的是为每个 invoke() JExpression 调用存储一个中间值：

JExpression rhsAssignmentExpression = JExpr._new(refArrayList);
for(String methodName : methodsToInvoke) {
    rhsAssignmentExpression = rhsAssignmentExpression.invoke(methodName).arg(JExpr._new(refObject));
}

这避免了每次重新定义调用，并有效地链接调用。 结果如下：

public class GeneratedClass {

    public ArrayList objects;

    public ArrayList getNewObjectList() {
        objects = new ArrayList().add(new Object()).add(new Object()).add(new Object());
        return objects;
    }

}