io.polaris.builder.code.dto.TableDto Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of polaris-builder-core Show documentation
There is a newer version: 3.2.1
package io.polaris.builder.code.dto;

import io.polaris.builder.code.config.ConfigColumn;
import io.polaris.builder.dbv.DbCommentSplits;
import io.polaris.core.string.Strings;
import io.polaris.core.tuple.Tuple2;
import com.thoughtworks.xstream.annotations.XStreamAlias;
import com.thoughtworks.xstream.annotations.XStreamAsAttribute;
import com.thoughtworks.xstream.annotations.XStreamImplicit;
import com.thoughtworks.xstream.annotations.XStreamOmitField;
import lombok.Data;
import lombok.ToString;
import org.apache.commons.lang3.SerializationUtils;

import java.io.Serializable;
import java.util.*;
import java.util.function.Function;

/**
 * @author Qt
 */
@SuppressWarnings("ALL")
@Data
@XStreamAlias("table")
public class TableDto implements Serializable {
	private static final long serialVersionUID = 1L;
	/** 表名 */
	@XStreamAsAttribute
	private String name;
	/** 注释 */
	@XStreamAsAttribute
	private String comment;

	/** 所有列 */
	@ToString.Exclude
	@XStreamAlias(("columns"))
	@XStreamImplicit(itemFieldName = "column", keyFieldName = "name")
	private List columns = new ArrayList<>();


	@XStreamOmitField
	private String catalog = "";
	@XStreamOmitField
	private String schema = "";
	/** 主键列 */
	@ToString.Exclude
	@XStreamOmitField
	private List pkColumns = new ArrayList<>();
	/** 非主键列 */
	@ToString.Exclude
	@XStreamOmitField
	private List normalColumns = new ArrayList();
	@XStreamOmitField
	private Set columnJavaTypes = new LinkedHashSet<>();

	@XStreamOmitField
	private Map property;
	/** 代码生成时对应的包名 */
	@XStreamOmitField
	private String javaPackageName = "";
	/** 代码生成时对应的包名 */
	@XStreamOmitField
	private String javaPackageDir = "";

	/** 表名映射后的类名 */
	@XStreamOmitField
	private String javaClassName;
	/** 表名映射后的类的实例变量名 */
	@XStreamOmitField
	private String javaVariableName;
	@XStreamOmitField
	private String xmlName;
	@XStreamOmitField
	private String label;
	@XStreamOmitField
	private String remark;

	/**
	 * 预处理,对类名、变量名、主键列、非主键列等的处理
	 */
	public void prepare4Java(Function tableNameTrimmer, Function columnNameTrimmer, Map columnMap) {
		{
			Tuple2 tuple = DbCommentSplits.split(this.comment);
			this.label = tuple.getFirst();
			this.remark = tuple.getSecond();
		}
		if (columns == null) {
			columns = new ArrayList<>();
		}
		if (pkColumns == null) {
			pkColumns = new ArrayList<>();
		}
		if (normalColumns == null) {
			normalColumns = new ArrayList<>();
		}
		if (columnJavaTypes == null) {
			columnJavaTypes = new LinkedHashSet<>();
		}

		for (ColumnDto col : columns) {
			col.prepare4Java(columnNameTrimmer, columnMap);
			if (col.getJavaType().contains(".")) {
				columnJavaTypes.add(col.getJavaType());
			}
		}

		if (pkColumns.isEmpty()) {
			for (ColumnDto col : columns) {
				if (col.isPrimary()) {
					pkColumns.add(col);
				}
			}
		} else {
			for (ColumnDto col : pkColumns) {
				col.prepare4Java(columnNameTrimmer, columnMap);
				if (col.getJavaType().contains(".")) {
					columnJavaTypes.add(col.getJavaType());
				}
			}
		}

		if (normalColumns.isEmpty()) {
			for (ColumnDto col : columns) {
				if (!col.isPrimary()) {
					normalColumns.add(col);
				}
			}
		} else {
			for (ColumnDto col : normalColumns) {
				col.prepare4Java(columnNameTrimmer, columnMap);
				if (col.getJavaType().contains(".")) {
					columnJavaTypes.add(col.getJavaType());
				}
			}
		}
		javaPackageDir = javaPackageName.replace('.', '/');

		String name = tableNameTrimmer.apply(this.name);
		char[] nameChars = name.toCharArray();
		StringBuilder sb = new StringBuilder();
		boolean flag = false;
		for (int i = 1; i < nameChars.length; i++) {
			if (nameChars[i] == '_') {
				flag = true;
			} else {
				if (flag) {
					sb.append(Character.toUpperCase(nameChars[i]));
					flag = false;
				} else {
					sb.append(Character.toLowerCase(nameChars[i]));
				}
			}
		}
		javaClassName = Character.toUpperCase(nameChars[0]) + sb.toString();
		javaVariableName = Character.toLowerCase(nameChars[0]) + sb.toString();
		xmlName = name.toLowerCase().replace("_", "-");
	}

	@Override
	public TableDto clone() {
		return SerializationUtils.clone(this);
	}


	public boolean containsColumnName(String columnName){
		for (ColumnDto column : columns) {
			if (column.getName().equals(columnName)){
				return true;
			}
		}
		return false;
	}

	public boolean containsColumnVariableName(String columnVariableName){
		for (ColumnDto column : columns) {
			if (column.getJavaVariableName().equals(columnVariableName)){
				return true;
			}
		}
		return false;
	}
}