Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.landawn.abacus.util.CodeGenerator2 Maven / Gradle / Ivy
/*
* Copyright (C) 2015 HaiYang Li
*
* 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 com.landawn.abacus.util;
import static com.landawn.abacus.util.CodeGenerator2.EntityMode.EXTEND_DIRTY_MARKER;
import static com.landawn.abacus.util.CodeGenerator2.EntityMode.IMPL_DIRTY_MARKER;
import static com.landawn.abacus.util.WD._PERIOD;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Serializable;
import java.io.Writer;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.NavigableMap;
import java.util.NavigableSet;
import java.util.Objects;
import java.util.Set;
import java.util.SortedMap;
import java.util.SortedSet;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicInteger;
import javax.xml.bind.annotation.XmlTransient;
import org.w3c.dom.Comment;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import com.landawn.abacus.DirtyMarker;
import com.landawn.abacus.annotation.Id;
import com.landawn.abacus.core.AbstractDirtyMarker;
import com.landawn.abacus.core.DirtyMarkerImpl;
import com.landawn.abacus.exception.UncheckedIOException;
import com.landawn.abacus.logging.Logger;
import com.landawn.abacus.logging.LoggerFactory;
import com.landawn.abacus.metadata.Column;
import com.landawn.abacus.metadata.ColumnType;
import com.landawn.abacus.metadata.Database;
import com.landawn.abacus.metadata.EntityDefXmlEle.DatabaseEle;
import com.landawn.abacus.metadata.EntityDefXmlEle.EntityDefEle;
import com.landawn.abacus.metadata.EntityDefXmlEle.EntityDefEle.EntityEle;
import com.landawn.abacus.metadata.EntityDefXmlEle.EntityDefEle.EntityEle.PropertyEle;
import com.landawn.abacus.metadata.EntityDefXmlEle.EntityDefEle.EntityEle.TableEle;
import com.landawn.abacus.metadata.EntityDefXmlEle.EntityDefEle.EntityEle.TableEle.ColumnEle;
import com.landawn.abacus.metadata.EntityDefinition;
import com.landawn.abacus.metadata.EntityDefinitionFactory;
import com.landawn.abacus.metadata.Property;
import com.landawn.abacus.metadata.Table;
import com.landawn.abacus.metadata.sql.SQLEntityDefinition;
import com.landawn.abacus.metadata.sql.SQLEntityDefinitionFactory;
import com.landawn.abacus.type.HBaseColumnType;
import com.landawn.abacus.type.ObjectType;
import com.landawn.abacus.type.Type;
import com.landawn.abacus.type.TypeType;
import com.landawn.abacus.util.CodeGenerator.ParentPropertyMode;
// TODO: Auto-generated Javadoc
/**
*
* @author Haiyang Li
* @since 0.8
*/
public final class CodeGenerator2 {
private static final Logger logger = LoggerFactory.getLogger(CodeGenerator2.class);
/** Field UTF_8. */
private static final String UTF_8 = "UTF-8";
/**
* To generate the entity classes which not depend on N.class in the codes of hashcode/equals/toString methods, specify the parameter utilClassForHashEqualsToString in method entityDefinitionXml2Class with this value _N.
*/
public static final Class _N = _N.class;
private static final String POSTFIX_OF_JAVA_FILE = ".java";
/**
* Field TABLE_NAME2ENTITY_NAME. (value is ""tableName2EntityName"")
*/
private static final String TABLE_NAME2ENTITY_NAME = "tableName2EntityName";
/**
* Field COLUMN_NAME2PROP_NAME. (value is ""columnName2PropName"")
*/
private static final String COLUMN_NAME2PROP_NAME = "columnName2PropName";
/**
* Field PROP_NAME2VAR_NAME. (value is ""propName2VarName"")
*/
private static final String PROP_NAME2VAR_NAME = "propName2VarName";
/**
* Field PROP_NAME2METHOD_NAME. (value is ""propName2MethodName"")
*/
private static final String PROP_NAME2METHOD_NAME = "propName2MethodName";
/**
* Field DOMAIN_NAME_VAR. (value is """_DN"")
*/
private static final String DOMAIN_NAME_VAR = "_DN";
/**
* Field ENTITY_NAME_VAR. (value is ""__"")
*/
private static final String ENTITY_NAME_VAR = "__";
/**
* Field PROP_NAME_LIST. (value is ""_PNL"")
*/
private static final String PROP_NAME_LIST = "_PNL";
/**
* Field PROP_NAME_MAP. (value is ""_PNM"")
*/
// private static final String PROP_NAME_MAP = "_PNM";
/**
* Field PROP_COLUMN_MAP. (value is ""_PCM"")
*/
// private static final String PROP_COLUMN_MAP = "_PCM";
/**
* Field COLUMN_NAME_LIST. (value is ""_CNL"")
*/
private static final String COLUMN_NAME_LIST = "_CNL";
/**
* Field COLUMN_NAME_MAP. (value is ""_CNM"")
*/
// private static final String COLUMN_NAME_MAP = "_CNM";
/**
* Field COLUMN_PROP_MAP. (value is ""_CPM"")
*/
// private static final String COLUMN_PROP_MAP = "_CPM";
/**
* Field POSTFIX_OF_PROP_NAME_LIST. (value is ""PNL"")
*/
private static final String POSTFIX_OF_PROP_NAME_LIST = "PNL";
/**
* Field POSTFIX_OF_COLUMN_NAME_LIST. (value is ""CNL"")
*/
private static final String POSTFIX_OF_COLUMN_NAME_LIST = "CNL";
/**
* The field name of DirtyMarker implementation in a generated entity class which implements the DirtyMarker interface. (value is ""dirtyMarkerImpl"")
*/
private static final String DIRTY_MARKER_IMPL_FIELD_NAME = "dirtyMarkerImpl";
/**
* Field USUAL_TYPES.
*/
private static Set USUAL_TYPES = N.newLinkedHashSet();
static {
USUAL_TYPES.add("java.lang");
USUAL_TYPES.add("java.util");
USUAL_TYPES.add("java.util.concurrent");
USUAL_TYPES.add("java.time");
USUAL_TYPES.add("java.io");
USUAL_TYPES.add("java.nio");
USUAL_TYPES.add("java.sql");
USUAL_TYPES.add("java.net");
USUAL_TYPES.add("java.math");
USUAL_TYPES.add("javax.xml");
USUAL_TYPES.add(HBaseColumn.class.getCanonicalName());
}
private static Map JAVA_TYPE_PROP_NAME = new HashMap<>();
static {
JAVA_TYPE_PROP_NAME.put(boolean.class.getName(), "boolean_");
JAVA_TYPE_PROP_NAME.put(char.class.getName(), "char_");
JAVA_TYPE_PROP_NAME.put(byte.class.getName(), "byte_");
JAVA_TYPE_PROP_NAME.put(short.class.getName(), "short_");
JAVA_TYPE_PROP_NAME.put(int.class.getName(), "int_");
JAVA_TYPE_PROP_NAME.put(long.class.getName(), "long_");
JAVA_TYPE_PROP_NAME.put(float.class.getName(), "float_");
JAVA_TYPE_PROP_NAME.put(double.class.getName(), "double_");
JAVA_TYPE_PROP_NAME.put("class", "class_");
}
/**
* Constructor for CodeGenerator.
*/
private CodeGenerator2() {
// No instance.
}
/**
* Database 2 xml.
*
* @param database
* @param xml
* @param databaseElementAttrs
*/
static void database2Xml(final Database database, final File xml, final Map databaseElementAttrs) {
Document doc = XMLUtil.createDOMParser(true, true).newDocument();
Element databaseNode = doc.createElement(DatabaseEle.DATABASE);
database2Node(database, databaseNode, databaseElementAttrs);
doc.appendChild(databaseNode);
XMLUtil.transform(doc, xml);
}
/**
* Database 2 node.
*
* @param database
* @param node
* @param databaseElementAttrs
* @return
*/
static Element database2Node(final Database database, final Element node, final Map databaseElementAttrs) {
Document doc = node.getOwnerDocument();
Element databaseNode = null;
if (DatabaseEle.DATABASE.equals(node.getTagName())) {
databaseNode = node;
} else {
databaseNode = doc.createElement(DatabaseEle.DATABASE);
node.appendChild(databaseNode);
}
setAttributes(databaseNode, databaseElementAttrs);
databaseNode.setAttribute(DatabaseEle.NAME, database.getName());
Collection tableNames = database.getTableNameList();
for (String tableName : tableNames) {
Table table = database.getTable(tableName);
Element tableNode = table2Node(table, databaseNode);
tableNode.removeAttribute(DatabaseEle.DATABASE);
}
return databaseNode;
}
/**
* Table 2 node.
*
* @param table
* @param node
* @return
*/
static Element table2Node(final Table table, final Element node) {
Document doc = node.getOwnerDocument();
Element tableNode = null;
if (TableEle.TABLE.equals(node.getTagName())) {
tableNode = node;
} else {
tableNode = doc.createElement(TableEle.TABLE);
node.appendChild(tableNode);
}
for (Map.Entry entry : table.getAttributes().entrySet()) {
tableNode.setAttribute(entry.getKey(), entry.getValue());
}
Collection columnNames = table.getColumnNameList();
for (String columnName : columnNames) {
Element columnNode = doc.createElement(ColumnEle.COLUMN);
Column column = table.getColumn(columnName);
for (Map.Entry entry : column.getAttributes().entrySet()) {
columnNode.setAttribute(entry.getKey(), entry.getValue());
}
columnNode.setAttribute(ColumnEle.NAME, column.getName());
tableNode.appendChild(columnNode);
}
return tableNode;
}
/**
* Table 2 xml.
*
* @param table
* @param xml
*/
static void table2Xml(final Table table, final File xml) {
Document doc = XMLUtil.createDOMParser(true, true).newDocument();
Element tableNode = doc.createElement(TableEle.TABLE);
table2Node(table, tableNode);
doc.appendChild(tableNode);
XMLUtil.transform(doc, xml);
}
/**
* Database 2 entity definition xml.
*
* @param database
* @param entityDefinitionXml
*/
public static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml) {
database2EntityDefinitionXml(database, entityDefinitionXml, null);
}
/**
* Database 2 entity definition xml.
*
* @param database
* @param entityDefinitionXml
* @param entityDefAttrs
*/
public static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml, final Map entityDefAttrs) {
Map newEntityNames = new LinkedHashMap<>();
Map newPropNames = new LinkedHashMap<>();
database2EntityDefinitionXml(database, entityDefinitionXml, entityDefAttrs, newEntityNames, newPropNames, null);
}
/**
* Database 2 entity definition xml.
*
* @param database
* @param entityDefinitionXml
* @param entityDefAttrs
* @param newEntityNames
* @param newPropNames
* @param excludedeEntityPropNames
*/
public static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml, final Map entityDefAttrs,
final Map newEntityNames, final Map newPropNames, final Map> excludedeEntityPropNames) {
Map> newEntityPropNames = new LinkedHashMap<>();
Collection tables = database.getTableList();
for (Table table : tables) {
String entityName = table.getName();
if (newEntityNames.get(entityName) != null) {
entityName = newEntityNames.get(entityName);
}
newEntityPropNames.put(entityName, newPropNames);
}
database2EntityDefinitionXml(database, entityDefinitionXml, entityDefAttrs, null, newEntityNames, null, newEntityPropNames, excludedeEntityPropNames);
}
/**
* Database 2 entity definition xml.
*
* @param database
* @param entityDefinitionXml
* @param entityDefAttrs
* @param tableName2EntityName
* @param columnName2PropName
* @param excludedeEntityPropNames
*/
public static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml, final Map entityDefAttrs,
Method tableName2EntityName, final Method columnName2PropName, final Map> excludedeEntityPropNames) {
if (tableName2EntityName == null) {
tableName2EntityName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, TABLE_NAME2ENTITY_NAME, String.class);
}
Map entityColumnName2PropName = new LinkedHashMap<>();
Collection tables = database.getTableList();
for (Table table : tables) {
String entityName = (String) ClassUtil.invokeMethod(tableName2EntityName, table.getName());
entityColumnName2PropName.put(entityName, columnName2PropName);
}
database2EntityDefinitionXml(database, entityDefinitionXml, entityDefAttrs, tableName2EntityName, null, entityColumnName2PropName, null,
excludedeEntityPropNames);
}
/**
* Database 2 entity definition xml.
*
* @param database
* @param entityDefinitionXml
* @param entityDefAttrs
* @param tableName2EntityName
* @param newEntityNames
* @param entityColumnName2PropName
* @param newEntityPropNames
* @param excludedeEntityPropNames
*/
static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml, final Map entityDefAttrs,
Method tableName2EntityName, Map newEntityNames, Map entityColumnName2PropName,
Map> newEntityPropNames, Map> excludedeEntityPropNames) {
if (tableName2EntityName == null) {
tableName2EntityName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, TABLE_NAME2ENTITY_NAME, String.class);
}
if (newEntityNames == null) {
newEntityNames = new LinkedHashMap<>();
}
if (entityColumnName2PropName == null) {
entityColumnName2PropName = new LinkedHashMap<>();
}
if (newEntityPropNames == null) {
newEntityPropNames = new LinkedHashMap<>();
}
if (excludedeEntityPropNames == null) {
excludedeEntityPropNames = new LinkedHashMap<>();
}
Document doc = XMLUtil.createDOMParser(true, true).newDocument();
Element entityDefNode = doc.createElement(EntityDefEle.ENTITY_DEF);
Document existedDoc = null;
if (entityDefinitionXml.exists()) {
try {
existedDoc = XMLUtil.createDOMParser(true, true).parse(entityDefinitionXml);
} catch (Exception e) {
throw N.toRuntimeException(e);
}
NodeList nodes = existedDoc.getElementsByTagName(EntityDefEle.ENTITY_DEF);
if ((nodes != null) && (nodes.getLength() > 0)) {
NamedNodeMap attrs = nodes.item(0).getAttributes();
for (int i = 0; i < attrs.getLength(); i++) {
entityDefNode.setAttribute(attrs.item(i).getNodeName(), attrs.item(i).getNodeValue());
}
}
}
setAttributes(entityDefNode, entityDefAttrs);
String packageAttr = entityDefNode.getAttribute(EntityDefEle.PACKAGE);
if ((packageAttr == null) || (packageAttr.length() == 0)) {
entityDefNode.setAttribute(EntityDefEle.PACKAGE, database.getName());
}
Map existedEntityEleList = getEntityElementList(existedDoc);
Set entityNameList = N.newLinkedHashSet();
Collection tables = database.getTableList();
for (Table table : tables) {
String entityName = (String) ClassUtil.invokeMethod(tableName2EntityName, table.getName());
if (newEntityNames.get(entityName) != null) {
entityName = newEntityNames.get(entityName);
}
Method columnName2PropName = entityColumnName2PropName.get(entityName);
if (columnName2PropName == null) {
for (String key : entityColumnName2PropName.keySet()) {
if (entityName.matches(key)) {
columnName2PropName = entityColumnName2PropName.get(key);
}
}
}
if (columnName2PropName == null) {
columnName2PropName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, COLUMN_NAME2PROP_NAME, String.class);
}
Map newPropNames = newEntityPropNames.get(entityName);
if (newPropNames == null) {
for (String key : newEntityPropNames.keySet()) {
if (entityName.matches(key)) {
newPropNames = newEntityPropNames.get(key);
}
}
}
Set excludedPropNames = excludedeEntityPropNames.get(entityName);
if (excludedPropNames == null) {
for (String key : excludedeEntityPropNames.keySet()) {
if (entityName.matches(key)) {
excludedPropNames = excludedeEntityPropNames.get(key);
}
}
}
Element importedEntityEelement = getEntityEleByName(entityName, existedEntityEleList);
EntityDefinition entityDef = new SQLEntityDefinition(entityName, table, columnName2PropName);
entityDefinition2Node(entityDef, entityDefNode, newPropNames, excludedPropNames, importedEntityEelement);
entityNameList.add(entityName.toUpperCase());
}
for (String entityEleName : existedEntityEleList.keySet()) {
if (!entityNameList.contains(entityEleName.toUpperCase())) {
Node importedNode = doc.importNode(existedEntityEleList.get(entityEleName), true);
entityDefNode.appendChild(importedNode);
}
}
Comment comment = doc.createComment("Generated by Abacus");
doc.appendChild(comment);
doc.appendChild(entityDefNode);
XMLUtil.transform(doc, entityDefinitionXml);
}
/**
* Sets the attributes.
*
* @param element
* @param attrs
*/
static void setAttributes(final Element element, final Map attrs) {
if (attrs != null) {
for (String attrName : attrs.keySet()) {
element.setAttribute(attrName, attrs.get(attrName));
}
}
}
/**
* Entity definition 2 node.
*
* @param entityDef
* @param node
* @param newPropNames
* @param excludedPropNames
* @param importedEntityEelement
*/
static void entityDefinition2Node(final EntityDefinition entityDef, final Element node, Map newPropNames, Set excludedPropNames,
final Element importedEntityEelement) {
Document doc = node.getOwnerDocument();
Element entityNode = null;
if (newPropNames == null) {
newPropNames = new LinkedHashMap<>();
}
if (excludedPropNames == null) {
excludedPropNames = N.newLinkedHashSet();
} else {
Set tempList = N.newLinkedHashSet(excludedPropNames);
for (String propName : excludedPropNames) {
tempList.add(propName.toUpperCase());
}
excludedPropNames = tempList;
}
if (EntityEle.ENTITY.equals(node.getTagName())) {
entityNode = node;
} else {
entityNode = doc.createElement(EntityEle.ENTITY);
node.appendChild(doc.createComment(entityDef.getName()));
node.appendChild(entityNode);
}
for (Map.Entry entry : entityDef.getAttributes().entrySet()) {
entityNode.setAttribute(entry.getKey(), entry.getValue());
}
if (entityDef.getIdPropertyList().size() > 0) {
String idAttr = "";
for (Property idProp : entityDef.getIdPropertyList()) {
if (excludedPropNames.contains(idProp.getName()) || excludedPropNames.contains(idProp.getName().toUpperCase())) {
continue;
}
if (idAttr.length() > 0) {
idAttr += ", ";
}
idAttr += idProp.getName();
}
if (N.notNullOrEmpty(idAttr)) {
entityNode.setAttribute(EntityEle.ID, idAttr);
} else {
entityNode.removeAttribute(EntityEle.ID);
}
}
Element propsNode = doc.createElement(EntityEle.PROPERTIES);
Collection props = entityDef.getPropertyList();
for (Property prop : props) {
String propName = prop.getName();
if (excludedPropNames.contains(propName) || excludedPropNames.contains(propName.toUpperCase())) {
continue;
}
if ((newPropNames.get(propName) != null)) {
propName = newPropNames.get(propName);
}
Element propNode = null;
if (prop.isList()) {
propNode = doc.createElement(PropertyEle.LIST);
} else if (prop.isSet()) {
propNode = doc.createElement(PropertyEle.SET);
} else {
propNode = doc.createElement(PropertyEle.PROPERTY);
}
for (Map.Entry entry : prop.getAttributes().entrySet()) {
propNode.setAttribute(entry.getKey(), entry.getValue());
}
propNode.setAttribute(PropertyEle.NAME, propName);
propNode.setAttribute(PropertyEle.TYPE, prop.getType().name());
propNode.removeAttribute(PropertyEle.COLLECTION);
propsNode.appendChild(propNode);
}
entityNode.appendChild(propsNode);
if (importedEntityEelement != null) {
NamedNodeMap attrs = importedEntityEelement.getAttributes();
for (int i = 0; i < attrs.getLength(); i++) {
Node attr = attrs.item(i);
entityNode.setAttribute(attr.getNodeName(), attr.getNodeValue());
}
try {
List propElementList = XMLUtil.getElementsByTagName(propsNode, PropertyEle.PROPERTY);
NodeList nodeList = importedEntityEelement.getChildNodes();
for (int i = 0; i < nodeList.getLength(); i++) {
if (nodeList.item(i) instanceof Element) {
Element ele = (Element) nodeList.item(i);
if (EntityEle.PROPERTIES.equals(ele.getTagName())) {
List importedPropEleList = new ArrayList<>();
NodeList propNodeList = ele.getChildNodes();
for (int j = 0; j < propNodeList.getLength(); j++) {
if (propNodeList.item(j) instanceof Element) {
importedPropEleList.add((Element) propNodeList.item(j));
}
}
for (Element importedPropEle : importedPropEleList) {
boolean isExistedProp = false;
if (importedPropEle.getTagName().equals(PropertyEle.PROPERTY)) {
String columnName = importedPropEle.getAttribute(PropertyEle.COLUMN);
for (int k = 0; k < propElementList.size(); k++) {
Element propEle = (propElementList.get(k));
if (columnName.equalsIgnoreCase(propEle.getAttribute(PropertyEle.COLUMN))) {
NamedNodeMap nnm = importedPropEle.getAttributes();
for (int j = 0; j < nnm.getLength(); j++) {
propEle.setAttribute(nnm.item(j).getNodeName(), nnm.item(j).getNodeValue());
}
isExistedProp = true;
break;
}
}
}
if (!isExistedProp) {
Node importedNode = doc.importNode(importedPropEle, true);
propsNode.appendChild(importedNode);
}
}
} else {
// objectNode.getOwnerDocument().adoptNode(addedEle);
Node importedNode = doc.importNode(ele, true);
entityNode.appendChild(importedNode);
}
}
}
} catch (Exception e) {
throw N.toRuntimeException(e);
}
}
}
/**
* Gets the entity element list.
*
* @param doc
* @return
*/
private static Map getEntityElementList(final Document doc) {
Map result = new LinkedHashMap<>();
if (doc != null) {
List entityDefElementList = XMLUtil.getElementsByTagName(doc.getDocumentElement(), EntityDefEle.ENTITY_DEF);
List entityElementList = null;
if (entityDefElementList.size() > 0) {
entityElementList = XMLUtil.getElementsByTagName((entityDefElementList.get(0)), EntityEle.ENTITY);
} else {
entityElementList = XMLUtil.getElementsByTagName(doc.getDocumentElement(), EntityEle.ENTITY);
}
for (int i = 0; i < entityElementList.size(); i++) {
String entityName = XMLUtil.getAttribute(entityElementList.get(i), EntityEle.NAME);
result.put(entityName, entityElementList.get(i));
}
}
return result;
}
/**
* Gets the entity ele by name.
*
* @param entityName
* @param entityEleList
* @return
*/
private static Element getEntityEleByName(final String entityName, final Map entityEleList) {
Element importedElement = null;
for (String entityEleName : entityEleList.keySet()) {
if (entityName.equalsIgnoreCase(entityEleName)) {
importedElement = entityEleList.get(entityEleName);
break;
}
}
return importedElement;
}
/**
* Gets the simple prop name table class name.
*
* @param className
* @return
*/
protected static String getSimplePropNameTableClassName(final String className) {
String simpleClassName = className;
int index = className.lastIndexOf(WD._PERIOD);
if (index > -1) {
simpleClassName = className.substring(index + 1);
}
return simpleClassName;
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, null, null, true);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, propName2VarName, propName2MethodName, fluentSetMethod, null);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
* @param extendedClass
* @param implementedInterfaces
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod, final Class extendedClass,
final Class... implementedInterfaces) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, propName2VarName, propName2MethodName, fluentSetMethod, extendedClass,
N.asList(implementedInterfaces), true, true);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
* @param extendedClass
* @param implementedInterfaces
* @param generateHashEqualsMethod
* @param generateToStringMethod
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod, final Class extendedClass,
final List> implementedInterfaces, final boolean generateHashEqualsMethod, final boolean generateToStringMethod) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, propName2VarName, propName2MethodName, fluentSetMethod, extendedClass,
implementedInterfaces, generateHashEqualsMethod, generateToStringMethod, generateHashEqualsMethod, generateToStringMethod, Objects.class);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
* @param extendedClass
* @param implementedInterfaces
* @param generateHashEqualsMethod
* @param generateToStringMethod
* @param hashEqualsWithParentProperties
* @param toStringWithParentProperties
* @param utilClassForHashEqualsToString
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod, final Class extendedClass,
final List> implementedInterfaces, final boolean generateHashEqualsMethod, final boolean generateToStringMethod,
final boolean hashEqualsWithParentProperties, final boolean toStringWithParentProperties, final Class utilClassForHashEqualsToString) {
final EntityDefinitionFactory entityDefinitionFactory = SQLEntityDefinitionFactory.newInstance(domainName, entityDefinitionXml);
Map> entityExtendedClasses = null;
if (extendedClass != null) {
entityExtendedClasses = new LinkedHashMap<>();
for (EntityDefinition entityDef : entityDefinitionFactory.getDefinitionList()) {
entityExtendedClasses.put(entityDef.getName(), extendedClass);
}
}
Map>> entityImplementedInterfaces = null;
if (N.notNullOrEmpty(implementedInterfaces)) {
entityImplementedInterfaces = new LinkedHashMap<>();
for (EntityDefinition entityDef : entityDefinitionFactory.getDefinitionList()) {
entityImplementedInterfaces.put(entityDef.getName(), implementedInterfaces);
}
}
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, propName2VarName, propName2MethodName, fluentSetMethod,
entityExtendedClasses, entityImplementedInterfaces, generateHashEqualsMethod, generateToStringMethod, hashEqualsWithParentProperties,
toStringWithParentProperties, utilClassForHashEqualsToString);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
* @param entityExtendedClasses
* @param entityImplementedInterfaces
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod, final Map> entityExtendedClasses,
final Map>> entityImplementedInterfaces) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, propName2VarName, propName2MethodName, fluentSetMethod,
entityExtendedClasses, entityImplementedInterfaces, true, true);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
* @param entityExtendedClasses
* @param entityImplementedInterfaces
* @param generateHashEqualsMethod
* @param generateToStringMethod
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod, final Map> entityExtendedClasses,
final Map>> entityImplementedInterfaces, final boolean generateHashEqualsMethod, final boolean generateToStringMethod) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, propName2VarName, propName2MethodName, fluentSetMethod,
entityExtendedClasses, entityImplementedInterfaces, generateHashEqualsMethod, generateToStringMethod, generateHashEqualsMethod,
generateToStringMethod, Objects.class);
}
/**
* Entity definition xml 2 class.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod true to generate the setter methods with returning this to support fluent setter. For example: account.setFirstName("fn").setLastName("ln")...
* @param entityExtendedClasses
* @param entityImplementedInterfaces
* @param generateHashEqualsMethod
* @param generateToStringMethod
* @param hashEqualsWithParentProperties
* @param toStringWithParentProperties
* @param utilClassForHashEqualsToString is Objects.class by default. It can also be N.class or any classes else which provide the {@code hashCode/equals/toString} method.
* Or specify CodeGenerator._N or your own utility class to generate entity classes which not dependent on abacus-common.jar for Methods {@code hashCode/equals/toString}.
*/
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode,
Method propName2VarName, Method propName2MethodName, final boolean fluentSetMethod, Map> entityExtendedClasses,
Map>> entityImplementedInterfaces, final boolean generateHashEqualsMethod, final boolean generateToStringMethod,
final boolean hashEqualsWithParentProperties, final boolean toStringWithParentProperties, final Class utilClassForHashEqualsToString) {
final EntityDefinitionFactory entityDefinitionFactory = SQLEntityDefinitionFactory.newInstance(domainName, entityDefinitionXml);
final String pkgName = entityDefinitionFactory.getAttribute(EntityDefEle.PACKAGE);
if ((pkgName == null) || (pkgName.length() == 0)) {
throw new RuntimeException("the package attribute in 'entityDef' element can't be null or empty");
}
//
// if (generatePropNameTable && ((domainName == null) || (domainName.length() == 0))) {
// throw new RuntimeException("the domain attribute in 'entityDef' element can't be null or empty");
// }
if (entityMode == null) {
throw new RuntimeException("entityMode can't be null");
}
if (propName2VarName == null) {
propName2VarName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, PROP_NAME2VAR_NAME, String.class);
}
if (propName2MethodName == null) {
propName2MethodName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, PROP_NAME2METHOD_NAME, String.class);
}
if (entityExtendedClasses == null) {
entityExtendedClasses = new LinkedHashMap<>();
}
if (entityImplementedInterfaces == null) {
entityImplementedInterfaces = new LinkedHashMap<>();
}
final Class utilClass = utilClassForHashEqualsToString == null ? Objects.class : utilClassForHashEqualsToString;
final String classFilePath = ClassUtil.makePackageFolder(srcPath, pkgName);
if (_N.equals(utilClass)) {
File utilClassFile = new File(classFilePath + ClassUtil.getSimpleClassName(_N) + POSTFIX_OF_JAVA_FILE);
if (!utilClassFile.exists()) {
String sourceCode = CodeGenerator._N_STRING.replaceFirst("package com.landawn.abacus.util;",
N.isNullOrEmpty(pkgName) ? "" : "package " + pkgName + ";");
try {
IOUtil.write(utilClassFile, sourceCode);
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
}
String propNameTableClass = null;
if (entityMode != EntityMode.POJO) {
entityDefinitionXml2PropNameTable(domainName, entityDefinitionXml, srcPath, propName2VarName, utilClass.equals(Objects.class) ? _N : utilClass);
propNameTableClass = getPropNameTableClassName(entityDefinitionFactory);
}
final ExecutorService executorService = Executors.newFixedThreadPool(8);
final AtomicInteger activeThreadNum = new AtomicInteger();
for (EntityDefinition entityDef : entityDefinitionFactory.getDefinitionList()) {
if (entityDef.isSliceEntity()) {
continue;
}
final EntityDefinition entityDef2 = entityDef;
final String propNameTableClass2 = propNameTableClass;
final Method propName2VarName2 = propName2VarName;
final Method propName2MethodName2 = propName2MethodName;
final File classFile = new File(classFilePath + entityDef.getName() + POSTFIX_OF_JAVA_FILE);
final Class extendedClass = entityExtendedClasses.get(entityDef.getName());
final List> implementedInterfaces = entityImplementedInterfaces.get(entityDef.getName());
activeThreadNum.incrementAndGet();
executorService.execute(new Runnable() {
@Override
public void run() {
try {
entityDefinition2Class(entityDef2, classFile, entityMode, propNameTableClass2, propName2VarName2, propName2MethodName2, extendedClass,
implementedInterfaces, fluentSetMethod, generateHashEqualsMethod, generateToStringMethod, hashEqualsWithParentProperties,
toStringWithParentProperties, utilClass);
} finally {
activeThreadNum.decrementAndGet();
}
}
});
}
while (activeThreadNum.get() > 0) {
N.sleep(1);
}
}
/**
* Gets the prop name table class name.
*
* @param entityDefinitionFactory
* @return
*/
private static String getPropNameTableClassName(final EntityDefinitionFactory entityDefinitionFactory) {
String propNameTableClass = POSTFIX_OF_PROP_NAME_LIST;
String domainName = entityDefinitionFactory.domainName();
if (N.notNullOrEmpty(domainName)) {
propNameTableClass = StringUtil.capitalize(domainName) + propNameTableClass;
}
String pkgName = entityDefinitionFactory.getAttribute(EntityDefEle.PACKAGE);
if (pkgName != null) {
propNameTableClass = pkgName + '.' + propNameTableClass;
}
return propNameTableClass;
}
/**
* Entity definition 2 class.
*
* @param entityDef
* @param classFile
* @param entityMode
* @param propNameTableClass
* @param propName2VarName
* @param propName2MethodName
* @param extendedClass
* @param implementedInterfaces
* @param fluentSetMethod
* @param generateHashEqualsMethod
* @param generateToStringMethod
* @param hashEqualsWithParentProperties
* @param toStringWithParentProperties
* @param utilClass
*/
static void entityDefinition2Class(final EntityDefinition entityDef, final File classFile, final EntityMode entityMode, final String propNameTableClass,
final Method propName2VarName, final Method propName2MethodName, Class extendedClass, List> implementedInterfaces,
final boolean fluentSetMethod, boolean generateHashEqualsMethod, boolean generateToStringMethod, boolean hashEqualsWithParentProperties,
boolean toStringWithParentProperties, final Class utilClass) {
final String pkgName = getPackageName(entityDef);
String attr = entityDef.getAttribute("extends");
if (N.notNullOrEmpty(attr)) {
String className = attr;
try {
try {
ClassUtil.forClass(className);
} catch (Exception e) {
className = attr.indexOf('.') > 0 ? attr : pkgName + "." + attr;
}
extendedClass = ClassUtil.forClass(className);
} catch (Exception e) {
// ignore for second time run.
if (logger.isWarnEnabled()) {
logger.warn("No class found by name: " + className + ". Please run it again after the generated codes are compiled");
}
}
}
attr = entityDef.getAttribute("implements");
if (N.notNullOrEmpty(attr)) {
implementedInterfaces = new ArrayList<>();
final List strList = Splitter.with(",").trimResults().split(attr);
for (String str : strList) {
String className = str;
try {
try {
ClassUtil.forClass(className);
} catch (Exception e) {
className = str.indexOf('.') > 0 ? str : pkgName + "." + str;
}
implementedInterfaces.add(ClassUtil.forClass(className));
} catch (Exception e) {
// ignore for second time run.
if (logger.isWarnEnabled()) {
logger.warn("No class found by name: " + className + ". Please run it again after the generated codes are compiled");
}
}
}
}
if (implementedInterfaces == null) {
implementedInterfaces = new ArrayList<>();
}
String generateHashEqualsMethodAttr = entityDef.getAttribute("generateHashEqualsMethod");
if (N.notNullOrEmpty(generateHashEqualsMethodAttr)) {
generateHashEqualsMethod = Boolean.valueOf(generateHashEqualsMethodAttr);
}
String generateToStringMethodAttr = entityDef.getAttribute("generateToStringMethod");
if (N.notNullOrEmpty(generateToStringMethodAttr)) {
generateToStringMethod = Boolean.valueOf(generateToStringMethodAttr);
}
String hashEqualsWithParentPropertiesAttr = entityDef.getAttribute("hashEqualsWithParentProperties");
if (N.notNullOrEmpty(hashEqualsWithParentPropertiesAttr)) {
hashEqualsWithParentProperties = Boolean.valueOf(hashEqualsWithParentPropertiesAttr);
}
String toStringWithParentPropertiesAttr = entityDef.getAttribute("toStringWithParentProperties");
if (N.notNullOrEmpty(toStringWithParentPropertiesAttr)) {
toStringWithParentProperties = Boolean.valueOf(toStringWithParentPropertiesAttr);
}
Writer fileWrite = null;
try {
if (EntityMode.POJO == entityMode || EntityMode.POJO_WITH_PROP_NAME_TABLE == entityMode) {
// do something.
} else if (EntityMode.EXTEND_DIRTY_MARKER == entityMode) {
if (extendedClass == null) {
extendedClass = AbstractDirtyMarker.class;
} else {
if (!AbstractDirtyMarker.class.isAssignableFrom(extendedClass)) {
throw new RuntimeException(
"the class: " + extendedClass.getCanonicalName() + " must extend " + AbstractDirtyMarker.class.getCanonicalName());
}
}
} else if (EntityMode.IMPL_DIRTY_MARKER == entityMode) {
if (extendedClass == null) {
extendedClass = DirtyMarkerImpl.class;
}
if (implementedInterfaces.size() == 0) {
implementedInterfaces.add(DirtyMarker.class);
}
}
Map> annotationLinesMapper = new LinkedHashMap<>();
Set> annotationImportClasses = N.newLinkedHashSet();
if (classFile.exists()) {
List lines = IOUtil.readAllLines(classFile);
List annotationLines = null;
for (String line : lines) {
if (line.startsWith("import")) {
String str = line.trim().substring(6);
str = str.trim();
str = str.substring(0, str.length() - 1);
try {
Class cls = ClassUtil.forClass(str);
if (cls.isAnnotation()) {
annotationImportClasses.add(cls);
}
} catch (Exception e) {
// ignore.
}
} else if (N.notNullOrEmpty(line) && line.length() > 4) {
boolean isAnnotationLine = false;
for (int i = 0, len = line.length(); i < len; i++) {
char ch = line.charAt(i);
if (ch == ' ' || ch == '\t') {
continue;
}
if (ch == '@') {
isAnnotationLine = true;
}
break;
}
if (isAnnotationLine) {
if (annotationLines == null) {
annotationLines = new ArrayList<>();
}
annotationLines.add(line);
} else if (N.notNullOrEmpty(annotationLines) && line.trim().length() > 0) {
annotationLinesMapper.put(line, annotationLines);
annotationLines = null;
}
}
}
} else {
classFile.createNewFile();
}
fileWrite = newFileWriter(classFile);
writeFileHead(fileWrite);
writePackageName(pkgName, fileWrite);
final Map> importedClasses = writeImport(entityDef, pkgName, propNameTableClass, entityMode, extendedClass, implementedInterfaces,
fileWrite, generateHashEqualsMethod, generateToStringMethod, utilClass, annotationImportClasses);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
writeClassComment(fileWrite);
writeClass(entityDef, pkgName, propNameTableClass, propName2VarName, propName2MethodName, entityMode, extendedClass, implementedInterfaces, "",
fileWrite, generateHashEqualsMethod, generateToStringMethod, hashEqualsWithParentProperties, toStringWithParentProperties, utilClass,
fluentSetMethod, importedClasses);
fileWrite.flush();
if (N.notNullOrEmpty(annotationLinesMapper)) {
IOUtil.close(fileWrite);
List lines = IOUtil.readAllLines(classFile);
List newLines = new ArrayList<>(lines.size() + 64);
for (String line : lines) {
List annotationLines = annotationLinesMapper.get(line);
if (N.notNullOrEmpty(annotationLines)) {
if (annotationLines.size() == 2) {
N.println(annotationLines);
}
for (int len = newLines.size(), i = len - 1; i > N.max(0, len - 5); i--) {
String str = newLines.get(i);
if (str.indexOf('@') > 0) {
if (annotationLines.size() > 0 && !annotationLines.remove(str)) {
String annoLine = annotationLines.iterator().next();
String prefixSpaces = annoLine.substring(0, annoLine.indexOf('@'));
str = prefixSpaces + str.trim();
annotationLines.remove(str);
}
} else if (str.trim().length() > 0) {
break;
}
}
if (N.notNullOrEmpty(annotationLines)) {
// String prevLine = newLines.get(newLines.size() - 1).trim();
//
// if (prevLine.length() > 0 && (prevLine.charAt(0) != '/' && prevLine.charAt(0) != '*')) {
// newLines.add(N.EMPTY_STRING);
// }
newLines.addAll(annotationLines);
}
}
newLines.add(line);
}
fileWrite = newFileWriter(classFile);
IOUtil.writeLines(fileWrite, newLines);
fileWrite.flush();
}
} catch (Exception e) {
throw N.toRuntimeException(e);
} finally {
IOUtil.close(fileWrite);
}
}
/**
* Entity definition xml 2 prop name table.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
*/
public static void entityDefinitionXml2PropNameTable(final String domainName, final File entityDefinitionXml, final String srcPath) {
entityDefinitionXml2PropNameTable(domainName, entityDefinitionXml, srcPath, null, _N);
}
/**
* Entity definition xml 2 prop name table.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param propName2VarName
* @param utilClass specify CodeGenerator._N or your own utility class to generate entity classes which not dependent on abacus-common.jar
*/
public static void entityDefinitionXml2PropNameTable(String domainName, final File entityDefinitionXml, final String srcPath, Method propName2VarName,
Class utilClass) {
if (propName2VarName == null) {
propName2VarName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, PROP_NAME2VAR_NAME, String.class);
}
if (utilClass == null) {
utilClass = _N;
}
domainName = N.nullToEmpty(domainName);
EntityDefinitionFactory entityDefinitionFactory = SQLEntityDefinitionFactory.newInstance(domainName, entityDefinitionXml);
String pkgName = entityDefinitionFactory.getAttribute(EntityDefEle.PACKAGE);
String classFilePath = ClassUtil.makePackageFolder(srcPath, pkgName);
// if (utilClass.equals(_N)) {
// File utilClassFile = new File(classFilePath + ClassUtil.getSimpleClassName(_N) + POSTFIX_OF_JAVA_FILE);
// if (!utilClassFile.exists()) {
// String sourceCode = _N_STRING.replaceFirst("com.landawn.abacus.util", pkgName);
// IOUtil.write(utilClassFile, sourceCode);
// }
// }
String propNameTableClassName = getSimplePropNameTableClassName(getPropNameTableClassName(entityDefinitionFactory));
File propNameDefClassFile = new File(classFilePath + propNameTableClassName + POSTFIX_OF_JAVA_FILE);
entityDefinition2PropNameTable(entityDefinitionFactory, propNameDefClassFile, pkgName, propName2VarName);
}
/**
* Entity definition 2 prop name table.
*
* @param entityDefinitionFactory
* @param propNameDefClassFile
* @param pkgName
* @param propName2VarName
*/
static void entityDefinition2PropNameTable(final EntityDefinitionFactory entityDefinitionFactory, final File propNameDefClassFile, final String pkgName,
final Method propName2VarName) {
if (propNameDefClassFile.exists()) {
propNameDefClassFile.delete();
}
Writer fileWrite = null;
try {
propNameDefClassFile.createNewFile();
fileWrite = newFileWriter(propNameDefClassFile);
writeFileHead(fileWrite);
writePackageName(pkgName, fileWrite);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write("import " + Arrays.class.getCanonicalName() + ";" + IOUtil.LINE_SEPARATOR);
fileWrite.write("import " + Collections.class.getCanonicalName() + ";" + IOUtil.LINE_SEPARATOR);
fileWrite.write("import " + List.class.getCanonicalName() + ";" + IOUtil.LINE_SEPARATOR);
// if (utilClass.equals(N.class)) {
// fileWrite.write("import " + Map.class.getCanonicalName() + ";" + N.LINE_SEPARATOR);
// fileWrite.write("import " + BiMap.class.getCanonicalName() + ";" + N.LINE_SEPARATOR);
// }
// fileWrite.write(N.LINE_SEPARATOR);
// String utilClassName = ClassUtil.getCanonicalClassName(utilClass);
// if (pkgName.equals(utilClassName.substring(0, utilClassName.lastIndexOf('.'))) || utilClass.equals(_N)) {
// // ignore
// } else {
// fileWrite.write("import " + ClassUtil.getCanonicalClassName(utilClass) + ";" + N.LINE_SEPARATOR);
// fileWrite.write(N.LINE_SEPARATOR);
// }
writeClassComment(fileWrite);
String clazzName = propNameDefClassFile.getName().substring(0, propNameDefClassFile.getName().indexOf(_PERIOD));
fileWrite.write("public interface " + clazzName + " {");
String headSpace = "";
String domainName = entityDefinitionFactory.domainName();
writeDomainPropNameClass(domainName, fileWrite, headSpace);
EntityDefinition[] entityDefinitions = entityDefinitionFactory.getDefinitionList()
.toArray(new EntityDefinition[entityDefinitionFactory.getDefinitionList().size()]);
writePropNameTable(entityDefinitions, propName2VarName, headSpace, fileWrite);
// -------- write global domain property name.
Map allPropNamesMap = new LinkedHashMap<>();
Map allPropColumnNameMap = new LinkedHashMap<>();
for (EntityDefinition entityDef : entityDefinitions) {
if (entityDef.isSliceEntity()) {
continue;
}
for (Property prop : entityDef.getPropertyList()) {
allPropNamesMap.put(prop.getName().toUpperCase(), prop.getName());
if (prop.getColumnType() == ColumnType.TABLE_COLUMN) {
allPropColumnNameMap.put(prop.getName().toUpperCase(), prop.getColumnName());
}
}
}
List keys = new ArrayList<>(allPropNamesMap.keySet());
N.sort(keys);
for (String key : keys) {
String propName = allPropNamesMap.get(key);
String propVarName = ClassUtil.invokeMethod(propName2VarName, propName);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(" public static final String " + propVarName + " = \"" + propName + "\".intern();");
fileWrite.write(IOUtil.LINE_SEPARATOR);
}
// if (utilClass.equals(N.class)) {
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(" /**" + N.LINE_SEPARATOR);
// fileWrite.write(" * Mapping of property name to column name " + N.LINE_SEPARATOR);
// fileWrite.write(" */" + N.LINE_SEPARATOR);
// fileWrite.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// fileWrite.write(" public static final BiMap " + PROP_COLUMN_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map) " + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(");
//
// int i = 0;
// for (String key : keys) {
// String propName = allPropNamesMap.get(key);
// String propVarName = N.invokeMethod(propName2VarName, propName);
// String columnName = allPropColumnNameMap.get(key);
//
// if (N.isNullOrEmpty(columnName)) {
// continue;
// }
//
// if (i++ > 0) {
// fileWrite.write(", ");
// }
//
// fileWrite.write(propVarName + ", \"" + columnName + "\".intern()");
// }
//
// fileWrite.write("));" + N.LINE_SEPARATOR);
// }
fileWrite.write("}" + IOUtil.LINE_SEPARATOR);
fileWrite.flush();
} catch (IOException e) {
throw new UncheckedIOException(e);
} finally {
IOUtil.close(fileWrite);
}
}
/**
* Write prop name table.
*
* @param entityDefs
* @param propName2VarName
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writePropNameTable(final EntityDefinition[] entityDefs, final Method propName2VarName, final String headSpace, final Writer fileWrite)
throws IOException {
for (EntityDefinition entityDef : entityDefs) {
if (entityDef.isSliceEntity()) {
continue;
}
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public static interface " + entityDef.getName() + POSTFIX_OF_PROP_NAME_LIST + " {");
writePropNameField(entityDef, propName2VarName, headSpace + " ", fileWrite);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// write slice entities
for (EntityDefinition hed : entityDef.getSliceEntityList()) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public static interface " + hed.getName() + POSTFIX_OF_PROP_NAME_LIST + " {");
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " * Name of \"" + hed.getName() + "\" entity. " + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " */" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public static final String " + ENTITY_NAME_VAR + " = \"" + hed.getName() + "\";" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
}
/**
* Write prop name field.
*
* @param entityDef
* @param propName2VarName
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writePropNameField(final EntityDefinition entityDef, final Method propName2VarName, final String headSpace, final Writer fileWrite)
throws IOException {
final StringBuilder sb = Objectory.createStringBuilder();
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Name of \"" + entityDef.getName() + "\" entity. " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " public static final String " + ENTITY_NAME_VAR + " = \"" + entityDef.getName() + "\".intern();" + IOUtil.LINE_SEPARATOR);
String _propNameList = "";
String _columnNameList = "";
// String _propNameMap = "";
// String _propColumnNameMap = "";
// String _columnNameMap = "";
for (Property prop : entityDef.getPropertyList()) {
String propVarName = ClassUtil.invokeMethod(propName2VarName, prop.getName());
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Name of \"" + prop.getName() + "\" property. " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * type: " + prop.getType().name() + ". " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * column: \"" + prop.getAttribute(PropertyEle.COLUMN) + "\". " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
// sb.append(headSpace + " public static final String " + propVarName + " = \"" + prop.getName() + "\".intern();" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final String _" + propVarName + " = (_ + \"." + prop.getName() + "\").intern();" + N.LINE_SEPARATOR);
sb.append(headSpace + " public static final String " + propVarName + " = (" + ENTITY_NAME_VAR + " + \"." + prop.getName() + "\").intern();"
+ IOUtil.LINE_SEPARATOR);
_propNameList += (", " + propVarName);
// _propNameMap += (", " + propVarName + ", \"" + prop.getName() + "\".intern()");
if (prop.getColumnType() == ColumnType.TABLE_COLUMN) {
String columnName = "\"" + entityDef.getTableName() + "." + prop.getColumnName() + "\".intern()";
_columnNameList += (", " + columnName + "");
// _propColumnNameMap += (", " + propVarName + ", " + columnName);
// _columnNameMap += (", " + columnName + ", \"" + prop.getColumnName() + "\".intern()");
}
}
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Immutable property name list" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " public static final List " + PROP_NAME_LIST + " = Collections.unmodifiableList(Arrays.asList("
+ _propNameList.substring(2) + "));" + IOUtil.LINE_SEPARATOR);
// if (utilClass.equals(N.class)) {
// sb.append(N.LINE_SEPARATOR);
// sb.append(headSpace + " /**" + N.LINE_SEPARATOR);
// sb.append(headSpace + " * Immutable property name to entityName.propName mapper" + N.LINE_SEPARATOR);
// sb.append(headSpace + " */" + N.LINE_SEPARATOR);
// sb.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final BiMap " + PROP_NAME_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map)" + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(" + _propNameMap.substring(2) + "));"
// + N.LINE_SEPARATOR);
//
// sb.append(N.LINE_SEPARATOR);
// sb.append(headSpace + " /**" + N.LINE_SEPARATOR);
// sb.append(headSpace + " * Immutable property name to column name mapper" + N.LINE_SEPARATOR);
// sb.append(headSpace + " */" + N.LINE_SEPARATOR);
// sb.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final BiMap " + PROP_COLUMN_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map)" + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(" + _propColumnNameMap.substring(2) + "));"
// + N.LINE_SEPARATOR);
// }
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Immutable column name list" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " public static final List " + COLUMN_NAME_LIST + " = Collections.unmodifiableList(Arrays.asList("
+ _columnNameList.substring(2) + "));" + IOUtil.LINE_SEPARATOR);
// if (utilClass.equals(N.class)) {
// sb.append(N.LINE_SEPARATOR);
// sb.append(headSpace + " /**" + N.LINE_SEPARATOR);
// sb.append(headSpace + " * Immutable column name to tableName.columnName mapper" + N.LINE_SEPARATOR);
// sb.append(headSpace + " */" + N.LINE_SEPARATOR);
// sb.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final BiMap " + COLUMN_NAME_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map)" + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(" + _columnNameMap.substring(2) + "));"
// + N.LINE_SEPARATOR);
// }
fileWrite.write(sb.toString());
Objectory.recycle(sb);
}
/**
* Entity definition xml 2 column name table.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
*/
public static void entityDefinitionXml2ColumnNameTable(final String domainName, final File entityDefinitionXml, final String srcPath) {
entityDefinitionXml2ColumnNameTable(domainName, entityDefinitionXml, srcPath, null, _N);
}
/**
* Entity definition xml 2 column name table.
*
* @param domainName
* @param entityDefinitionXml
* @param srcPath
* @param columnName2VarName
* @param utilClass specify CodeGenerator._N or your own utility class to generate entity classes which not dependent on abacus-common.jar
*/
public static void entityDefinitionXml2ColumnNameTable(final String domainName, final File entityDefinitionXml, final String srcPath,
Method columnName2VarName, Class utilClass) {
if (columnName2VarName == null) {
columnName2VarName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, PROP_NAME2VAR_NAME, String.class);
}
if (utilClass == null) {
utilClass = _N;
}
final EntityDefinitionFactory entityDefinitionFactory = SQLEntityDefinitionFactory.newInstance(domainName, entityDefinitionXml);
final Collection entityDefs = entityDefinitionFactory.getDefinitionList();
final String pkgName = entityDefinitionFactory.getAttribute(EntityDefEle.PACKAGE);
final String classFilePath = ClassUtil.makePackageFolder(srcPath, pkgName);
// if (utilClass.equals(_N)) {
// File utilClassFile = new File(classFilePath + ClassUtil.getSimpleClassName(_N) + POSTFIX_OF_JAVA_FILE);
// if (!utilClassFile.exists()) {
// String sourceCode = _N_STRING.replaceFirst("com.landawn.abacus.util", pkgName);
// IOUtil.write(utilClassFile, sourceCode);
// }
// }
String columnNameTableClassName = POSTFIX_OF_COLUMN_NAME_LIST;
if (N.notNullOrEmpty(domainName)) {
columnNameTableClassName = StringUtil.capitalize(domainName) + columnNameTableClassName;
}
File columnNameTableClassFile = new File(classFilePath + columnNameTableClassName + POSTFIX_OF_JAVA_FILE);
if (columnNameTableClassFile.exists()) {
columnNameTableClassFile.delete();
}
Writer fileWrite = null;
try {
columnNameTableClassFile.createNewFile();
fileWrite = newFileWriter(columnNameTableClassFile);
writeFileHead(fileWrite);
writePackageName(pkgName, fileWrite);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write("import " + Arrays.class.getCanonicalName() + ";" + IOUtil.LINE_SEPARATOR);
fileWrite.write("import " + Collections.class.getCanonicalName() + ";" + IOUtil.LINE_SEPARATOR);
fileWrite.write("import " + List.class.getCanonicalName() + ";" + IOUtil.LINE_SEPARATOR);
// if (utilClass.equals(N.class)) {
// fileWrite.write("import " + Map.class.getCanonicalName() + ";" + N.LINE_SEPARATOR);
// fileWrite.write("import " + BiMap.class.getCanonicalName() + ";" + N.LINE_SEPARATOR);
// }
// fileWrite.write(N.LINE_SEPARATOR);
// String utilClassName = ClassUtil.getCanonicalClassName(utilClass);
// if (pkgName.equals(utilClassName.substring(0, utilClassName.lastIndexOf('.'))) || utilClass.equals(_N)) {
// // ignore
// } else {
// fileWrite.write("import " + ClassUtil.getCanonicalClassName(utilClass) + ";" + N.LINE_SEPARATOR);
// fileWrite.write(N.LINE_SEPARATOR);
// }
writeClassComment(fileWrite);
String clazzName = columnNameTableClassFile.getName().substring(0, columnNameTableClassFile.getName().indexOf(_PERIOD));
fileWrite.write("public interface " + clazzName + " {");
String headSpace = "";
writeDomainPropNameClass(domainName, fileWrite, headSpace);
writeColumnNameTable(entityDefs, columnName2VarName, headSpace, fileWrite);
// -------- write global domain property name.
Map allColumnNamesMap = new LinkedHashMap<>();
Map allColumnPropNameMap = new LinkedHashMap<>();
for (EntityDefinition entityDef : entityDefs) {
for (Property prop : entityDef.getPropertyList()) {
if (prop.getColumnType() == ColumnType.ENTITY || N.isNullOrEmpty(prop.getColumnName())) {
continue;
}
String columnName = prop.getColumnName();
allColumnNamesMap.put(columnName.toUpperCase(), columnName);
allColumnPropNameMap.put(columnName.toUpperCase(), prop.getName());
}
}
List keys = new ArrayList<>(allColumnNamesMap.keySet());
N.sort(keys);
for (String key : keys) {
String columnName = allColumnNamesMap.get(key);
String columnVarName = ClassUtil.invokeMethod(columnName2VarName, columnName);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(" public static final String " + columnVarName + " = \"" + columnName + "\".intern();");
fileWrite.write(IOUtil.LINE_SEPARATOR);
}
// if (utilClass.equals(N.class)) {
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(" /**" + N.LINE_SEPARATOR);
// fileWrite.write(" * Mapping of column name to property name " + N.LINE_SEPARATOR);
// fileWrite.write(" */" + N.LINE_SEPARATOR);
// fileWrite.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// fileWrite.write(" public static final BiMap " + COLUMN_PROP_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map) " + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(");
//
// int i = 0;
// for (String key : keys) {
// String columnName = allColumnNamesMap.get(key);
// String columnVarName = N.invokeMethod(columnName2VarName, columnName);
// String propName = allColumnPropNameMap.get(key);
//
// if (i++ > 0) {
// fileWrite.write(", ");
// }
//
// fileWrite.write(columnVarName + ", \"" + propName + "\".intern()");
// }
//
// fileWrite.write("));" + N.LINE_SEPARATOR);
// }
fileWrite.write("}" + IOUtil.LINE_SEPARATOR);
fileWrite.flush();
} catch (IOException e) {
throw new UncheckedIOException(e);
} finally {
IOUtil.close(fileWrite);
}
}
/**
* Write column name table.
*
* @param entityDefs
* @param columnName2VarName
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeColumnNameTable(final Collection entityDefs, final Method columnName2VarName, final String headSpace,
final Writer fileWrite) throws IOException {
for (EntityDefinition entityDef : entityDefs) {
if (entityDef.isSliceEntity()) {
continue;
}
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public static interface " + entityDef.getName() + POSTFIX_OF_COLUMN_NAME_LIST + " {");
writeColumnNameField(entityDef, columnName2VarName, headSpace + " ", fileWrite);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
/**
* Write column name field.
*
* @param entityDef
* @param columnName2VarName
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeColumnNameField(final EntityDefinition entityDef, final Method columnName2VarName, final String headSpace, final Writer fileWrite)
throws IOException {
final StringBuilder sb = Objectory.createStringBuilder();
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Name of \"" + entityDef.getTableName() + "\" table. " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
sb.append(
headSpace + " public static final String " + ENTITY_NAME_VAR + " = \"" + entityDef.getTableName() + "\".intern();" + IOUtil.LINE_SEPARATOR);
String _columnNameList = "";
String _propNameList = "";
// String _columnNameMap = "";
// String _columnPropNameMap = "";
// String _propNameMap = "";
for (Property prop : entityDef.getPropertyList()) {
if (prop.getColumnType().isEntity()) {
continue;
}
String columnName = prop.getColumnName();
if (N.isNullOrEmpty(columnName)) {
continue;
}
String columnVarName = ClassUtil.invokeMethod(columnName2VarName, columnName);
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Name of \"" + columnName + "\" column. " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Java type: " + prop.getType().name() + ". " + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
// sb.append(headSpace + " public static final String " + propVarName + " = \"" + columnName + "\".intern();" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final String _" + propVarName + " = (_ + \"." + columnName + "\").intern();" + N.LINE_SEPARATOR);
sb.append(headSpace + " public static final String " + columnVarName + " = (" + ENTITY_NAME_VAR + " + \"." + columnName + "\").intern();"
+ IOUtil.LINE_SEPARATOR);
_columnNameList += (", " + columnVarName);
String propName = "\"" + entityDef.getName() + "." + prop.getName() + "\".intern()";
_propNameList += (", " + propName);
// _columnNameMap += (", " + columnVarName + ", \"" + columnName + "\".intern()");
// _columnPropNameMap += (", " + columnVarName + ", " + propName);
// _propNameMap += (", " + propName + ", \"" + prop.getName() + "\".intern()");
}
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Immutable column name list" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " public static final List " + COLUMN_NAME_LIST + " = Collections.unmodifiableList(Arrays.asList("
+ _columnNameList.substring(2) + "));" + IOUtil.LINE_SEPARATOR);
// if (utilClass.equals(N.class)) {
// sb.append(N.LINE_SEPARATOR);
// sb.append(headSpace + " /**" + N.LINE_SEPARATOR);
// sb.append(headSpace + " * Immutable column name to tableName.columnName mapper" + N.LINE_SEPARATOR);
// sb.append(headSpace + " */" + N.LINE_SEPARATOR);
// sb.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final BiMap " + COLUMN_NAME_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map)" + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(" + _columnNameMap.substring(2) + "));"
// + N.LINE_SEPARATOR);
//
// sb.append(N.LINE_SEPARATOR);
// sb.append(headSpace + " /**" + N.LINE_SEPARATOR);
// sb.append(headSpace + " * Immutable column name to property name mapper" + N.LINE_SEPARATOR);
// sb.append(headSpace + " */" + N.LINE_SEPARATOR);
// sb.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final BiMap " + COLUMN_PROP_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map)" + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(" + _columnPropNameMap.substring(2) + "));"
// + N.LINE_SEPARATOR);
// }
sb.append(IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " /**" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " * Immutable property name list" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " */" + IOUtil.LINE_SEPARATOR);
sb.append(headSpace + " public static final List " + PROP_NAME_LIST + " = Collections.unmodifiableList(Arrays.asList("
+ _propNameList.substring(2) + "));" + IOUtil.LINE_SEPARATOR);
// if (utilClass.equals(N.class)) {
// sb.append(N.LINE_SEPARATOR);
// sb.append(headSpace + " /**" + N.LINE_SEPARATOR);
// sb.append(headSpace + " * Immutable property name to entityName.propName mapper" + N.LINE_SEPARATOR);
// sb.append(headSpace + " */" + N.LINE_SEPARATOR);
// sb.append(headSpace + " @SuppressWarnings(\"rawtypes\")" + N.LINE_SEPARATOR);
// sb.append(headSpace + " public static final BiMap " + PROP_NAME_MAP + " = " + N.getSimpleClassName(utilClass)
// + ".newImmutableBiMapForInterface((Map)" + N.getSimpleClassName(utilClass) + ".asLinkedHashMap(" + _propNameMap.substring(2) + "));"
// + N.LINE_SEPARATOR);
// }
fileWrite.write(sb.toString());
Objectory.recycle(sb);
}
/**
* Generate SQL mapper id table.
*
* @param filePath
* @param srcPath
* @param pkgName
* @param className
* @param id2VarName
*/
public static void generateSQLMapperIdTable(final String filePath, final String srcPath, final String pkgName, final String className,
final Method id2VarName) {
generateSQLMapperIdTable(SQLMapper.fromFile(filePath), srcPath, pkgName, className, id2VarName);
}
/**
* Generate SQL mapper id table.
*
* @param sqlMapper
* @param srcPath
* @param pkgName
* @param className
* @param id2VarName
*/
public static void generateSQLMapperIdTable(final SQLMapper sqlMapper, final String srcPath, final String pkgName, final String className,
Method id2VarName) {
String classFilePath = ClassUtil.makePackageFolder(srcPath, pkgName);
File sqlMapperTableClassFile = new File(classFilePath + className + POSTFIX_OF_JAVA_FILE);
if (sqlMapperTableClassFile.exists()) {
sqlMapperTableClassFile.delete();
}
if (id2VarName == null) {
id2VarName = ClassUtil.getDeclaredMethod(CodeGenerator2.class, PROP_NAME2VAR_NAME, String.class);
}
Writer fileWrite = null;
try {
sqlMapperTableClassFile.createNewFile();
fileWrite = newFileWriter(sqlMapperTableClassFile);
writeFileHead(fileWrite);
writePackageName(pkgName, fileWrite);
fileWrite.write(IOUtil.LINE_SEPARATOR);
writeClassComment(fileWrite);
fileWrite.write("public final class " + className + " {");
fileWrite.write(IOUtil.LINE_SEPARATOR);
for (String id : sqlMapper.keySet()) {
String propVarName = ClassUtil.invokeMethod(id2VarName, id);
fileWrite.write(" public static final String " + propVarName + " = \"" + id + "\";" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write("}" + IOUtil.LINE_SEPARATOR);
fileWrite.flush();
} catch (IOException e) {
throw new UncheckedIOException(e);
} finally {
IOUtil.close(fileWrite);
}
}
/**
*
* @param srcDir
* @param packageName
* @param className
* @param fields
*/
public static void generateEntity(File srcDir, String packageName, String className, Map fields) {
generateEntity(srcDir, packageName, className, fields, false, false, false, Objects.class);
}
/**
*
* @param srcDir
* @param packageName
* @param className
* @param fields key is field name, value is field type, which must be Class, Type or the name of them.
* It must be LinkedHashMap.
* @param constructor
* @param copyMethod
* @param fluentSetter
* @param utilClassForHashEqualsToString is Objects.class by default. It can also be N.class or any classes else which provide the {@code hashCode/equals/toString} method.
* Or specify CodeGenerator._N or your own utility class to generate entity classes which not dependent on abacus-common.jar for Methods {@code hashCode/equals/toString}.
*/
public static void generateEntity(File srcDir, String packageName, String className, Map fields, final boolean constructor,
final boolean copyMethod, final boolean fluentSetter, final Class utilClassForHashEqualsToString) {
generateEntity(srcDir, packageName, className, fields, constructor, copyMethod, fluentSetter, null, null, null, null, utilClassForHashEqualsToString);
}
/**
*
* @param srcDir
* @param packageName
* @param className
* @param fields key is field name, value is field type, which must be Class, Type or the name of them.
* It must be LinkedHashMap.
* @param constructor
* @param copyMethod
* @param fluentSetter
* @param fieldName2MethodName
* @param parentClass
* @param parentPropertyModeForHashEquals
* @param parentPropertyModeForToString
* @param utilClassForHashEqualsToString is Objects.class by default. It can also be N.class or any classes else which provide the {@code hashCode/equals/toString} method.
* Or specify CodeGenerator._N or your own utility class to generate entity classes which not dependent on abacus-common.jar for Methods {@code hashCode/equals/toString}.
*/
public static void generateEntity(File srcDir, String packageName, String className, Map fields, final boolean constructor,
final boolean copyMethod, final boolean fluentSetter, final Map fieldName2MethodName, Class parentClass,
final ParentPropertyMode parentPropertyModeForHashEquals, final ParentPropertyMode parentPropertyModeForToString,
final Class utilClassForHashEqualsToString) {
if (fields instanceof LinkedHashMap == false) {
throw new IllegalArgumentException("'fields has to be LinkedHashMap to keep the order");
}
if (N.isNullOrEmpty(fields)) {
return;
}
final Class utilClass = utilClassForHashEqualsToString == null ? Objects.class : utilClassForHashEqualsToString;
final Map> fieldTypes = N.newLinkedHashMap(fields.size());
Type type = null;
for (Map.Entry entry : fields.entrySet()) {
if (entry.getValue() instanceof Type) {
type = (Type) entry.getValue();
} else if (entry.getValue() instanceof Class) {
type = N.typeOf((Class) entry.getValue());
} else if (entry.getValue() instanceof String) {
String typeName = (String) entry.getValue();
final TypeAttrParser attrs = TypeAttrParser.parse(typeName);
if (N.notNullOrEmpty(attrs.getTypeParameters())) {
if (attrs.getTypeParameters().length == 1) {
Class eleClass = null;
try {
eleClass = ClassUtil.forClass(attrs.getTypeParameters()[0]);
} catch (Exception e) {
// ignore.
}
if (eleClass == null && N.notNullOrEmpty(packageName)) {
try {
eleClass = ClassUtil.forClass(packageName + "." + attrs.getTypeParameters()[0]);
} catch (Exception e) {
// ignore.
}
if (eleClass != null) {
typeName = attrs.getClassName() + "<" + eleClass.getCanonicalName() + ">";
}
}
} else if (attrs.getTypeParameters().length == 2) {
String keyTypeName = attrs.getTypeParameters()[0];
String valueTypeName = attrs.getTypeParameters()[1];
Class keyClass = null;
Class valueClass = null;
try {
keyClass = ClassUtil.forClass(attrs.getTypeParameters()[0]);
} catch (Exception e) {
// ignore.
}
try {
valueClass = ClassUtil.forClass(attrs.getTypeParameters()[1]);
} catch (Exception e) {
// ignore.
}
if ((keyClass == null || valueClass == null) && N.notNullOrEmpty(packageName)) {
if (keyClass == null) {
try {
keyClass = ClassUtil.forClass(packageName + "." + attrs.getTypeParameters()[0]);
if (keyClass != null) {
keyTypeName = keyClass.getCanonicalName();
}
} catch (Exception e) {
// ignore.
}
}
if (valueClass == null) {
try {
valueClass = ClassUtil.forClass(packageName + "." + attrs.getTypeParameters()[1]);
if (valueClass != null) {
valueTypeName = valueClass.getCanonicalName();
}
} catch (Exception e) {
// ignore.
}
}
if (keyClass != null && valueClass != null) {
typeName = attrs.getClassName() + "<" + keyTypeName + ", " + valueTypeName + ">";
}
}
}
}
type = N.typeOf(typeName);
} else {
throw new IllegalArgumentException("Only Class, Type or the name of them are supported");
}
fieldTypes.put(entry.getKey(), type);
}
final File dirFile = new File(srcDir.getAbsolutePath()
+ (N.isNullOrEmpty(packageName) ? "" : IOUtil.FILE_SEPARATOR + StringUtil.replaceAll(packageName, ".", IOUtil.FILE_SEPARATOR)));
if (dirFile.exists() == false) {
dirFile.mkdirs();
}
if (_N.equals(utilClass)) {
File utilClassFile = new File(dirFile.getAbsolutePath() + IOUtil.FILE_SEPARATOR + ClassUtil.getSimpleClassName(_N) + POSTFIX_OF_JAVA_FILE);
if (!utilClassFile.exists()) {
String sourceCode = CodeGenerator._N_STRING.replaceFirst("package com.landawn.abacus.util;",
N.isNullOrEmpty(packageName) ? "" : "package " + packageName + ";");
try {
IOUtil.write(utilClassFile, sourceCode);
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
}
final File classFile = new File(dirFile.getAbsolutePath() + IOUtil.FILE_SEPARATOR + className + ".java");
IOUtil.createIfNotExists(classFile);
final BufferedWriter writer = IOUtil.newBufferedWriter(classFile);
try {
try {
IOUtil.writeLine(writer, "package " + packageName + ";");
} catch (IOException e) {
throw new UncheckedIOException(e);
}
// IOUtil.writeLine(writer, N.EMPTY_STRING);
// write imports
final Multimap> packClasses = new Multimap<>(SortedMap.class, Set.class);
for (Map.Entry> entry : fieldTypes.entrySet()) {
readPackageTypes(entry.getValue(), packageName, packClasses);
}
if (parentClass != null) {
Package pkg = parentClass.getPackage();
if (pkg == null || pkg.getName().equals(packageName) || parentClass.getCanonicalName().startsWith("java.lang.")) {
// ignore
} else {
packClasses.put(pkg.getName(), parentClass.getCanonicalName());
}
}
if (!ClassUtil.getPackageName(utilClass).equals(packageName) && !utilClass.equals(_N)) {
packClasses.put(ClassUtil.getPackageName(utilClass), utilClass.getCanonicalName());
}
// if (N.notNullOrEmpty(fields) && ClassUtil.getPackageName(com.landawn.abacus.annotation.Type.class).equals(packageName) == false) {
// packClasses.put(ClassUtil.getPackageName(com.landawn.abacus.annotation.Type.class),
// com.landawn.abacus.annotation.Type.class.getCanonicalName());
// }
final Map> importedClasses = new LinkedHashMap<>();
try {
for (Set clsNames : packClasses.values()) {
for (String clsName : clsNames) {
// IOUtil.writeLine(writer, "import " + clsName + ";");
Class cls = ClassUtil.forClass(clsName);
writeClassImport(writer, cls, importedClasses);
}
// IOUtil.writeLine(writer, N.EMPTY_STRING);
}
} catch (IOException e) {
throw new UncheckedIOException(e);
}
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, N.EMPTY_STRING);
final String iden = " ";
IOUtil.writeLine(writer, "/**");
IOUtil.writeLine(writer, " * Generated by Abacus.");
IOUtil.writeLine(writer, " * @version ${version}");
IOUtil.writeLine(writer, " */");
if (parentClass == null) {
IOUtil.writeLine(writer, "public class " + className + " {");
} else {
IOUtil.writeLine(writer, "public class " + className + " extends " + parentClass.getSimpleName() + " {");
}
if (parentClass != null && Serializable.class.isAssignableFrom(parentClass)) {
long hashCode = className.hashCode();
for (Map.Entry> entry : fieldTypes.entrySet()) {
hashCode += hashCode * 31 + entry.getKey().hashCode();
hashCode += hashCode * 31 + entry.getValue().name().hashCode();
}
long serialVersionUID = 0;
if (String.valueOf(Long.MAX_VALUE).length() > (String.valueOf(hashCode).length() + 1)) {
serialVersionUID = Long.valueOf(String.valueOf(hashCode) + fieldTypes.size());
} else {
serialVersionUID = Long.valueOf(String.valueOf(hashCode + fieldTypes.size()));
}
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "private static final long serialVersionUID = " + serialVersionUID + "L;");
}
IOUtil.writeLine(writer, N.EMPTY_STRING);
for (Map.Entry> entry : fieldTypes.entrySet()) {
IOUtil.writeLine(writer, iden + "private " + getSimpleType(entry.getValue(), null, packageName, importedClasses) + " " + entry.getKey() + ";");
}
writeClassMethod(null, className, parentClass, packageName, fieldTypes, constructor, copyMethod, fluentSetter, parentPropertyModeForHashEquals,
parentPropertyModeForToString, fieldName2MethodName, importedClasses, utilClass, writer);
IOUtil.writeLine(writer, "}");
} catch (NoSuchFieldException | SecurityException | IOException e) {
throw N.toRuntimeException(e);
} finally {
IOUtil.closeQuietly(writer);
}
}
/**
* Read package types.
*
* @param type
* @param packageName
* @param packClasses
*/
private static void readPackageTypes(Type type, String packageName, final Multimap> packClasses) {
final Package pkg = type.clazz().getPackage();
if (pkg == null || pkg.getName().equals(packageName) || type.clazz().getCanonicalName().startsWith("java.lang.")) {
// ignore.
} else {
packClasses.put(pkg.getName(), type.clazz().getCanonicalName());
}
if (N.notNullOrEmpty(type.getParameterTypes())) {
for (Type paramType : type.getParameterTypes()) {
readPackageTypes(paramType, packageName, packClasses);
}
}
}
// public static void generateEntity(File srcDir, String packageName, Map> classNameFields) {
// generateEntity(srcDir, packageName, classNameFields, false, false, false);
// }
//
// /**
// *
// * @param srcDir
// * @param packageName
// * @param classNameFields
// * @param constructor
// * @param copyMethod
// * @param fluentSetter
// */
// public static void generateEntity(File srcDir, String packageName, Map> classNameFields, final boolean constructor,
// final boolean copyMethod, final boolean fluentSetter) {
// generateEntity(srcDir, packageName, classNameFields, constructor, copyMethod, fluentSetter, null, ParentPropertyMode.NONE, ParentPropertyMode.NONE);
// }
//
// /**
// *
// * @param srcDir
// * @param packageName
// * @param classNameFields
// * @param constructor
// * @param copyMethod
// * @param fluentSetter
// * @param parentClass
// * @param parentPropertyModeForHashEquals
// * @param parentPropertyModeForToString
// */
// public static void generateEntity(File srcDir, String packageName, Map> classNameFields, final boolean constructor,
// final boolean copyMethod, final boolean fluentSetter, Class parentClass, final ParentPropertyMode parentPropertyModeForHashEquals,
// final ParentPropertyMode parentPropertyModeForToString) {
//
// for (String className : classNameFields.keySet()) {
// generateEntity(srcDir, packageName, className, classNameFields.get(className), constructor, copyMethod, fluentSetter, parentClass,
// parentPropertyModeForHashEquals, parentPropertyModeForToString);
// }
// }
// /**
// * Generate and Print out the methods according to fields defined the in specified class.
// *
// * @param cls
// */
// public static void printClassMethod(final Class cls) {
// printClassMethod(cls, false, false, false, null, null);
// }
//
// /**
// * Generate and Print out the methods according to fields defined the in specified class.
// *
// * @param cls
// * @param constructor generate constructor
// * @param copyMethod generate the copy method.
// * @param fluentSetter
// * @param ignoreFieldNames
// * @param fieldName2MethodName
// */
// public static void printClassMethod(final Class cls, final boolean constructor, final boolean copyMethod, final boolean fluentSetter,
// Set ignoreFieldNames, final Map fieldName2MethodName) {
// printClassMethod(cls, constructor, copyMethod, fluentSetter, ignoreFieldNames, fieldName2MethodName, ParentPropertyMode.FIRST,
// ParentPropertyMode.FIRST);
// }
//
// /**
// * Generate and Print out the methods according to fields defined the in specified class.
// *
// * @param cls
// * @param constructor
// * @param copyMethod
// * @param fluentSetter
// * @param ignoreFieldNames
// * @param fieldName2MethodName
// * @param parentPropertyModeForHashEquals
// * @param parentPropertyModeForToString
// */
// public static void printClassMethod(final Class cls, final boolean constructor, final boolean copyMethod, final boolean fluentSetter,
// Set ignoreFieldNames, final Map fieldName2MethodName, final ParentPropertyMode parentPropertyModeForHashEquals,
// final ParentPropertyMode parentPropertyModeForToString) {
// if (ignoreFieldNames == null) {
// ignoreFieldNames = N.newLinkedHashSet();
// }
//
// final Map> fieldTypes = new LinkedHashMap<>();
//
// for (Field field : cls.getDeclaredFields()) {
// final String fieldName = field.getName();
// if (Modifier.isStatic(field.getModifiers()) || ignoreFieldNames.contains(fieldName)) {
// continue;
// } else {
// fieldTypes.put(fieldName, N.typeOf(field.getType()));
// }
// }
//
// try (Writer writer = new OutputStreamWriter(System.out)) {
// printClassMethod(cls, ClassUtil.getSimpleClassName(cls), cls.getSuperclass(), cls.getPackage() == null ? null : ClassUtil.getPackageName(cls),
// fieldTypes, constructor, copyMethod, fluentSetter, parentPropertyModeForHashEquals, parentPropertyModeForToString, fieldName2MethodName,
// new LinkedHashMap>(), writer);
// } catch (IOException | NoSuchFieldException | SecurityException e) {
// throw N.toRuntimeException(e);
// }
// }
/**
* Write class method.
*
* @param cls
* @param className
* @param parentClass
* @param pkgName
* @param fieldTypes
* @param constructor
* @param copyMethod
* @param fluentSetter
* @param parentPropertyModeForHashEquals
* @param parentPropertyModeForToString
* @param fieldName2MethodName
* @param importedClasses
* @param utilClass
* @param writer
* @throws NoSuchFieldException the no such field exception
* @throws SecurityException the security exception
* @throws IOException
*/
private static void writeClassMethod(Class cls, final String className, final Class parentClass, final String pkgName,
final Map> fieldTypes, final boolean constructor, final boolean copyMethod, final boolean fluentSetter,
ParentPropertyMode parentPropertyModeForHashEquals, ParentPropertyMode parentPropertyModeForToString, Map fieldName2MethodName,
final Map> importedClasses, final Class utilClass, Writer writer) throws NoSuchFieldException, SecurityException, IOException {
if (N.isNullOrEmpty(fieldTypes) && fluentSetter == false && constructor == false && copyMethod == false) {
return;
}
if (parentPropertyModeForHashEquals == null) {
parentPropertyModeForHashEquals = ParentPropertyMode.NONE;
}
if (parentPropertyModeForToString == null) {
parentPropertyModeForToString = ParentPropertyMode.NONE;
}
if (fieldName2MethodName == null) {
fieldName2MethodName = new LinkedHashMap<>();
}
final String utilClassName = utilClass.getSimpleName();
final List parentGetterMethods = new ArrayList<>();
final Map parentSettterMethods = new LinkedHashMap<>();
final List> allClasses = new ArrayList<>();
if (parentClass != null) {
allClasses.add(parentClass);
while (allClasses.get(allClasses.size() - 1).getSuperclass() != null) {
allClasses.add(allClasses.get(allClasses.size() - 1).getSuperclass());
}
for (Class superClass : allClasses) {
parentGetterMethods.addAll(ClassUtil.getPropGetMethods(superClass).values());
parentSettterMethods.putAll(ClassUtil.getPropSetMethods(superClass));
}
}
final String iden = " ";
if (constructor) {
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "public " + className + "() {");
if (parentClass != null && AbstractDirtyMarker.class.isAssignableFrom(parentClass)) {
IOUtil.writeLine(writer, iden + iden + "super(" + className + ".class.getSimpleName());");
}
IOUtil.writeLine(writer, iden + "}");
IOUtil.writeLine(writer, N.EMPTY_STRING);
String parameterStr = "";
String signValues = "";
for (Map.Entry entry : parentSettterMethods.entrySet()) {
if (parameterStr.length() > 0) {
parameterStr += ", ";
}
parameterStr += (getParameterTypeName(pkgName, entry));
if (signValues.length() > 0) {
signValues += IOUtil.LINE_SEPARATOR;
}
signValues += (iden + iden + "this." + entry.getValue().getName() + "(" + entry.getKey() + ");");
}
for (Map.Entry> entry : fieldTypes.entrySet()) {
if (parameterStr.length() > 0) {
parameterStr += ", ";
}
parameterStr += (getSimpleType(entry.getValue(), null, pkgName, importedClasses) + " " + entry.getKey());
if (signValues.length() > 0) {
signValues += IOUtil.LINE_SEPARATOR;
}
signValues += (iden + iden + "this." + entry.getKey() + " = " + entry.getKey() + ";");
}
if (parameterStr.length() > 0) {
IOUtil.writeLine(writer, iden + "public " + className + "(" + parameterStr + ") {");
if (parentClass != null && AbstractDirtyMarker.class.isAssignableFrom(parentClass)) {
IOUtil.writeLine(writer, iden + iden + "super(" + className + ".class.getSimpleName());");
IOUtil.writeLine(writer, N.EMPTY_STRING);
}
IOUtil.writeLine(writer, signValues);
IOUtil.writeLine(writer, iden + "}");
}
}
if (fluentSetter && N.notNullOrEmpty(parentSettterMethods)) {
for (Map.Entry entry : parentSettterMethods.entrySet()) {
if (parentClass.isAssignableFrom(entry.getValue().getReturnType()) == false) {
continue;
}
final String methodName = entry.getValue().getName();
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "public " + className + " " + methodName + "(" + getParameterTypeName(pkgName, entry) + ") {");
IOUtil.writeLine(writer, iden + iden + "super." + methodName + "(" + entry.getKey() + ");");
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + iden + "return this;");
IOUtil.writeLine(writer, iden + "}");
}
}
for (Map.Entry> entry : fieldTypes.entrySet()) {
final String fieldName = entry.getKey();
final String simpleTypeName = getSimpleType(entry.getValue(), null, pkgName, importedClasses);
// final String getPrefix = boolean.class.equals(entry.getValue().getTypeClass()) || Boolean.class.equals(entry.getValue().getTypeClass()) ? "is"
// : "get";
final String postfix = fieldName2MethodName.containsKey(fieldName) ? fieldName2MethodName.get(fieldName)
: (StringUtil.isAllUpperCase(fieldName) ? fieldName : StringUtil.capitalize(fieldName));
IOUtil.writeLine(writer, N.EMPTY_STRING);
// final String annoTypeName = getAnnoType(entry.getValue(), pkgName, importedClasses);
// IOUtil.writeLine(writer, iden + "@Type(\"" + entry.getValue().getName() + "\")");
// if (!entry.getValue().getName().equals(annoTypeName) || N.notNullOrEmpty(entry.getValue().getParameterTypes())) {
// IOUtil.writeLine(writer, iden + "@Type(\"" + annoTypeName + "\")");
// }
IOUtil.writeLine(writer, iden + "public " + simpleTypeName + " get" + postfix + "() {");
IOUtil.writeLine(writer, iden + iden + "return " + fieldName + ";");
IOUtil.writeLine(writer, iden + "}");
if (cls == null || Modifier.isFinal(cls.getDeclaredField(fieldName).getModifiers()) == false) {
IOUtil.writeLine(writer, N.EMPTY_STRING);
if (fluentSetter) {
IOUtil.writeLine(writer, iden + "public " + className + " set" + postfix + "(" + simpleTypeName + " " + fieldName + ") {");
} else {
IOUtil.writeLine(writer, iden + "public void set" + postfix + "(" + simpleTypeName + " " + fieldName + ") {");
}
if (parentClass != null && AbstractDirtyMarker.class.isAssignableFrom(parentClass)) {
IOUtil.writeLine(writer, iden + iden + "super.setUpdatedPropName(\"" + fieldName + "\");");
}
IOUtil.writeLine(writer, iden + iden + "this." + fieldName + " = " + fieldName + ";");
if (fluentSetter) {
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + iden + "return this;");
}
IOUtil.writeLine(writer, iden + "}");
}
}
if (copyMethod) {
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "public " + className + " copy() {");
IOUtil.writeLine(writer, iden + iden + "final " + className + " copy = new " + className + "();");
IOUtil.writeLine(writer, N.EMPTY_STRING);
for (Method method : parentGetterMethods) {
IOUtil.writeLine(writer,
iden + iden + "copy." + ClassUtil.getPropSetMethod(method.getDeclaringClass(), ClassUtil.getPropNameByMethod(method)).getName()
+ "(this." + method.getName() + "());");
}
for (Map.Entry> entry : fieldTypes.entrySet()) {
IOUtil.writeLine(writer, iden + iden + "copy." + entry.getKey() + " = this." + entry.getKey() + ";");
}
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + "return copy;");
IOUtil.writeLine(writer, iden + "}");
}
if (N.notNullOrEmpty(fieldTypes)) {
{
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "@Override");
IOUtil.writeLine(writer, iden + "public int hashCode() {");
IOUtil.writeLine(writer, iden + iden + "int h = 17;");
if (parentPropertyModeForHashEquals == ParentPropertyMode.FIRST && parentGetterMethods.size() > 0) {
for (Method method : parentGetterMethods) {
IOUtil.writeLine(writer, iden + iden + "h = 31 * h + " + utilClassName + ".hashCode(" + method.getName() + "());");
}
}
for (Map.Entry> entry : fieldTypes.entrySet()) {
IOUtil.writeLine(writer, iden + iden + "h = 31 * h + " + utilClassName + ".hashCode(" + entry.getKey() + ");");
}
if (parentPropertyModeForHashEquals == ParentPropertyMode.LAST && parentGetterMethods.size() > 0) {
for (Method method : parentGetterMethods) {
IOUtil.writeLine(writer, iden + iden + "h = 31 * h + " + utilClassName + "hashCode(" + method.getName() + "());");
}
}
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + "return h;");
IOUtil.writeLine(writer, iden + "}");
}
{
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "@Override");
IOUtil.writeLine(writer, iden + "public boolean equals(Object obj) {");
IOUtil.writeLine(writer, iden + iden + "if (this == obj) {");
IOUtil.writeLine(writer, iden + iden + iden + "return true;");
IOUtil.writeLine(writer, iden + iden + "}");
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + "if (obj instanceof " + className + ") {");
IOUtil.writeLine(writer, iden + iden + iden + "final " + className + " other = (" + className + ") obj;");
int i = 0;
if (parentPropertyModeForHashEquals == ParentPropertyMode.FIRST && parentGetterMethods.size() > 0) {
for (Method method : parentGetterMethods) {
if (i++ == 0) {
if (i == parentGetterMethods.size() + fieldTypes.size()) {
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + iden + "return " + utilClassName + ".equals(" + method.getName()
+ "(), other." + method.getName() + "());");
} else {
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + iden + "return " + utilClassName + ".equals(" + method.getName()
+ "(), other." + method.getName() + "())");
}
} else {
if (i == parentGetterMethods.size() + fieldTypes.size()) {
IOUtil.writeLine(writer, iden + iden + iden + iden + "&& " + utilClassName + ".equals(" + method.getName() + "(), other."
+ method.getName() + "());");
} else {
IOUtil.writeLine(writer, iden + iden + iden + iden + "&& " + utilClassName + ".equals(" + method.getName() + "(), other."
+ method.getName() + "())");
}
}
}
}
for (Map.Entry> entry : fieldTypes.entrySet()) {
final String fieldName = entry.getKey();
if (i++ == 0) {
if (i == fieldTypes.size() && (parentGetterMethods.size() == 0 || parentPropertyModeForHashEquals != ParentPropertyMode.LAST)) {
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + iden + "return " + utilClassName + ".equals(" + fieldName
+ ", other." + fieldName + ");");
} else {
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + iden + "return " + utilClassName + ".equals(" + fieldName
+ ", other." + fieldName + ")");
}
} else {
if (((parentPropertyModeForHashEquals != ParentPropertyMode.FIRST && i == fieldTypes.size())
|| (parentPropertyModeForHashEquals == ParentPropertyMode.FIRST && i == fieldTypes.size() + parentGetterMethods.size()))
&& (parentGetterMethods.size() == 0 || parentPropertyModeForHashEquals != ParentPropertyMode.LAST)) {
IOUtil.writeLine(writer,
iden + iden + iden + iden + "&& " + utilClassName + ".equals(" + fieldName + ", other." + fieldName + ");");
} else {
IOUtil.writeLine(writer, iden + iden + iden + iden + "&& " + utilClassName + ".equals(" + fieldName + ", other." + fieldName + ")");
}
}
}
if (parentPropertyModeForHashEquals == ParentPropertyMode.LAST && parentGetterMethods.size() > 0) {
for (Method method : parentGetterMethods) {
if (i++ == 0) {
if (i == parentGetterMethods.size() + fieldTypes.size()) {
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + iden + "return " + utilClassName + ".equals(" + method.getName()
+ "(), other." + method.getName() + "());");
} else {
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + iden + "return " + utilClassName + ".equals(" + method.getName()
+ "(), other." + method.getName() + "())");
}
} else {
if (i == parentGetterMethods.size() + fieldTypes.size()) {
IOUtil.writeLine(writer, iden + iden + iden + iden + "&& " + utilClassName + ".equals(" + method.getName() + "(), other."
+ method.getName() + "());");
} else {
IOUtil.writeLine(writer, iden + iden + iden + iden + "&& " + utilClassName + ".equals(" + method.getName() + "(), other."
+ method.getName() + "())");
}
}
}
}
IOUtil.writeLine(writer, iden + iden + "}");
IOUtil.writeLine(writer, IOUtil.LINE_SEPARATOR + iden + iden + "return false;");
IOUtil.writeLine(writer, iden + "}");
}
{
final StringBuilder sb = new StringBuilder();
IOUtil.writeLine(writer, N.EMPTY_STRING);
sb.append(iden + "@Override" + IOUtil.LINE_SEPARATOR);
sb.append(iden + "public String toString() {" + IOUtil.LINE_SEPARATOR);
int i = 0;
if (parentPropertyModeForToString == ParentPropertyMode.FIRST && parentGetterMethods.size() > 0) {
for (Method method : parentGetterMethods) {
if (i++ == 0) {
sb.append(iden + iden + "return \"{" + ClassUtil.getPropNameByMethod(method) + "=\" + " + utilClassName + ".toString("
+ method.getName() + "())");
} else {
sb.append(IOUtil.LINE_SEPARATOR + iden + iden + " + \", " + ClassUtil.getPropNameByMethod(method) + "=\" + " + utilClassName
+ ".toString(" + method.getName() + "())");
}
if (i == parentGetterMethods.size() + fieldTypes.size()) {
if (i > 1) {
sb.append(IOUtil.LINE_SEPARATOR + iden + iden + " + \"}\";" + IOUtil.LINE_SEPARATOR);
} else {
sb.append(" + \"}\";" + IOUtil.LINE_SEPARATOR);
}
}
}
}
for (Map.Entry> entry : fieldTypes.entrySet()) {
final String fieldName = entry.getKey();
if (i++ == 0) {
sb.append(iden + iden + "return \"{" + fieldName + "=\" + " + utilClassName + ".toString(" + fieldName + ")");
} else {
sb.append(IOUtil.LINE_SEPARATOR + iden + iden + " + \", " + fieldName + "=\" + " + utilClassName + ".toString(" + fieldName
+ ")");
}
if ((((parentPropertyModeForToString == null || parentPropertyModeForToString == ParentPropertyMode.NONE)
|| (parentPropertyModeForToString == ParentPropertyMode.LAST && parentGetterMethods.size() == 0)) && i == fieldTypes.size())
|| (parentPropertyModeForToString == ParentPropertyMode.FIRST && i == parentGetterMethods.size() + fieldTypes.size())) {
if (i > 1) {
sb.append(IOUtil.LINE_SEPARATOR + iden + iden + " + \"}\";" + IOUtil.LINE_SEPARATOR);
} else {
sb.append(" + \"}\";" + IOUtil.LINE_SEPARATOR);
}
}
}
if (parentPropertyModeForToString == ParentPropertyMode.LAST && parentGetterMethods.size() > 0) {
for (Method method : parentGetterMethods) {
if (i++ == 0) {
sb.append(iden + iden + "return \"{" + ClassUtil.getPropNameByMethod(method) + "=\" + " + utilClassName + ".toString("
+ method.getName() + "())");
} else {
sb.append(IOUtil.LINE_SEPARATOR + iden + iden + " + \", " + ClassUtil.getPropNameByMethod(method) + "=\" + " + utilClassName
+ ".toString(" + method.getName() + "())");
}
if (i == parentGetterMethods.size() + fieldTypes.size()) {
if (i > 1) {
sb.append(IOUtil.LINE_SEPARATOR + iden + iden + " + \"}\";" + IOUtil.LINE_SEPARATOR);
} else {
sb.append(" + \"}\";" + IOUtil.LINE_SEPARATOR);
}
}
}
}
sb.append(iden + "}");
IOUtil.writeLine(writer, sb.toString());
}
}
}
/**
* Gets the parameter type name.
*
* @param pkgName
* @param entry
* @return
*/
private static String getParameterTypeName(final String pkgName, Map.Entry entry) {
String paraTypeName = ClassUtil.getParameterizedTypeNameByMethod(entry.getValue());
if (N.notNullOrEmpty(pkgName)) {
String tmp = pkgName + ".";
int idx = 0;
char ch = 0;
while ((idx = paraTypeName.indexOf(tmp, idx)) >= 0) {
for (int i = idx + tmp.length(), len = paraTypeName.length(); i < len; i++) {
ch = paraTypeName.charAt(i);
if ((Character.isLetterOrDigit(ch) || ch == '$' || ch == '_') && i != len - 1) {
continue;
} else if (ch == '.') {
idx = i;
break;
} else {
paraTypeName = paraTypeName.replace(paraTypeName.substring(idx, i), paraTypeName.substring(idx + tmp.length(), i));
idx += (i - idx - tmp.length());
break;
}
}
}
}
return paraTypeName + " " + entry.getKey();
}
/**
* Gets the anno type.
*
* @param type
* @param packageName
* @param importedClasses
* @return
*/
static String getAnnoType(Type type, final String packageName, Map> importedClasses) {
if (N.isNullOrEmpty(type.getParameterTypes()) && !type.clazz().equals(Object.class)) {
return type.name();
}
String typeName = type.name();
final StringBuilder sb = new StringBuilder();
int start = 0;
for (int i = 0, len = typeName.length(); i < len; i++) {
char ch = typeName.charAt(i);
if (ch == '<' || ch == '>' || ch == ' ' || ch == ',') {
String str = typeName.substring(start, i);
if (str.length() > 0 && N.typeOf(str).clazz().equals(Object.class)) {
String pkgName = packageName;
for (Map.Entry> entry : importedClasses.entrySet()) {
if (entry.getValue().getSimpleName().equals(str)) {
pkgName = ClassUtil.getPackageName(entry.getValue());
break;
}
}
if (N.notNullOrEmpty(pkgName) && !N.typeOf(pkgName + "." + str).clazz().equals(Object.class)) {
sb.append(pkgName + "." + str);
} else {
sb.append(str);
}
} else {
sb.append(str);
}
sb.append(ch);
start = i + 1;
}
}
if (start < typeName.length()) {
String str = typeName.substring(start);
if (str.length() > 0 && N.typeOf(str).clazz().equals(Object.class)) {
String pkgName = packageName;
for (Map.Entry> entry : importedClasses.entrySet()) {
if (entry.getValue().getSimpleName().equals(str)) {
pkgName = ClassUtil.getPackageName(entry.getValue());
break;
}
}
if (N.notNullOrEmpty(pkgName) && !N.typeOf(pkgName + "." + str).clazz().equals(Object.class)) {
sb.append(pkgName + "." + str);
} else {
sb.append(str);
}
} else {
sb.append(str);
}
}
return sb.toString();
}
// /**
// *
// * @param pkgName
// */
// public static void printUtilClassForHashEqualsToString(final String pkgName) {
// printUtilClassForHashEqualsToString(pkgName, "_N");
// }
//
// /**
// *
// * @param pkgName
// * @param utilClassName
// */
// public static void printUtilClassForHashEqualsToString(final String pkgName, final String utilClassName) {
// if (N.isNullOrEmpty(pkgName)) {
// N.println(_N_STRING.replaceFirst("package com.landawn.abacus.util;", "").replaceAll("_N", utilClassName));
// } else {
// N.println(_N_STRING.replaceFirst("com.landawn.abacus.util", pkgName).replaceAll("_N", utilClassName));
// }
// }
/**
*
* @param classFile
* @return Writer
* @throws IOException Signals that an I/O exception has occurred.
*/
private static Writer newFileWriter(final File classFile) throws IOException {
if (!classFile.exists()) {
classFile.createNewFile();
}
return new BufferedWriter(new OutputStreamWriter(new FileOutputStream(classFile), UTF_8));
}
/**
*
* @param domainName
* @param fileWrite
* @param headSpace
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeDomainPropNameClass(final String domainName, final Writer fileWrite, final String headSpace) throws IOException {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public static final String " + DOMAIN_NAME_VAR + " = \"" + domainName + "\".intern();" + IOUtil.LINE_SEPARATOR);
}
/**
* Upper case the first char of the table name.
*
* @param tableName
* @return String
*/
static String tableName2EntityName(final String tableName) {
return StringUtil.capitalize(formalizePropName(tableName));
}
/**
* SubString {@code columnName} from first under score if it exists. lower the first char of the subString.
*
* @param columnName
* @return String
*/
static String columnName2PropName(final String columnName) {
String propName = formalizePropName(columnName);
return JAVA_TYPE_PROP_NAME.containsKey(propName) ? JAVA_TYPE_PROP_NAME.get(propName) : propName;
}
/**
* Prop name 2 var name.
*
* @param propName
* @return
*/
static String propName2VarName(final String propName) {
return ClassUtil.toUpperCaseWithUnderscore(ClassUtil.formalizePropName(propName));
}
/**
* Prop name 2 method name.
*
* @param propName
* @return
*/
static String propName2MethodName(final String propName) {
return StringUtil.capitalize(propName);
}
/**
* Prop name 2 field name.
*
* @param propName
* @return
*/
static String propName2FieldName(final String propName) {
if (propName.toUpperCase().equals(propName)) {
return propName.toLowerCase();
}
return propName;
}
/**
*
* @param st
* @return String
*/
private static String formalizePropName(String st) {
String temp = st;
while ((st.length() > 0) && !Character.isLetter(st.charAt(0))) {
st = st.substring(1);
}
if (st.length() == 0) {
throw new RuntimeException("NOT able to generate valid entity/property name by table/column name: " + temp);
}
String nst = "";
boolean isAllUpperCase = true;
boolean hasUnderScore = false;
char c = 0;
for (int i = 0; i < st.length(); i++) {
c = st.charAt(i);
if (Character.isLowerCase(c)) {
isAllUpperCase = false;
}
if (c == WD._UNDERSCORE) {
hasUnderScore = true;
}
}
if (hasUnderScore) {
if (isAllUpperCase) {
for (int i = 0; i < st.length(); i++) {
c = st.charAt(i);
if (c == WD._UNDERSCORE) {
i++;
if (i < st.length()) {
c = st.charAt(i);
nst += Character.toUpperCase(c);
}
} else {
nst += Character.toLowerCase(c);
}
}
} else {
for (int i = 0; i < st.length(); i++) {
c = st.charAt(i);
if (c == WD._UNDERSCORE) {
i++;
if (i < st.length()) {
c = st.charAt(i);
nst += Character.toUpperCase(c);
}
} else {
nst += c;
}
}
}
} else if (isAllUpperCase) {
nst = st.toLowerCase();
} else {
nst = st;
}
return StringUtil.uncapitalize(nst);
}
/**
* Write file head.
*
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeFileHead(final Writer fileWrite) throws IOException {
fileWrite.write("/*" + IOUtil.LINE_SEPARATOR);
fileWrite.write(" * Generated by Abacus." + IOUtil.LINE_SEPARATOR);
fileWrite.write(" */" + IOUtil.LINE_SEPARATOR);
}
/**
* Write package name.
*
* @param pkgName
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writePackageName(final String pkgName, final Writer fileWrite) throws IOException {
if (N.notNullOrEmpty(pkgName)) {
fileWrite.write("package " + pkgName + ";" + IOUtil.LINE_SEPARATOR);
}
}
/**
*
* @param entityDef
* @param pkgName
* @param propNameTableClass
* @param entityMode
* @param extendedClass
* @param implementedInterfaces
* @param fileWrite
* @param generateHashEqualsMethod
* @param generateToStringMethod
* @param utilClass
* @param annotationImportClasses
* @return
* @throws IOException Signals that an I/O exception has occurred.
*/
private static Map> writeImport(final EntityDefinition entityDef, final String pkgName, final String propNameTableClass,
final EntityMode entityMode, final Class extendedClass, final List> implementedInterfaces, final Writer fileWrite,
final boolean generateHashEqualsMethod, final boolean generateToStringMethod, final Class utilClass, final Set> annotationImportClasses)
throws IOException {
final Map> importedClasses = new LinkedHashMap<>();
if ((IMPL_DIRTY_MARKER == entityMode)) {
writeClassImport(fileWrite, Collection.class, importedClasses);
writeClassImport(fileWrite, Set.class, importedClasses);
}
for (Property prop : entityDef.getPropertyList()) {
if (prop.isCollection()) {
writeClassImport(fileWrite, prop.getType().clazz(), importedClasses);
}
}
Set> sqlTypes = getUsualType(entityDef);
for (Class sqlType : sqlTypes) {
if (!sqlType.getCanonicalName().startsWith("java.lang.")) {
writeClassImport(fileWrite, sqlType, importedClasses);
}
}
if ((IMPL_DIRTY_MARKER == entityMode)) {
writeClassImport(fileWrite, DirtyMarkerImpl.class, importedClasses);
}
if ((generateHashEqualsMethod || generateToStringMethod) && (hasHashEqualsProperty(entityDef) || hasToStringProperty(entityDef))) {
String utilClassName = ClassUtil.getCanonicalClassName(utilClass);
if (pkgName.equals(utilClassName.substring(0, utilClassName.lastIndexOf('.'))) || _N.equals(utilClass)) {
// ignore
} else {
writeClassImport(fileWrite, utilClass, importedClasses);
}
}
for (Property prop : entityDef.getPropertyList()) {
if (prop.getType().name().contains("HBaseColumn<")) {
writeClassImport(fileWrite, HBaseColumn.class, importedClasses);
if (prop.getType().isMap() || prop.getType().isCollection()) {
writeClassImport(fileWrite, N.class, importedClasses);
}
}
}
if (N.notNullOrEmpty(entityDef.getIdPropertyList())) {
writeClassImport(fileWrite, Id.class, importedClasses);
}
if (N.notNullOrEmpty(entityDef.getPropertyList())) {
writeClassImport(fileWrite, com.landawn.abacus.annotation.Column.class, importedClasses);
}
if (extendedClass != null) {
writeClassImport(fileWrite, extendedClass, importedClasses);
}
if (implementedInterfaces != null) {
for (Class clazz : implementedInterfaces) {
writeClassImport(fileWrite, clazz, importedClasses);
}
}
if (propNameTableClass != null) {
fileWrite.write(IOUtil.LINE_SEPARATOR + "import " + propNameTableClass + ";");
}
if ((IMPL_DIRTY_MARKER == entityMode) || ((EXTEND_DIRTY_MARKER == entityMode) && (entityDef.getEntiyPropertyList().size() > 0))) {
writeClassImport(fileWrite, XmlTransient.class, importedClasses);
}
if (N.notNullOrEmpty(annotationImportClasses)) {
for (Class cls : annotationImportClasses) {
writeClassImport(fileWrite, cls, importedClasses);
}
}
if (N.notNullOrEmpty(entityDef.getPropertyList())) {
for (Property prop : entityDef.getPropertyList()) {
if (prop.getType().isGenericType()) {
writeClassImport(fileWrite, com.landawn.abacus.annotation.Type.class, importedClasses);
break;
}
}
}
return importedClasses;
}
/**
* Write class import.
*
* @param fileWrite
* @param cls
* @param importedClasses
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeClassImport(final Writer fileWrite, Class cls, final Map> importedClasses) throws IOException {
// if (!cls.isInterface()) {
// if (Map.class.isAssignableFrom(cls)) {
// cls = Map.class;
// } else if (List.class.isAssignableFrom(cls)) {
// cls = List.class;
// } else if (Set.class.isAssignableFrom(cls)) {
// cls = Set.class;
// } else if (Queue.class.isAssignableFrom(cls)) {
// cls = Queue.class;
// } else if (Collection.class.isAssignableFrom(cls)) {
// Class tmp = Collection.class;
//
// Class[] interfaceClasses = cls.getInterfaces();
// for (Class interfaceClass : interfaceClasses) {
// if (Collection.class.isAssignableFrom(interfaceClass) && !interfaceClass.equals(Collection.class)) {
// tmp = interfaceClass;
//
// break;
// }
// }
//
// cls = tmp;
// }
// }
if ((cls != null) && !importedClasses.containsKey(cls.getSimpleName())) {
fileWrite.write(IOUtil.LINE_SEPARATOR + "import " + ClassUtil.getCanonicalClassName(cls) + ";");
importedClasses.put(cls.getSimpleName(), cls);
}
}
/**
*
* @param entityDef
* @param pkgName
* @param propNameTableClass
* @param propName2VarName
* @param propName2MethodName
* @param entityMode
* @param extendedClass
* @param implementedInterfaces
* @param headSpace
* @param fileWrite
* @param generateHashEqualsMethod
* @param generateToStringMethod
* @param hashEqualsWithParentProperties
* @param toStringWithParentProperties
* @param utilClass
* @param fluentSetMethod
* @param importedClasses
* @throws IOException Signals that an I/O exception has occurred.
* @throws IllegalAccessException the illegal access exception
* @throws InvocationTargetException the invocation target exception
*/
private static void writeClass(final EntityDefinition entityDef, final String pkgName, final String propNameTableClass, final Method propName2VarName,
final Method propName2MethodName, final EntityMode entityMode, Class extendedClass, final List> implementedInterfaces,
final String headSpace, final Writer fileWrite, final boolean generateHashEqualsMethod, final boolean generateToStringMethod,
final boolean hashEqualsWithParentProperties, final boolean toStringWithParentProperties, final Class utilClass, final boolean fluentSetMethod,
final Map> importedClasses) throws IOException, IllegalAccessException, InvocationTargetException {
writeClassHead(entityDef, propNameTableClass, extendedClass, implementedInterfaces, headSpace, fileWrite);
if (entityMode.equals(EntityMode.IMPL_DIRTY_MARKER) && (extendedClass != null) && !DirtyMarkerImpl.class.isAssignableFrom(extendedClass)) {
extendedClass = DirtyMarkerImpl.class;
}
// writePropNameField(entityDef, propName2VarName, fileWrite);
// writePropNameField(entityDef, propName2VarName, headSpace, fileWrite);
// =================================================================
// writeClass(entityDef, fileWrite, isMarkable);
writeField(entityDef, pkgName, entityMode, extendedClass, implementedInterfaces, headSpace, fileWrite, importedClasses);
// write default constructor.
writeDefaultConstructor(entityDef, entityMode, headSpace, fileWrite);
// write id properties constructor.
writeIdPropertyConstructor(entityDef, pkgName, propName2MethodName, headSpace, fileWrite, importedClasses);
if (entityDef.getPropertyList().size() > 0) {
// write full properties constructor.
writeFullPropertyConstructor(entityDef, pkgName, propName2MethodName, headSpace, fileWrite, importedClasses);
}
// write copy constructor.
// writeCopyConstructor(entityDef, pkgName, headSpace, fileWrite);
writeDirtyMarkerMethod(entityDef, entityMode, headSpace, fileWrite);
// write get/set methods
writeGetSetMethod(entityDef, pkgName, extendedClass, entityMode, propName2VarName, propName2MethodName, fluentSetMethod, headSpace, fileWrite,
importedClasses);
if (entityDef.getPropertyList().size() > 0) {
if (generateHashEqualsMethod) {
// write hashCode() method
writeHashCodeMethod(entityDef, headSpace, fileWrite, extendedClass, hashEqualsWithParentProperties, utilClass);
// if ((entityMode == EXTEND_DIRTY_MARKER) || (entityMode ==
// IMPL_DIRTY_MARKER)
// || (entityMode == POJO)) {
// write equals method
writeEqualMethod(entityDef, headSpace, fileWrite, extendedClass, hashEqualsWithParentProperties, utilClass);
}
if (generateToStringMethod) {
// write toString() method
writeToStringMethod(entityDef, headSpace, fileWrite, extendedClass, toStringWithParentProperties, utilClass);
// } else if (entityMode == IMPL_ACTIVE_RECORD) {
// writeHES(headSpace, fileWrite);
// }
}
}
// =================================================================
fileWrite.write(headSpace + "}");
}
/**
*
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeClassComment(final Writer fileWrite) throws IOException {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write("/**" + IOUtil.LINE_SEPARATOR);
fileWrite.write(" * Generated by Abacus." + IOUtil.LINE_SEPARATOR);
fileWrite.write(" * @version ${version}" + IOUtil.LINE_SEPARATOR);
fileWrite.write(" */" + IOUtil.LINE_SEPARATOR);
}
/**
* Write class head.
*
* @param entityDef
* @param propNameTableClass
* @param extendedClass
* @param implementsInterface
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeClassHead(final EntityDefinition entityDef, final String propNameTableClass, final Class extendedClass,
final List> implementsInterface, final String headSpace, final Writer fileWrite) throws IOException {
final StringBuilder sb = Objectory.createStringBuilder();
boolean isAbstract = N.parseBoolean(entityDef.getAttribute("abstract"));
if (isAbstract) {
sb.append(headSpace + "public abstract class " + entityDef.getName());
} else {
sb.append(headSpace + "public class " + entityDef.getName());
}
if ((extendedClass != null) && !DirtyMarkerImpl.class.isAssignableFrom(extendedClass)) {
if (sb.indexOf(" extends ") > 0) {
sb.append(", " + ClassUtil.getSimpleClassName(extendedClass));
} else {
sb.append(" extends " + ClassUtil.getSimpleClassName(extendedClass));
}
}
if (propNameTableClass != null) {
String simplePropNameTableClass = propNameTableClass;
int index = propNameTableClass.lastIndexOf(".");
if (index > -1) {
simplePropNameTableClass = propNameTableClass.substring(index + 1);
}
if (sb.indexOf(" implements ") > 0) {
sb.append(", " + simplePropNameTableClass + WD._PERIOD + entityDef.getName() + POSTFIX_OF_PROP_NAME_LIST);
} else {
sb.append(" implements " + simplePropNameTableClass + WD._PERIOD + entityDef.getName() + POSTFIX_OF_PROP_NAME_LIST);
}
}
if ((implementsInterface != null) && (implementsInterface.size() > 0)) {
for (Class cls : implementsInterface) {
if (sb.indexOf(" implements ") > 0) {
sb.append(", " + ClassUtil.getSimpleClassName(cls));
} else {
sb.append(" implements " + ClassUtil.getSimpleClassName(cls));
}
}
}
sb.append(" {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(sb.toString());
Objectory.recycle(sb);
}
// /**
// * Method writeStaticField.
// *
// * @param entityDef
// * SQLEntityDefinition
// * @param entityMode
// * EntityMode
// * @param headSpace
// * String
// * @param fileWrite
// * Writer
// * @throws IOException
// */
// private static void writeStaticField(EntityDefinition entityDef, EntityMode
// entityMode, String headSpace, Writer fileWrite)
// throws IOException {
// if ((EntityMode.EXTEND_ACTIVE_RECORD == entityMode) ||
// (EntityMode.IMPL_ACTIVE_RECORD == entityMode)) {
// if (entityDef.getEntityType() == EntityType.ENTITY) {
// fileWrite.write(D.LINE_SEPARATOR + headSpace + " public static final "
// + AbstractActiveRecord.class.getName() + "."
// + AbstractActiveRecord.Calculator.class.getName() + " Cal = new "
// + AbstractActiveRecord.class.getName() + "."
// + AbstractActiveRecord.Calculator.class.getName() + "(" +
// DOMAIN_NAME_VAR + ", _);"
// + D.LINE_SEPARATOR);
// }
// }
// }
/**
*
* @param entityDef
* @param pkgName
* @param entityMode
* @param extendedClass
* @param implementedInterfaces
* @param headSpace
* @param fileWrite
* @param importedClasses
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeField(final EntityDefinition entityDef, final String pkgName, final EntityMode entityMode, final Class extendedClass,
final List> implementedInterfaces, final String headSpace, final Writer fileWrite, final Map> importedClasses)
throws IOException {
boolean isSerializable = isSerializable(extendedClass, implementedInterfaces);
if (isSerializable) {
writeSerialVersionUID(entityDef, headSpace, fileWrite);
}
// writeStaticField(entityDef, entityMode, headSpace, fileWrite);
if (entityMode == IMPL_DIRTY_MARKER) {
fileWrite.write(headSpace + " private final " + ClassUtil.getSimpleClassName(extendedClass) + " " + DIRTY_MARKER_IMPL_FIELD_NAME + " = new "
+ ClassUtil.getSimpleClassName(extendedClass) + "(" + ENTITY_NAME_VAR + ");" + IOUtil.LINE_SEPARATOR + IOUtil.LINE_SEPARATOR);
}
Collection propList = entityDef.getPropertyList();
for (Property prop : propList) {
String fieldName = propName2FieldName(prop.getName());
String type = getSimpleType(null, prop, pkgName, importedClasses);
fileWrite.write(IOUtil.LINE_SEPARATOR);
if (prop.isId()) {
fileWrite.write(headSpace + " @Id" + IOUtil.LINE_SEPARATOR);
}
if (prop.getColumnType() == ColumnType.TABLE_COLUMN) {
fileWrite.write(headSpace + " @Column(\"" + prop.getColumnName() + "\")" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " private " + type + " " + fieldName + ";" + IOUtil.LINE_SEPARATOR);
}
}
/**
* Checks if is serializable.
*
* @param extendedClass
* @param implementedInterfaces
* @return true, if is serializable
*/
private static boolean isSerializable(final Class extendedClass, final List> implementedInterfaces) {
if ((extendedClass != null) && !DirtyMarkerImpl.class.isAssignableFrom(extendedClass) && Serializable.class.isAssignableFrom(extendedClass)) {
return true;
}
if ((implementedInterfaces != null) && (implementedInterfaces.size() > 0)) {
for (Class clazz : implementedInterfaces) {
if (Serializable.class.equals(clazz) || Serializable.class.isAssignableFrom(clazz)) {
return true;
}
}
}
return false;
}
/**
* Write serial version UID.
*
* @param entityDef
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeSerialVersionUID(final EntityDefinition entityDef, final String headSpace, final Writer fileWrite) throws IOException {
long hashCode = entityDef.getName().hashCode();
for (String propName : entityDef.getPropertyNameList()) {
if (propName.hashCode() > 0) {
hashCode += propName.hashCode();
} else {
hashCode -= propName.hashCode();
}
}
long serialVersionUID = 0;
if (String.valueOf(Long.MAX_VALUE).length() > (String.valueOf(hashCode).length() + 1)) {
serialVersionUID = Long.valueOf(String.valueOf(hashCode) + entityDef.getPropertyList().size());
} else {
serialVersionUID = Long.valueOf(String.valueOf(hashCode + entityDef.getPropertyList().size()));
}
serialVersionUID = Long.valueOf(String.valueOf(hashCode) + entityDef.getPropertyList().size());
fileWrite.write(
headSpace + " private static final long serialVersionUID = " + serialVersionUID + "L;" + IOUtil.LINE_SEPARATOR + IOUtil.LINE_SEPARATOR);
}
/**
* Write default constructor.
*
* @param entityDef
* @param entityMode
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeDefaultConstructor(final EntityDefinition entityDef, final EntityMode entityMode, final String headSpace, final Writer fileWrite)
throws IOException {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + "() {" + IOUtil.LINE_SEPARATOR);
/*
* if (inheritLevel == IMPL_ACTIVE_RECORD) { fileWrite.write(headSpace + " activeRecordImpl = new " +
* inheritClass.getCanonicalName() + "(this);" + N.LINE_SEPARATOR); } else if (inheritLevel ==
* IMPL_DIRTY_MARKER) { fileWrite.write(headSpace + " dirtyMarkerImpl = new
* " + inheritClass.getCanonicalName() + "();" + N.LINE_SEPARATOR); }
*/
if (entityMode == EntityMode.EXTEND_DIRTY_MARKER) {
fileWrite.write(headSpace + " super(" + ENTITY_NAME_VAR + ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
/**
* Write id property constructor.
*
* @param entityDef
* @param pkgName
* @param propName2MethodName
* @param headSpace
* @param fileWrite
* @param importedClasses
* @return
* @throws IOException Signals that an I/O exception has occurred.
*/
private static Collection writeIdPropertyConstructor(final EntityDefinition entityDef, final String pkgName, final Method propName2MethodName,
final String headSpace, final Writer fileWrite, final Map> importedClasses) throws IOException {
final List idPropList = entityDef.getIdPropertyList();
final List idPropNames = new ArrayList<>(idPropList.size());
if ((idPropList.size() > 0) && (idPropList.size() < entityDef.getPropertyList().size())) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + "(");
fileWrite.write(getIdParaString(pkgName, idPropList, importedClasses));
fileWrite.write(") {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " this();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
for (Property idProp : idPropList) {
String fieldName = propName2FieldName(idProp.getName());
fileWrite.write(headSpace + " set" + ClassUtil.invokeMethod(propName2MethodName, idProp.getName()) + "(" + fieldName + ");"
+ IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
List tmp = new ArrayList<>(entityDef.getPropertyNameList());
for (Property idProp : idPropList) {
idPropNames.add(idProp.getName());
tmp.remove(idProp.getName());
}
boolean writeNonIdConstructor = false;
if (tmp.size() > entityDef.getIdPropertyList().size()) {
writeNonIdConstructor = true;
} else {
for (int i = 0; i < tmp.size(); i++) {
if (!entityDef.getProperty(tmp.get(i)).getType().clazz().isAssignableFrom(entityDef.getIdPropertyList().get(i).getType().clazz())) {
writeNonIdConstructor = true;
break;
}
}
}
if (writeNonIdConstructor) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + "(");
int length = 8;
String headEmpty = "";
for (int j = 0; j < length; j++) {
headEmpty += " ";
}
int i = 1;
// tempList.addAll(0, entityDef.getIdPropNameList());
for (String propName : tmp) {
Property prop = entityDef.getProperty(propName);
String para = getSimpleType(null, prop, pkgName, importedClasses) + " " + propName2FieldName(prop.getName());
if (i < tmp.size()) {
para += ", ";
}
length += para.length();
if (length > 80) {
length = 8 + para.length();
para = IOUtil.LINE_SEPARATOR + headEmpty + para;
}
fileWrite.write(para);
i++;
}
fileWrite.write(") {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " this();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
for (String propName : tmp) {
Property prop = entityDef.getProperty(propName);
fileWrite.write(headSpace + " set" + ClassUtil.invokeMethod(propName2MethodName, prop.getName()) + "("
+ propName2FieldName(prop.getName()) + ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
return idPropNames;
}
/**
* Write full property constructor.
*
* @param entityDef
* @param pkgName
* @param propName2MethodName
* @param headSpace
* @param fileWrite
* @param importedClasses
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeFullPropertyConstructor(final EntityDefinition entityDef, final String pkgName, final Method propName2MethodName,
final String headSpace, final Writer fileWrite, final Map> importedClasses) throws IOException {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + "(");
int length = 8;
String headEmpty = "";
for (int j = 0; j < length; j++) {
headEmpty += " ";
}
List tempList = new ArrayList<>(entityDef.getPropertyNameList());
for (Property idProp : entityDef.getIdPropertyList()) {
tempList.remove(idProp.getName());
}
int i = 0;
for (Property idProp : entityDef.getIdPropertyList()) {
tempList.add(i++, idProp.getName());
}
i = 0;
for (String propName : tempList) {
Property prop = entityDef.getProperty(propName);
String para = getSimpleType(null, prop, pkgName, importedClasses) + " " + propName2FieldName(prop.getName());
if (++i < tempList.size()) {
para += ", ";
}
length += para.length();
if (length > 80) {
length = 8 + para.length();
para = IOUtil.LINE_SEPARATOR + headEmpty + para;
}
fileWrite.write(para);
}
fileWrite.write(") {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " this();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
for (String propName : tempList) {
Property prop = entityDef.getProperty(propName);
fileWrite.write(headSpace + " set" + ClassUtil.invokeMethod(propName2MethodName, prop.getName()) + "(" + propName2FieldName(prop.getName())
+ ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
/**
*
* @param entityDef
* @param pkgName
* @param extendedClass
* @param entityMode
* @param propName2VarName
* @param propName2MethodName
* @param fluentSetMethod
* @param headSpace
* @param fileWrite
* @param importedClasses
* @throws IOException Signals that an I/O exception has occurred.
*/
/*
* static void writeCopyConstructor(EntityDefinition entityDef, String pkgName, String headSpace, Writer
* fileWrite) throws IOException { String objectFieldName = N.lowCaseFirstChar(entityDef.getName());
* fileWrite.write(N.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + "(" + entityDef.getName() +
* " " + objectFieldName + ") {" + N.LINE_SEPARATOR); fileWrite.write(headSpace + " this();" +
* N.LINE_SEPARATOR);
*
* for (Property prop : entityDef.getPropertyList()) { fileWrite.write(headSpace + " set" +
* getMethodName(prop) + "(" + objectFieldName + "." + prop.getName() + ");" + N.LINE_SEPARATOR); }
*
* fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR); }
*/
@SuppressWarnings("rawtypes")
private static void writeGetSetMethod(final EntityDefinition entityDef, final String pkgName, final Class extendedClass, final EntityMode entityMode,
final Method propName2VarName, final Method propName2MethodName, final boolean fluentSetMethod, final String headSpace, final Writer fileWrite,
final Map> importedClasses) throws IOException {
final Collection props = entityDef.getPropertyList();
if (fluentSetMethod) {
try {
final Class superClass = extendedClass == null && N.notNullOrEmpty(entityDef.getAttribute("extends"))
? ClassUtil.forClass(entityDef.getAttribute("extends"))
: extendedClass;
if (superClass != null) {
Map propSetMethodMap = ClassUtil.getPropSetMethods(superClass);
for (String propName : propSetMethodMap.keySet()) {
if (entityDef.getProperty(propName) != null) {
continue;
}
final String methodName = ClassUtil.invokeMethod(propName2MethodName, propName);
final String fieldName = propName2FieldName(propName);
String typeName = ClassUtil.getParameterizedTypeNameByMethod(propSetMethodMap.get(propName)).replaceAll("java.lang.", "");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + typeName + " "
+ fieldName + ") {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " super.set" + methodName + "(" + fieldName + ");" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
} catch (Exception e) {
logger.error("Failed to generate fluent set methods for parent properties", e);
}
}
for (Property prop : props) {
final String methodName = ClassUtil.invokeMethod(propName2MethodName, prop.getName());
final String fieldName = propName2FieldName(prop.getName());
final String propNameVar = ClassUtil.invokeMethod(propName2VarName, prop.getName());
final String simpleTypeName = getSimpleType(null, prop, pkgName, importedClasses);
// fileWrite.write(N.LINE_SEPARATOR + headSpace + " @Type(\"" + prop.getType().getName() + "\")");
// if (N.notNullOrEmpty(prop.getType().getParameterTypes())) {
// fileWrite.write(N.LINE_SEPARATOR + headSpace + " @Type(\"" + prop.getType().getName() + "\")");
// }
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + simpleTypeName + " get" + methodName + "() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + fieldName + ";" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + simpleTypeName + " "
+ fieldName + ") {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public void set" + methodName + "(" + simpleTypeName + " " + fieldName + ") {"
+ IOUtil.LINE_SEPARATOR);
}
if (EntityMode.POJO != entityMode && EntityMode.POJO_WITH_PROP_NAME_TABLE != entityMode) {
String word = EXTEND_DIRTY_MARKER == entityMode ? "super" : DIRTY_MARKER_IMPL_FIELD_NAME;
fileWrite.write(headSpace + " " + word + ".setUpdatedPropName(" + propNameVar + ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " this." + fieldName + " = " + fieldName + ";" + IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
if (prop.getType().clazz().equals(HBaseColumn.class)) {
HBaseColumnType hbaseColumnType = (HBaseColumnType) prop.getType();
if (hbaseColumnType.getParameterTypes()[0].isEntity()) {
throw new RuntimeException("Family/Entity property can't/unnecessary to be HBaseColumn");
}
String valueTypeName = N.isWrapperType(hbaseColumnType.getElementType().clazz())
? N.unwrap(hbaseColumnType.getElementType().clazz()).getSimpleName()
: hbaseColumnType.getElementType().name();
// =========================
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " /**");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " * Returns the (first) column or an empty column if it's null.");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " */");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + hbaseColumnType.name() + " " + fieldName + "() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return (" + hbaseColumnType.name() + ") (this." + fieldName + " == null ? "
+ HBaseColumn.class.getSimpleName() + ".emptyOf(" + valueTypeName + ".class) : " + fieldName + ");" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + valueTypeName
+ " value) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(
IOUtil.LINE_SEPARATOR + headSpace + " public void set" + methodName + "(" + valueTypeName + " value) {" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " set" + methodName + "(" + HBaseColumn.class.getSimpleName() + ".valueOf(value));" + IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + valueTypeName
+ " value, long version) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public void set" + methodName + "(" + valueTypeName + " value, long version) {"
+ IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " set" + methodName + "(" + HBaseColumn.class.getSimpleName() + ".valueOf(value, version));"
+ IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
} else if ((Collection.class.isAssignableFrom(prop.getType().clazz()) && prop.getType().getParameterTypes()[0].clazz().equals(HBaseColumn.class))
|| (Map.class.isAssignableFrom(prop.getType().clazz()) && prop.getType().getParameterTypes()[1].clazz().equals(HBaseColumn.class))) {
HBaseColumnType hbaseColumnType = null;
if (Collection.class.isAssignableFrom(prop.getType().clazz())) {
hbaseColumnType = (HBaseColumnType) prop.getType().getParameterTypes()[0];
} else {
hbaseColumnType = (HBaseColumnType) prop.getType().getParameterTypes()[1];
}
if (hbaseColumnType.getParameterTypes()[0].isEntity()) {
throw new RuntimeException("Family/Entity property can't/unnecessary to be HBaseColumn");
}
String valueTypeName = N.isWrapperType(hbaseColumnType.getElementType().clazz())
? N.unwrap(hbaseColumnType.getElementType().clazz()).getSimpleName()
: hbaseColumnType.getElementType().name();
// =========================
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " /**");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " * Returns the (first) column or an empty column if it's null.");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " */");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + hbaseColumnType.name() + " " + fieldName + "() {" + IOUtil.LINE_SEPARATOR);
if (Collection.class.isAssignableFrom(prop.getType().clazz())) {
fileWrite.write(headSpace + " return (" + hbaseColumnType.name() + ") (N.isNullOrEmpty(" + fieldName + ") ? "
+ HBaseColumn.class.getSimpleName() + ".emptyOf(" + valueTypeName + ".class) : " + fieldName + ".iterator().next());"
+ IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(headSpace + " return (" + hbaseColumnType.name() + ") (N.isNullOrEmpty(" + fieldName + ") ? "
+ HBaseColumn.class.getSimpleName() + ".emptyOf(" + valueTypeName + ".class) : " + fieldName + ".values().iterator().next());"
+ IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + valueTypeName
+ " value) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(
IOUtil.LINE_SEPARATOR + headSpace + " public void set" + methodName + "(" + valueTypeName + " value) {" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " set" + methodName + "(" + HBaseColumn.class.getSimpleName() + ".valueOf(value));" + IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + valueTypeName
+ " value, long version) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public void set" + methodName + "(" + valueTypeName + " value, long version) {"
+ IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " set" + methodName + "(" + HBaseColumn.class.getSimpleName() + ".valueOf(value, version));"
+ IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " set" + methodName + "(" + hbaseColumnType.name()
+ " hbaseColumn) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public void set" + methodName + "(" + hbaseColumnType.name() + " hbaseColumn) {"
+ IOUtil.LINE_SEPARATOR);
}
if (EntityMode.POJO != entityMode && EntityMode.POJO_WITH_PROP_NAME_TABLE != entityMode) {
String word = EXTEND_DIRTY_MARKER == entityMode ? "super" : DIRTY_MARKER_IMPL_FIELD_NAME;
fileWrite.write(headSpace + " " + word + ".setUpdatedPropName(" + propNameVar + ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " if (" + fieldName + " == null) {" + IOUtil.LINE_SEPARATOR);
if (prop.getType().clazz().equals(SortedSet.class) || prop.getType().clazz().equals(NavigableSet.class)) {
fileWrite.write(headSpace + " " + fieldName + " = new java.util.TreeSet<" + prop.getType().getParameterTypes()[0].name()
+ ">(HBaseColumn.DESC_HBASE_COLUMN_COMPARATOR);" + IOUtil.LINE_SEPARATOR);
} else if (prop.getType().clazz().equals(SortedMap.class) || prop.getType().clazz().equals(NavigableMap.class)) {
fileWrite.write(headSpace + " " + fieldName + " = new java.util.TreeMap<" + prop.getType().getParameterTypes()[0].name() + ", "
+ prop.getType().getParameterTypes()[1].name() + ">(HBaseColumn.DESC_HBASE_VERSION_COMPARATOR);" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(headSpace + " " + fieldName + " = N.newInstance(" + ClassUtil.getCanonicalClassName(prop.getType().clazz())
+ ".class);" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " } else {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + fieldName + ".clear();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
if (Collection.class.isAssignableFrom(prop.getType().clazz())) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " " + fieldName + ".add(hbaseColumn);" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(
IOUtil.LINE_SEPARATOR + headSpace + " " + fieldName + ".put(hbaseColumn.version(), hbaseColumn);" + IOUtil.LINE_SEPARATOR);
}
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " add" + methodName + "(" + valueTypeName
+ " value) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(
IOUtil.LINE_SEPARATOR + headSpace + " public void add" + methodName + "(" + valueTypeName + " value) {" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " add" + methodName + "(" + HBaseColumn.class.getSimpleName() + ".valueOf(value));" + IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " add" + methodName + "(" + valueTypeName
+ " value, long version) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public void add" + methodName + "(" + valueTypeName + " value, long version) {"
+ IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " add" + methodName + "(" + HBaseColumn.class.getSimpleName() + ".valueOf(value, version));"
+ IOUtil.LINE_SEPARATOR);
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// =========================
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " add" + methodName + "(" + hbaseColumnType.name()
+ " hbaseColumn) {" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public void add" + methodName + "(" + hbaseColumnType.name() + " hbaseColumn) {"
+ IOUtil.LINE_SEPARATOR);
}
if (EntityMode.POJO != entityMode && EntityMode.POJO_WITH_PROP_NAME_TABLE != entityMode) {
String word = EXTEND_DIRTY_MARKER == entityMode ? "super" : DIRTY_MARKER_IMPL_FIELD_NAME;
fileWrite.write(headSpace + " " + word + ".setUpdatedPropName(" + propNameVar + ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " if (" + fieldName + " == null) {" + IOUtil.LINE_SEPARATOR);
if (prop.getType().clazz().equals(SortedSet.class) || prop.getType().clazz().equals(NavigableSet.class)) {
fileWrite.write(headSpace + " " + fieldName + " = new java.util.TreeSet<" + prop.getType().getParameterTypes()[0].name()
+ ">(HBaseColumn.DESC_HBASE_COLUMN_COMPARATOR);" + IOUtil.LINE_SEPARATOR);
} else if (prop.getType().clazz().equals(SortedMap.class) || prop.getType().clazz().equals(NavigableMap.class)) {
fileWrite.write(headSpace + " " + fieldName + " = new java.util.TreeMap<" + prop.getType().getParameterTypes()[0].name() + ", "
+ prop.getType().getParameterTypes()[1].name() + ">(HBaseColumn.DESC_HBASE_VERSION_COMPARATOR);" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(headSpace + " " + fieldName + " = N.newInstance(" + ClassUtil.getCanonicalClassName(prop.getType().clazz())
+ ".class);" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
if (Collection.class.isAssignableFrom(prop.getType().clazz())) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " " + fieldName + ".add(hbaseColumn);" + IOUtil.LINE_SEPARATOR);
} else {
fileWrite.write(
IOUtil.LINE_SEPARATOR + headSpace + " " + fieldName + ".put(hbaseColumn.version(), hbaseColumn);" + IOUtil.LINE_SEPARATOR);
}
if (fluentSetMethod) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return this;" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
}
/**
* Write hash code method.
*
* @param entityDef
* @param headSpace
* @param fileWrite
* @param extendedClass
* @param hashEqualsWithParentProperties
* @param utilClass
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeHashCodeMethod(final EntityDefinition entityDef, final String headSpace, final Writer fileWrite, final Class extendedClass,
final boolean hashEqualsWithParentProperties, final Class utilClass) throws IOException {
if (hasHashEqualsProperty(entityDef) == false) {
return;
}
final String utilClassName = ClassUtil.getSimpleClassName(utilClass);
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public int hashCode() {" + IOUtil.LINE_SEPARATOR);
String st = headSpace + " int h = 17;";
for (Property prop : entityDef.getIdPropertyList()) {
String attr = prop.getAttribute("hashEquals");
if (attr == null || Boolean.valueOf(attr)) {
st += (IOUtil.LINE_SEPARATOR + headSpace + " ");
st += ("h = 31 * h + " + utilClassName + ".hashCode(" + propName2FieldName(prop.getName()) + ");");
}
}
for (Property prop : entityDef.getPropertyList()) {
if (!prop.isId()) {
String attr = prop.getAttribute("hashEquals");
if (attr == null || Boolean.valueOf(attr)) {
st += (IOUtil.LINE_SEPARATOR + headSpace + " ");
st += ("h = 31 * h + " + utilClassName + ".hashCode(" + propName2FieldName(prop.getName()) + ");");
}
}
}
if (extendedClass != null && hashEqualsWithParentProperties) {
Map propMethodMap = ClassUtil.getPropGetMethods(extendedClass);
for (Method method : propMethodMap.values()) {
st += (IOUtil.LINE_SEPARATOR + headSpace + " ");
st += ("h = 31 * h + " + utilClassName + ".hashCode(" + method.getName() + "());");
}
}
st += IOUtil.LINE_SEPARATOR;
st += (IOUtil.LINE_SEPARATOR + headSpace + " return h;");
fileWrite.write(st + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
/**
* Write equal method.
*
* @param entityDef
* @param headSpace
* @param fileWrite
* @param extendedClass
* @param hashEqualsWithParentProperties
* @param utilClass
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeEqualMethod(final EntityDefinition entityDef, final String headSpace, final Writer fileWrite, final Class extendedClass,
boolean hashEqualsWithParentProperties, final Class utilClass) throws IOException {
if (hasHashEqualsProperty(entityDef) == false) {
return;
}
final String utilClassName = ClassUtil.getSimpleClassName(utilClass);
String hashEqualsWithParentPropertiesAttr = entityDef.getAttribute("hashEqualsWithParentProperties");
if (N.notNullOrEmpty(hashEqualsWithParentPropertiesAttr)) {
hashEqualsWithParentProperties = Boolean.valueOf(hashEqualsWithParentPropertiesAttr);
}
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean equals(Object obj) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " if (this == obj) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return true;" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " if (obj instanceof " + entityDef.getName() + ") {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " final " + entityDef.getName() + " other = (" + entityDef.getName() + ") obj;" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return ");
int i = 0;
for (Property prop : entityDef.getIdPropertyList()) {
String attr = prop.getAttribute("hashEquals");
if (attr == null || Boolean.valueOf(attr)) {
if (i++ > 0) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " && ");
}
String fieldName = propName2FieldName(prop.getName());
fileWrite.write(utilClassName + ".equals(" + fieldName + ", other." + fieldName + ")");
}
}
for (Property prop : entityDef.getPropertyList()) {
if (!prop.isId()) {
String attr = prop.getAttribute("hashEquals");
if (attr == null || Boolean.valueOf(attr)) {
if (i++ > 0) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " && ");
}
String fieldName = propName2FieldName(prop.getName());
fileWrite.write(utilClassName + ".equals(" + fieldName + ", other." + fieldName + ")");
}
}
}
if (extendedClass != null && hashEqualsWithParentProperties) {
Map propMethodMap = ClassUtil.getPropGetMethods(extendedClass);
for (Method method : propMethodMap.values()) {
if (i++ > 0) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " && ");
}
String methodName = method.getName() + "()";
fileWrite.write(utilClassName + ".equals(" + methodName + ", other." + methodName + ")");
}
}
fileWrite.write(";" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return false;" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
/**
* Write to string method.
*
* @param entityDef
* @param headSpace
* @param fileWrite
* @param extendedClass
* @param toStringWithParentProperties
* @param utilClass
* @throws IOException Signals that an I/O exception has occurred.
*/
private static void writeToStringMethod(final EntityDefinition entityDef, final String headSpace, final Writer fileWrite, final Class extendedClass,
final boolean toStringWithParentProperties, final Class utilClass) throws IOException {
if (hasToStringProperty(entityDef) == false) {
return;
}
final String utilClassName = ClassUtil.getSimpleClassName(utilClass);
String st = "";
int i = 0;
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public String toString() {" + IOUtil.LINE_SEPARATOR);
/*
* if (entityDef.getEntityType() == EntityType.ENTITY) { st = headSpace + " return \"" + entityDef.getName() +
* "={\"" + N.LINE_SEPARATOR + headSpace + " "; } else { st = headSpace + " return \"{\"" + N.LINE_SEPARATOR +
* headSpace + " "; }
*/
if (extendedClass != null && toStringWithParentProperties) {
Map propMethodMap = ClassUtil.getPropGetMethods(extendedClass);
for (Method method : propMethodMap.values()) {
String propName = ClassUtil.getPropNameByMethod(method);
if (i++ == 0) {
st += (" return \"{" + propName + "=\"" + " + " + utilClassName + ".toString(" + method.getName() + "())");
} else {
st += (" + \", " + propName + "=\"" + " + " + utilClassName + ".toString(" + method.getName() + "())");
}
st += (IOUtil.LINE_SEPARATOR + headSpace + " ");
}
}
for (Property prop : entityDef.getIdPropertyList()) {
String attr = prop.getAttribute("toString");
if (attr == null || Boolean.valueOf(attr)) {
if (i++ == 0) {
st += (" return \"{" + propName2FieldName(prop.getName()) + "=\"" + " + " + utilClassName + ".toString("
+ propName2FieldName(prop.getName()) + ")");
} else {
st += (" + \", " + propName2FieldName(prop.getName()) + "=\"" + " + " + utilClassName + ".toString("
+ propName2FieldName(prop.getName()) + ")");
}
st += (IOUtil.LINE_SEPARATOR + headSpace + " ");
}
}
for (Property prop : entityDef.getPropertyList()) {
if (!prop.isId()) {
String attr = prop.getAttribute("toString");
if (attr == null || Boolean.valueOf(attr)) {
if (i++ == 0) {
st += (" return \"{" + propName2FieldName(prop.getName()) + "=\"" + " + " + utilClassName + ".toString("
+ propName2FieldName(prop.getName()) + ")");
} else {
st += (" + \", " + propName2FieldName(prop.getName()) + "=\"" + " + " + utilClassName + ".toString("
+ propName2FieldName(prop.getName()) + ")");
}
st += (IOUtil.LINE_SEPARATOR + headSpace + " ");
}
}
}
st = st.substring(0, st.length() - (IOUtil.LINE_SEPARATOR + headSpace + " ").length());
if (st.indexOf(" + \",") > 0) {
st += (IOUtil.LINE_SEPARATOR + headSpace + " ") + " + \"}\";";
} else {
st += " + \"}\";";
}
fileWrite.write(st + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
/**
* Checks for hash equals property.
*
* @param entityDef
* @return true, if successful
*/
private static boolean hasHashEqualsProperty(final EntityDefinition entityDef) {
final Collection props = entityDef.getPropertyList();
for (Property prop : props) {
String attr = prop.getAttribute("hashEquals");
if (attr == null || Boolean.valueOf(attr)) {
return true;
}
}
return false;
}
/**
* Checks for to string property.
*
* @param entityDef
* @return true, if successful
*/
private static boolean hasToStringProperty(final EntityDefinition entityDef) {
final Collection props = entityDef.getPropertyList();
for (Property prop : props) {
String attr = prop.getAttribute("toString");
if (attr == null || Boolean.valueOf(attr)) {
return true;
}
}
return false;
}
/**
* Write dirty marker method.
*
* @param entityDef
* @param entityMode
* @param headSpace
* @param fileWrite
* @throws IOException Signals that an I/O exception has occurred.
*/
/*
* static void writeHES(String headSpace, Writer fileWrite) throws IOException { fileWrite.write(N.LINE_SEPARATOR);
* fileWrite.write(headSpace + " public int hashCode() {" + N.LINE_SEPARATOR); fileWrite.write(headSpace +
* " return activeRecordImpl.hashCode();" + N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" +
* N.LINE_SEPARATOR);
*
* fileWrite.write(N.LINE_SEPARATOR); fileWrite.write(headSpace + " public boolean equals(Object obj) {" +
* N.LINE_SEPARATOR); fileWrite.write(headSpace + " return activeRecordImpl.equals(anObject);" +
* N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
*
* fileWrite.write(N.LINE_SEPARATOR); fileWrite.write(headSpace + " public String toString() {" +
* N.LINE_SEPARATOR); fileWrite.write(headSpace + " return activeRecordImpl.toString();" + N.LINE_SEPARATOR);
* fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR); }
*/
private static void writeDirtyMarkerMethod(final EntityDefinition entityDef, final EntityMode entityMode, final String headSpace, final Writer fileWrite)
throws IOException {
if (!(EntityMode.POJO.equals(entityMode) || EntityMode.POJO_WITH_PROP_NAME_TABLE.equals(entityMode))) {
String word = EXTEND_DIRTY_MARKER == entityMode ? "super" : DIRTY_MARKER_IMPL_FIELD_NAME;
if (entityMode == EntityMode.IMPL_DIRTY_MARKER) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public String entityName() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + ENTITY_NAME_VAR + ";" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (((EntityMode.IMPL_DIRTY_MARKER == entityMode) || (EntityMode.EXTEND_DIRTY_MARKER == entityMode))
&& (entityDef.getEntiyPropertyList().size() > 0)) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean isDirty() {" + IOUtil.LINE_SEPARATOR);
String temp = headSpace + " return " + word + ".isDirty()";
Collection props = entityDef.getPropertyList();
for (Property prop : props) {
if (prop.getColumnType().isEntity() && !prop.isCollection()) {
temp += (IOUtil.LINE_SEPARATOR + headSpace + " || (" + prop.getName() + " == null ? false : " + prop.getName()
+ ".isDirty())");
}
}
for (Property prop : props) {
if (prop.isCollection()) {
temp += (IOUtil.LINE_SEPARATOR + headSpace + " || (" + prop.getName() + " == null ? false : " + word + ".isEntityDirty("
+ prop.getName() + "))");
}
}
temp += (";" + IOUtil.LINE_SEPARATOR);
fileWrite.write(temp);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
} else if (entityMode == IMPL_DIRTY_MARKER) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean isDirty() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + word + ".isDirty();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (entityMode == IMPL_DIRTY_MARKER) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean isDirty(String propName) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + word + ".isDirty(propName);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (((EntityMode.IMPL_DIRTY_MARKER == entityMode) || (EntityMode.EXTEND_DIRTY_MARKER == entityMode))
&& (entityDef.getEntiyPropertyList().size() > 0)) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void markDirty(boolean isDirty) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + word + ".markDirty(isDirty);" + IOUtil.LINE_SEPARATOR);
Collection props = entityDef.getPropertyList();
for (Property prop : props) {
if (prop.getColumnType().isEntity() && !prop.isCollection()) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " if (" + prop.getName() + " != null) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + prop.getName() + ".markDirty(isDirty);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
for (Property prop : props) {
if (prop.isCollection()) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " if (" + prop.getName() + " != null) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + word + ".markEntityDirty(" + prop.getName() + ", isDirty);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
} else if (entityMode == IMPL_DIRTY_MARKER) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void markDirty(boolean isDirty) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + word + ".markDirty(isDirty);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (entityMode == IMPL_DIRTY_MARKER) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void markDirty(String propName, boolean isDirty) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + word + ".markDirty(propName, isDirty);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void markDirty(Collection propNames, boolean isDirty) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + word + ".markDirty(propNames, isDirty);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public Set signedPropNames() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + word + ".signedPropNames();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public Set dirtyPropNames() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + word + ".dirtyPropNames();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void freeze() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " " + word + ".freeze();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean frozen() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + word + ".frozen();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " @XmlTransient" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public long version() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return " + word + ".version();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
// if (inheritLevel == IMPL_DIRTY_MARKER) {
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " public int hashCode() {" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " return dirtyMarkerImpl.hashCode(this);"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
//
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " public boolean equals(Object obj) {"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace
// + " return dirtyMarkerImpl.equals(this, anObject);" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
//
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " public String toString() {" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " return dirtyMarkerImpl.toString(this);"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// }
/*
* if ((inheritLevel == IMPL_ACTIVE_RECORD) || (inheritLevel == IMPL_DIRTY_MARKER)) {
* fileWrite.write(N.LINE_SEPARATOR); fileWrite.write(headSpace + " public void clear() {" +
* N.LINE_SEPARATOR); fileWrite.write(headSpace + " " + word + ".clear();" + N.LINE_SEPARATOR);
* fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR); }
*/
// if (hasCollectionProp) {
// // write isDirty(Collection objects)
// // method
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace
// + " private boolean hasDirtyEntity(Collection entities) {" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " if (entities == null) {" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " return false;" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " } else {" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " for (" +
// DirtyMarker.class.getName()
// + " entity : entities) {" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " if (entity.isDirty()) {"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " return true;" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " return false;" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
//
// // write isDirty(Collection entities,
// // String propName) method
// /*
// * fileWrite.write(N.LINE_SEPARATOR); fileWrite.write(headSpace +
// "
// * private boolean isDirty(Collection
// * entities, String propName) {" + N.LINE_SEPARATOR);
// * fileWrite.write(headSpace + " if (objects == null) {" +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " return false;"
// +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " } else {" +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " for
// * (DirtyMarker entity : entities) {" + N.LINE_SEPARATOR);
// * fileWrite.write(headSpace + " if (entity.isDirty(propName)) {"
// +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " return true;"
// +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " return false;"
// +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" +
// * N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" +
// * N.LINE_SEPARATOR);
// */
//
// // write markDirty(Collection entities,
// // boolean isDirty) method
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace
// + " private void markEntityDirty(Collection entities, boolean isDirty) {"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " if (entities != null) {" +
// N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " for (" +
// DirtyMarker.class.getName()
// + " entity : entities) {" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " entity.markDirty(isDirty);"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// }
}
}
/**
* Gets the simple type.
*
* @param type
* @param prop
* @param pkgName
* @param importedClasses
* @return
*/
private static String getSimpleType(Type type, final Property prop, final String pkgName, final Map> importedClasses) {
type = type == null ? prop.getType() : type;
final Class typeClass = type.clazz();
String typeName = null;
if (type.isGenericType()) {
typeName = type.name();
} else {
Class clazz = type.clazz();
typeName = Object.class.equals(clazz) && !type.name().equals(ObjectType.OBJECT) ? type.name() : clazz.getCanonicalName();
}
if (typeClass.isArray()) {
String componentClassName = StringUtil.substring(typeName, 0, typeName.indexOf('['));
if (importedClasses.containsKey(componentClassName)) {
typeName = typeName.replaceAll(componentClassName.substring(0, componentClassName.lastIndexOf('.') + 1), "");
}
} else if (typeName.startsWith("java.lang.") || (importedClasses.containsValue(typeClass) && N.notNullOrEmpty(ClassUtil.getPackageName(typeClass)))) {
typeName = typeName.replace(ClassUtil.getPackageName(typeClass) + ".", "");
}
if (type.isGenericType()) {
Type[] paramTypes = type.getParameterTypes();
if (N.notNullOrEmpty(paramTypes)) {
String tmp = typeName.substring(0, typeName.indexOf('<')) + "<";
for (int i = 0, len = paramTypes.length; i < len; i++) {
if (i > 0) {
tmp += ", ";
}
tmp += getSimpleType(paramTypes[i], null, pkgName, importedClasses);
}
tmp += ">";
typeName = tmp;
// for (Type paraType : parameterTypes) {
// try {
// if (isUsualType(paraType.getTypeClass().getCanonicalName())) {
// typeName = typeName.replace(ClassUtil.getPackageName(paraType.getTypeClass()) + ".", "");
// }
// } catch (Exception e) {
// // ignore;
// }
// }
}
}
if ((ClassUtil.getPackageName(TypeType.class) + "." + TypeType.TYPE).equals(typeName)) {
typeName = typeName + "";
}
if (N.notNullOrEmpty(pkgName)) {
if (prop != null && prop.getColumnType().isEntity()) {
EntityDefinition columnEntityDef = prop.getColumnEntityDef();
String objectType = pkgName + _PERIOD + columnEntityDef.getName();
typeName = typeName.replace(objectType, columnEntityDef.getName());
} else {
if ((typeName.indexOf(pkgName) == 0) && (typeName.lastIndexOf(_PERIOD) == pkgName.length())) {
typeName = typeName.replace(pkgName + _PERIOD, "");
}
}
}
return typeName;
}
/**
*
* @param entityDef
* @return Set
*/
private static Set> getUsualType(final EntityDefinition entityDef) {
Set> set = N.newLinkedHashSet();
for (Property prop : entityDef.getPropertyList()) {
if (prop.getColumnType().isEntity()) {
//
} else {
getUsualType(set, prop.getType());
}
}
return set;
}
/**
* Gets the usual type.
*
* @param set
* @param type
* @return
*/
private static void getUsualType(final Set> set, final Type type) {
if (isUsualType(type.clazz().getCanonicalName())) {
set.add(type.clazz());
}
if (type.isGenericType()) {
Type[] parameterTypes = type.getParameterTypes();
if (N.notNullOrEmpty(parameterTypes)) {
for (Type e : parameterTypes) {
getUsualType(set, e);
}
}
}
}
/**
* Checks if is usual type.
*
* @param canonicalName
* @return true, if is usual type
*/
private static boolean isUsualType(String canonicalName) {
int lastIndex = canonicalName.lastIndexOf('.');
return lastIndex > 0 && USUAL_TYPES.contains(canonicalName.substring(0, lastIndex));
}
/**
* Gets the package name.
*
* @param entityDef
* @return
*/
/*
* static String getSimpleType(EntityDefinition entityDef, String pkgName) { String type =
* entityDef.getJavaType(); type = type.replace(pkgName + _PERIOD + entityDef.getName(), entityDef.getName());
*
* if (type.startsWith("java.lang.")) { type = type.replace("java.lang.", ""); }
*
* if (type.startsWith("java.util.")) { type = type.replace("java.util.", ""); }
*
* return type; }
*/
private static String getPackageName(final EntityDefinition entityDef) {
return entityDef.getAttribute(EntityDefEle.PACKAGE);
}
/**
* Gets the id para string.
*
* @param pkgName
* @param idPropList
* @param importedClasses
* @return
*/
private static String getIdParaString(final String pkgName, final List idPropList, final Map> importedClasses) {
String idPropParaStr = "";
if (idPropList.size() > 0) {
int length = 8;
String headEmpty = "";
for (int j = 0; j < length; j++) {
headEmpty += " ";
}
int i = 1;
for (Property idProp : idPropList) {
String fieldName = propName2FieldName(idProp.getName());
idPropParaStr += (getSimpleType(null, idProp, pkgName, importedClasses) + " " + fieldName);
if (i < idPropList.size()) {
idPropParaStr += ", ";
}
length += idPropParaStr.length();
if (length > 80) {
length = 8 + idPropParaStr.length();
idPropParaStr += (IOUtil.LINE_SEPARATOR + headEmpty + idPropParaStr);
}
i++;
}
}
return idPropParaStr;
}
/**
* The Enum EntityMode.
*
* @author Haiyang Li
* @version $Revision: 0.8 $
*/
public enum EntityMode {
/** The pojo. */
POJO,
/** The pojo with prop name table. */
POJO_WITH_PROP_NAME_TABLE,
/** The impl dirty marker. */
IMPL_DIRTY_MARKER,
/** The extend dirty marker. */
EXTEND_DIRTY_MARKER;
}
}
