fix springmvc by removing import of Object

This commit is contained in:
wing328
2015-09-30 16:28:17 +08:00
parent d9bfb884da
commit 17bdd73891
36 changed files with 87 additions and 49 deletions

View File

@@ -12,6 +12,7 @@ import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Iterator;
public class SpringMVCServerCodegen extends JavaClientCodegen implements CodegenConfig {
protected String invokerPackage = "io.swagger.api";
@@ -187,5 +188,17 @@ public class SpringMVCServerCodegen extends JavaClientCodegen implements Codegen
public void setConfigPackage(String configPackage) {
this.configPackage = configPackage;
}
@Override
public Map<String, Object> postProcessModels(Map<String, Object> objs) {
// remove the import of "Object" to avoid compilation error
List<Map<String, String>> imports = (List<Map<String, String>>) objs.get("imports");
Iterator<Map<String, String>> iterator = imports.iterator();
while (iterator.hasNext()) {
String _import = iterator.next().get("import");
if (_import.endsWith(".Object")) iterator.remove();
}
return objs;
}
}