波总,我想把_Generator生成的文件格式从.java改为.groovy,请问在哪里改?

波总,我想把_Generator生成的文件格式从.java改为.groovy,请问在哪里改?

评论区

杜福忠

2018-03-03 09:46

比如BaseModelGenerator这个生成器, 你只需要继承 BaseModelGenerator 这个类, 然后重写writeToFile方法:

/**
* base model 覆盖写入
*/
protected void writeToFile(TableMeta tableMeta) throws IOException {
File dir = new File(baseModelOutputDir);
if (!dir.exists()) {
dir.mkdirs();
}

String target = baseModelOutputDir + File.separator + tableMeta.baseModelName + ".groovy";
FileWriter fw = new FileWriter(target);
try {
fw.write(tableMeta.baseModelContent);
}
finally {
fw.close();
}
}

其他类,一样操作。
然后使用时传入你自己的类
Generator gen = new Generator(dataSource, YouBaseModelGenerator, YouModelGenerator)
就OK了

热门反馈

扫码入社