com.landawn.abacus.util.CodeGenerator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of abacus-android-se-jdk7 Show documentation
Show all versions of abacus-android-se-jdk7 Show documentation
A general programming library in Java
/*
* 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.CodeGenerator.EntityMode.EXTEND_ACTIVE_RECORD;
import static com.landawn.abacus.util.CodeGenerator.EntityMode.EXTEND_DIRTY_MARKER;
import static com.landawn.abacus.util.CodeGenerator.EntityMode.IMPL_ACTIVE_RECORD;
import static com.landawn.abacus.util.CodeGenerator.EntityMode.IMPL_DIRTY_MARKER;
import static com.landawn.abacus.util.D._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.Field;
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.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
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.ActiveRecord;
import com.landawn.abacus.DataSet;
import com.landawn.abacus.DirtyMarker;
import com.landawn.abacus.EntityId;
import com.landawn.abacus.LockMode;
import com.landawn.abacus.condition.Condition;
import com.landawn.abacus.core.AbstractActiveRecord;
import com.landawn.abacus.core.AbstractDirtyMarker;
import com.landawn.abacus.core.ActiveRecordImpl;
import com.landawn.abacus.core.DirtyMarkerImpl;
import com.landawn.abacus.core.Seid;
import com.landawn.abacus.exception.AbacusException;
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;
/**
*
* @since 0.8
*
* @author Haiyang Li
*/
public final class CodeGenerator {
private static final Logger logger = LoggerFactory.getLogger(CodeGenerator.class);
/**
* 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 UTF_8
*/
private static final String UTF_8 = "UTF-8";
/**
* 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";
/**
* The field name of ActiveRecord implementation in a generated entity class which implements the ActiveRecord interface. (value is ""activeRecordImpl"")
*/
private static final String ACTIVE_RECORD_IMPL_FIELD_NAME = "activeRecordImpl";
/**
* Field USUAL_TYPES.
*/
private static Set USUAL_TYPES = new LinkedHashSet<>();
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 CodeGenerator() {
// No instance.
}
static String makePackageFolder(String srcPath, final String pkgName) {
srcPath = (srcPath.endsWith("/") || srcPath.endsWith("\\")) ? srcPath : (srcPath + File.separator);
String classFilePath = (pkgName == null) ? srcPath : (srcPath + pkgName.replace('.', File.separatorChar) + File.separator);
File classFileFolder = new File(classFilePath);
if (!classFileFolder.exists()) {
classFileFolder.mkdirs();
}
return classFilePath;
}
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);
}
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;
}
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;
}
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);
}
public static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml) {
database2EntityDefinitionXml(database, entityDefinitionXml, null);
}
public static void database2EntityDefinitionXml(final Database database, final File entityDefinitionXml, final Map entityDefAttrs) {
Map newEntityNames = new HashMap<>();
Map newPropNames = new HashMap<>();
database2EntityDefinitionXml(database, entityDefinitionXml, entityDefAttrs, newEntityNames, newPropNames, null);
}
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 HashMap<>();
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);
}
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(CodeGenerator.class, TABLE_NAME2ENTITY_NAME, String.class);
}
Map entityColumnName2PropName = new HashMap<>();
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);
}
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(CodeGenerator.class, TABLE_NAME2ENTITY_NAME, String.class);
}
if (newEntityNames == null) {
newEntityNames = new HashMap<>();
}
if (entityColumnName2PropName == null) {
entityColumnName2PropName = new HashMap<>();
}
if (newEntityPropNames == null) {
newEntityPropNames = new HashMap<>();
}
if (excludedeEntityPropNames == null) {
excludedeEntityPropNames = new HashMap<>();
}
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 new AbacusException(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 = new HashSet<>();
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(CodeGenerator.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);
}
static void setAttributes(final Element element, final Map attrs) {
if (attrs != null) {
for (String attrName : attrs.keySet()) {
element.setAttribute(attrName, attrs.get(attrName));
}
}
}
/**
* @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 HashMap<>();
}
if (excludedPropNames == null) {
excludedPropNames = new HashSet<>();
} else {
Set tempList = new HashSet<>(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 new AbacusException(e);
}
}
}
private static Map getEntityElementList(final Document doc) {
Map result = new HashMap<>();
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;
}
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;
}
protected static String getSimplePropNameTableClassName(final String className) {
String simpleClassName = className;
int index = className.lastIndexOf(D._PERIOD);
if (index > -1) {
simpleClassName = className.substring(index + 1);
}
return simpleClassName;
}
public static void entityDefinitionXml2Class(final String domainName, final File entityDefinitionXml, final String srcPath, final EntityMode entityMode) {
entityDefinitionXml2Class(domainName, entityDefinitionXml, srcPath, entityMode, null, null, true);
}
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);
}
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);
}
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);
}
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 HashMap<>();
for (EntityDefinition entityDef : entityDefinitionFactory.getDefinitionList()) {
entityExtendedClasses.put(entityDef.getName(), extendedClass);
}
}
Map>> entityImplementedInterfaces = null;
if (N.notNullOrEmpty(implementedInterfaces)) {
entityImplementedInterfaces = new HashMap<>();
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);
}
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);
}
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);
}
/**
*
* @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 AbacusUtil.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) {
if (entityMode == EntityMode.EXTEND_ACTIVE_RECORD || entityMode == EntityMode.IMPL_ACTIVE_RECORD) {
try {
ClassUtil.forClass("com.landawn.abacus.core.EntityManagerImpl");
} catch (Exception e) {
throw new AbacusException("Active Record model is not supported in this version");
}
}
final EntityDefinitionFactory entityDefinitionFactory = SQLEntityDefinitionFactory.newInstance(domainName, entityDefinitionXml);
final String pkgName = entityDefinitionFactory.getAttribute(EntityDefEle.PACKAGE);
if ((pkgName == null) || (pkgName.length() == 0)) {
throw new AbacusException("the package attribute in 'entityDef' element can't be null or empty");
}
//
// if (generatePropNameTable && ((domainName == null) || (domainName.length() == 0))) {
// throw new AbacusException("the domain attribute in 'entityDef' element can't be null or empty");
// }
if (entityMode == null) {
throw new AbacusException("entityMode can't be null");
}
if (propName2VarName == null) {
propName2VarName = ClassUtil.getDeclaredMethod(CodeGenerator.class, PROP_NAME2VAR_NAME, String.class);
}
if (propName2MethodName == null) {
propName2MethodName = ClassUtil.getDeclaredMethod(CodeGenerator.class, PROP_NAME2METHOD_NAME, String.class);
}
if (entityExtendedClasses == null) {
entityExtendedClasses = new HashMap<>();
}
if (entityImplementedInterfaces == null) {
entityImplementedInterfaces = new HashMap<>();
}
final Class utilClass = utilClassForHashEqualsToString == null ? Objects.class : utilClassForHashEqualsToString;
final String classFilePath = makePackageFolder(srcPath, pkgName);
if (_N.equals(utilClass)) {
File utilClassFile = new File(classFilePath + ClassUtil.getSimpleClassName(_N) + POSTFIX_OF_JAVA_FILE);
if (!utilClassFile.exists()) {
String sourceCode = _N_STRING.replaceFirst("package com.landawn.abacus.util;", N.isNullOrEmpty(pkgName) ? "" : "package " + pkgName + ";");
IOUtil.write(utilClassFile, sourceCode);
}
}
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);
}
}
private static String getPropNameTableClassName(final EntityDefinitionFactory entityDefinitionFactory) {
String propNameTableClass = POSTFIX_OF_PROP_NAME_LIST;
String domainName = entityDefinitionFactory.domainName();
if (N.notNullOrEmpty(domainName)) {
propNameTableClass = N.capitalize(domainName) + propNameTableClass;
}
String pkgName = entityDefinitionFactory.getAttribute(EntityDefEle.PACKAGE);
if (pkgName != null) {
propNameTableClass = pkgName + '.' + propNameTableClass;
}
return propNameTableClass;
}
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 (Throwable e) {
className = attr.indexOf('.') > 0 ? attr : pkgName + "." + attr;
}
extendedClass = ClassUtil.forClass(className);
} catch (Throwable 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(",").trim(true).split(attr);
for (String str : strList) {
String className = str;
try {
try {
ClassUtil.forClass(className);
} catch (Throwable e) {
className = str.indexOf('.') > 0 ? str : pkgName + "." + str;
}
implementedInterfaces.add(ClassUtil.forClass(className));
} catch (Throwable 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_ACTIVE_RECORD == entityMode) {
if (extendedClass == null) {
extendedClass = AbstractActiveRecord.class;
} else {
if (!AbstractActiveRecord.class.isAssignableFrom(extendedClass)) {
throw new AbacusException(
"the class: " + extendedClass.getCanonicalName() + " must extend " + AbstractActiveRecord.class.getCanonicalName());
}
}
} else if (EntityMode.IMPL_ACTIVE_RECORD == entityMode) {
if (extendedClass == null) {
extendedClass = ActiveRecordImpl.class;
}
if (implementedInterfaces.size() == 0) {
implementedInterfaces.add(ActiveRecord.class);
}
} else if (EntityMode.EXTEND_DIRTY_MARKER == entityMode) {
if (extendedClass == null) {
extendedClass = AbstractDirtyMarker.class;
} else {
if (!AbstractDirtyMarker.class.isAssignableFrom(extendedClass)) {
throw new AbacusException(
"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 HashMap<>();
Set> annotationImportClasses = new LinkedHashSet<>();
if (classFile.exists()) {
List lines = IOUtil.readLines(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.readLines(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 new AbacusException(e);
} finally {
IOUtil.close(fileWrite);
}
}
public static void entityDefinitionXml2PropNameTable(final String domainName, final File entityDefinitionXml, final String srcPath) {
entityDefinitionXml2PropNameTable(domainName, entityDefinitionXml, srcPath, null, _N);
}
/**
* @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 AbacusUtil.java
*/
public static void entityDefinitionXml2PropNameTable(String domainName, final File entityDefinitionXml, final String srcPath, Method propName2VarName,
Class utilClass) {
if (propName2VarName == null) {
propName2VarName = ClassUtil.getDeclaredMethod(CodeGenerator.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 = 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, utilClass);
}
static void entityDefinition2PropNameTable(final EntityDefinitionFactory entityDefinitionFactory, final File propNameDefClassFile, final String pkgName,
final Method propName2VarName, final Class utilClass) {
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, utilClass);
// -------- 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);
}
}
private static void writePropNameTable(final EntityDefinition[] entityDefs, final Method propName2VarName, final String headSpace, final Writer fileWrite,
final Class utilClass) 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, utilClass);
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);
}
}
}
/**
*
* @param entityDef
* @param propName2VarName
* @param headSpace
* @param fileWrite
* @throws IOException
*/
private static void writePropNameField(final EntityDefinition entityDef, final Method propName2VarName, final String headSpace, final Writer fileWrite,
final Class utilClass) throws IOException {
final StringBuilder sb = ObjectFactory.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());
ObjectFactory.recycle(sb);
}
public static void entityDefinitionXml2ColumnNameTable(final String domainName, final File entityDefinitionXml, final String srcPath) {
entityDefinitionXml2ColumnNameTable(domainName, entityDefinitionXml, srcPath, null, _N);
}
/**
*
* @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 AbacusUtil.java
*/
public static void entityDefinitionXml2ColumnNameTable(final String domainName, final File entityDefinitionXml, final String srcPath,
Method columnName2VarName, Class utilClass) {
if (columnName2VarName == null) {
columnName2VarName = ClassUtil.getDeclaredMethod(CodeGenerator.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 = 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 = N.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, utilClass);
// -------- 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);
}
}
private static void writeColumnNameTable(final Collection entityDefs, final Method columnName2VarName, final String headSpace,
final Writer fileWrite, final Class utilClass) 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, utilClass);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
private static void writeColumnNameField(final EntityDefinition entityDef, final Method columnName2VarName, final String headSpace, final Writer fileWrite,
final Class utilClass) throws IOException {
final StringBuilder sb = ObjectFactory.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());
ObjectFactory.recycle(sb);
}
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);
}
public static void generateSQLMapperIdTable(final SQLMapper sqlMapper, final String srcPath, final String pkgName, final String className,
Method id2VarName) {
String classFilePath = makePackageFolder(srcPath, pkgName);
File sqlMapperTableClassFile = new File(classFilePath + className + POSTFIX_OF_JAVA_FILE);
if (sqlMapperTableClassFile.exists()) {
sqlMapperTableClassFile.delete();
}
if (id2VarName == null) {
id2VarName = ClassUtil.getDeclaredMethod(CodeGenerator.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);
}
}
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 AbacusUtil.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 AbacusUtil.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 = new LinkedHashMap<>(N.initHashCapacity(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 (Throwable e) {
// ignore.
}
if (eleClass == null && N.notNullOrEmpty(packageName)) {
try {
eleClass = ClassUtil.forClass(packageName + "." + attrs.getTypeParameters()[0]);
} catch (Throwable 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 (Throwable e) {
// ignore.
}
try {
valueClass = ClassUtil.forClass(attrs.getTypeParameters()[1]);
} catch (Throwable 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 (Throwable e) {
// ignore.
}
}
if (valueClass == null) {
try {
valueClass = ClassUtil.forClass(packageName + "." + attrs.getTypeParameters()[1]);
if (valueClass != null) {
valueTypeName = valueClass.getCanonicalName();
}
} catch (Throwable 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 + N.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 = _N_STRING.replaceFirst("package com.landawn.abacus.util;",
N.isNullOrEmpty(packageName) ? "" : "package " + packageName + ";");
IOUtil.write(utilClassFile, sourceCode);
}
}
final File classFile = new File(dirFile.getAbsolutePath() + IOUtil.FILE_SEPARATOR + className + ".java");
IOUtil.createIfNotExists(classFile);
final BufferedWriter writer = IOUtil.newBufferedWriter(classFile);
try {
IOUtil.writeLine(writer, "package " + packageName + ";");
// 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 e) {
throw N.toRuntimeException(e);
} finally {
IOUtil.closeQuietly(writer);
}
}
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);
}
}
}
/**
* Write the generated methods by the fields defined the in specified class to the source file.
*
*
* Add below comments to specified the section where the generated methods should be written to
*
* =====>
*
* <=====
*
*
* @param srcDir
* @param cls
*/
public static void writeClassMethod(final File srcDir, final Class cls) {
writeClassMethod(srcDir, cls, false, false, false, null, null, Objects.class);
}
/**
* Write the generated methods by the fields defined the in specified class to the source file.
*
*
* Add below comments to specified the section where the generated methods should be written to
*
* =====>
*
* <=====
*
*
* @param srcDir
* @param cls
* @param constructor generate constructor
* @param copyMethod generate the copy method.
* @param fluentSetter
* @param ignoreFieldNames
* @param fieldName2MethodName
* @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 AbacusUtil.jar for Methods {@code hashCode/equals/toString}.
*/
public static void writeClassMethod(final File srcDir, final Class cls, final boolean constructor, final boolean copyMethod, final boolean fluentSetter,
Set ignoreFieldNames, final Map fieldName2MethodName, final Class utilClassForHashEqualsToString) {
writeClassMethod(srcDir, cls, constructor, copyMethod, fluentSetter, ignoreFieldNames, fieldName2MethodName, ParentPropertyMode.FIRST,
ParentPropertyMode.FIRST, utilClassForHashEqualsToString);
}
/**
* Write the generated methods by the fields defined the in specified class to the source file.
*
*
* Add below comments to specified the section where the generated methods should be written to
*
* =====>
*
* <=====
*
*
* @param srcDir
* @param cls
* @param constructor
* @param copyMethod
* @param fluentSetter
* @param ignoreFieldNames
* @param fieldName2MethodName
* @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 AbacusUtil.jar for Methods {@code hashCode/equals/toString}.
*/
public static void writeClassMethod(final File srcDir, final Class cls, final boolean constructor, final boolean copyMethod, final boolean fluentSetter,
Set ignoreFieldNames, final Map fieldName2MethodName, final ParentPropertyMode parentPropertyModeForHashEquals,
final ParentPropertyMode parentPropertyModeForToString, final Class utilClassForHashEqualsToString) {
final Package pkg = cls.getPackage();
final String clsSourcePath = srcDir.getAbsolutePath()
+ (pkg == null ? "" : IOUtil.FILE_SEPARATOR + N.replaceAll(pkg.getName(), ".", IOUtil.FILE_SEPARATOR)) + IOUtil.FILE_SEPARATOR
+ cls.getSimpleName() + ".java";
final File clsSourceFile = new File(clsSourcePath);
if (clsSourceFile.exists() == false) {
throw new RuntimeException("No source file found by path: " + clsSourcePath + " for class: " + cls.getCanonicalName());
}
if (ignoreFieldNames == null) {
ignoreFieldNames = new HashSet<>();
}
final String simpleClassName = cls.getSimpleName();
final List lines = ImmutableList.of(IOUtil.readLines(clsSourceFile));
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()));
}
}
if (N.isNullOrEmpty(fieldTypes)) {
return;
}
final BiMap> importedClasses = new BiMap<>();
boolean hasGenericTypeField = false;
for (int i = 0, size = lines.size(); i < size; i++) {
final String tmp = "class " + simpleClassName;
while (i < size && lines.get(i).indexOf(tmp) < 0) {
String line = lines.get(i);
if (line.startsWith("import ") && line.endsWith(";") && line.indexOf(" static ") < 0) {
String clsName = N.between(line, ' ', line.lastIndexOf(';')).get();
importedClasses.put(clsName, ClassUtil.forClass(clsName));
}
i++;
}
int start = i + 1;
while (i < size && (lines.get(i).indexOf(") {") < 0 && lines.get(i).indexOf("}") < 0)) {
i++;
}
int end = i;
if (start < size && end <= size && start < end) {
for (int j = start; j < end; j++) {
String line = lines.get(j).trim().replaceAll(" ", " ");
int lastIndex = line.lastIndexOf(' ');
String fieldName = null;
if (lastIndex > 0 && line.endsWith(";") && fieldTypes.containsKey((fieldName = line.substring(lastIndex + 1, line.length() - 1)))) {
int firstIdx = line.indexOf('<');
if (firstIdx > 0) {
hasGenericTypeField = true;
int fromIndex = line.lastIndexOf(' ', firstIdx < 0 ? lastIndex - 1 : firstIdx);
if (fromIndex >= 0) {
String typeName = line.substring(fromIndex + 1, lastIndex);
if ((typeName = typeName.trim()).length() > 0) {
// for java.util.Date
if (importedClasses.containsKey("java.util.Date")) {
typeName = N.replaceAll(typeName, "", "");
typeName = N.replaceAll(typeName, "", " java.util.Date>");
typeName = N.replaceAll(typeName, ",Date>", ",java.util.Date>");
typeName = N.replaceAll(typeName, " Date,", " java.util.Date,");
typeName = N.replaceAll(typeName, ",Date,", ",java.util.Date,");
typeName = N.replaceAll(typeName, " utilClass = utilClassForHashEqualsToString == null ? Objects.class : utilClassForHashEqualsToString;
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 = _N_STRING.replaceFirst("package com.landawn.abacus.util;",
N.isNullOrEmpty(packageName) ? "" : "package " + packageName + ";");
IOUtil.write(utilClassFile, sourceCode);
}
}
try (StringWriter writer = new StringWriter()) {
writeClassMethod(cls, ClassUtil.getSimpleClassName(cls), cls.getSuperclass(), packageName, fieldTypes, constructor, copyMethod, fluentSetter,
parentPropertyModeForHashEquals, parentPropertyModeForToString, fieldName2MethodName, importedClasses, utilClass, writer);
int start = -1, end = -1;
for (int i = 0, size = lines.size(); i < size; i++) {
if (lines.get(i).indexOf("=====>") > 0) {
start = i;
} else if (lines.get(i).indexOf("<=====") > 0) {
end = i;
}
}
final List newLines = new ArrayList<>();
if (start >= 0 && end >= 0) {
newLines.addAll(lines.subList(0, start + 1));
newLines.add(writer.toString());
newLines.addAll(lines.subList(end, lines.size()));
} else {
for (int i = 0, size = lines.size(); i < size; i++) {
if (lines.get(i).indexOf(") {") > 0) {
String tmp = lines.get(i).trim();
if (tmp.startsWith("public ") && (tmp.indexOf(cls.getSimpleName()) > 0 || tmp.indexOf(" get") > 0 || tmp.indexOf(" set") > 0)) {
start = i;
while (start-- > 0 && lines.get(start).trim().startsWith("@")) {
}
while (lines.get(start).trim().length() == 0 && start-- > 0) {
}
break;
}
}
}
if (start < 0) {
for (int i = lines.size() - 1; i >= 0; i--) {
if (lines.get(i).trim().startsWith("}")) {
start = i - 1;
break;
}
}
}
newLines.addAll(lines.subList(0, start + 1));
newLines.add(writer.toString());
newLines.add("}");
}
if (!ClassUtil.getPackageName(utilClass).equals(packageName) && !_N.equals(utilClass) && fieldTypes.size() > 0) {
String importUtilClass = "import " + utilClass.getCanonicalName() + ";";
for (int i = 0, size = newLines.size(); i < size; i++) {
if (newLines.get(i).indexOf(importUtilClass) >= 0) {
break;
} else if (newLines.get(i).indexOf("public ") >= 0) {
int ins = 0;
if (newLines.get(i - 1).trim().length() > 0) {
newLines.add(i + ins++, N.EMPTY_STRING);
}
newLines.add(i + ins++, importUtilClass);
newLines.add(i + ins++, N.EMPTY_STRING);
break;
}
}
}
if (hasGenericTypeField) {
// String importTypeClass = "import com.landawn.abacus.annotation.Type;";
//
// for (int i = 0, size = newLines.size(); i < size; i++) {
// if (newLines.get(i).indexOf(importTypeClass) >= 0) {
// break;
// } else if (newLines.get(i).indexOf("public ") >= 0) {
// int ins = 0;
// if (newLines.get(i - 1).trim().length() > 0) {
// newLines.add(i + ins++, N.EMPTY_STRING);
// }
// newLines.add(i + ins++, importTypeClass);
// newLines.add(i + ins++, N.EMPTY_STRING);
// break;
// }
// }
}
if (newLines.get(newLines.size() - 1).startsWith("}") && newLines.get(newLines.size() - 2).endsWith(IOUtil.LINE_SEPARATOR)) {
newLines.set(newLines.size() - 2, N.chop(newLines.get(newLines.size() - 2)));
}
IOUtil.writeLines(clsSourceFile, newLines);
// Add annotation back.
Map> annoMap = new HashMap<>();
for (int i = 0, len = lines.size(); i < len; i++) {
String line = lines.get(i);
if (line.trim().startsWith("@")) {
Set tmp = new LinkedHashSet<>();
tmp.add(line);
while (++i < len && lines.get(i).trim().startsWith("@")) {
tmp.add(lines.get(i));
}
while (i < len && lines.get(i).trim().equals("")) {
i++;
}
annoMap.put(lines.get(i), tmp);
}
}
if (N.notNullOrEmpty(annoMap)) {
final List finalLines = new ArrayList<>();
final List srcLines = IOUtil.readLines(clsSourceFile);
for (int i = 0, len = srcLines.size(); i < len; i++) {
final String line = srcLines.get(i);
final Set annons = annoMap.get(line);
if (N.notNullOrEmpty(annons)) {
int j = finalLines.size() - 1;
while (j >= 0 && finalLines.get(j).trim().startsWith("@")) {
annons.add(finalLines.remove(j--));
}
finalLines.addAll(annoMap.get(line));
}
finalLines.add(line);
}
IOUtil.writeLines(clsSourceFile, finalLines);
}
} catch (IOException | NoSuchFieldException | SecurityException e) {
throw N.toRuntimeException(e);
}
}
// 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 = new HashSet<>();
// }
//
// 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);
// }
// }
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 {
if (N.isNullOrEmpty(fieldTypes)) {
return;
}
if (parentPropertyModeForHashEquals == null) {
parentPropertyModeForHashEquals = ParentPropertyMode.NONE;
}
if (parentPropertyModeForToString == null) {
parentPropertyModeForToString = ParentPropertyMode.NONE;
}
if (fieldName2MethodName == null) {
fieldName2MethodName = new HashMap<>();
}
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.getPropGetMethodList(superClass).values());
parentSettterMethods.putAll(ClassUtil.getPropSetMethodList(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 : 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();
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;
}
}
}
}
IOUtil.writeLine(writer, N.EMPTY_STRING);
IOUtil.writeLine(writer, iden + "public " + className + " " + methodName + "(" + paraTypeName + " " + entry.getKey() + ") {");
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)
: (N.isAllUpperCase(fieldName) ? fieldName : N.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 + "}");
}
{
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());
}
}
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 srcDir
* @param pkgName
*/
public static void writeUtilClassForHashEqualsToString(final File srcDir, final String pkgName) {
writeUtilClassForHashEqualsToString(srcDir, pkgName, "_N");
}
/**
*
* @param srcDir
* @param pkgName
* @param utilClassName
*/
public static void writeUtilClassForHashEqualsToString(final File srcDir, final String pkgName, final String utilClassName) {
final String utilClassFilePath = srcDir.getAbsolutePath()
+ (N.isNullOrEmpty(pkgName) ? "" : IOUtil.FILE_SEPARATOR + N.replaceAll(pkgName, ".", IOUtil.FILE_SEPARATOR)) + IOUtil.FILE_SEPARATOR
+ utilClassName + ".java";
final File utilClassFile = new File(utilClassFilePath);
if (utilClassFile.exists() == false && IOUtil.createIfNotExists(utilClassFile) == false) {
throw new RuntimeException("Failed to create new File by path: " + utilClassFilePath);
}
if (N.isNullOrEmpty(pkgName)) {
IOUtil.write(utilClassFile, _N_STRING.replaceFirst("package com.landawn.abacus.util;", "").replaceAll("_N", utilClassName));
} else {
IOUtil.write(utilClassFile, _N_STRING.replaceFirst("com.landawn.abacus.util", pkgName).replaceAll("_N", utilClassName));
}
}
// /**
// *
// * @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));
// }
// }
/**
* Method newFileWriter.
*
* @param classFile
* @return Writer
* @throws IOException
*/
private static Writer newFileWriter(final File classFile) throws IOException {
if (!classFile.exists()) {
classFile.createNewFile();
}
return new BufferedWriter(new OutputStreamWriter(new FileOutputStream(classFile), UTF_8));
}
/**
* Method writeDomanPropNameClass.
*
* @param domainName
* @param fileWrite
* @param headSpace
* @throws IOException
*/
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 N.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;
}
/**
*
* @param propName
* @return
*/
static String propName2VarName(final String propName) {
return ClassUtil.toUpperCaseWithUnderscore(ClassUtil.formalizePropName(propName));
}
static String propName2MethodName(final String propName) {
return N.capitalize(propName);
}
static String propName2FieldName(final String propName) {
if (propName.toUpperCase().equals(propName)) {
return propName.toLowerCase();
}
return propName;
}
/**
* Method rule.
*
* @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 AbacusException("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 == D._UNDERSCORE) {
hasUnderScore = true;
}
}
if (hasUnderScore) {
if (isAllUpperCase) {
for (int i = 0; i < st.length(); i++) {
c = st.charAt(i);
if (c == D._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 == D._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 N.uncapitalize(nst);
}
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);
}
private static void writePackageName(final String pkgName, final Writer fileWrite) throws IOException {
if (N.notNullOrEmpty(pkgName)) {
fileWrite.write("package " + pkgName + ";" + IOUtil.LINE_SEPARATOR);
}
}
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);
} else if ((IMPL_ACTIVE_RECORD == entityMode)) {
writeClassImport(fileWrite, Collection.class, importedClasses);
writeClassImport(fileWrite, List.class, importedClasses);
writeClassImport(fileWrite, Set.class, importedClasses);
writeClassImport(fileWrite, Map.class, importedClasses);
} else if ((EXTEND_ACTIVE_RECORD == entityMode)) {
writeClassImport(fileWrite, Collection.class, importedClasses);
writeClassImport(fileWrite, List.class, importedClasses);
writeClassImport(fileWrite, Map.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);
} else if ((IMPL_ACTIVE_RECORD == entityMode)) {
// writeClassImport(fileWrite, EntityManager.class, importedClasses);
writeClassImport(fileWrite, LockMode.class, importedClasses);
writeClassImport(fileWrite, DataSet.class, importedClasses);
writeClassImport(fileWrite, Condition.class, importedClasses);
writeClassImport(fileWrite, ActiveRecord.class, importedClasses);
writeClassImport(fileWrite, ActiveRecordImpl.class, importedClasses);
if (entityDef.getIdPropertyList().size() > 0) {
writeClassImport(fileWrite, EntityId.class, importedClasses);
writeClassImport(fileWrite, Seid.class, importedClasses);
}
} else if ((EXTEND_ACTIVE_RECORD == entityMode)) {
writeClassImport(fileWrite, DataSet.class, importedClasses);
writeClassImport(fileWrite, Condition.class, importedClasses);
writeClassImport(fileWrite, AbstractActiveRecord.class, importedClasses);
if (entityDef.getIdPropertyList().size() > 0) {
writeClassImport(fileWrite, Seid.class, importedClasses);
writeClassImport(fileWrite, EntityId.class, importedClasses);
}
}
if ((EXTEND_ACTIVE_RECORD == entityMode) || (IMPL_ACTIVE_RECORD == entityMode)) {
writeClassImport(fileWrite, Nullable.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 (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 ((EXTEND_ACTIVE_RECORD == entityMode) || (IMPL_ACTIVE_RECORD == entityMode)) {
fileWrite.write(IOUtil.LINE_SEPARATOR + "import static " + propNameTableClass + "." + DOMAIN_NAME_VAR + ";");
}
}
if ((IMPL_DIRTY_MARKER == entityMode) || (IMPL_ACTIVE_RECORD == 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;
}
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);
}
}
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_ACTIVE_RECORD) && (extendedClass != null) && !ActiveRecordImpl.class.isAssignableFrom(extendedClass)) {
extendedClass = ActiveRecordImpl.class;
} else 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);
// write static(load, find) methods
if ((IMPL_ACTIVE_RECORD == entityMode) || (EXTEND_ACTIVE_RECORD == entityMode)) {
writeStaticMethod(entityDef, pkgName, propName2VarName, entityMode, extendedClass, headSpace, fileWrite, importedClasses);
}
// write objectManager method.
writeActiveRecordMethod(entityMode, headSpace, fileWrite);
writeDirtyMarkerMethod(entityDef, entityMode, headSpace, fileWrite);
// write get/set methods
writeGetSetMethod(entityDef, pkgName, extendedClass, entityMode, propName2VarName, propName2MethodName, fluentSetMethod, headSpace, fileWrite,
importedClasses);
// write copy method
// only write copy method for entity class.
if (entityMode == EntityMode.EXTEND_ACTIVE_RECORD || entityMode == EntityMode.IMPL_ACTIVE_RECORD
|| Boolean.valueOf(entityDef.getAttribute("generateCopyMethod"))) {
writeCopyMethod(entityDef, headSpace, fileWrite);
}
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 + "}");
}
private static void writeCopyMethod(final EntityDefinition entityDef, final String headSpace, final Writer fileWrite) throws IOException {
boolean isAbstract = N.asBoolean(entityDef.getAttribute("abstract"));
if (isAbstract) {
return;
}
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " public " + entityDef.getName() + " copy() {");
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " final " + entityDef.getName() + " copy = new " + entityDef.getName() + "();"
+ IOUtil.LINE_SEPARATOR);
int counter = 0;
for (Property prop : entityDef.getIdPropertyList()) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " copy." + propName2FieldName(prop.getName()) + " = this."
+ propName2FieldName(prop.getName()) + ";");
counter++;
}
for (Property prop : entityDef.getPropertyList()) {
if (!prop.isId()) {
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " copy." + propName2FieldName(prop.getName()) + " = this."
+ propName2FieldName(prop.getName()) + ";");
counter++;
}
}
if (counter > 0) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
}
fileWrite.write(IOUtil.LINE_SEPARATOR + headSpace + " return copy;" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
/**
* Method writeClassComment2.
*
* @param fileWrite
* @throws IOException
*/
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);
}
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 = ObjectFactory.createStringBuilder();
boolean isAbstract = N.asBoolean(entityDef.getAttribute("abstract"));
if (isAbstract) {
sb.append(headSpace + "public abstract class " + entityDef.getName());
} else {
sb.append(headSpace + "public class " + entityDef.getName());
}
if ((extendedClass != null) && !ActiveRecordImpl.class.isAssignableFrom(extendedClass) && !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 + D._PERIOD + entityDef.getName() + POSTFIX_OF_PROP_NAME_LIST);
} else {
sb.append(" implements " + simplePropNameTableClass + D._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());
ObjectFactory.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
* @throws IOException
*/
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 || ((entityMode == EntityMode.EXTEND_DIRTY_MARKER) || (entityMode == EntityMode.EXTEND_ACTIVE_RECORD))) {
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);
} else if (entityMode == IMPL_ACTIVE_RECORD) {
fileWrite.write(headSpace + " private final " + ClassUtil.getSimpleClassName(extendedClass) + " " + ACTIVE_RECORD_IMPL_FIELD_NAME + " = new "
+ ClassUtil.getSimpleClassName(extendedClass) + "(this);" + 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(headSpace + " private " + type + " " + fieldName + ";" + IOUtil.LINE_SEPARATOR);
}
}
private static boolean isSerializable(final Class extendedClass, final List> implementedInterfaces) {
if ((extendedClass != null) && !DirtyMarkerImpl.class.isAssignableFrom(extendedClass) && !ActiveRecordImpl.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;
}
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);
}
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 || entityMode == EntityMode.EXTEND_ACTIVE_RECORD)) {
fileWrite.write(headSpace + " super(" + ENTITY_NAME_VAR + ");" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
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;
}
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);
}
/**
* Method writeCopyConstructor.
*
* @param entityDef
* @param pkgName
* @param headSpace
* @param fileWrite
* @throws IOException
*/
/*
* 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.getPropSetMethodList(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 (Throwable 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_ACTIVE_RECORD == entityMode) || (EXTEND_DIRTY_MARKER == entityMode)) ? "super"
: ((IMPL_ACTIVE_RECORD == entityMode) ? ACTIVE_RECORD_IMPL_FIELD_NAME : 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 AbacusException("Family/Entity property can't/unnecessary to be HBaseColumn");
}
String valueTypeName = N.isPrimitiveWapper(hbaseColumnType.getElementType().clazz())
? N.primitiveOf(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 AbacusException("Family/Entity property can't/unnecessary to be HBaseColumn");
}
String valueTypeName = N.isPrimitiveWapper(hbaseColumnType.getElementType().clazz())
? N.primitiveOf(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_ACTIVE_RECORD == entityMode) || (EXTEND_DIRTY_MARKER == entityMode)) ? "super"
: ((IMPL_ACTIVE_RECORD == entityMode) ? ACTIVE_RECORD_IMPL_FIELD_NAME : 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_ACTIVE_RECORD == entityMode) || (EXTEND_DIRTY_MARKER == entityMode)) ? "super"
: ((IMPL_ACTIVE_RECORD == entityMode) ? ACTIVE_RECORD_IMPL_FIELD_NAME : 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);
}
}
}
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.getPropGetMethodList(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);
}
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.getPropGetMethodList(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);
}
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.getPropGetMethodList(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);
}
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;
}
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;
}
private static void writeActiveRecordMethod(final EntityMode entityMode, final String headSpace, final Writer fileWrite)
throws IllegalAccessException, InvocationTargetException, IOException {
if (IMPL_ACTIVE_RECORD.equals(entityMode)) {
boolean hasOutObjectProp = true;
boolean hasOutObjectCollProp = true;
if (hasOutObjectProp) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void add(ActiveRecord entity, boolean isNew) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.add(entity, isNew);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void add(ActiveRecord entity, boolean isNew, Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.add(entity, isNew, options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (hasOutObjectCollProp) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void addAll(Collection entities, boolean isNew) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.addAll(entities, isNew);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void addAll(Collection entities, boolean isNew, Map options) {"
+ IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.addAll(entities, isNew, options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (hasOutObjectProp) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void remove(ActiveRecord entity, boolean isDelete) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.remove(entity, isDelete);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(
headSpace + " public void remove(ActiveRecord entity, boolean isDelete, Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.remove(entity, isDelete, options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
if (hasOutObjectCollProp) {
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(
headSpace + " public void removeAll(Collection entities, boolean isDelete) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.removeAll(entities, isDelete);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(
headSpace + " public void removeAll(Collection entities, boolean isDelete, Map options) {"
+ IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.removeAll(entities, isDelete, options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void store() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.store();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void store(Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.store(options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public int update() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.update();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public int update(Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.update(options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean refresh() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.refresh();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean refresh(Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.refresh(options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public int delete() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.delete();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public int delete(Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.delete(options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
//
// fileWrite.write(D.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " public Nullable fetch(String propName) {" + D.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " return activeRecordImpl.fetch(propName);" +
// D.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + D.LINE_SEPARATOR);
//
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " public void modify(String propName, Object propValue) {" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " activeRecordImpl.modify(propName, propValue);" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean lockRecord(LockMode lockMode) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.lockRecord(lockMode);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean lockRecord(LockMode lockMode, Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.lockRecord(lockMode, options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean unlockRecord() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.unlockRecord();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public boolean unlockRecord(Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.unlockRecord(options);" + 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 getRecordVersion() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.getRecordVersion();" + 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 getRecordVersion(Map options) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " return activeRecordImpl.getRecordVersion(options);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
//
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace
// +
// " public boolean lockTable(LockMode lockMode, Map options) {"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " return activeRecordImpl.lockTable(lockMode, options);"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
//
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace
// +
// " public boolean unlockTable(Map options) {"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " return activeRecordImpl.unlockTable(options);"
// + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
/*
* fileWrite.write(N.LINE_SEPARATOR); fileWrite.write(headSpace + " public long getTableVersion(Map options) {" + N.LINE_SEPARATOR); fileWrite.write(headSpace + " return
* activeRecordImpl.getTableVersion(options);" + N.LINE_SEPARATOR); fileWrite.write(headSpace + " }" +
* N.LINE_SEPARATOR);
*/
/*
* fileWrite.write(N.LINE_SEPARATOR); fileWrite.write(headSpace +
* " public long version() {" + N.LINE_SEPARATOR); fileWrite.write(headSpace +
* " return activeRecordImpl.version(this);" + N.LINE_SEPARATOR); fileWrite.write(headSpace + " }"
* + N.LINE_SEPARATOR);
*/
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void merge(Object sourceEntity) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.merge(sourceEntity);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " @SuppressWarnings(\"unchecked\")" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " public " + entityDef.getName() + " copy() {" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " return " + "(" + entityDef.getName() + ") activeRecordImpl.copy();" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
// fileWrite.write(N.LINE_SEPARATOR);
// fileWrite.write(headSpace +
// " @SuppressWarnings(\"unchecked\")" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " public " + entityDef.getName() +
// " clone() {" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " return " + "(" +
// entityDef.getName()
// + ") activeRecordImpl.clone(this);" + N.LINE_SEPARATOR);
// fileWrite.write(headSpace + " }" + N.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void erase(String... propNames) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.erase(propNames);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void erase(Collection propNames) {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.erase(propNames);" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
fileWrite.write(IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " public void eraseAll() {" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " activeRecordImpl.eraseAll();" + IOUtil.LINE_SEPARATOR);
fileWrite.write(headSpace + " }" + IOUtil.LINE_SEPARATOR);
}
}
/*
* 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_ACTIVE_RECORD == entityMode) || (EXTEND_DIRTY_MARKER == entityMode)) ? "super"
: ((IMPL_ACTIVE_RECORD == entityMode) ? ACTIVE_RECORD_IMPL_FIELD_NAME : 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) || (entityMode == IMPL_ACTIVE_RECORD)) {
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_ACTIVE_RECORD) || (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) || (entityMode == IMPL_ACTIVE_RECORD)) {
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_ACTIVE_RECORD) || (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);
// }
}
}
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 = N.substring(typeName, 0, typeName.indexOf('[')).get();
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 + "