Improved code
This commit is contained in:
parent
81e163b7c9
commit
9e73e053ea
|
@ -23,14 +23,17 @@ public class TreeGenerator extends Generator {
|
||||||
private static final Logger logger = LoggerFactory.getLogger(TreeGenerator.class);
|
private static final Logger logger = LoggerFactory.getLogger(TreeGenerator.class);
|
||||||
|
|
||||||
public static final String EXTENSION = ".rst";
|
public static final String EXTENSION = ".rst";
|
||||||
|
public static final String FILENAME_SEPARATOR = "-";
|
||||||
|
|
||||||
protected RegistrationProvider rp;
|
protected RegistrationProvider rp;
|
||||||
protected boolean splittedFiles;
|
protected boolean splittedFiles;
|
||||||
|
protected String desiredFileName;
|
||||||
|
|
||||||
public TreeGenerator(RegistrationProvider rp) {
|
public TreeGenerator(RegistrationProvider rp) {
|
||||||
super();
|
super();
|
||||||
this.rp = rp;
|
this.rp = rp;
|
||||||
this.splittedFiles = false;
|
this.splittedFiles = false;
|
||||||
|
this.desiredFileName = rp.getModelName().toLowerCase().replaceAll(" ", FILENAME_SEPARATOR);
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setSplittedFiles(boolean splittedFiles) {
|
public void setSplittedFiles(boolean splittedFiles) {
|
||||||
|
@ -40,6 +43,10 @@ public class TreeGenerator extends Generator {
|
||||||
this.splittedFiles = splittedFiles;
|
this.splittedFiles = splittedFiles;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public void setDesiredFileName(String desiredFileName) {
|
||||||
|
this.desiredFileName = desiredFileName;
|
||||||
|
}
|
||||||
|
|
||||||
public void elaborateTree(final AccessType at, Tree<Class<Element>> tree, File file, RegistrationProvider rp) throws Exception {
|
public void elaborateTree(final AccessType at, Tree<Class<Element>> tree, File file, RegistrationProvider rp) throws Exception {
|
||||||
logger.info("Going to elaborate the following type tree\n{}", tree.toString());
|
logger.info("Going to elaborate the following type tree\n{}", tree.toString());
|
||||||
|
|
||||||
|
@ -73,11 +80,10 @@ public class TreeGenerator extends Generator {
|
||||||
}
|
}
|
||||||
|
|
||||||
protected String getFileName(AccessType accessType) {
|
protected String getFileName(AccessType accessType) {
|
||||||
String modelName = rp.getModelName();
|
|
||||||
StringBuffer stringBuffer = new StringBuffer();
|
StringBuffer stringBuffer = new StringBuffer();
|
||||||
stringBuffer.append(modelName.toLowerCase().replaceAll(" ", "_"));
|
stringBuffer.append(desiredFileName);
|
||||||
if(accessType!=null && splittedFiles) {
|
if(accessType!=null && splittedFiles) {
|
||||||
stringBuffer.append("_");
|
stringBuffer.append(FILENAME_SEPARATOR);
|
||||||
stringBuffer.append(accessType.getName().toLowerCase());
|
stringBuffer.append(accessType.getName().toLowerCase());
|
||||||
}
|
}
|
||||||
stringBuffer.append(EXTENSION);
|
stringBuffer.append(EXTENSION);
|
||||||
|
|
|
@ -21,7 +21,12 @@ public class GenerateTest {
|
||||||
.load(RegistrationProvider.class);
|
.load(RegistrationProvider.class);
|
||||||
for(RegistrationProvider rp : registrationProviders) {
|
for(RegistrationProvider rp : registrationProviders) {
|
||||||
TreeGenerator treeGenerator = new TreeGenerator(rp);
|
TreeGenerator treeGenerator = new TreeGenerator(rp);
|
||||||
treeGenerator.setSplittedFiles(!(rp instanceof ISModelRegistrationProvider));
|
if(rp instanceof ISModelRegistrationProvider) {
|
||||||
|
treeGenerator.setSplittedFiles(false);
|
||||||
|
treeGenerator.setDesiredFileName("is-model");
|
||||||
|
}else {
|
||||||
|
treeGenerator.setSplittedFiles(true);
|
||||||
|
}
|
||||||
treeGenerator.generate();
|
treeGenerator.generate();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue