mybatisplus自定义生成injectionconfig
时间: 2023-05-23 19:07:35 浏览: 627
A: 对于Mybatis-Plus中的代码生成器,可以通过自定义InjectionConfig来实现自定义的生成操作。以下是一个简单的示例:
```
//新建自定义的InjectionConfig
InjectionConfig injectionConfig = new InjectionConfig() {
@Override
public void initMap() {
//这里可以定义需要生成的文件
Map<String, Object> map = new HashMap<>();
map.put("customXml", "/templates/mapper/custom.xml.vm");
this.setMap(map);
}
};
//配置生成xml文件的输出路径
List<FileOutConfig> fileOutConfigList = new ArrayList<>();
fileOutConfigList.add(new FileOutConfig("/templates/mapper/custom.xml.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
return projectPath + "/src/main/resources/mapper/" + tableInfo.getEntityName() + "CustomMapper.xml";
}
});
injectionConfig.setFileOutConfigList(fileOutConfigList);
//将自定义的InjectionConfig设置到代码生成器中
AutoGenerator generator = new AutoGenerator();
generator.setGlobalConfig(globalConfig);
generator.setDataSource(dataSourceConfig);
generator.setPackageInfo(packageConfig);
generator.setStrategy(strategyConfig);
generator.setTemplate(templateConfig);
generator.setCfg(injectionConfig);
generator.execute();
```
其中,自定义的InjectionConfig需要实现initMap方法来定义需要生成的文件路径和模板引擎。通过配置FileOutConfig来指定模板引擎和输出路径。最后将自定义的InjectionConfig设置到代码生成器中即可。
阅读全文