使用代码模型添加导入

时间:2015-06-12 09:44:48

标签: java sun-codemodel jcodemodel

我正在尝试使用代码模型 导入我的代码中的类。 这是我的代码。

JCodeModel model = new JCodeModel();
JClass mapper = model.directClass("com.another.Mapper");
JDefinedClass dc = model._class("com.example.Something");
JMethod method = dc.method(JMod.PUBLIC | JMod.STATIC, Void.TYPE,
        "testMethod");
JBlock executerBlock = method.body();
    executerBlock.directStatement("Mapper.get()");
File file = new File("./src");
file.mkdirs();
model.build(file);

现在我得到以下课程。

package com.example;
public class Something {
    public static void testMethod() {
    Mapper.get()
    }
}

但实际上我需要的是,

package com.example;
import com.another.Mapper;
public class Something {
    public static void testMethod() {
    Mapper.get()
    }
}

除非使用,否则导入不会发生。我怎样才能进行此导入。

1 个答案:

答案 0 :(得分:0)

要将Mapper添加为导入,您需要拨打invoke() / staticInvoke()而不是directStatement()

JClass mapper = model.directClass("com.another.Mapper");
JDefinedClass dc = model._class("com.example.Something");
JMethod method = dc.method(JMod.PUBLIC | JMod.STATIC, Void.TYPE, "testMethod");
JBlock executerBlock = method.body();
executerBlock.staticInvoke(mapper, "get");