package org.jooq.meta;
import static org.jooq.meta.AbstractDatabase.fetchedSize;
import static org.jooq.meta.AbstractDatabase.getDefinition;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.jooq.tools.JooqLogger;
import org.jooq.tools.StringUtils;
public abstract class AbstractElementContainerDefinition<E extends TypedElementDefinition<?>>
extends AbstractDefinition {
protected static final Pattern PRECISION_SCALE = Pattern.compile("\\((\\d+)\\s*(?:,\\s*(\\d+))?\\)");
private static final JooqLogger log = JooqLogger.getLogger(AbstractElementContainerDefinition.class);
private List<E> elements;
public AbstractElementContainerDefinition(SchemaDefinition schema, String name, String comment) {
this(schema, null, name, comment);
}
public AbstractElementContainerDefinition(SchemaDefinition schema, PackageDefinition pkg, String name, String comment) {
super(schema.getDatabase(), schema, pkg, name, comment, null, null);
}
public AbstractElementContainerDefinition(SchemaDefinition schema, PackageDefinition pkg, String name, String comment, String source) {
super(schema.getDatabase(), schema, pkg, name, comment, null, source);
}
protected final List<E> getElements() {
if (elements == null) {
elements = new ArrayList<>();
try {
Database db = getDatabase();
List<E> e = getElements0();
if (this instanceof TableDefinition) {
boolean hasIdentity = false;
for (E c : e) {
boolean isIdentity = ((ColumnDefinition) c).isIdentity();
if (isIdentity) {
if (hasIdentity) {
log.warn("Multiple identities", "Table " + getOutputName() + " has multiple identity columns. Only the first one is considered.");
break;
}
hasIdentity = true;
}
}
}
if (this instanceof TableDefinition && db.getIncludeExcludeColumns()) {
elements = db.filterExcludeInclude(e);
log.info("Columns fetched", fetchedSize(e, elements));
}
else {
elements = e;
}
db.sort(elements);
}
catch (Exception e) {
log.error("Error while initialising type", e);
}
}
return elements;
}
protected final E getElement(String name) {
return getElement(name, false);
}
protected final E getElement(String name, boolean ignoreCase) {
return getDefinition(getElements(), name, ignoreCase);
}
protected final E getElement(int index) {
return getElements().get(index);
}
protected abstract List<E> getElements0() throws SQLException;
protected Number parsePrecision(String typeName) {
if (typeName.contains("(")) {
Matcher m = PRECISION_SCALE.matcher(typeName);
if (m.find()) {
if (!StringUtils.isBlank(m.group(1))) {
return Integer.valueOf(m.group(1));
}
}
}
return 0;
}
protected Number parseScale(String typeName) {
if (typeName.contains("(")) {
Matcher m = PRECISION_SCALE.matcher(typeName);
if (m.find()) {
if (!StringUtils.isBlank(m.group(2))) {
return Integer.valueOf(m.group(2));
}
}
}
return 0;
}
protected String parseTypeName(String typeName) {
return typeName.replace(" NOT NULL", "");
}
protected boolean parseNotNull(String typeName) {
return typeName.toUpperCase().contains("NOT NULL");
}
}