io.polaris.builder.code.dto.ColumnDto 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.JdbcTypes;
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.XStreamOmitField;
import lombok.Data;
import org.apache.commons.lang3.SerializationUtils;

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

/**
 * @author Qt
 */
@SuppressWarnings({"AlibabaAvoidCommentBehindStatement"})
@Data
@XStreamAlias("column")
public class ColumnDto implements Serializable {
	private static final long serialVersionUID = 1L;
	@XStreamAsAttribute
	private String name; // 列名
	@XStreamAsAttribute
	@XStreamOmitField
	private Integer type; // 列类型 java.sql.Types
	@XStreamAsAttribute
	private String comment; // 注释
	@XStreamAsAttribute
	private String defaultValue; // 默认值
	@XStreamAsAttribute
	private boolean primary = false; // 是否主键列
	@XStreamAsAttribute
	private boolean nullable = true; // 是否可为空值
	@XStreamAsAttribute
	private int columnSize;
	@XStreamAsAttribute
	private int decimalDigits;
	@XStreamAsAttribute
	private boolean generated = false;
	@XStreamAsAttribute
	private boolean autoincrement = false;

	@XStreamAsAttribute
	@XStreamAlias("jdbcType")
	private String jdbcType; // 列的JDBC类型
	@XStreamAsAttribute
	@XStreamAlias("javaType")
	private String javaType; // 列的映射类型

	@XStreamOmitField
	private String javaTypeSimpleName;
	@XStreamOmitField
	private String javaClassName; // 列的映射类名,一般列只作为类的字段,不会用到此值
	@XStreamOmitField
	private String javaVariableName; // 列的映射变量名
	@XStreamOmitField
	private String xmlName;
	@XStreamOmitField
	private String trimName;
	@XStreamOmitField
	private String label;
	@XStreamOmitField
	private String remark;


	public boolean isNotNull() {
		return !nullable;
	}

	public void setNotNull(boolean notNull) {
		this.nullable = !notNull;
	}

	/**
	 * 预处理,对列的映射变量名、映射类型等的处理
	 */
	public void prepare4Java(Function columnNameTrimmer, Map columnMap) {
		{
			Tuple2 tuple = DbCommentSplits.split(this.comment);
			this.label = tuple.getFirst();
			this.remark = tuple.getSecond();
		}

		{
			String name = columnNameTrimmer.apply(this.name);
			this.trimName = name;
			this.xmlName = name.toLowerCase().replace("_", "-");
			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]));
					}
				}
			}
			if (Strings.isEmpty(this.javaClassName)) {
				this.javaClassName = Character.toUpperCase(nameChars[0]) + sb.toString();
			}
			if (Strings.isEmpty(this.javaVariableName)) {
				this.javaVariableName = Character.toLowerCase(nameChars[0]) + sb.toString();
			}
		}

		prepare4Type();

		ConfigColumn configColumn = columnMap.get(this.name);
		if (configColumn != null && Strings.isNotBlank(configColumn.getJavaType())) {
			this.javaType = configColumn.getJavaType();
		} else {
			Class c = JdbcTypes.getCustomJavaType(this.type);
			if (c != null) {
				this.javaType = c.getName();
			} else {
				if (Strings.isEmpty(this.javaType)) {
					c = JdbcTypes.getJavaType(this.type, this.columnSize, this.decimalDigits);
					if (c == null) {
						throw new IllegalArgumentException("不支持的JdbcType：" + this.jdbcType);
					}
					this.javaType = c.getName();
				}
			}
		}

		{
			if (this.javaType.matches("java\\.lang\\.\\w+")) {
				this.javaType = this.javaType.replace("java.lang.", "");
				this.javaTypeSimpleName = this.javaType;
			} else {
				int i = this.javaType.lastIndexOf(".");
				if (i >= 0) {
					this.javaTypeSimpleName = this.javaType.substring(i + 1);
				} else {
					this.javaTypeSimpleName = this.javaType;
				}
			}
		}
	}

	public void prepare4Type() {
		if (Strings.isNotEmpty(this.jdbcType)) {
			this.type = JdbcTypes.getTypeValue(this.jdbcType);
			if (this.type == null) {
				throw new IllegalArgumentException("未知JdbcType");
			}
		} else {
			if (this.type == null) {
				throw new IllegalArgumentException("未知JdbcType");
			}
			this.jdbcType = JdbcTypes.getTypeName(this.type, "VARCHAR");
		}

		if (Strings.isEmpty(this.javaType)) {
			Class c = JdbcTypes.getJavaType(this.type, this.columnSize, this.decimalDigits);
			if (c != null) {
				this.javaType = c.getName();
			}
		}
	}

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