modules/registry/registry-tools/jpa-gen/src/main/java/generators/JPAClassGenerator.java

/**
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 generators;
import java.util.ArrayList;
import java.util.List;

import model.JPAClassField;
import model.JPAClassModel;
import model.JPAPKClassModel;
import model.SQLData;


/*
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 *
 */

public class JPAClassGenerator extends AbstractGenerator{
//    private static final Logger log = LoggerFactory.getLogger(JPAClassGenerator.class);
	private String jpaClassPackageName;
	
    public JPAClassModel createJPAClassModel(SQLData sqlData){
		JPAClassModel model = new JPAClassModel();
		model.generatePKClass=sqlData.getPrimaryKeys().size()>1;
		model.tableName=sqlData.getTableName();
		model.className = convertToJavaTitleCaseStringConvention(sqlData.getTableName());
		if (model.generatePKClass) {
			model.pkClassModel.className=model.className+"_PK";
		}
		for (String field : sqlData.getFieldData().keySet()) {
		    String dataType = null;
		    SQLGenerator.DataTypes sqlDataType = SQLGenerator.DataTypes.valueOf(sqlData.getFieldData().get(field).get(0));
		    switch (sqlDataType){
		    case LONGTEXT:case VARCHAR:
		    	dataType="String"; break;
		    case INTEGER: 
		    	dataType="int"; break;
		    case SMALLINT:
		    	dataType="boolean"; break;
		    case TIMESTAMP: 
		    	dataType="Timestamp"; break;
		    case CLOB:
		    	dataType="String"; break;
		    }
		    String fieldTitleString = convertToJavaTitleCaseStringConvention(field);
		    String fieldName = convertToJavaVariableNameCaseStringConvention(field);

		    
		    boolean foriegnKey = sqlData.getForiegnKeys().containsKey(field);
			JPAClassField jpaField = new JPAClassField(field,fieldName,dataType,fieldTitleString,sqlData.getPrimaryKeys().contains(field),
		    		foriegnKey,(foriegnKey?sqlData.getForiegnKeys().get(field).jpaClassName:null),
		    		(foriegnKey?sqlData.getForiegnKeys().get(field).jpaResourceClassName:null));
			model.fields.add(jpaField);
		    if (model.generatePKClass){
		    	if (sqlData.getPrimaryKeys().contains(field)){
		    		model.pkClassModel.pkFields.add(jpaField);
		    	}
		    }
		    
		}
		return model;
	}
    
	public String generateJPAClass(JPAClassModel model){
		String classStr = null;
		String pkClassName = null;
		classStr=addLines(classStr,"@DataCache");
		classStr=addLines(classStr,"@Entity");
		classStr=addLines(classStr,"@Table(name = \""+model.tableName+"\")");
		String className = model.className;
		if (model.generatePKClass) {
			pkClassName=model.pkClassModel.className;
			classStr = addLines(classStr,"@IdClass("+pkClassName+".class)");
		}
		classStr=addLines(classStr,"public class "+className+" implements Serializable {");
		
		List<String> columnFields=new ArrayList<String>();
		List<String> fieldGetters=new ArrayList<String>();
		List<String> fieldSetters=new ArrayList<String>();
		for (JPAClassField jpaField : model.fields) {
			String field=jpaField.tableColumnName;
		    String fieldString=null;
		    
		    String fieldName = jpaField.fieldName;
		    String dataType = jpaField.fieldDataType;
		    String fieldTitleString = jpaField.fieldTitle;

		    
		    fieldString=addLines(fieldString, tabs(1));
		    if (jpaField.primaryKey){
		    	fieldString=addLines(fieldString,tabs(1)+"@Id");	
		    }
		    fieldString=addLines(fieldString,tabs(1)+"@Column(name = \""+field+"\")");
			fieldString=addLines(fieldString,tabs(1)+createFieldVarString(dataType ,fieldName));
		    columnFields.add(fieldString);
		    
		    
		    fieldGetters.add(createGetterString(1, fieldName,dataType, fieldTitleString));

		    fieldSetters.add(createSetterString(1, fieldName,dataType, fieldTitleString));
		    
		    if (jpaField.foriegnKey){
			    fieldString=null;
			    
			    fieldName = createVarNameFromClassName(jpaField.foriegnKeyJPAClass);
			    dataType = jpaField.foriegnKeyJPAClass;
			    fieldTitleString = jpaField.foriegnKeyJPAClass;
			    
			    
			    fieldString=addLines(fieldString, tabs(1));
			    fieldString=addLines(fieldString,tabs(1)+"@ManyToOne(cascade= CascadeType.MERGE)");
			    fieldString=addLines(fieldString,tabs(1)+"@JoinColumn(name = \""+jpaField.tableColumnName+"\")");
				fieldString=addLines(fieldString,tabs(1)+createFieldVarString(dataType ,fieldName));
				columnFields.add(fieldString);
				
			    fieldGetters.add(createGetterString(1, fieldName,dataType, fieldTitleString));

			    fieldSetters.add(createSetterString(1, fieldName,dataType, fieldTitleString));
		    }
		}
		classStr=addLines(classStr,columnFields.toArray(new String[]{}));
		classStr=addLines(classStr,fieldGetters.toArray(new String[]{}));
		classStr=addLines(classStr,fieldSetters.toArray(new String[]{}));
		
		classStr=addLines(classStr,"}");
		return classStr;
	}

	public String generateJPAPKClass(JPAPKClassModel model){
		if (model.pkFields.size()==0){
			return "";
		}
		String classStr=null;
		classStr=addLines(classStr,"public class "+model.className+" implements Serializable {");
		
		List<String> columnFields=new ArrayList<String>();
		List<String> fieldGetters=new ArrayList<String>();
		List<String> fieldSetters=new ArrayList<String>();
		List<String> parameterList=new ArrayList<String>();
		String constructorMethod=null;
		for (JPAClassField jpaField : model.pkFields) {
		    
		    String dataType = jpaField.fieldDataType;
		    String fieldTitleString = jpaField.fieldTitle;
		    String fieldName = jpaField.fieldName;
		    
		    String fieldString=tabs(1)+createFieldVarString(dataType ,fieldName);
		    columnFields.add(fieldString);
		    
		    
		    fieldGetters.add(createGetterString(1, fieldName, dataType, fieldTitleString));

		    fieldSetters.add(createSetterString(1, fieldName,	dataType, fieldTitleString));

		    parameterList.add(dataType+" "+fieldName);
		    constructorMethod=addLines(constructorMethod, tabs(2)+"this."+fieldName+" = "+fieldName+";");
		}
		classStr=addLines(classStr,columnFields.toArray(new String[]{}));
		String constructorParametersString=commaSeperatedString(parameterList,", ");
		constructorMethod=addLines(tabs(1), tabs(1)+"public "+model.className+"("+constructorParametersString+"){",constructorMethod);
		constructorMethod=addLines(constructorMethod, tabs(1)+"}");
		String emptyConstructorMethod=null;
		emptyConstructorMethod=addLines(tabs(1),tabs(1)+"public "+model.className+"(){",tabs(1)+"}");
		
		classStr=addLines(classStr,emptyConstructorMethod);
		classStr=addLines(classStr,constructorMethod);
		



		classStr=addLines(classStr,tabs(1));
		classStr=addLines(classStr,tabs(1)+"@Override");
		classStr=addLines(classStr,tabs(1)+"public boolean equals(Object o) {");
		classStr=addLines(classStr,tabs(2)+"return false;");
		classStr=addLines(classStr,tabs(1)+"}");

		classStr=addLines(classStr,tabs(1));
		classStr=addLines(classStr,tabs(1)+"@Override");
		classStr=addLines(classStr,tabs(1)+"public int hashCode() {");
		classStr=addLines(classStr,tabs(2)+"return 1;");
		classStr=addLines(classStr,tabs(1)+"}");
	    
		classStr=addLines(classStr,fieldGetters.toArray(new String[]{}));
		classStr=addLines(classStr,fieldSetters.toArray(new String[]{}));
		
		classStr=addLines(classStr,"}");
		return classStr;
	}

	public String generatePersistenceXmlEntry(JPAClassModel model){
		String xmlEntry=null;
		xmlEntry=addLines(xmlEntry,"<persistence xmlns=\"http://java.sun.com/xml/ns/persistence\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" version=\"1.0\">");
		xmlEntry=addLines(xmlEntry,tabs(1)+"<persistence-unit name=\"appcatalog_data\">");
		xmlEntry=addLines(xmlEntry,tabs(2)+"<provider>org.apache.openjpa.persistence.PersistenceProviderImpl</provider>");
		xmlEntry=addLines(xmlEntry,tabs(2)+"<class>"+getJpaClassPackageName()+"."+model.className+"</class>");
		xmlEntry=addLines(xmlEntry,tabs(2)+"<exclude-unlisted-classes>true</exclude-unlisted-classes>");
		xmlEntry=addLines(xmlEntry,tabs(1)+"</persistence-unit>");
		xmlEntry=addLines(xmlEntry,"</persistence>");
		return xmlEntry;
	}
	
	public String getJpaClassPackageName() {
		return jpaClassPackageName;
	}

	public void setJpaClassPackageName(String jpaClassPackageName) {
		this.jpaClassPackageName = jpaClassPackageName;
	}
	
}