/*
* Copyright 2009 The Apache Software Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.mybatis.generator.codegen.mybatis3;
import java.util.ArrayList;
import java.util.List;
import org.mybatis.generator.api.GeneratedJavaFile;
import org.mybatis.generator.api.GeneratedXmlFile;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.ProgressCallback;
import org.mybatis.generator.api.dom.java.CompilationUnit;
import org.mybatis.generator.api.dom.xml.Document;
import org.mybatis.generator.codegen.AbstractGenerator;
import org.mybatis.generator.codegen.AbstractJavaClientGenerator;
import org.mybatis.generator.codegen.AbstractJavaGenerator;
import org.mybatis.generator.codegen.AbstractXmlGenerator;
import org.mybatis.generator.codegen.mybatis3.javamapper.AnnotatedClientGenerator;
import org.mybatis.generator.codegen.mybatis3.javamapper.JavaMapperGenerator;
import org.mybatis.generator.codegen.mybatis3.javamapper.MixedClientGenerator;
import org.mybatis.generator.codegen.mybatis3.model.BaseRecordGenerator;
import org.mybatis.generator.codegen.mybatis3.model.ExampleGenerator;
import org.mybatis.generator.codegen.mybatis3.model.PrimaryKeyGenerator;
import org.mybatis.generator.codegen.mybatis3.model.RecordWithBLOBsGenerator;
import org.mybatis.generator.codegen.mybatis3.xmlmapper.XMLMapperGenerator;
import org.mybatis.generator.config.PluginConfiguration;
import org.mybatis.generator.internal.ObjectFactory;
/**
*
* @author Jeff Butler
*
*/
public class IntrospectedTableMyBatis3Impl extends IntrospectedTable {
protected List<AbstractJavaGenerator> javaModelGenerators;
protected List<AbstractJavaGenerator> clientGenerators;
protected AbstractXmlGenerator xmlMapperGenerator;
public IntrospectedTableMyBatis3Impl() {
super(TargetRuntime.MYBATIS3);
javaModelGenerators = new ArrayList<AbstractJavaGenerator>();
clientGenerators = new ArrayList<AbstractJavaGenerator>();
}
@Override
public void calculateGenerators(List<String> warnings, ProgressCallback progressCallback) {
calculateJavaModelGenerators(warnings, progressCallback);
AbstractJavaClientGenerator javaClientGenerator = calculateClientGenerators(warnings, progressCallback);
calculateXmlMapperGenerator(javaClientGenerator, warnings, progressCallback);
}
protected void calculateXmlMapperGenerator(AbstractJavaClientGenerator javaClientGenerator, List<String> warnings,
ProgressCallback progressCallback) {
if (javaClientGenerator == null) {
xmlMapperGenerator = new XMLMapperGenerator();
} else {
xmlMapperGenerator = javaClientGenerator.getMatchedXMLGenerator();
}
initializeAbstractGenerator(xmlMapperGenerator, warnings, progressCallback);
}
/**
*
* @param warnings
* @param progressCallback
* @return true if an XML generator is required
*/
protected AbstractJavaClientGenerator calculateClientGenerators(List<String> warnings,
ProgressCallback progressCallback) {
AbstractJavaClientGenerator javaGenerator = createJavaClientGenerator();
if (javaGenerator == null) {
return null;
}
initializeAbstractGenerator(javaGenerator, warnings, progressCallback);
clientGenerators.add(javaGenerator);
return javaGenerator;
}
private AbstractJavaClientGenerator createJavaClientGenerator() {
if (context.getJavaClientGeneratorConfiguration() == null) {
return null;
}
String type = context.getJavaClientGeneratorConfiguration().getConfigurationType();
AbstractJavaClientGenerator javaGenerator;
if ("XMLMAPPER".equalsIgnoreCase(type)) { //$NON-NLS-1$
javaGenerator = new JavaMapperGenerator();
} else if ("MIXEDMAPPER".equalsIgnoreCase(type)) { //$NON-NLS-1$
javaGenerator = new MixedClientGenerator();
} else if ("ANNOTATEDMAPPER".equalsIgnoreCase(type)) { //$NON-NLS-1$
javaGenerator = new AnnotatedClientGenerator();
} else if ("MAPPER".equalsIgnoreCase(type)) { //$NON-NLS-1$
javaGenerator = new JavaMapperGenerator();
} else {
javaGenerator = (AbstractJavaClientGenerator) ObjectFactory.createInternalObject(type);
}
return javaGenerator;
}
protected void calculateJavaModelGenerators(List<String> warnings, ProgressCallback progressCallback) {
if (getRules().generateExampleClass()) {
AbstractJavaGenerator javaGenerator = new ExampleGenerator();
initializeAbstractGenerator(javaGenerator, warnings, progressCallback);
javaModelGenerators.add(javaGenerator);
}
if (getRules().generatePrimaryKeyClass()) {
AbstractJavaGenerator javaGenerator = new PrimaryKeyGenerator();
initializeAbstractGenerator(javaGenerator, warnings, progressCallback);
javaModelGenerators.add(javaGenerator);
}
if (getRules().generateBaseRecordClass()) {
AbstractJavaGenerator javaGenerator = new BaseRecordGenerator();
initializeAbstractGenerator(javaGenerator, warnings, progressCallback);
javaModelGenerators.add(javaGenerator);
}
if (getRules().generateRecordWithBLOBsClass()) {
AbstractJavaGenerator javaGenerator = new RecordWithBLOBsGenerator();
initializeAbstractGenerator(javaGenerator, warnings, progressCallback);
javaModelGenerators.add(javaGenerator);
}
}
protected void initializeAbstractGenerator(AbstractGenerator abstractGenerator, List<String> warnings,
ProgressCallback progressCallback) {
if (abstractGenerator == null) {
return;
}
abstractGenerator.setContext(context);
abstractGenerator.setIntrospectedTable(this);
abstractGenerator.setProgressCallback(progressCallback);
abstractGenerator.setWarnings(warnings);
}
@Override
public List<GeneratedJavaFile> getGeneratedJavaFiles() {
List<GeneratedJavaFile> answer = new ArrayList<GeneratedJavaFile>();
boolean isExample = true;
List<PluginConfiguration> list = context.getPluginConfigurations();
for (int i = 0; i < list.size(); i++) {
PluginConfiguration p = list.get(i);
if (p.toXmlElement().getFormattedContent(0).contains("CriteriaPlugin")) {
isExample = false;
}
}
for (AbstractJavaGenerator javaGenerator : javaModelGenerators) {
List<CompilationUnit> compilationUnits = javaGenerator.getCompilationUnits();
for (CompilationUnit compilationUnit : compilationUnits) {
if(compilationUnit.getFormattedContent().contains("oredCriteria")){
if(!isExample){
break;
}
}
GeneratedJavaFile gjf = new GeneratedJavaFile(compilationUnit, context
.getJavaModelGeneratorConfiguration().getTargetProject());
answer.add(gjf);
}
}
for (AbstractJavaGenerator javaGenerator : clientGenerators) {
List<CompilationUnit> compilationUnits = javaGenerator.getCompilationUnits();
for (CompilationUnit compilationUnit : compilationUnits) {
GeneratedJavaFile gjf = new GeneratedJavaFile(compilationUnit, context
.getJavaClientGeneratorConfiguration().getTargetProject());
answer.add(gjf);
}
}
return answer;
}
@Override
public List<GeneratedXmlFile> getGeneratedXmlFiles() {
List<GeneratedXmlFile> answer = new ArrayList<GeneratedXmlFile>();
if (xmlMapperGenerator != null) {
Document document = xmlMapperGenerator.getDocument();
GeneratedXmlFile gxf = new GeneratedXmlFile(document, getMyBatis3XmlMapperFileName(),
getMyBatis3XmlMapperPackage(), context.getSqlMapGeneratorConfiguration().getTargetProject(), false);
if (context.getPlugins().sqlMapGenerated(gxf, this)) {
answer.add(gxf);
}
}
return answer;
}
@Override
public int getGenerationSteps() {
return javaModelGenerators.size() + clientGenerators.size() + (xmlMapperGenerator == null ? 0 : 1);
}
@Override
public boolean isJava5Targeted() {
return true;
}
@Override
public boolean requiresXMLGenerator() {
AbstractJavaClientGenerator javaClientGenerator = createJavaClientGenerator();
if (javaClientGenerator == null) {
return true;
} else {
return javaClientGenerator.requiresXMLGenerator();
}
}
}