All Downloads are FREE. Search and download functionalities are using the official Maven repository.

gu.sql2java.generator.Table Maven / Gradle / Ivy

The newest version!
package gu.sql2java.generator;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.sql.DatabaseMetaData;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.TreeMap;
import java.util.TreeSet;
import java.util.LinkedHashSet;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;
import java.util.Random;
import java.util.Vector;
import java.util.stream.Collectors;

import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;
import org.apache.commons.lang.builder.ToStringBuilder;

import com.gitee.l0km.com4j.base.VolatileVariable;
import com.google.common.base.Function;
import com.google.common.base.Joiner;
import com.google.common.base.Predicate;
import com.google.common.base.Strings;
import com.google.common.collect.Collections2;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import com.google.common.collect.Iterators;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Ordering;
import com.google.common.collect.Sets;
import gu.sql2java.excel.annotations.ExcelSheet;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkArgument;

import static com.google.common.base.Strings.nullToEmpty;import static gu.sql2java.generator.ArraySupport.bitSet;
import static gu.sql2java.generator.ArraySupport.toHex;;

public class Table {
	private Hashtable colHash = new Hashtable();
	private Vector cols = new Vector();
	private Hashtable indHash = new Hashtable();
	private Vector indices = new Vector();
	private Hashtable indUniqueHash = new Hashtable();
	private Vector uniqueIndices = new Vector();
	private Hashtable indNonUniHash = new Hashtable();
	private Vector nonUniqueIndices = new Vector();
	private Vector priKey = new Vector();
	private Column autoincrement;
	private String catalog;
	private String schema;
	private String name;
	private String type;
	private SqlComment remarks;
	private Database database;
	private Vector foreignKeys = new Vector();
	private Vector importedKeys = new Vector();
	/** FK_NAME 为索引保存所有 foreign keys */
	private TreeMap fkNameMap = new TreeMap();
	private List procedures = new ArrayList();
	private HashMap procHash = new HashMap();
	private Random aleatorio = new Random(new Date().getTime());
	/** 
	 * 从配置文件中读取的字段能见度定义,字段名--能见度映射
	 */
	private final VolatileVariable> visibilityOfColumns = new VolatileVariable>(()->{
		Map map =new LinkedHashMap<>();
		for(ColumnVisibility visibility:ColumnVisibility.values()) {
			if(ColumnVisibility.DEFAULT.equals(visibility)) {
				continue;
			}
			Set names = getColumnsAsList().stream().map(Column::getName).collect(Collectors.toSet());
			/** 按ColumnVisibility中枚举变量定义顺序为优先级,生成字段可见度映射  */
			CodeWriter.getPropertyExplodedAsSet("table."+name+".scope."+visibility.name().toLowerCase())
				.stream()
				/** 排除无效字段名 */
				.filter(names::contains)
				.forEach(c->map.putIfAbsent(c, visibility));
		}
		return Collections.unmodifiableMap(map);
	});
	
	@Override
	public String toString() {
		return new ToStringBuilder(this)
				.append("catalog",catalog)
				.append("schema", schema)
				.append("name", name)
				.append("type", type)
				.append("remarks", remarks)
				.toString();
	}

	@Override
	public int hashCode() {
		return new HashCodeBuilder()
				.append(catalog)
				.append(indices)
				.append(name)
				.append(schema)
				.toHashCode();
	}

	@Override
	public boolean equals(Object obj) {
		if(super.equals(obj))return true;
		if(!(obj instanceof Table))return false;
		Table other = (Table)obj;
		return new EqualsBuilder()
			.append(catalog, other.catalog)
			.append(indices, other.indices)
			.append(name, other.name)
			.append(schema, other.schema)
			.append(type, other.type)
			.isEquals();
	}

	public boolean isRelationTable() {
		return this.foreignKeys.size() == 2;
	}
	
	/**
	 * 代替{@link #isRelationTable()},判断当前表是否为联接表
	 */
	public boolean isJunctionTable() {
		if(fkNameMap.size() ==2){
			HashSet fkColumns = Sets.newHashSet();
			for(ForeignKey fks:fkNameMap.values()){
				fkColumns.addAll(fks.columns);
			}
			Vector pkColumns = getPrimaryKeysAsList();
			if(pkColumns.size() == fkColumns.size()){
				fkColumns.retainAll(getPrimaryKeysAsList());
				return fkColumns.size() ==pkColumns.size();	
			}			
		}			
		return false;
	}
	/**
	 * 判断是否为联接表,且主键为两个字段
	 */
	public boolean isSampleJunctionTable() {
		return isJunctionTable() && 2 == countPrimaryKeys();
	}
	/**
	 * 判断外键是否为自引用
	 */
	public boolean isSelfRef(ForeignKey fk) {
		if(null == fk)return false;
		return fk.getForeignTable().equals(this);
	}
	/** 返回所有自引用外键 */
	public List getSelfRefKeys(){
		return ImmutableList.copyOf(Collections2.filter(this.fkNameMap.values(), new Predicate(){
			@Override
			public boolean apply(ForeignKey input) {
				return isSelfRef(input);
			}}));
	}
	public boolean relationConnectsTo(Table otherTable) {
		if (this.equals(otherTable)) {
			return false;
		}
		int nbImported = this.importedKeys.size();
		for (int i = 0; i < nbImported; ++i) {
			Column c = (Column) this.importedKeys.get(i);
			if (!c.getTableName().equals(otherTable.getName()))
				continue;
			return true;
		}
		return false;
	}

	public Table[] linkedTables(Database pDatabase, Table pTable) {
		Vector vector = new Vector
(); int nbImported = this.importedKeys.size(); for (int iIndex = 0; iIndex < nbImported; ++iIndex) { Table pTableToAdd; Column pColumn = (Column) this.importedKeys.get(iIndex); if (pColumn.getTableName().equals(pTable.getName()) || vector.contains(pTableToAdd = pDatabase.getTable(pColumn.getTableName()))) continue; vector.add(pTableToAdd); } return vector.toArray(new Table[vector.size()]); } /** * 代替 {@link #linkedTables(Database, Table)}
* 对于联接表(junction)返回{@code pTable}联接的表
* @param pTable * @return 当前对象不是连接表或{@code pTable}不属于联接表时返回{@code null} */ public Table tableOfJunction(Table pTable) { if(this.isJunctionTable()){ for(ForeignKey fk:this.fkNameMap.values()){ Table jtable = fk.getForeignTable(); if(!jtable.equals(pTable)) return jtable; } } return null; } public Column getForeignKeyFor(Table pTable) { int nbImported = this.importedKeys.size(); for (int iIndex = 0; iIndex < nbImported; ++iIndex) { Column pColumn = (Column) this.importedKeys.get(iIndex); if (!pColumn.getTableName().equals(pTable.getName())) continue; return pColumn; } return null; } /** 返回当前对象的关联表 */ public List
getJunctionTables() { List
tabs = this.getDatabase().getJunctionTables(); if(isJunctionTable()){ // 当前表是关联表,则返回空 tabs.clear(); return tabs; } for(Iterator
itor = tabs.iterator();itor.hasNext();){ Table table = itor.next(); if(!table.getForeignTablesAsList().contains(this)){ itor.remove(); } } return tabs; } public void setCatalog(String catalog) { this.catalog = catalog; } public void setSchema(String schema) { this.schema = schema; } public void setName(String name) { this.name = name; } public void setType(String type) { this.type = type; } public void setRemarks(String remarks) { this.remarks = new SqlComment(remarks); } public String getCatalog() { return this.catalog; } public String getSchema() { return this.schema; } public String getName() { return this.name; } public String getType() { return this.type; } public Column[] getColumns() { return this.getColumnsAsList().toArray(new Column[this.cols.size()]); } public Vector getColumnsAsList() { Collections.sort(this.cols); return this.cols; } public List getColumnsExceptPrimaryAsList() { return Lists.newArrayList(Collections2.filter(getColumnsAsList(), new Predicate(){ @Override public boolean apply(Column input) { return !input.isPrimaryKey(); }})); } public Column[] getColumnsExceptPrimary() { return getColumnsExceptPrimaryAsList().toArray(new Column[0]); } public Column getColumn(String columnName) { return (Column) this.colHash.get(columnName.toLowerCase()); } public void addColumn(Column column) { this.colHash.put(column.getName().toLowerCase(), column); this.cols.addElement(column); } public void removeColumn(Column column) { this.cols.removeElement((Object) column); this.colHash.remove(column.getName().toLowerCase()); } public Index[] getUniqueIndices() { return this.uniqueIndices.toArray(new Index[this.uniqueIndices.size()]); } public Index[] getNonUniqueIndices() { return this.nonUniqueIndices.toArray( new Index[this.nonUniqueIndices.size()]); } public int countIndices() { return this.indices.size(); } public Index[] getIndices() { return this.indices.toArray( new Index[this.indices.size()]); } public List getIndicesAsList(Boolean unique) { Iteratoritor = this.indices.iterator(); if(Boolean.TRUE == unique) itor = Iterators.filter(itor, new Predicate(){ @Override public boolean apply(Index arg0) { return arg0.isUnique(); }}); return ImmutableList.copyOf(itor); } public List getIndicesAsList(){ return getIndicesAsList(false); } public List getUniqueIndicesAsList(){ return getIndicesAsList(true); } public Index getIndex(String indName) { return (Index) this.indHash.get(indName.toLowerCase()); } public void addIndex(Index index) { this.indHash.put(index.getName().toLowerCase(), index); this.indices.add(index); if (index.isUnique()) { this.indUniqueHash.put(index.getName().toLowerCase(), index); this.uniqueIndices.add(index); } else { this.indNonUniHash.put(index.getName().toLowerCase(), index); this.nonUniqueIndices.add(index); } } public void removeIndex(Index index) { this.indices.remove((Object) index); this.indHash.remove(index.getName().toLowerCase()); if (index.isUnique()) { this.uniqueIndices.remove((Object) index); this.indUniqueHash.remove(index.getName().toLowerCase()); } else { this.nonUniqueIndices.remove((Object) index); this.indNonUniHash.remove(index.getName().toLowerCase()); } } public Column[] getPrimaryKeys() { return this.priKey.toArray( new Column[this.priKey.size()]); } public Vector getPrimaryKeysAsList() { return new Vector(this.priKey); } public boolean hasCompositeKey() { if (this.priKey.size() > 1) { return true; } return false; } public Column getPrimaryKey() throws RuntimeException { if (this.priKey.size() != 1) { throw new RuntimeException("Table " + this.getName() + " has a composite key, not a unique primary key"); } return (Column) this.priKey.get(0); } public void addPrimaryKey(Column column) { this.priKey.addElement(column); column.isPrimaryKey(true); } public Column[] getImportedKeys() { return this.importedKeys.toArray( new Column[this.importedKeys.size()]); } public void addImportedKey(Column column) { if (!this.importedKeys.contains((Object) column)) { this.importedKeys.addElement(column); } } public int countColumns() { return this.cols.size(); } public int countPrimaryKeys() { return this.priKey.size(); } public boolean hasPrimaryKey() { return this.countPrimaryKeys() > 0; } public int countImportedKeys() { return this.importedKeys.size(); } public boolean hasImportedKeys() { return this.countImportedKeys() > 0; } public int countForeignKeys() { return this.foreignKeys.size(); } public boolean hasForeignKeys() { return this.countForeignKeys() > 0; } public void addForeignKey(Column col, String fkName, short keySeq, Table.ForeignKeyRule updateRule, Table.ForeignKeyRule deleteRule) { checkNotNull(col); checkArgument(!Strings.isNullOrEmpty(fkName)); checkArgument(keySeq>0,"the argument 'keySeq' must >0"); if (!this.foreignKeys.contains(col)) { this.foreignKeys.add(col); } if(!this.fkNameMap.containsKey(fkName)) this.fkNameMap.put(fkName, new ForeignKey(fkName,updateRule,deleteRule,col)); Vector fkCols = fkNameMap.get(fkName).columns; if(keySeq > fkCols.size()){ fkCols.setSize(keySeq); } fkCols.set(keySeq-1, col); } /** * 返回所有 foreign key name ( FK_NAME ) */ public Vector getFkMapNames() { return new Vector(this.fkNameMap.keySet()); } /** * 检索外键引用指定表(tableName)的所有 FK_NAME
* 没有结果则返回空数组 * @param tableName */ public Vector getFkMapNames(String tableName) { Vector names=new Vector(); for(ForeignKey fk:this.fkNameMap.values()){ for(Column col:fk.columns.get(0).getForeignKeys()){ if(col.getTableName().equals(tableName)){ names.add(fk.fkName); break; } } } Collections.sort(names); return names; } /** * 检索指定 FK_NAME 包含的所有字段
* 没有结果则返回空数组 * @param fkName */ public Vector getForeignKeysByFkName(String fkName) { ForeignKey keys=this.fkNameMap.get(fkName); return null==keys?new Vector():new Vector(keys.columns); } /** * 检索指定 FK_NAME 的{@link ForeignKey}对象
* @param fkName * @see java.util.concurrent.ConcurrentHashMap#get(java.lang.Object) */ public ForeignKey getForeignKey(String fkName) { return fkNameMap.get(fkName); } /** * 返回{@code table}对应的所有{@link ForeignKey}对象 * @param table */ public List getForeignKeys(final Table table){ return Lists.newArrayList(Collections2.filter(this.fkNameMap.values(), new Predicate(){ @Override public boolean apply(ForeignKey input) { return input.getForeignTable().equals(table); }})); } public List getImportedFoeignKeysAsList(){ ArrayList list = new ArrayList(); for(Table table:getImportedTablesAsList()){ list.addAll(table.getForeignKeys(this)); } return list; } public List getForeignKeysAsList(){ return ImmutableList.copyOf(this.fkNameMap.values()); } /** * 返回 所有需要输出foreign key listener的 {@link ForeignKey}对象 */ public List getForeignKeysForListener(){ Collection c = Maps.filterEntries(fkNameMap, new Predicate>(){ @Override public boolean apply(Entry input) { ForeignKey fk = input.getValue(); return fk.updateRule.isNoAction() && !Strings.isNullOrEmpty(fk.deleteRule.getEventOfDeleteRule()); }}).values(); // 排序输出 return Ordering.natural().onResultOf(new Function(){ @Override public String apply(ForeignKey input) { return input.fkName; }}).sortedCopy(c); } /** * 判断 FK_NAME 包含的所有字段是否都允许为null * @param fkName */ public boolean isNullable(String fkName){ for(Column column: getForeignKeysByFkName(fkName)){ if(column.isNotNull())return false; } return true; } /** * 返回 FK_NAME 包含的所有字段中不允许为null的所有字段 * @param fkName */ public Vector noNullableColumns(String fkName){ Vector keys = getForeignKeysByFkName(fkName); for(Iterator itor = keys.iterator();itor.hasNext();){ Column column = itor.next(); if(DatabaseMetaData.columnNullable !=column.getNullable())continue; itor.remove(); } return keys; } private String toUniversalFkName(String fkName) { ForeignKey key = this.fkNameMap.get(fkName); if(null!=key){ return key.getUniversalName(); } return ""; } public Table getForeignTableByFkName(String fkName){ Vector keys = this.getForeignKeysByFkName(fkName); return 0==keys.size()? null:keys.get(0).getForeignColumn().getTable(); } public Column[] getForeignKeys() { return this.foreignKeys.toArray( new Column[this.foreignKeys.size()]); } public boolean isForeignKey(Column col) { return this.foreignKeys.contains((Object) col); } public int countManyToManyTables() { return this.getManyToManyTables().length; } public boolean hasManyToManyTables() { return this.countManyToManyTables() > 0; } public Table[] getManyToManyTables() { Vector
vector = new Vector
(); Table[] linkedTables = this.getImportedTables(); System.out.println(this.getName() + " getManyToManyTables, linked tables = " + linkedTables.length); for (int iIndex = 0; iIndex < linkedTables.length; ++iIndex) { System.out.println(this.getName() + " " + linkedTables[iIndex].getName() + " relation table ?"); if (!linkedTables[iIndex].isRelationTable()) continue; Table[] relationLinkedTable = linkedTables[iIndex].getForeignTables(); System.out.println(this.getName() + " " + linkedTables[iIndex].getName() + " has " + relationLinkedTable.length + " foreign table"); for (int i = 0; i < relationLinkedTable.length; ++i) { System.out.println(this.getName() + " " + i + " " + relationLinkedTable[i].getName() + " is relation table"); if (relationLinkedTable[i].equals(this) || vector.contains(relationLinkedTable[i])) continue; vector.add(relationLinkedTable[i]); } } return vector.toArray(new Table[vector.size()]); } public int countLinkedTables() { return this.getLinkedTables().length; } public boolean hasLinkedTables() { return this.countLinkedTables() > 0; } public Table[] getLinkedTables() { Vector
vector = new Vector
(); int nbImported = this.importedKeys.size(); for (int iIndex = 0; iIndex < nbImported; ++iIndex) { Table pTableToAdd; Column column = (Column) this.importedKeys.get(iIndex); if (column.getTableName().equals(this.getName()) || vector.contains(pTableToAdd = column.getTable())) continue; vector.add(pTableToAdd); } int nbForeign = this.foreignKeys.size(); for (int iIndex2 = 0; iIndex2 < nbForeign; ++iIndex2) { Table pTableToAdd; Column column = (Column) this.foreignKeys.get(iIndex2); if ((column = column.getForeignColumn()).getTableName().equals(this.getName()) || vector.contains(pTableToAdd = column.getTable())) continue; vector.add(pTableToAdd); } for(Table jtable:getJunctionTables()){ Table pTableToAdd = jtable.tableOfJunction(this); if(vector.contains(pTableToAdd)){ continue; } vector.add(pTableToAdd); } return vector.toArray(new Table[vector.size()]); } public int countImportedTables() { return this.getImportedTables().length; } public boolean hasImportedTables() { return this.countImportedTables() > 0; } public List
getImportedTablesAsList() { LinkedHashSet
tables = Sets.newLinkedHashSet(Lists.transform(this.importedKeys, new Function(){ @Override public Table apply(Column t) { return t.getTable(); }})); return Lists.newArrayList(tables); } public Table[] getImportedTables() { return getImportedTablesAsList().toArray(new Table[0]); } public int countForeignTables() { return this.getForeignTables().length; } public boolean hasForeignTables() { return this.countForeignTables() > 0; } public List
getForeignTablesAsList() { LinkedHashSet
tables = Sets.newLinkedHashSet(Lists.transform(this.foreignKeys, new Function(){ @Override public Table apply(Column t) { return t.getForeignColumn().getTable(); }})); return Lists.newArrayList(tables); } public Table[] getForeignTables() { return getForeignTablesAsList().toArray(new Table[0]); } public Set
getAccociatedTables() { List
accociatedTables = getImportedTablesAsList(); accociatedTables.addAll(getForeignTablesAsList()); return FluentIterable.from(accociatedTables).toSortedSet(Ordering.natural().onResultOf(Table::getName)); } public Table getRelationTable(Table targetTable) { System.out.println("getRelationTable " + this.getName() + "<->" + targetTable.getName() + ")"); Table[] importedTables = this.getImportedTables(); for (int iIndex = 0; iIndex < importedTables.length; ++iIndex) { Table[] foreignTables = importedTables[iIndex].getForeignTables(); for (int iIndex2 = 0; iIndex2 < foreignTables.length; ++iIndex2) { if (!foreignTables[iIndex2].getName().equalsIgnoreCase(this.getName())) continue; return importedTables[iIndex]; } } return targetTable; } public int countProcedures() { return this.procedures.size(); } public boolean hasProcedures() { return this.countProcedures() > 0; } public Procedure[] getProcedures() { return this.procedures.toArray( new Procedure[this.procedures.size()]); } public void addProcedure(Procedure procedure) { if (null == this.procHash.get(procedure.getName())) { this.procedures.add(procedure); this.procHash.put(procedure.getName(), procedure); } } public String[] getLinkedPackages() { Vector vector = new Vector(); Table[] linkedTables = this.getLinkedTables(); for (int iIndex = 0; iIndex < linkedTables.length; ++iIndex) { if (vector.contains(linkedTables[iIndex].getPackage())) continue; vector.add(linkedTables[iIndex].getPackage()); } return vector.toArray(new String[vector.size()]); } public String getPackage() { String basePackage = CodeWriter.getProperty((String) "codewriter.package"); return checkNotNull(basePackage,"NOT DEFINED codewriter.package"); } public String getPackagePath() { return this.getPackage().replace('.', '/') + "/"; } public Column getFirstColumn() { return (Column) this.cols.get(0); } public String getRemarks() { return this.remarks == null ? "" : this.remarks.getRemarks(); } public boolean hasRemarks(){ return !getRemarks().isEmpty(); } public String getJavaName() { return this.convertName(""); } /** * @since 3.31.1 */ public String getCorename(){ return this.getName().replaceFirst(this.getDatabase().getSamePrefix(), ""); } public String getBasename(Boolean nsp){ String basename = Boolean.TRUE.equals(nsp) ? this.getCorename() : this.getName(); return "".equals(CodeWriter.getClassPrefix()) ? basename : CodeWriter.getClassPrefix() + "_" + basename; } public String convertName(String value,Boolean nsp) { String basename = getBasename(nsp); if ("".equals(value)) { return StringUtilities.convertName((String) basename, false); } return StringUtilities.convertName((String) (basename + "_" + value), false); } public String convertName(String value) { return convertName(value,false); } public String convertNameNSP(String value) { return convertName(value,true); } public String asClass(String suffix) { return this.convertName(suffix); } public String asCoreClass() { return this.convertName(""); } public String asCoreClassNSP() { return this.convertNameNSP(""); } public String asCoreClass(Boolean nsp) { return this.convertName("",nsp); } public String asBeanClass() { return this.convertName("Bean"); } public String asFullBeanClass() { return this.getPackage() + "." + this.asBeanClass(); } public String asBeanClassNSP() { return this.convertNameNSP("Bean"); } public String asBeanClass(Boolean nsp) { return convertName("Bean",nsp); } public String asConstClass() { return this.convertName("Const"); } public String asConstClass(boolean nsp) { return this.convertName("Const",nsp); } public String asConstClassNSP() { return asConstClass(true); } public String asCacheClass() { return this.convertName("Cache"); } public String asCacheClass(boolean nsp) { return this.convertName("Cache",nsp); } public String asRelationnalBeanClass() { return this.convertName("Relationnal_Bean"); } public String asHibernateManagerClass() { return this.convertName("Hibernate_Manager"); } public String asIteratorClass() { return this.convertName("Iterator"); } public String asFactoryClass() { return this.convertName("Factory"); } public String asHttpFactoryClass() { return this.convertName("Http_Factory"); } public String asComparatorClass() { return this.convertName("Comparator"); } public String asComparatorClass(Boolean nsp) { return convertName("Comparator",nsp); } public String asListenerClass() { return this.convertName("Listener"); } public String asListenerClassNSP() { return this.convertNameNSP("Listener"); } public String asRendererClass() { return this.convertName("Renderer"); } public String asExceptionClass() { return this.convertName("Exception"); } public String asWidgetClass() { return this.convertName("Widget"); } public String asWidgetFactoryClass() { return this.convertName("Widget_Factory"); } public String asActionClass() { return this.convertName("Action"); } public String asActionTestClass() { return this.convertName("Action_Test"); } public String asControllerClass() { return this.convertName("Controller"); } public String asControllerTestClass() { return this.convertName("Controller_Test"); } public String asFormControllerClass() { return this.convertName("Form_Controller"); } public String asFormControllerTestClass() { return this.convertName("Form_Controller_Test"); } public String asDAOClass() { return this.convertName("D_A_O"); } public String asDAOTestClass() { return this.convertName("D_A_O_Test"); } public String asDAOHibernateClass() { return this.convertName("D_A_O_Hibernate"); } public String asManagerClass() { return this.convertName("Manager"); } public String asManagerClass(Boolean nsp) { return this.convertName("Manager",nsp); } public String asManagerClassNSP() { return this.convertNameNSP("Manager"); } public String asManagerInterfaceNSP() { return "I"+asManagerClassNSP(); } public String asManagerImplClass() { return this.convertName("Manager_Impl"); } public String asMetaDataClassNSP() { return asCoreClass(true) + "MetaData"; } public String asManagerTestClass() { return this.convertName("Manager_Test"); } public String asCacheManagerClass() { return this.convertName("cache_manager"); } public String asCacheManagerClassNSP() { return this.convertNameNSP("cache_manager"); } public String asCacheManagerClass(boolean nsp) { return this.convertName("cache_manager",nsp); } public String asVar(String prefix,String suffix){ return StringUtilities.convertName( prefix + getBasename(true) + suffix,true); } public String asVar(String prefix){ return asVar(prefix,""); } public String asVar(){ return asVar("",""); } /** * @since 3.31.0 */ public String asConstantVar(String prefix,String suffix,Boolean nsp) { return nullToEmpty(prefix)+getBasename(nsp).toUpperCase()+nullToEmpty(suffix); } /** * @since 3.31.0 */ public String asConstantVar(String suffix) { return asConstantVar("",suffix,true); } /** * @since 3.31.0 */ public String asConstantVar() { return asConstantVar("","",true); } public String asVarBean(){ return asVar("","_Bean"); } public String asVarManager(){ return asVar("","_Manager"); } public String asConverterVar(){ return "converter" + asBeanClassNSP() ; } public String asConverterConstVar(){ return "converter_".concat(asBeanClassNSP()).toUpperCase(); } public String asCacheVarName(){ return StringUtilities.convertName( getBasename(true) + "_cache",true); } public String asCacheVarSetMethod(){ return StringUtilities.convertName("set_" + getBasename(true) + "_cache",true); } public String asCacheVarGetMethod(){ return StringUtilities.convertName("get_" + getBasename(true) + "_cache",true); } public String asInstanceMethod(Boolean nsp){ return "instanceOf" + asManagerClass(nsp); } public String asModelClass() { return this.convertName("Model"); } public String asPKClass() { return this.convertName("P_K"); } public String asTblClass() { return this.convertName("Tbl"); } public Column getVersionColumn() { int nbCols = this.cols.size(); for (int i = 0; i < nbCols; ++i) { Column c = (Column) this.cols.get(i); if (!c.isVersion()) continue; return c; } throw new IllegalArgumentException("No version column for table " + this.getName()); } public boolean hasVersionColumn() { try { this.getVersionColumn(); return true; } catch (IllegalArgumentException e) { return false; } } public long getSerialVersionUID() { return this.aleatorio.nextLong(); } /** * 生成MD5校验码 * * @param source */ private static byte[] getMD5(byte[] source) { if (null==source) return null; try { MessageDigest md = MessageDigest.getInstance("MD5"); return md.digest(source); } catch (NoSuchAlgorithmException e) { throw new RuntimeException(e); } } /** * 将字节数组转为long
* 如果input为null,或offset指定的剩余数组长度不足8字节则抛出异常 * @param input * @param offset 起始偏移量 * @param littleEndian 输入数组是否小端模式 */ public static long longFrom8Bytes(byte[] input, int offset, boolean littleEndian){ if(offset <0 || offset+8>input.length) throw new IllegalArgumentException(String.format("less than 8 bytes from index %d is insufficient for long",offset)); long value=0; for(int count=0;count<8;++count){ int shift=(littleEndian?count:(7-count))<<3; value |=((long)0xff<< shift) & ((long)input[offset+count] << shift); } return value; } /** * 根据输入的String返回唯一的UID(long) */ public long getSerialVersionUID(String input){ byte[] md5 = getMD5(input.getBytes()); return longFrom8Bytes(md5,0, false) ^ longFrom8Bytes(md5,8, false); } public String asFkVar(String fkName){ return StringUtilities.convertName(toUniversalFkName(fkName),true); } public String asIKVar(String fkName){ Table foreignTable = this.getForeignTableByFkName(fkName); return null==foreignTable ? "" : StringUtilities.convertName(toUniversalFkName(fkName) + "_of_" +getBasename(true), true); } public String asFKConst(String fkName){ return (this.name+"_FK_" + toUniversalFkName(fkName)).toUpperCase(); } public String asIKConst(String fkName){ Table foreignTable = this.getForeignTableByFkName(fkName); return (null==foreignTable?"":foreignTable.getName() + "_IK_" + this.name +"_" + toUniversalFkName(fkName)).toUpperCase(); } public String asRefArg(String fkName){ return StringUtilities.convertName("ref_"+ this.getForeignTableByFkName(fkName).asCoreClassNSP() +"_by_" + toUniversalFkName(fkName),true); } public String asImpArg(String fkName){ return StringUtilities.convertName("imp_"+ this.asCoreClassNSP() +"_by_" + toUniversalFkName(fkName),true); } public String getReferencedVarName(String fkName){ return StringUtilities.convertName("referenced_by_" + toUniversalFkName(fkName),true); } public String getReferencedVarGetMethod(String fkName) { return StringUtilities.convertName("get_" + "referenced_by_" + toUniversalFkName(fkName),true); } /** * @since 3.25.0 */ public String readReferencedVarGetMethod(String fkName) { return StringUtilities.convertName("read_" + "referenced_by_" + toUniversalFkName(fkName),true); } public String getReferencedVarSetMethod(String fkName){ return StringUtilities.convertName("set_" + "referenced_by_" + toUniversalFkName(fkName),true); } public String getImportedBeansGetMethod(String fkName) { return "get" + this.asBeanClassNSP() + "s" + StringUtilities.convertName("By_" + toUniversalFkName(fkName),false); } public String getImportedBeansSetMethod(String fkName) { return "set" + this.asBeanClassNSP() + "s" + StringUtilities.convertName("By_" + toUniversalFkName(fkName),false); } public String getImportedBeansDelMethod(String fkName) { return "delete" + this.asBeanClassNSP() + "s" + StringUtilities.convertName("By_" + toUniversalFkName(fkName),false); } /** * @since 3.20.0 */ public String readImportedBeansGetMethod(String fkName) { return "read" + this.asBeanClassNSP() + "s" + StringUtilities.convertName("By_" + toUniversalFkName(fkName),false); } public String getForeignKeyListenerVar(String fkName) { return "foreignKeyListener" + StringUtilities.convertName("By_" + toUniversalFkName(fkName),false); } /** 根据字段是否有default value生成initialized字段初始值 */ public String maskInitializeWithDefaultValue(){ final byte[] array = new byte[stateBitMaxCount()/Byte.SIZE]; Arrays.fill(array, (byte)0); Column[] columns = getColumns(); for(int i=0;i0 ? indent: "").append( String.format("ArraySupport.bitReset(%s,%s);\n", varName,column.getIDConstName())); } return buffer.toString(); } public int stateBitMaxCount(){ return (countColumns() + Byte.SIZE -1)/Byte.SIZE*Byte.SIZE; } /** * 包含foreign key信息的数据对象 * @author guyadong * */ public static class ForeignKey{ /** foreign key name */ final String fkName; /** columns of foreign key ,in KEY_SEQ order*/ final Vector columns = new Vector(); /** UPDATE_RULE */ final Table.ForeignKeyRule updateRule; /** DELETE_RULE */ final Table.ForeignKeyRule deleteRule; public ForeignKey(String fkName, Table.ForeignKeyRule updateRule, Table.ForeignKeyRule deleteRule, Column columns) { checkArgument(!Strings.isNullOrEmpty(fkName)); checkNotNull(columns); this.fkName = fkName; this.columns.add(columns); this.updateRule = checkNotNull(updateRule); this.deleteRule = checkNotNull(deleteRule); } public Table getForeignTable(){ return columns.get(0).getForeignColumn().getTable(); } public Table getTable(){ return columns.get(0).getTable(); } /** 返回主键{@code primaryColumn}对应的字段 */ public Column foreignColumnOf(Column primaryColumn){ for(Column column:columns){ if(column.getForeignColumn().equals(primaryColumn))return column; } return null; } @Override public boolean equals(Object obj) { if(super.equals(obj))return true; if(!(obj instanceof ForeignKey))return false; ForeignKey other = (ForeignKey)obj; return new EqualsBuilder() .append(fkName, other.fkName) .append(columns, other.columns) .append(updateRule, other.updateRule) .append(deleteRule, other.deleteRule) .isEquals(); } @Override public String toString() { return new ToStringBuilder(this) .append("fkName",fkName) .append("columns", columns) .append("updateRule", updateRule) .append("deleteRule", deleteRule) .toString(); } @Override public int hashCode() { return new HashCodeBuilder() .append(fkName) .append(columns) .append(updateRule) .append(deleteRule) .toHashCode(); } public String getFkName() { return fkName; } public Vector getColumns() { return columns; } public Table.ForeignKeyRule getUpdateRule() { return updateRule; } public Table.ForeignKeyRule getDeleteRule() { return deleteRule; } public String asFkVar(){ return StringUtilities.convertName(getUniversalName(),true); } public String asIkVar(){ return StringUtilities.convertName(getUniversalName() + "_of_" +getTable().getBasename(true), true); } public String getUniversalName() { return Joiner.on('_').join(Lists.transform( columns, new Function(){ @Override public String apply(Column input) { return input.getName(); }})); } } public static enum ForeignKeyRule{ CASCADE("DELETE"),RESTRICT,SET_NULL("UPDATE"),NO_ACTION,SET_DEFAULT("UPDATE"); private final String eventOfDeleteRule; private ForeignKeyRule(){ this(""); } private ForeignKeyRule(String event) { this.eventOfDeleteRule = event; } public boolean isNoAction(){ return this == NO_ACTION || this == RESTRICT; } public boolean equals(String value){ try{ if(Strings.isNullOrEmpty(value))return false; return this == ForeignKeyRule.valueOf(value.toUpperCase()); }catch(Exception e){ return false; } } public String getEventOfDeleteRule() { return eventOfDeleteRule; } } public String getCyeleTestMethod(ForeignKey fk) { return getSelftMethod(fk,"is_cycle_on_"); } public String getTopMethod(ForeignKey fk) { return getSelftMethod(fk,"top_of_"); } public String getLevelMethod(ForeignKey fk) { return getSelftMethod(fk,"level_of_"); } public String getListMethod(ForeignKey fk) { return getSelftMethod(fk,"list_of_"); } public String getCheckNotCycleMethod(ForeignKey fk) { return getSelftMethod(fk,"check_cycle_of_"); } public String getChildListMethod(ForeignKey fk) { return getSelftMethod(fk,"child_list_by_"); } /** * @param fk * @since 3.23.0 */ public String getChildrenMethod(ForeignKey fk) { return getSelftMethod(fk,"children_by_"); } /** * @param fk * @since 3.23.0 */ public String getChildrenOfBeansMethod(ForeignKey fk) { return getSelftMethod(fk,"children_of_beans_by_"); } /** * @param fk * @since 3.23.0 */ public String getChildrenOfPksMethod(ForeignKey fk) { return getSelftMethod(fk,"children_of_pks_by_"); } private String getSelftMethod(ForeignKey fk,String prefix) { if(null == fk )return null; return StringUtilities.convertName( prefix + getSelfFkSuffix(fk), true); } public String getSelfFkSuffix(ForeignKey fk) { if(null == fk )return null; return Joiner.on('_').join(Lists.transform(fk.columns, new Function(){ @Override public String apply(Column input) { return input.getName(); }})); } public String getGetManagerMethod(){ return "get" + asManagerClassNSP(); } public boolean isHasMaxSize(){ return Iterables.tryFind(cols, new Predicate(){ @Override public boolean apply(Column input) { return input.isMaxSize(); }}).isPresent(); } public boolean isHasNotNullNoDef(){ return Iterables.tryFind(cols, new Predicate(){ @Override public boolean apply(Column input) { return input.isNotNull() && (input.getOriginalDefaultValue() == null); }}).isPresent(); } public boolean isHasCrossableDefaultvalue(){ return Iterables.tryFind(cols, new Predicate(){ @Override public boolean apply(Column input) { return input.isCrossableDefaultvalue() ; }}).isPresent(); } public boolean isNeedInvalidValueAnn(){ return Iterables.tryFind(cols, new Predicate(){ @Override public boolean apply(Column input) { return !input.getInvalidValueAnn().isEmpty(); }}).isPresent(); } /** * @see gu.sql2java.generator.SqlComment#getDescName() */ public String getDescName(){ return remarks.getDescName(); } /** * @see gu.sql2java.generator.SqlComment#getNamesTagMap() */ public Map getNamesTagMap() { return remarks.getNamesTagMap(); } public String getExcelSheet(){ return remarks.getExcelAnnotation(ExcelSheet.class, buffer->{buffer.append("sheetName=\"").append(name).append("\"");return 1;}); } private Boolean hasCustomSuperClass; /** * @since 3.31.0 */ public boolean isHasCustomSuperClass() { if(null == hasCustomSuperClass) { synchronized (this) { if(null == hasCustomSuperClass) { hasCustomSuperClass = false; if(!CodeWriter.getSuperClassPackage().isEmpty()) { hasCustomSuperClass = CodeWriter.getSuperClassTablenames().contains(getBasename(true)); } } } } return hasCustomSuperClass; } /** * 返回从配置文件中读取的字段能见度定义,字段名--能见度映射 * @since 3.32.0 */ public Map getVisibilityOfColumns() { return visibilityOfColumns.get(); } /** * 返回指定可见度的字段名集合 * @param visibility * @since 3.32.0 */ public Set getColumnsByVisibility(ColumnVisibility visibility) { return Arrays.stream(getColumns()) .filter(c->c.getVisibility().equals(visibility)) .collect(Collectors.toCollection(TreeSet::new)); /** * 不能直接使用Vector对象的作流处理,否则会抛出 ConcurrentModificationException 异常 * 原因不明 */ // // return getColumnsAsList().stream() // .filter(c->c.getVisibility().equals(visibility)) // .collect(Collectors.toCollection(TreeSet::new)); } /** * 返回指定可见度的字段名集合 * @param visibility * @since 3.32.0 */ public Set getColumnsByVisibility(String visibility) { try { return getColumnsByVisibility(ColumnVisibility.valueOf(visibility)); } catch (Exception e) { e.printStackTrace(); return Collections.emptySet(); } } }