com.mysql.cj.xdevapi.AbstractColumnDef Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mysql-connector-java
Show all versions of mysql-connector-java
JDBC Type 4 driver for MySQL
/*
Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
The MySQL Connector/J is licensed under the terms of the GPLv2
, like most MySQL Connectors.
There are special exceptions to the terms and conditions of the GPLv2 as it is applied to
this software, see the FOSS License Exception
.
This program is free software; you can redistribute it and/or modify it under the terms
of the GNU General Public License as published by the Free Software Foundation; version 2
of the License.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with this
program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth
Floor, Boston, MA 02110-1301 USA
*/
package com.mysql.cj.xdevapi;
import java.util.Arrays;
import java.util.stream.Collectors;
import com.mysql.cj.api.xdevapi.ColumnDefinition;
import com.mysql.cj.api.xdevapi.Type;
import com.mysql.cj.core.Messages;
import com.mysql.cj.core.exceptions.ExceptionFactory;
import com.mysql.cj.core.exceptions.WrongArgumentException;
import com.mysql.cj.core.util.StringUtils;
public abstract class AbstractColumnDef> implements ColumnDefinition {
protected String name;
protected Type type;
protected Number length;
protected Boolean notNull;
protected boolean uniqueIndex = false;
protected boolean primaryKey = false;
protected String comment = null;
protected Boolean unsigned;
protected Number decimals;
protected String charset = null;
protected String collation = null;
protected Boolean binary;
protected String[] values;
abstract T self();
@Override
public T notNull() {
this.notNull = true;
return self();
}
@Override
public T uniqueIndex() {
this.uniqueIndex = true;
return self();
}
@Override
public T primaryKey() {
this.primaryKey = true;
return self();
}
@Override
public T comment(String cmt) {
this.comment = cmt;
return self();
}
@Override
public T unsigned() {
this.unsigned = true;
return self();
}
@Override
public T decimals(int val) {
this.decimals = val;
return self();
}
@Override
public T charset(String charsetName) {
this.charset = charsetName;
return self();
}
@Override
public T collation(String collationName) {
this.collation = collationName;
return self();
}
@Override
public T binary() {
this.binary = true;
return self();
}
@Override
public T values(String... val) {
this.values = val;
return self();
}
protected String getMysqlType() {
StringBuilder sb = new StringBuilder();
String mysqlTypeName;
switch (this.type) {
case STRING:
mysqlTypeName = "VARCHAR";
break;
case BYTES:
mysqlTypeName = "VARBINARY";
break;
default:
mysqlTypeName = this.type.name();
break;
}
sb.append(mysqlTypeName);
if (this.length != null) {
switch (this.type) {
case JSON:
case GEOMETRY:
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.0", new String[] { "Length parameter", mysqlTypeName, this.name }));
default:
sb.append("(").append(this.length);
if (this.decimals != null) {
switch (this.type) {
case DECIMAL:
case DOUBLE:
case FLOAT:
sb.append(", ").append(this.decimals);
break;
default:
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.0", new String[] { "Decimals parameter", mysqlTypeName, this.name }));
}
}
sb.append(")");
break;
}
} else {
switch (this.type) {
case ENUM:
case SET:
sb.append(Arrays.stream(this.values).map(v -> StringUtils.quoteIdentifier(v, "'", true)).collect(Collectors.joining(",", "(", ")")));
break;
default:
if (this.values != null) {
throw ExceptionFactory.createException(WrongArgumentException.class, Messages.getString("ColumnDefinition.0",
new String[] { Arrays.stream(this.values).collect(Collectors.joining(", ")), mysqlTypeName, this.name }));
}
if (this.decimals != null) {
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.1", new String[] { this.name }));
}
break;
}
}
if (this.unsigned != null && this.unsigned) {
switch (this.type) {
case TINYINT:
case SMALLINT:
case MEDIUMINT:
case INT:
case BIGINT:
case DOUBLE:
case FLOAT:
case DECIMAL:
sb.append(" UNSIGNED");
break;
default:
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.0", new String[] { "UNSIGNED", mysqlTypeName, this.name }));
}
}
if (this.binary != null && this.binary) {
if (this.type == Type.STRING) {
sb.append(" BINARY");
} else {
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.0", new String[] { "BINARY", mysqlTypeName, this.name }));
}
}
if (this.charset != null && !this.charset.isEmpty()) {
switch (this.type) {
case STRING:
case ENUM:
case SET:
sb.append(" CHARACTER SET ").append(this.charset);
break;
default:
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.0", new String[] { "CHARACTER SET", mysqlTypeName, this.name }));
}
}
if (this.collation != null && !this.collation.isEmpty()) {
switch (this.type) {
case STRING:
case ENUM:
case SET:
sb.append(" COLLATE ").append(this.collation);
break;
default:
throw ExceptionFactory.createException(WrongArgumentException.class,
Messages.getString("ColumnDefinition.0", new String[] { "COLLATE", mysqlTypeName, this.name }));
}
}
return sb.toString();
}
}