Example 1 with UniqueKeyDefinition
use of org.jooq.meta.UniqueKeyDefinition in project jOOQ by jOOQ.
the class JavaGenerator method printTableJPAAnnotation.
protected void printTableJPAAnnotation(JavaWriter out, TableDefinition table) {
SchemaDefinition schema = table.getSchema();
int indent = out.indent();
if (generateJPAAnnotations()) {
// Since JPA 1.0
out.println("@%s", out.ref("jakarta.persistence.Entity"));
// Since JPA 1.0
out.println("@%s(", out.ref("jakarta.persistence.Table"));
out.print("name = \"", out.ref("jakarta.persistence.Table"));
out.print(escapeString(table.getName()));
out.print("\"");
if (!schema.isDefaultSchema()) {
out.println(",");
out.print("schema = \"");
out.print(escapeString(schema.getOutputName()));
out.print("\"");
}
List<UniqueKeyDefinition> keys = table.getUniqueKeys();
if (!keys.isEmpty()) {
out.println(",");
out.print("uniqueConstraints = ");
out.println(scala ? "Array(" : kotlin ? "[" : "{");
for (int i = 0; i < keys.size(); i++) {
UniqueKeyDefinition uk = keys.get(i);
out.print(scala ? "new " : kotlin ? "" : "@").print(out.ref("jakarta.persistence.UniqueConstraint")).print("(");
if (!StringUtils.isBlank(uk.getOutputName()))
out.print("name = \"" + escapeString(uk.getOutputName()) + "\", ");
out.print("columnNames = ").print(scala ? "Array(" : kotlin ? "[ " : "{ ");
List<ColumnDefinition> columns = uk.getKeyColumns();
for (int j = 0; j < columns.size(); j++) {
out.print(j > 0 ? ", " : "");
out.print("\"");
out.print(escapeString(columns.get(j).getName()));
out.print("\"");
}
out.print(scala ? ")" : kotlin ? " ]" : " }").print(")").println(i < keys.size() - 1 ? "," : "");
}
out.print(scala ? ")" : kotlin ? "]" : "}");
}
if (StringUtils.isBlank(generateJPAVersion()) || "2.1".compareTo(generateJPAVersion()) <= 0) {
List<IndexDefinition> indexes = table.getIndexes();
if (!indexes.isEmpty()) {
out.println(",");
out.print("indexes = ").println(scala ? "Array(" : kotlin ? "[" : "{");
for (int i = 0; i < indexes.size(); i++) {
IndexDefinition index = indexes.get(i);
out.print(scala ? "new " : kotlin ? "" : "@");
out.print(out.ref("jakarta.persistence.Index"));
out.print("(name = \"").print(escapeString(index.getOutputName())).print("\"");
if (index.isUnique())
out.print(", unique = true");
out.print(", columnList = \"");
List<IndexColumnDefinition> columns = index.getIndexColumns();
for (int j = 0; j < columns.size(); j++) {
IndexColumnDefinition column = columns.get(j);
if (j > 0)
out.print(", ");
out.print(escapeString(column.getOutputName()));
if (column.getSortOrder() == SortOrder.ASC)
out.print(" ASC");
else if (column.getSortOrder() == SortOrder.DESC)
out.print(" DESC");
}
out.print("\")").println(i < indexes.size() - 1 ? "," : "");
}
out.print(scala ? ")" : kotlin ? "]" : "}");
}
}
out.println();
out.println(")");
}
// [#10196] The above logic triggers an indent level of -1, incorrectly
out.indent(indent);
}
Also used :
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
SchemaDefinition(org.jooq.meta.SchemaDefinition)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
IndexDefinition(org.jooq.meta.IndexDefinition)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
Example 2 with UniqueKeyDefinition
use of org.jooq.meta.UniqueKeyDefinition in project jOOQ by jOOQ.
the class JavaGenerator method generateDao.
protected void generateDao(TableDefinition table, JavaWriter out) {
UniqueKeyDefinition key = table.getPrimaryKey();
if (key == null) {
log.info("Skipping DAO generation", out.file().getName());
return;
}
final String className = getStrategy().getJavaClassName(table, Mode.DAO);
final List<String> interfaces = out.ref(getStrategy().getJavaClassImplements(table, Mode.DAO));
final String tableRecord = out.ref(getStrategy().getFullJavaClassName(table, Mode.RECORD));
final String daoImpl = out.ref(DAOImpl.class);
final String tableIdentifier = out.ref(getStrategy().getFullJavaIdentifier(table), 2);
String tType = (scala || kotlin ? "Unit" : "Void");
String pType = out.ref(getStrategy().getFullJavaClassName(table, Mode.POJO));
List<ColumnDefinition> keyColumns = key.getKeyColumns();
if (keyColumns.size() == 1) {
tType = getJavaType(keyColumns.get(0).getType(resolver(out)), out, Mode.POJO);
} else if (keyColumns.size() <= Constants.MAX_ROW_DEGREE) {
StringBuilder generics = new StringBuilder();
forEach(keyColumns, "", ", ", (column, separator) -> {
generics.append(out.ref(getJavaType(column.getType(resolver(out)), out)));
if (kotlin)
generics.append("?");
generics.append(separator);
});
if (scala)
tType = Record.class.getName() + keyColumns.size() + "[" + generics + "]";
else
tType = Record.class.getName() + keyColumns.size() + "<" + generics + ">";
} else {
tType = Record.class.getName();
}
tType = out.ref(tType);
printPackage(out, table, Mode.DAO);
generateDaoClassJavadoc(table, out);
printClassAnnotations(out, table, Mode.DAO);
if (generateSpringAnnotations())
out.println("@%s", out.ref("org.springframework.stereotype.Repository"));
if (scala)
out.println("%sclass %s(configuration: %s) extends %s[%s, %s, %s](%s, classOf[%s], configuration)[[before= with ][separator= with ][%s]] {", visibility(), className, Configuration.class, daoImpl, tableRecord, pType, tType, tableIdentifier, pType, interfaces);
else if (kotlin)
out.println("%sopen class %s(configuration: %s?) : %s<%s, %s, %s>(%s, %s::class.java, configuration)[[before=, ][%s]] {", visibility(), className, Configuration.class, daoImpl, tableRecord, pType, tType, tableIdentifier, pType, interfaces);
else
out.println("%sclass %s extends %s<%s, %s, %s>[[before= implements ][%s]] {", visibility(), className, daoImpl, tableRecord, pType, tType, interfaces);
// Default constructor
// -------------------
out.javadoc("Create a new %s without any configuration", className);
if (scala) {
out.println("%sdef this() = this(null)", visibility());
} else if (kotlin) {
out.println("%sconstructor(): this(null)", visibility());
} else {
out.println("%s%s() {", visibility(), className);
out.println("super(%s, %s.class);", tableIdentifier, pType);
out.println("}");
}
if (!scala && !kotlin) {
out.javadoc("Create a new %s with an attached configuration", className);
printDaoConstructorAnnotations(table, out);
out.println("%s%s(%s configuration) {", visibility(), className, Configuration.class);
out.println("super(%s, %s.class, configuration);", tableIdentifier, pType);
out.println("}");
}
// -------------------------------
if (scala) {
out.println();
out.print("%soverride def getId(o: %s): %s = ", visibilityPublic(), pType, tType);
} else if (kotlin) {
out.println();
out.print("%soverride fun getId(o: %s): %s? = ", visibilityPublic(), pType, tType);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s getId(%s object) {", visibilityPublic(), tType, pType);
}
if (keyColumns.size() == 1) {
if (scala)
out.println("o.%s", getStrategy().getJavaGetterName(keyColumns.get(0), Mode.POJO));
else if (kotlin)
out.println("o.%s", getStrategy().getJavaMemberName(keyColumns.get(0), Mode.POJO));
else
out.println("return object.%s();", generatePojosAsJavaRecordClasses() ? getStrategy().getJavaMemberName(keyColumns.get(0), Mode.POJO) : getStrategy().getJavaGetterName(keyColumns.get(0), Mode.POJO));
} else // [#2574] This should be replaced by a call to a method on the target table's Key type
{
StringBuilder params = new StringBuilder();
forEach(keyColumns, "", ", ", (column, separator) -> {
if (scala)
params.append("o.").append(getStrategy().getJavaGetterName(column, Mode.POJO));
else if (kotlin)
params.append("o.").append(getStrategy().getJavaMemberName(column, Mode.POJO));
else
params.append("object.").append(generatePojosAsJavaRecordClasses() ? getStrategy().getJavaMemberName(column, Mode.POJO) : getStrategy().getJavaGetterName(column, Mode.POJO)).append("()");
params.append(separator);
});
if (scala || kotlin)
out.println("compositeKeyRecord(%s)", params.toString());
else
out.println("return compositeKeyRecord(%s);", params.toString());
}
if (scala || kotlin) {
} else
out.println("}");
List<Definition> embeddablesAndUnreplacedColumns = embeddablesAndUnreplacedColumns(table);
for (Definition column : embeddablesAndUnreplacedColumns) {
final String colName = column.getOutputName();
final String colClass = getStrategy().getJavaClassName(column);
final String colMemberUC = StringUtils.toUC(getStrategy().getJavaMemberName(column), getStrategy().getTargetLocale());
final String colTypeFull = getJavaType(column, out, Mode.POJO);
final String colTypeRecord = out.ref(getJavaType(column, out, Mode.RECORD));
final String colType = out.ref(colTypeFull);
final String colIdentifier = out.ref(getStrategy().getFullJavaIdentifier(column), colRefSegments(column));
// -----------------------
if (!printDeprecationIfUnknownType(out, colTypeFull))
out.javadoc("Fetch records that have <code>%s BETWEEN lowerInclusive AND upperInclusive</code>", colName);
if (scala) {
if (column instanceof EmbeddableDefinition)
out.println("%sdef fetchRangeOf%s(lowerInclusive: %s, upperInclusive: %s): %s[%s] = fetchRange(%s, new %s(lowerInclusive), new %s(upperInclusive))", visibility(), colMemberUC, colType, colType, List.class, pType, colIdentifier, colTypeRecord, colTypeRecord);
else
out.println("%sdef fetchRangeOf%s(lowerInclusive: %s, upperInclusive: %s): %s[%s] = fetchRange(%s, lowerInclusive, upperInclusive)", visibility(), colClass, colType, colType, List.class, pType, colIdentifier);
} else if (kotlin) {
if (column instanceof EmbeddableDefinition)
out.println("%sfun fetchRangeOf%s(lowerInclusive: %s?, upperInclusive: %s?): %s<%s> = fetchRange(%s, if (lowerInclusive != null) %s(lowerInclusive) else null, if (upperInclusive != null) %s(upperInclusive) else null)", visibility(), colMemberUC, colType, colType, out.ref(KLIST), pType, colIdentifier, colTypeRecord, colTypeRecord);
else
out.println("%sfun fetchRangeOf%s(lowerInclusive: %s?, upperInclusive: %s?): %s<%s> = fetchRange(%s, lowerInclusive, upperInclusive)", visibility(), colClass, colType, colType, out.ref(KLIST), pType, colIdentifier);
} else {
printNonnullAnnotation(out);
if (column instanceof EmbeddableDefinition) {
out.println("%s%s<%s> fetchRangeOf%s(%s lowerInclusive, %s upperInclusive) {", visibility(), List.class, pType, colMemberUC, colType, colType);
out.println("return fetchRange(%s, new %s(lowerInclusive), new %s(upperInclusive));", colIdentifier, colTypeRecord, colTypeRecord);
} else {
out.println("%s%s<%s> fetchRangeOf%s(%s lowerInclusive, %s upperInclusive) {", visibility(), List.class, pType, colClass, colType, colType);
out.println("return fetchRange(%s, lowerInclusive, upperInclusive);", colIdentifier);
}
out.println("}");
}
// -----------------------
if (!printDeprecationIfUnknownType(out, colTypeFull))
out.javadoc("Fetch records that have <code>%s IN (values)</code>", colName);
if (scala) {
if (column instanceof EmbeddableDefinition)
out.println("%sdef fetchBy%s(values: %s*): %s[%s] = fetch(%s, values.map(v => new %s(v)).toArray:_*)", visibility(), colMemberUC, colType, List.class, pType, colIdentifier, colTypeRecord);
else
out.println("%sdef fetchBy%s(values: %s*): %s[%s] = fetch(%s, values:_*)", visibility(), colClass, colType, List.class, pType, colIdentifier);
} else if (kotlin) {
String toTypedArray = PRIMITIVE_WRAPPERS.contains(colTypeFull) ? ".toTypedArray()" : "";
if (column instanceof EmbeddableDefinition)
out.println("%sfun fetchBy%s(vararg values: %s): %s<%s> = fetch(%s, values.map { %s(it) })", visibility(), colMemberUC, colType, out.ref(KLIST), pType, colIdentifier, colTypeRecord);
else
out.println("%sfun fetchBy%s(vararg values: %s): %s<%s> = fetch(%s, *values%s)", visibility(), colClass, colType, out.ref(KLIST), pType, colIdentifier, toTypedArray);
} else {
printNonnullAnnotation(out);
if (column instanceof EmbeddableDefinition) {
out.println("%s%s<%s> fetchBy%s(%s... values) {", visibility(), List.class, pType, colMemberUC, colType);
out.println("%s[] records = new %s[values.length];", colTypeRecord, colTypeRecord);
out.println();
out.println("for (int i = 0; i < values.length; i++)");
out.tab(1).println("records[i] = new %s(values[i]);", colTypeRecord);
out.println();
out.println("return fetch(%s, records);", colIdentifier);
} else {
out.println("%s%s<%s> fetchBy%s(%s... values) {", visibility(), List.class, pType, colClass, colType);
out.println("return fetch(%s, values);", colIdentifier);
}
out.println("}");
}
// -----------------------
ukLoop: if (column instanceof ColumnDefinition) {
for (UniqueKeyDefinition uk : ((ColumnDefinition) column).getKeys()) {
// If column is part of a single-column unique key...
if (uk.getKeyColumns().size() == 1 && uk.getKeyColumns().get(0).equals(column)) {
if (!printDeprecationIfUnknownType(out, colTypeFull))
out.javadoc("Fetch a unique record that has <code>%s = value</code>", colName);
if (scala) {
out.println("%sdef fetchOneBy%s(value: %s): %s = fetchOne(%s, value)", visibility(), colClass, colType, pType, colIdentifier);
} else if (kotlin) {
out.println("%sfun fetchOneBy%s(value: %s): %s? = fetchOne(%s, value)", visibility(), colClass, colType, pType, colIdentifier);
} else {
printNullableAnnotation(out);
out.println("%s%s fetchOneBy%s(%s value) {", visibility(), pType, colClass, colType);
out.println("return fetchOne(%s, value);", colIdentifier);
out.println("}");
if (!printDeprecationIfUnknownType(out, colTypeFull))
out.javadoc("Fetch a unique record that has <code>%s = value</code>", colName);
printNonnullAnnotation(out);
out.println("%s%s<%s> fetchOptionalBy%s(%s value) {", visibility(), Optional.class, pType, colClass, colType);
out.println("return fetchOptional(%s, value);", colIdentifier);
out.println("}");
}
break ukLoop;
}
}
}
}
generateDaoClassFooter(table, out);
out.println("}");
}
Also used :
RandomAccessFile(java.io.RandomAccessFile)
Arrays(java.util.Arrays)
TableRecordImpl(org.jooq.impl.TableRecordImpl)
DSL(org.jooq.impl.DSL)
SCALA(org.jooq.codegen.Language.SCALA)
StringUtils.isBlank(org.jooq.tools.StringUtils.isBlank)
Collectors.counting(java.util.stream.Collectors.counting)
Internal(org.jooq.impl.Internal)
POSTGRES(org.jooq.SQLDialect.POSTGRES)
DatatypeConverter(jakarta.xml.bind.DatatypeConverter)
AttributeDefinition(org.jooq.meta.AttributeDefinition)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
RoutineDefinition(org.jooq.meta.RoutineDefinition)
Matcher(java.util.regex.Matcher)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
SelectField(org.jooq.SelectField)
IdentityDefinition(org.jooq.meta.IdentityDefinition)
Arrays.asList(java.util.Arrays.asList)
Index(org.jooq.Index)
Map(java.util.Map)
BigInteger(java.math.BigInteger)
SQLDialect(org.jooq.SQLDialect)
Domain(org.jooq.Domain)
Database(org.jooq.meta.Database)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
CheckConstraintDefinition(org.jooq.meta.CheckConstraintDefinition)
Set(java.util.Set)
Constants(org.jooq.Constants)
AggregateFunction(org.jooq.AggregateFunction)
Result(org.jooq.Result)
Collectors.joining(java.util.stream.Collectors.joining)
GenerationOption(org.jooq.impl.QOM.GenerationOption)
Stream(java.util.stream.Stream)
TableField(org.jooq.TableField)
SQLDataType(org.jooq.impl.SQLDataType)
GeneratedAnnotationType(org.jooq.meta.jaxb.GeneratedAnnotationType)
SequenceDefinition(org.jooq.meta.SequenceDefinition)
Collectors.groupingBy(java.util.stream.Collectors.groupingBy)
Parameter(org.jooq.Parameter)
TableDefinition(org.jooq.meta.TableDefinition)
CatalogImpl(org.jooq.impl.CatalogImpl)
IndexDefinition(org.jooq.meta.IndexDefinition)
Sequence(org.jooq.Sequence)
ArrayList(java.util.ArrayList)
LinkedHashMap(java.util.LinkedHashMap)
Calendar(java.util.Calendar)
Records(org.jooq.Records)
Schema(org.jooq.Schema)
BiConsumer(java.util.function.BiConsumer)
RowId(org.jooq.RowId)
LinkedHashSet(java.util.LinkedHashSet)
MYSQL(org.jooq.SQLDialect.MYSQL)
Record(org.jooq.Record)
Mode(org.jooq.codegen.GeneratorStrategy.Mode)
TypeVariable(java.lang.reflect.TypeVariable)
IOException(java.io.IOException)
StringUtils(org.jooq.tools.StringUtils)
File(java.io.File)
YUGABYTEDB(org.jooq.SQLDialect.YUGABYTEDB)
VIRTUAL(org.jooq.impl.QOM.GenerationOption.VIRTUAL)
SchemaDefinition(org.jooq.meta.SchemaDefinition)
LazySupplier(org.jooq.impl.LazySupplier)
TableImpl(org.jooq.impl.TableImpl)
CatalogDefinition(org.jooq.meta.CatalogDefinition)
UniqueKey(org.jooq.UniqueKey)
Row(org.jooq.Row)
IntStream.rangeClosed(java.util.stream.IntStream.rangeClosed)
Time(java.sql.Time)
Table(org.jooq.Table)
DataTypeDefinition(org.jooq.meta.DataTypeDefinition)
DAOImpl(org.jooq.impl.DAOImpl)
PackageDefinition(org.jooq.meta.PackageDefinition)
DefaultDataType(org.jooq.impl.DefaultDataType)
UDT(org.jooq.UDT)
PostgresDatabase(org.jooq.meta.postgres.PostgresDatabase)
SortOrder(org.jooq.SortOrder)
DataType(org.jooq.DataType)
DSL.name(org.jooq.impl.DSL.name)
IdentityHashMap(java.util.IdentityHashMap)
JooqLogger(org.jooq.tools.JooqLogger)
SQLDialectNotSupportedException(org.jooq.exception.SQLDialectNotSupportedException)
Collections.nCopies(java.util.Collections.nCopies)
TimeZone(java.util.TimeZone)
Name(org.jooq.Name)
Timestamp(java.sql.Timestamp)
Collection(java.util.Collection)
Check(org.jooq.Check)
Field(org.jooq.Field)
CloseResult(org.jooq.codegen.GeneratorWriter.CloseResult)
PostgresRoutineDefinition(org.jooq.meta.postgres.PostgresRoutineDefinition)
Collectors(java.util.stream.Collectors)
UpdatableRecordImpl(org.jooq.impl.UpdatableRecordImpl)
TypedElementDefinition(org.jooq.meta.TypedElementDefinition)
JAVA(org.jooq.codegen.Language.JAVA)
DomainDefinition(org.jooq.meta.DomainDefinition)
Reflect(org.jooq.tools.reflect.Reflect)
List(java.util.List)
Modifier(java.lang.reflect.Modifier)
Function.identity(java.util.function.Function.identity)
Optional(java.util.Optional)
Pattern(java.util.regex.Pattern)
IntStream(java.util.stream.IntStream)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
KOTLIN(org.jooq.codegen.Language.KOTLIN)
UDTField(org.jooq.UDTField)
DefaultDataTypeDefinition(org.jooq.meta.DefaultDataTypeDefinition)
HashMap(java.util.HashMap)
STORED(org.jooq.impl.QOM.GenerationOption.STORED)
EnumType(org.jooq.EnumType)
ForeignKey(org.jooq.ForeignKey)
Function(java.util.function.Function)
PackageImpl(org.jooq.impl.PackageImpl)
Definition(org.jooq.meta.Definition)
HashSet(java.util.HashSet)
Identity(org.jooq.Identity)
TableLike(org.jooq.TableLike)
EmbeddableRecordImpl(org.jooq.impl.EmbeddableRecordImpl)
EmbeddableDefinition(org.jooq.meta.EmbeddableDefinition)
AbstractRoutine(org.jooq.impl.AbstractRoutine)
Function2(org.jooq.Function2)
ArrayDefinition(org.jooq.meta.ArrayDefinition)
UDTDefinition(org.jooq.meta.UDTDefinition)
AbstractTypedElementDefinition.getDataType(org.jooq.meta.AbstractTypedElementDefinition.getDataType)
JavaTypeResolver(org.jooq.meta.JavaTypeResolver)
ParameterDefinition(org.jooq.meta.ParameterDefinition)
UDTRecordImpl(org.jooq.impl.UDTRecordImpl)
ConstraintDefinition(org.jooq.meta.ConstraintDefinition)
LazySchema(org.jooq.impl.LazySchema)
DESC(org.jooq.SortOrder.DESC)
SchemaImpl(org.jooq.impl.SchemaImpl)
Date(java.sql.Date)
ReflectException(org.jooq.tools.reflect.ReflectException)
AbstractTypedElementDefinition(org.jooq.meta.AbstractTypedElementDefinition)
Configuration(org.jooq.Configuration)
Collectors.toList(java.util.stream.Collectors.toList)
Catalog(org.jooq.Catalog)
UDTImpl(org.jooq.impl.UDTImpl)
StopWatch(org.jooq.tools.StopWatch)
OrderField(org.jooq.OrderField)
TableOptions(org.jooq.TableOptions)
EnumDefinition(org.jooq.meta.EnumDefinition)
ForeignKeyDefinition(org.jooq.meta.ForeignKeyDefinition)
Collections(java.util.Collections)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
Configuration(org.jooq.Configuration)
AttributeDefinition(org.jooq.meta.AttributeDefinition)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
RoutineDefinition(org.jooq.meta.RoutineDefinition)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
IdentityDefinition(org.jooq.meta.IdentityDefinition)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
CheckConstraintDefinition(org.jooq.meta.CheckConstraintDefinition)
SequenceDefinition(org.jooq.meta.SequenceDefinition)
TableDefinition(org.jooq.meta.TableDefinition)
IndexDefinition(org.jooq.meta.IndexDefinition)
SchemaDefinition(org.jooq.meta.SchemaDefinition)
CatalogDefinition(org.jooq.meta.CatalogDefinition)
DataTypeDefinition(org.jooq.meta.DataTypeDefinition)
PackageDefinition(org.jooq.meta.PackageDefinition)
PostgresRoutineDefinition(org.jooq.meta.postgres.PostgresRoutineDefinition)
TypedElementDefinition(org.jooq.meta.TypedElementDefinition)
DomainDefinition(org.jooq.meta.DomainDefinition)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
DefaultDataTypeDefinition(org.jooq.meta.DefaultDataTypeDefinition)
Definition(org.jooq.meta.Definition)
EmbeddableDefinition(org.jooq.meta.EmbeddableDefinition)
ArrayDefinition(org.jooq.meta.ArrayDefinition)
UDTDefinition(org.jooq.meta.UDTDefinition)
ParameterDefinition(org.jooq.meta.ParameterDefinition)
ConstraintDefinition(org.jooq.meta.ConstraintDefinition)
AbstractTypedElementDefinition(org.jooq.meta.AbstractTypedElementDefinition)
EnumDefinition(org.jooq.meta.EnumDefinition)
ForeignKeyDefinition(org.jooq.meta.ForeignKeyDefinition)
EmbeddableDefinition(org.jooq.meta.EmbeddableDefinition)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
Record(org.jooq.Record)
Arrays.asList(java.util.Arrays.asList)
ArrayList(java.util.ArrayList)
List(java.util.List)
Collectors.toList(java.util.stream.Collectors.toList)
Example 3 with UniqueKeyDefinition
use of org.jooq.meta.UniqueKeyDefinition in project jOOQ by jOOQ.
the class JavaGenerator method generateTable.
protected void generateTable(TableDefinition table, JavaWriter out) {
final SchemaDefinition schema = table.getSchema();
final UniqueKeyDefinition primaryKey = table.getPrimaryKey();
final String className = getStrategy().getJavaClassName(table);
final String tableId = scala ? out.ref(getStrategy().getFullJavaIdentifier(table), 2) : getStrategy().getJavaIdentifier(table);
final String recordType = out.ref(getStrategy().getFullJavaClassName(table, Mode.RECORD));
final List<String> interfaces = out.ref(getStrategy().getJavaClassImplements(table, Mode.DEFAULT));
final String schemaId = out.ref(getStrategy().getFullJavaIdentifier(schema), 2);
final String tableType = table.isTemporary() ? "temporaryTable" : table.isView() ? "view" : table.isMaterializedView() ? "materializedView" : table.isTableValuedFunction() ? "function" : "table";
final List<ParameterDefinition> parameters = table.getParameters();
printPackage(out, table);
if (scala) {
out.println("object %s {", className);
printSingletonInstance(out, table);
out.println("}");
out.println();
}
generateTableClassJavadoc(table, out);
printClassAnnotations(out, table, Mode.DEFAULT);
if (scala) {
out.println("%sclass %s(", visibility(), className);
out.println("alias: %s,", Name.class);
out.println("child: %s[_ <: %s],", Table.class, Record.class);
out.println("path: %s[_ <: %s, %s],", ForeignKey.class, Record.class, recordType);
out.println("aliased: %s[%s],", Table.class, recordType);
out.println("parameters: %s[ %s[_] ]", out.ref("scala.Array"), Field.class);
out.println(")");
out.println("extends %s[%s](", TableImpl.class, recordType);
out.println("alias,");
out.println("%s,", schemaId);
out.println("child,");
out.println("path,");
out.println("aliased,");
out.println("parameters,");
out.println("%s.comment(\"%s\"),", DSL.class, escapeString(comment(table)));
if ((generateSourcesOnViews() || table.isSynthetic()) && table.isView() && table.getSource() != null)
out.println("%s.%s(\"%s\")", TableOptions.class, tableType, escapeString(table.getSource()));
else if (table.isSynthetic() && table.isTableValuedFunction() && table.getSource() != null)
out.println("%s.%s(\"%s\")", TableOptions.class, tableType, escapeString(table.getSource()));
else
out.println("%s.%s", TableOptions.class, tableType);
out.println(")[[before= with ][separator= with ][%s]] {", interfaces);
} else if (kotlin) {
out.println("%sopen class %s(", visibility(), className);
out.println("alias: %s,", Name.class);
out.println("child: %s<out %s>?,", Table.class, Record.class);
out.println("path: %s<out %s, %s>?,", ForeignKey.class, Record.class, recordType);
out.println("aliased: %s<%s>?,", Table.class, recordType);
out.println("parameters: Array<%s<*>?>?", Field.class);
out.println("): %s<%s>(", TableImpl.class, recordType);
out.println("alias,");
out.println("%s,", schemaId);
out.println("child,");
out.println("path,");
out.println("aliased,");
out.println("parameters,");
out.println("%s.comment(\"%s\"),", DSL.class, escapeString(comment(table)));
if ((generateSourcesOnViews() || table.isSynthetic()) && table.isView() && table.getSource() != null)
out.println("%s.%s(\"%s\")", TableOptions.class, tableType, escapeString(table.getSource()));
else if (table.isSynthetic() && table.isTableValuedFunction() && table.getSource() != null)
out.println("%s.%s(\"%s\")", TableOptions.class, tableType, escapeString(table.getSource()));
else
out.println("%s.%s()", TableOptions.class, tableType);
out.println(")[[before=, ][%s]] {", interfaces);
out.println("%scompanion object {", visibility());
printSingletonInstance(out, table);
out.println("}");
} else {
out.println("%sclass %s extends %s<%s>[[before= implements ][%s]] {", visibility(), className, TableImpl.class, recordType, interfaces);
out.printSerial();
printSingletonInstance(out, table);
}
printRecordTypeMethod(out, table);
if (table.isSynthetic()) {
if (scala) {
out.println();
out.println("protected override def isSynthetic(): Boolean = true");
} else if (kotlin) {
out.println();
out.println("protected override fun isSynthetic(): Boolean = true");
} else {
out.println();
out.override();
out.println("protected boolean isSynthetic() {");
out.println("return true;");
out.println("}");
}
}
for (ColumnDefinition column : table.getColumns()) {
final String columnTypeFull = getJavaType(column.getType(resolver(out)), out);
final String columnType = out.ref(columnTypeFull);
final String columnTypeRef = getJavaTypeReference(column.getDatabase(), column.getType(resolver(out)), out);
final String columnId = out.ref(getStrategy().getJavaIdentifier(column), colRefSegments(column));
final String columnName = column.getName();
final List<String> converter = out.ref(list(column.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(column.getType(resolver(out)).getBinding()));
final String columnVisibility = visibility();
if (!printDeprecationIfUnknownType(out, columnTypeFull))
out.javadoc("The column <code>%s</code>.[[before= ][%s]]", column.getQualifiedOutputName(), list(escapeEntities(comment(column))));
if (scala) {
out.println("%sval %s: %s[%s, %s] = createField(%s.name(\"%s\"), %s, \"%s\"" + converterTemplate(converter) + converterTemplate(binding) + ")", columnVisibility, scalaWhitespaceSuffix(columnId), TableField.class, recordType, columnType, DSL.class, columnName, columnTypeRef, escapeString(comment(column)), converter, binding);
} else if (kotlin) {
out.println("%sval %s: %s<%s, %s?> = createField(%s.name(\"%s\"), %s, this, \"%s\"" + converterTemplate(converter) + converterTemplate(binding) + ")", columnVisibility, columnId, TableField.class, recordType, columnType, DSL.class, columnName, columnTypeRef, escapeString(comment(column)), converter, binding);
} else {
String isStatic = generateInstanceFields() ? "" : "static ";
String tableRef = generateInstanceFields() ? "this" : out.ref(getStrategy().getJavaIdentifier(table), 2);
out.println("%s%sfinal %s<%s, %s> %s = createField(%s.name(\"%s\"), %s, %s, \"%s\"" + converterTemplate(converter) + converterTemplate(binding) + ");", columnVisibility, isStatic, TableField.class, recordType, columnType, columnId, DSL.class, columnName, columnTypeRef, tableRef, escapeString(comment(column)), converter, binding);
}
}
// [#2530] Embeddable types
for (EmbeddableDefinition embeddable : table.getReferencedEmbeddables()) {
final String columnId = out.ref(getStrategy().getJavaIdentifier(embeddable), colRefSegments(null));
final String columnType = out.ref(getStrategy().getFullJavaClassName(embeddable, Mode.RECORD));
final List<String> columnIds = new ArrayList<>();
for (EmbeddableColumnDefinition column : embeddable.getColumns()) columnIds.add(out.ref(getStrategy().getJavaIdentifier(column.getReferencingColumn()), colRefSegments(column.getReferencingColumn())));
out.javadoc("The embeddable type <code>%s</code>.[[before= ][%s]]", embeddable.getOutputName(), list(escapeEntities(referencingComment(embeddable))));
if (scala)
out.println("%sval %s: %s[%s, %s] = %s.createEmbeddable(%s.name(\"%s\"), classOf[%s], %s, this, [[%s]])", visibility(), scalaWhitespaceSuffix(columnId), TableField.class, recordType, columnType, Internal.class, DSL.class, escapeString(embeddable.getName()), columnType, embeddable.replacesFields(), columnIds);
else if (kotlin)
out.println("%sval %s: %s<%s, %s> = %s.createEmbeddable(%s.name(\"%s\"), %s::class.java, %s, this, [[%s]])", visibility(), columnId, TableField.class, recordType, columnType, Internal.class, DSL.class, escapeString(embeddable.getName()), columnType, embeddable.replacesFields(), columnIds);
else
out.println("%sfinal %s<%s, %s> %s = %s.createEmbeddable(%s.name(\"%s\"), %s.class, %s, this, [[%s]]);", visibility(), TableField.class, recordType, columnType, columnId, Internal.class, DSL.class, escapeString(embeddable.getName()), columnType, embeddable.replacesFields(), columnIds);
}
out.println();
// "called constructor's definition must precede calling constructor's definition"
if (scala) {
if (table.isTableValuedFunction()) {
out.println("private def this(alias: %s, aliased: %s[%s]) = this(alias, null, null, aliased, %s(", Name.class, Table.class, recordType, out.ref("scala.Array"));
forEach(parameters, (parameter, separator) -> {
final String paramTypeRef = getJavaTypeReference(parameter.getDatabase(), parameter.getType(resolver(out)), out);
final List<String> converter = out.ref(list(parameter.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(parameter.getType(resolver(out)).getBinding()));
out.println("%s.value(null, %s" + converterTemplateForTableValuedFunction(converter) + converterTemplateForTableValuedFunction(binding) + ")%s", DSL.class, paramTypeRef, converter, binding, separator);
});
out.println("))");
} else
out.println("private def this(alias: %s, aliased: %s[%s]) = this(alias, null, null, aliased, null)", Name.class, Table.class, recordType);
} else if (kotlin) {
if (table.isTableValuedFunction()) {
out.println("private constructor(alias: %s, aliased: %s<%s>?): this(alias, null, null, aliased, arrayOf(", Name.class, Table.class, recordType, Field.class, parameters.size());
forEach(parameters, (parameter, separator) -> {
final String paramTypeRef = getJavaTypeReference(parameter.getDatabase(), parameter.getType(resolver(out)), out);
final List<String> converter = out.ref(list(parameter.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(parameter.getType(resolver(out)).getBinding()));
out.println("%s.value(null, %s" + converterTemplateForTableValuedFunction(converter) + converterTemplateForTableValuedFunction(binding) + ")%s", DSL.class, paramTypeRef, converter, binding, separator);
});
out.println("))");
} else
out.println("private constructor(alias: %s, aliased: %s<%s>?): this(alias, null, null, aliased, null)", Name.class, Table.class, recordType);
out.println("private constructor(alias: %s, aliased: %s<%s>?, parameters: Array<%s<*>?>?): this(alias, null, null, aliased, parameters)", Name.class, Table.class, recordType, Field.class);
} else {
out.println("private %s(%s alias, %s<%s> aliased) {", className, Name.class, Table.class, recordType);
if (table.isTableValuedFunction()) {
out.println("this(alias, aliased, new %s[] {", Field.class);
forEach(parameters, (parameter, separator) -> {
final String paramTypeRef = getJavaTypeReference(parameter.getDatabase(), parameter.getType(resolver(out)), out);
final List<String> converter = out.ref(list(parameter.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(parameter.getType(resolver(out)).getBinding()));
out.println("%s.val(null, %s" + converterTemplateForTableValuedFunction(converter) + converterTemplateForTableValuedFunction(binding) + ")%s", DSL.class, paramTypeRef, converter, binding, separator);
});
out.println("});");
} else
out.println("this(alias, aliased, null);");
out.println("}");
out.println();
out.println("private %s(%s alias, %s<%s> aliased, %s<?>[] parameters) {", className, Name.class, Table.class, recordType, Field.class);
if ((generateSourcesOnViews() || table.isSynthetic()) && table.isView() && table.getSource() != null)
out.println("super(alias, null, aliased, parameters, %s.comment(\"%s\"), %s.%s(\"%s\"));", DSL.class, escapeString(comment(table)), TableOptions.class, tableType, escapeString(table.getSource()));
else if (table.isSynthetic() && table.isTableValuedFunction() && table.getSource() != null)
out.println("super(alias, null, aliased, parameters, %s.comment(\"%s\"), %s.%s(\"%s\"));", DSL.class, escapeString(comment(table)), TableOptions.class, tableType, escapeString(table.getSource()));
else
out.println("super(alias, null, aliased, parameters, %s.comment(\"%s\"), %s.%s());", DSL.class, escapeString(comment(table)), TableOptions.class, tableType);
out.println("}");
}
if (scala) {
out.javadoc("Create an aliased <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%sdef this(alias: %s) = this(%s.name(alias), %s)", visibility(), String.class, DSL.class, tableId);
out.javadoc("Create an aliased <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%sdef this(alias: %s) = this(alias, %s)", visibility(), Name.class, tableId);
} else if (kotlin) {
out.javadoc("Create an aliased <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%sconstructor(alias: %s): this(%s.name(alias))", visibility(), String.class, DSL.class);
out.javadoc("Create an aliased <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%sconstructor(alias: %s): this(alias, null)", visibility(), Name.class, tableId);
} else // be public, as tables are no longer singletons
if (generateInstanceFields()) {
out.javadoc("Create an aliased <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%s%s(%s alias) {", visibility(), className, String.class);
out.println("this(%s.name(alias), %s);", DSL.class, tableId);
out.println("}");
out.javadoc("Create an aliased <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%s%s(%s alias) {", visibility(), className, Name.class);
out.println("this(alias, %s);", tableId);
out.println("}");
}
if (scala) {
out.javadoc("Create a <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%sdef this() = this(%s.name(\"%s\"), null)", visibility(), DSL.class, escapeString(table.getOutputName()));
} else if (kotlin) {
out.javadoc("Create a <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%sconstructor(): this(%s.name(\"%s\"), null)", visibility(), DSL.class, escapeString(table.getOutputName()));
} else {
// be public, as tables are no longer singletons
if (generateInstanceFields()) {
out.javadoc("Create a <code>%s</code> table reference", table.getQualifiedOutputName());
out.println("%s%s() {", visibility(), className);
} else {
out.javadoc(NO_FURTHER_INSTANCES_ALLOWED);
out.println("private %s() {", className);
}
out.println("this(%s.name(\"%s\"), null);", DSL.class, escapeString(table.getOutputName()));
out.println("}");
}
if (generateImplicitJoinPathsToOne() && generateGlobalKeyReferences() && !table.isTableValuedFunction()) {
out.println();
if (scala) {
out.println("%sdef this(child: %s[_ <: %s], key: %s[_ <: %s, %s]) = this(%s.createPathAlias(child, key), child, key, %s, null)", visibility(), Table.class, Record.class, ForeignKey.class, Record.class, recordType, Internal.class, tableId);
} else if (kotlin) {
out.println("%sconstructor(child: %s<out %s>, key: %s<out %s, %s>): this(%s.createPathAlias(child, key), child, key, %s, null)", visibility(), Table.class, Record.class, ForeignKey.class, Record.class, recordType, Internal.class, tableId);
} else {
out.println("%s<O extends %s> %s(%s<O> child, %s<O, %s> key) {", visibility(), Record.class, className, Table.class, ForeignKey.class, recordType);
out.println("super(child, key, %s);", tableId);
out.println("}");
}
}
if (scala) {
out.println();
out.println("%soverride def getSchema: %s = if (aliased()) null else %s", visibilityPublic(), Schema.class, schemaId);
} else if (kotlin) {
out.println("%soverride fun getSchema(): %s? = if (aliased()) null else %s", visibilityPublic(), Schema.class, schemaId);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s getSchema() {", visibilityPublic(), Schema.class);
out.println("return aliased() ? null : %s;", schemaId);
out.println("}");
}
// Add index information
if (generateIndexes()) {
List<IndexDefinition> indexes = table.getIndexes();
if (!indexes.isEmpty()) {
if (generateGlobalIndexReferences()) {
final List<String> indexFullIds = kotlin ? out.ref(getStrategy().getFullJavaIdentifiers(indexes)) : out.ref(getStrategy().getFullJavaIdentifiers(indexes), 2);
if (scala) {
out.println();
out.println("%soverride def getIndexes: %s[%s] = %s.asList[ %s ]([[%s]])", visibilityPublic(), List.class, Index.class, Arrays.class, Index.class, indexFullIds);
} else if (kotlin) {
out.println("%soverride fun getIndexes(): %s<%s> = listOf([[%s]])", visibilityPublic(), out.ref(KLIST), Index.class, indexFullIds);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s> getIndexes() {", visibilityPublic(), List.class, Index.class);
out.println("return %s.asList([[%s]]);", Arrays.class, indexFullIds);
out.println("}");
}
} else {
if (scala) {
out.println();
out.println("%soverride def getIndexes: %s[%s] = %s.asList[%s](", visibilityPublic(), List.class, Index.class, Arrays.class, Index.class);
forEach(indexes, "", ", ", (index, separator) -> {
printCreateIndex(out, index);
out.println("%s", separator);
});
out.println(")");
} else if (kotlin) {
out.println("%soverride fun getIndexes(): %s<%s> = listOf(", visibilityPublic(), out.ref(KLIST), Index.class);
forEach(indexes, "", ", ", (index, separator) -> {
printCreateIndex(out, index);
out.println("%s", separator);
});
out.println(")");
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s> getIndexes() {", visibilityPublic(), List.class, Index.class);
out.println("return %s.asList(", Arrays.class);
forEach(indexes, "", ", ", (index, separator) -> {
printCreateIndex(out, index);
out.println("%s", separator);
});
out.println(");");
out.println("}");
}
}
}
}
// Add primary / unique / foreign key information
if (generateRelations()) {
IdentityDefinition identity = table.getIdentity();
// The identity column
if (identity != null) {
final String identityTypeFull = getJavaType(identity.getColumn().getType(resolver(out)), out);
final String identityType = out.ref(identityTypeFull);
if (scala) {
out.println();
printDeprecationIfUnknownType(out, identityTypeFull);
out.println("%soverride def getIdentity: %s[%s, %s] = super.getIdentity.asInstanceOf[ %s[%s, %s] ]", visibilityPublic(), Identity.class, recordType, identityType, Identity.class, recordType, identityType);
} else if (kotlin) {
printDeprecationIfUnknownType(out, identityTypeFull);
out.println("%soverride fun getIdentity(): %s<%s, %s?> = super.getIdentity() as %s<%s, %s?>", visibilityPublic(), Identity.class, recordType, identityType, Identity.class, recordType, identityType);
} else {
if (printDeprecationIfUnknownType(out, identityTypeFull))
out.override();
else
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s, %s> getIdentity() {", visibilityPublic(), Identity.class, recordType, identityType);
out.println("return (%s<%s, %s>) super.getIdentity();", Identity.class, recordType, identityType);
out.println("}");
}
}
// The primary / main unique key
if (primaryKey != null) {
final String keyFullId = generateGlobalKeyReferences() ? kotlin ? out.ref(getStrategy().getFullJavaIdentifier(primaryKey)) : out.ref(getStrategy().getFullJavaIdentifier(primaryKey), 2) : null;
if (scala) {
out.println();
out.print("%soverride def getPrimaryKey: %s[%s] = ", visibilityPublic(), UniqueKey.class, recordType);
if (keyFullId != null)
out.print("%s", keyFullId);
else
printCreateUniqueKey(out, primaryKey);
out.println();
} else if (kotlin) {
out.print("%soverride fun getPrimaryKey(): %s<%s> = ", visibilityPublic(), UniqueKey.class, recordType);
if (keyFullId != null)
out.print("%s", keyFullId);
else
printCreateUniqueKey(out, primaryKey);
out.println();
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s> getPrimaryKey() {", visibilityPublic(), UniqueKey.class, recordType);
out.print("return ");
if (keyFullId != null)
out.print("%s", keyFullId);
else
printCreateUniqueKey(out, primaryKey);
out.println(";");
out.println("}");
}
}
// The remaining unique keys
List<UniqueKeyDefinition> uniqueKeys = table.getUniqueKeys();
if (uniqueKeys.size() > 0) {
if (generateGlobalKeyReferences()) {
final List<String> keyFullIds = kotlin ? out.ref(getStrategy().getFullJavaIdentifiers(uniqueKeys)) : out.ref(getStrategy().getFullJavaIdentifiers(uniqueKeys), 2);
if (scala) {
out.println();
out.println("%soverride def getUniqueKeys: %s[ %s[%s] ] = %s.asList[ %s[%s] ]([[%s]])", visibilityPublic(), List.class, UniqueKey.class, recordType, Arrays.class, UniqueKey.class, recordType, keyFullIds);
} else if (kotlin) {
out.println("%soverride fun getUniqueKeys(): %s<%s<%s>> = listOf([[%s]])", visibilityPublic(), out.ref(KLIST), UniqueKey.class, recordType, keyFullIds);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s<%s>> getUniqueKeys() {", visibilityPublic(), List.class, UniqueKey.class, recordType);
out.println("return %s.asList([[%s]]);", Arrays.class, keyFullIds);
out.println("}");
}
} else {
if (scala) {
out.println();
out.println("%soverride def getUniqueKeys: %s[ %s[%s] ] = %s.asList[ %s[%s] ](", visibilityPublic(), List.class, UniqueKey.class, recordType, Arrays.class, UniqueKey.class, recordType);
forEach(uniqueKeys, "", ", ", (uniqueKey, separator) -> {
printCreateUniqueKey(out, uniqueKey);
out.println("%s", separator);
});
out.println(")");
} else if (kotlin) {
out.println("%soverride fun getUniqueKeys(): %s<%s<%s>> = listOf(", visibilityPublic(), out.ref(KLIST), UniqueKey.class, recordType);
forEach(uniqueKeys, "", ", ", (uniqueKey, separator) -> {
printCreateUniqueKey(out, uniqueKey);
out.println("%s", separator);
});
out.println(")");
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s<%s>> getUniqueKeys() {", visibilityPublic(), List.class, UniqueKey.class, recordType);
out.println("return %s.asList(", Arrays.class);
forEach(uniqueKeys, "", ", ", (uniqueKey, separator) -> {
printCreateUniqueKey(out, uniqueKey);
out.println("%s", separator);
});
out.println(");");
out.println("}");
}
}
}
// Foreign keys
List<ForeignKeyDefinition> foreignKeys = table.getForeignKeys();
// [#7554] [#8028] Not yet supported with global key references turned off
if (foreignKeys.size() > 0 && generateGlobalKeyReferences()) {
final List<String> keyFullIds = kotlin ? out.ref(getStrategy().getFullJavaIdentifiers(foreignKeys)) : out.ref(getStrategy().getFullJavaIdentifiers(foreignKeys), 2);
if (scala) {
out.println();
out.println("%soverride def getReferences: %s[ %s[%s, _] ] = %s.asList[ %s[%s, _] ]([[%s]])", visibilityPublic(), List.class, ForeignKey.class, recordType, Arrays.class, ForeignKey.class, recordType, keyFullIds);
} else if (kotlin) {
out.println("%soverride fun getReferences(): %s<%s<%s, *>> = listOf([[%s]])", visibilityPublic(), out.ref(KLIST), ForeignKey.class, recordType, keyFullIds);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s<%s, ?>> getReferences() {", visibilityPublic(), List.class, ForeignKey.class, recordType);
out.println("return %s.asList([[%s]]);", Arrays.class, keyFullIds);
out.println("}");
}
// Outbound (to-one) implicit join paths
if (generateImplicitJoinPathsToOne()) {
if (scala) {
} else {
out.println();
// [#8762] Cache these calls for much improved runtime performance!
for (ForeignKeyDefinition foreignKey : foreignKeys) {
final String referencedTableClassName = out.ref(getStrategy().getFullJavaClassName(foreignKey.getReferencedTable()));
final String keyMethodName = out.ref(getStrategy().getJavaMethodName(foreignKey));
// [#13008] Prevent conflicts with the below leading underscore
final String unquotedKeyMethodName = keyMethodName.replace("`", "");
if (kotlin)
out.println("private lateinit var _%s: %s", unquotedKeyMethodName, referencedTableClassName);
else
out.println("private transient %s _%s;", referencedTableClassName, keyMethodName);
}
}
Map<TableDefinition, Long> pathCounts = foreignKeys.stream().collect(groupingBy(ForeignKeyDefinition::getReferencedTable, counting()));
for (ForeignKeyDefinition foreignKey : foreignKeys) {
final String keyFullId = kotlin ? out.ref(getStrategy().getFullJavaIdentifier(foreignKey)) : out.ref(getStrategy().getFullJavaIdentifier(foreignKey), 2);
final String referencedTableClassName = out.ref(getStrategy().getFullJavaClassName(foreignKey.getReferencedTable()));
final String keyMethodName = out.ref(getStrategy().getJavaMethodName(foreignKey));
final String unquotedKeyMethodName = keyMethodName.replace("`", "");
out.javadoc("Get the implicit join path to the <code>" + foreignKey.getReferencedTable().getQualifiedName() + "</code> table" + (pathCounts.get(foreignKey.getReferencedTable()) > 1 ? ", via the <code>" + foreignKey.getInputName() + "</code> key" : "") + ".");
if (scala) {
out.println("%slazy val %s: %s = { new %s(this, %s) }", visibility(), scalaWhitespaceSuffix(keyMethodName), referencedTableClassName, referencedTableClassName, keyFullId);
} else if (kotlin) {
out.println("%sfun %s(): %s {", visibility(), keyMethodName, referencedTableClassName);
out.println("if (!this::_%s.isInitialized)", unquotedKeyMethodName);
out.println("_%s = %s(this, %s)", unquotedKeyMethodName, referencedTableClassName, keyFullId);
out.println();
out.println("return _%s;", unquotedKeyMethodName);
out.println("}");
} else {
out.println("%s%s %s() {", visibility(), referencedTableClassName, keyMethodName);
out.println("if (_%s == null)", keyMethodName);
out.println("_%s = new %s(this, %s);", keyMethodName, referencedTableClassName, keyFullId);
out.println();
out.println("return _%s;", keyMethodName);
out.println("}");
}
}
}
}
}
List<CheckConstraintDefinition> cc = table.getCheckConstraints();
if (!cc.isEmpty()) {
if (scala) {
out.println("%soverride def getChecks: %s[ %s[%s] ] = %s.asList[ %s[%s] ](", visibilityPublic(), List.class, Check.class, recordType, Arrays.class, Check.class, recordType);
} else if (kotlin) {
out.println("%soverride fun getChecks(): %s<%s<%s>> = listOf(", visibilityPublic(), out.ref(KLIST), Check.class, recordType);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s<%s>> getChecks() {", visibilityPublic(), List.class, Check.class, recordType);
out.println("return %s.asList(", Arrays.class);
}
forEach(cc, (c, separator) -> {
out.println("%s.createCheck(this, %s.name(\"%s\"), \"%s\", %s)%s", Internal.class, DSL.class, escapeString(c.getName()), escapeString(c.getCheckClause()), c.enforced(), separator);
});
if (scala || kotlin) {
out.println(")");
} else {
out.println(");");
out.println("}");
}
}
// can still be generated
versionLoop: for (String pattern : database.getRecordVersionFields()) {
Pattern p = Pattern.compile(pattern, Pattern.COMMENTS);
for (ColumnDefinition column : table.getColumns()) {
if ((p.matcher(column.getName()).matches() || p.matcher(column.getQualifiedName()).matches())) {
final String columnTypeFull = getJavaType(column.getType(resolver(out)), out);
final String columnType = out.ref(columnTypeFull);
final String columnId = getStrategy().getJavaIdentifier(column);
if (scala) {
printDeprecationIfUnknownType(out, columnTypeFull);
out.println("%soverride def getRecordVersion: %s[%s, %s] = %s", visibilityPublic(), TableField.class, recordType, columnType, columnId);
} else if (kotlin) {
printDeprecationIfUnknownType(out, columnTypeFull);
out.println("%soverride fun getRecordVersion(): %s<%s, %s?> = %s", visibilityPublic(), TableField.class, recordType, columnType, columnId);
} else {
if (printDeprecationIfUnknownType(out, columnTypeFull))
out.override();
else
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s, %s> getRecordVersion() {", visibilityPublic(), TableField.class, recordType, columnType);
out.println("return %s;", columnId);
out.println("}");
}
// Avoid generating this method twice
break versionLoop;
}
}
}
timestampLoop: for (String pattern : database.getRecordTimestampFields()) {
Pattern p = Pattern.compile(pattern, Pattern.COMMENTS);
for (ColumnDefinition column : table.getColumns()) {
if ((p.matcher(column.getName()).matches() || p.matcher(column.getQualifiedName()).matches())) {
final String columnTypeFull = getJavaType(column.getType(resolver(out)), out);
final String columnType = out.ref(columnTypeFull);
final String columnId = getStrategy().getJavaIdentifier(column);
if (scala) {
printDeprecationIfUnknownType(out, columnTypeFull);
out.println("%soverride def getRecordTimestamp: %s[%s, %s] = %s", visibilityPublic(), TableField.class, recordType, columnType, columnId);
} else if (kotlin) {
printDeprecationIfUnknownType(out, columnTypeFull);
out.println("%soverride fun getRecordTimestamp(): %s<%s, %s?> = %s", visibilityPublic(), TableField.class, recordType, columnType, columnId);
} else {
if (printDeprecationIfUnknownType(out, columnTypeFull))
out.override();
else
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s, %s> getRecordTimestamp() {", visibilityPublic(), TableField.class, recordType, columnType);
out.println("return %s;", columnId);
out.println("}");
}
// Avoid generating this method twice
break timestampLoop;
}
}
}
if (scala) {
out.print("%soverride def as(alias: %s): %s = ", visibilityPublic(), String.class, className);
if (table.isTableValuedFunction())
out.println("new %s(%s.name(alias), null, null, this, parameters)", className, DSL.class);
else
out.println("new %s(%s.name(alias), this)", className, DSL.class);
out.print("%soverride def as(alias: %s): %s = ", visibilityPublic(), Name.class, className);
if (table.isTableValuedFunction())
out.println("new %s(alias, null, null, this, parameters)", className);
else
out.println("new %s(alias, this)", className);
} else if (kotlin) {
out.print("%soverride fun `as`(alias: %s): %s = ", visibilityPublic(), String.class, className);
if (table.isTableValuedFunction())
out.println("%s(%s.name(alias), this, parameters)", className, DSL.class);
else
out.println("%s(%s.name(alias), this)", className, DSL.class);
out.print("%soverride fun `as`(alias: %s): %s = ", visibilityPublic(), Name.class, className);
if (table.isTableValuedFunction())
out.println("%s(alias, this, parameters)", className);
else
out.println("%s(alias, this)", className);
} else // type-safe table alias
if (generateInstanceFields()) {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s as(%s alias) {", visibilityPublic(), className, String.class);
if (table.isTableValuedFunction())
out.println("return new %s(%s.name(alias), this, parameters);", className, DSL.class);
else
out.println("return new %s(%s.name(alias), this);", className, DSL.class);
out.println("}");
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s as(%s alias) {", visibilityPublic(), className, Name.class);
if (table.isTableValuedFunction())
out.println("return new %s(alias, this, parameters);", className);
else
out.println("return new %s(alias, this);", className);
out.println("}");
}
if (scala) {
out.javadoc("Rename this table");
out.print("%soverride def rename(name: %s): %s = ", visibilityPublic(), String.class, className);
if (table.isTableValuedFunction())
out.println("new %s(%s.name(name), null, null, null, parameters)", className, DSL.class);
else
out.println("new %s(%s.name(name), null)", className, DSL.class);
out.javadoc("Rename this table");
out.print("%soverride def rename(name: %s): %s = ", visibilityPublic(), Name.class, className);
if (table.isTableValuedFunction())
out.println("new %s(name, null, null, null, parameters)", className);
else
out.println("new %s(name, null)", className);
} else if (kotlin) {
out.javadoc("Rename this table");
out.print("%soverride fun rename(name: %s): %s = ", visibilityPublic(), String.class, className);
if (table.isTableValuedFunction())
out.println("%s(%s.name(name), null, parameters)", className, DSL.class);
else
out.println("%s(%s.name(name), null)", className, DSL.class);
out.javadoc("Rename this table");
out.print("%soverride fun rename(name: %s): %s = ", visibilityPublic(), Name.class, className);
if (table.isTableValuedFunction())
out.println("%s(name, null, parameters)", className);
else
out.println("%s(name, null)", className);
} else // [#2921] With instance fields, tables can be renamed.
if (generateInstanceFields()) {
out.javadoc("Rename this table");
out.override();
printNonnullAnnotation(out);
out.println("%s%s rename(%s name) {", visibilityPublic(), className, String.class);
if (table.isTableValuedFunction())
out.println("return new %s(%s.name(name), null, parameters);", className, DSL.class);
else
out.println("return new %s(%s.name(name), null);", className, DSL.class);
out.println("}");
out.javadoc("Rename this table");
out.override();
printNonnullAnnotation(out);
out.println("%s%s rename(%s name) {", visibilityPublic(), className, Name.class);
if (table.isTableValuedFunction())
out.println("return new %s(name, null, parameters);", className);
else
out.println("return new %s(name, null);", className);
out.println("}");
}
// [#7809] fieldsRow()
// [#10481] Use the types from replaced embeddables if applicable
List<Definition> replacingEmbeddablesAndUnreplacedColumns = replacingEmbeddablesAndUnreplacedColumns(table);
int degree = replacingEmbeddablesAndUnreplacedColumns.size();
String rowType = refRowType(out, replacingEmbeddablesAndUnreplacedColumns);
String rowTypeContravariantJava = refRowType(out, replacingEmbeddablesAndUnreplacedColumns, s -> "? super " + s);
if (generateRecordsImplementingRecordN() && degree > 0 && degree <= Constants.MAX_ROW_DEGREE) {
final String rowNType = out.ref(Row.class.getName() + degree);
out.header("Row%s type methods", degree);
if (scala) {
out.println("%soverride def fieldsRow: %s[%s] = super.fieldsRow.asInstanceOf[ %s[%s] ]", visibilityPublic(), rowNType, rowType, rowNType, rowType);
} else if (kotlin) {
out.println("%soverride fun fieldsRow(): %s<%s> = super.fieldsRow() as %s<%s>", visibilityPublic(), rowNType, rowType, rowNType, rowType);
} else {
out.overrideInherit();
printNonnullAnnotation(out);
out.println("%s%s<%s> fieldsRow() {", visibilityPublic(), rowNType, rowType);
out.println("return (%s) super.fieldsRow();", rowNType);
out.println("}");
}
}
// [#1070] Table-valued functions should generate an additional set of call() methods
if (table.isTableValuedFunction()) {
for (boolean parametersAsField : new boolean[] { false, true }) {
// Don't overload no-args call() methods
if (parametersAsField && parameters.size() == 0)
break;
out.javadoc("Call this table-valued function");
if (scala) {
out.print("%sdef call(", visibility()).printlnIf(!parameters.isEmpty());
printParameterDeclarations(out, parameters, parametersAsField, " ");
out.print("): %s = ", className);
out.print("Option(new %s(%s.name(\"%s\"), null, null, null, %s(", className, DSL.class, escapeString(table.getOutputName()), out.ref("scala.Array")).printlnIf(!parameters.isEmpty());
forEach(parameters, (parameter, separator) -> {
final String paramArgName = getStrategy().getJavaMemberName(parameter);
final String paramTypeRef = getJavaTypeReference(parameter.getDatabase(), parameter.getType(resolver(out)), out);
final List<String> converter = out.ref(list(parameter.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(parameter.getType(resolver(out)).getBinding()));
if (parametersAsField)
out.println("%s%s", paramArgName, separator);
else
out.println("%s.value(%s, %s" + converterTemplateForTableValuedFunction(converter) + converterTemplateForTableValuedFunction(binding) + ")%s", DSL.class, paramArgName, paramTypeRef, converter, binding, separator);
});
out.println("))).map(r => if (aliased()) r.as(getUnqualifiedName) else r).get");
} else if (kotlin) {
out.print("%sfun call(", visibility()).printlnIf(!parameters.isEmpty());
printParameterDeclarations(out, parameters, parametersAsField, " ");
out.print("): %s = %s(%s.name(\"%s\"), null, arrayOf(", className, className, DSL.class, escapeString(table.getOutputName()), Field.class).printlnIf(!parameters.isEmpty());
forEach(parameters, (parameter, separator) -> {
final String paramArgName = getStrategy().getJavaMemberName(parameter);
final String paramTypeRef = getJavaTypeReference(parameter.getDatabase(), parameter.getType(resolver(out)), out);
final List<String> converter = out.ref(list(parameter.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(parameter.getType(resolver(out)).getBinding()));
if (parametersAsField)
out.println("%s%s", paramArgName, separator);
else
out.println("%s.value(%s, %s" + converterTemplateForTableValuedFunction(converter) + converterTemplateForTableValuedFunction(binding) + ")%s", DSL.class, paramArgName, paramTypeRef, converter, binding, separator);
});
out.println(")).let { if (aliased()) it.`as`(unqualifiedName) else it }");
} else {
out.print("%s%s call(", visibility(), className).printlnIf(!parameters.isEmpty());
printParameterDeclarations(out, parameters, parametersAsField, " ");
out.println(") {");
out.print("%s result = new %s(%s.name(\"%s\"), null, new %s[] {", className, className, DSL.class, escapeString(table.getOutputName()), Field.class).printlnIf(!parameters.isEmpty());
forEach(parameters, (parameter, separator) -> {
final String paramArgName = getStrategy().getJavaMemberName(parameter);
final String paramTypeRef = getJavaTypeReference(parameter.getDatabase(), parameter.getType(resolver(out)), out);
final List<String> converter = out.ref(list(parameter.getType(resolver(out)).getConverter()));
final List<String> binding = out.ref(list(parameter.getType(resolver(out)).getBinding()));
if (parametersAsField)
out.println("%s%s", paramArgName, separator);
else
out.println("%s.val(%s, %s" + converterTemplateForTableValuedFunction(converter) + converterTemplateForTableValuedFunction(binding) + ")%s", DSL.class, paramArgName, paramTypeRef, converter, binding, separator);
});
out.println("});");
out.println();
out.println("return aliased() ? result.as(getUnqualifiedName()) : result;");
out.println("}");
}
}
}
if (generateRecordsImplementingRecordN() && degree > 0 && degree <= Constants.MAX_ROW_DEGREE) {
out.javadoc("Convenience mapping calling {@link #convertFrom(%s)}.", Function.class);
if (scala) {
out.println("%sdef mapping[U](from: (" + rowType + ") => U): %s[U] = convertFrom(r => from.apply(" + rangeClosed(1, degree).mapToObj(i -> "r.value" + i + "()").collect(joining(", ")) + "))", visibility(), SelectField.class);
} else if (kotlin) {
out.println("%sfun <U> mapping(from: (" + rowType + ") -> U): %s<U> = convertFrom(%s.mapping(from))", visibility(), SelectField.class, Records.class);
} else {
out.println("%s<U> %s<U> mapping(%s<" + rowTypeContravariantJava + ", ? extends U> from) {", visibility(), SelectField.class, out.ref("org.jooq.Function" + degree));
out.println("return convertFrom(%s.mapping(from));", Records.class);
out.println("}");
}
out.javadoc("Convenience mapping calling {@link #convertFrom(%s, %s)}.", Class.class, Function.class);
if (scala) {
out.println("%sdef mapping[U](toType: %s[U], from: (" + rowType + ") => U): %s[U] = convertFrom(toType,r => from.apply(" + rangeClosed(1, degree).mapToObj(i -> "r.value" + i + "()").collect(joining(", ")) + "))", visibility(), Class.class, SelectField.class);
} else if (kotlin) {
out.println("%sfun <U> mapping(toType: %s<U>, from: (" + rowType + ") -> U): %s<U> = convertFrom(toType, %s.mapping(from))", visibility(), Class.class, SelectField.class, Records.class);
} else {
out.println("%s<U> %s<U> mapping(%s<U> toType, %s<" + rowTypeContravariantJava + ", ? extends U> from) {", visibility(), SelectField.class, Class.class, out.ref("org.jooq.Function" + degree));
out.println("return convertFrom(toType, %s.mapping(from));", Records.class);
out.println("}");
}
}
generateTableClassFooter(table, out);
out.println("}");
closeJavaWriter(out);
}
Also used :
RandomAccessFile(java.io.RandomAccessFile)
Arrays(java.util.Arrays)
TableRecordImpl(org.jooq.impl.TableRecordImpl)
DSL(org.jooq.impl.DSL)
SCALA(org.jooq.codegen.Language.SCALA)
StringUtils.isBlank(org.jooq.tools.StringUtils.isBlank)
Collectors.counting(java.util.stream.Collectors.counting)
Internal(org.jooq.impl.Internal)
POSTGRES(org.jooq.SQLDialect.POSTGRES)
DatatypeConverter(jakarta.xml.bind.DatatypeConverter)
AttributeDefinition(org.jooq.meta.AttributeDefinition)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
RoutineDefinition(org.jooq.meta.RoutineDefinition)
Matcher(java.util.regex.Matcher)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
SelectField(org.jooq.SelectField)
IdentityDefinition(org.jooq.meta.IdentityDefinition)
Arrays.asList(java.util.Arrays.asList)
Index(org.jooq.Index)
Map(java.util.Map)
BigInteger(java.math.BigInteger)
SQLDialect(org.jooq.SQLDialect)
Domain(org.jooq.Domain)
Database(org.jooq.meta.Database)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
CheckConstraintDefinition(org.jooq.meta.CheckConstraintDefinition)
Set(java.util.Set)
Constants(org.jooq.Constants)
AggregateFunction(org.jooq.AggregateFunction)
Result(org.jooq.Result)
Collectors.joining(java.util.stream.Collectors.joining)
GenerationOption(org.jooq.impl.QOM.GenerationOption)
Stream(java.util.stream.Stream)
TableField(org.jooq.TableField)
SQLDataType(org.jooq.impl.SQLDataType)
GeneratedAnnotationType(org.jooq.meta.jaxb.GeneratedAnnotationType)
SequenceDefinition(org.jooq.meta.SequenceDefinition)
Collectors.groupingBy(java.util.stream.Collectors.groupingBy)
Parameter(org.jooq.Parameter)
TableDefinition(org.jooq.meta.TableDefinition)
CatalogImpl(org.jooq.impl.CatalogImpl)
IndexDefinition(org.jooq.meta.IndexDefinition)
Sequence(org.jooq.Sequence)
ArrayList(java.util.ArrayList)
LinkedHashMap(java.util.LinkedHashMap)
Calendar(java.util.Calendar)
Records(org.jooq.Records)
Schema(org.jooq.Schema)
BiConsumer(java.util.function.BiConsumer)
RowId(org.jooq.RowId)
LinkedHashSet(java.util.LinkedHashSet)
MYSQL(org.jooq.SQLDialect.MYSQL)
Record(org.jooq.Record)
Mode(org.jooq.codegen.GeneratorStrategy.Mode)
TypeVariable(java.lang.reflect.TypeVariable)
IOException(java.io.IOException)
StringUtils(org.jooq.tools.StringUtils)
File(java.io.File)
YUGABYTEDB(org.jooq.SQLDialect.YUGABYTEDB)
VIRTUAL(org.jooq.impl.QOM.GenerationOption.VIRTUAL)
SchemaDefinition(org.jooq.meta.SchemaDefinition)
LazySupplier(org.jooq.impl.LazySupplier)
TableImpl(org.jooq.impl.TableImpl)
CatalogDefinition(org.jooq.meta.CatalogDefinition)
UniqueKey(org.jooq.UniqueKey)
Row(org.jooq.Row)
IntStream.rangeClosed(java.util.stream.IntStream.rangeClosed)
Time(java.sql.Time)
Table(org.jooq.Table)
DataTypeDefinition(org.jooq.meta.DataTypeDefinition)
DAOImpl(org.jooq.impl.DAOImpl)
PackageDefinition(org.jooq.meta.PackageDefinition)
DefaultDataType(org.jooq.impl.DefaultDataType)
UDT(org.jooq.UDT)
PostgresDatabase(org.jooq.meta.postgres.PostgresDatabase)
SortOrder(org.jooq.SortOrder)
DataType(org.jooq.DataType)
DSL.name(org.jooq.impl.DSL.name)
IdentityHashMap(java.util.IdentityHashMap)
JooqLogger(org.jooq.tools.JooqLogger)
SQLDialectNotSupportedException(org.jooq.exception.SQLDialectNotSupportedException)
Collections.nCopies(java.util.Collections.nCopies)
TimeZone(java.util.TimeZone)
Name(org.jooq.Name)
Timestamp(java.sql.Timestamp)
Collection(java.util.Collection)
Check(org.jooq.Check)
Field(org.jooq.Field)
CloseResult(org.jooq.codegen.GeneratorWriter.CloseResult)
PostgresRoutineDefinition(org.jooq.meta.postgres.PostgresRoutineDefinition)
Collectors(java.util.stream.Collectors)
UpdatableRecordImpl(org.jooq.impl.UpdatableRecordImpl)
TypedElementDefinition(org.jooq.meta.TypedElementDefinition)
JAVA(org.jooq.codegen.Language.JAVA)
DomainDefinition(org.jooq.meta.DomainDefinition)
Reflect(org.jooq.tools.reflect.Reflect)
List(java.util.List)
Modifier(java.lang.reflect.Modifier)
Function.identity(java.util.function.Function.identity)
Optional(java.util.Optional)
Pattern(java.util.regex.Pattern)
IntStream(java.util.stream.IntStream)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
KOTLIN(org.jooq.codegen.Language.KOTLIN)
UDTField(org.jooq.UDTField)
DefaultDataTypeDefinition(org.jooq.meta.DefaultDataTypeDefinition)
HashMap(java.util.HashMap)
STORED(org.jooq.impl.QOM.GenerationOption.STORED)
EnumType(org.jooq.EnumType)
ForeignKey(org.jooq.ForeignKey)
Function(java.util.function.Function)
PackageImpl(org.jooq.impl.PackageImpl)
Definition(org.jooq.meta.Definition)
HashSet(java.util.HashSet)
Identity(org.jooq.Identity)
TableLike(org.jooq.TableLike)
EmbeddableRecordImpl(org.jooq.impl.EmbeddableRecordImpl)
EmbeddableDefinition(org.jooq.meta.EmbeddableDefinition)
AbstractRoutine(org.jooq.impl.AbstractRoutine)
Function2(org.jooq.Function2)
ArrayDefinition(org.jooq.meta.ArrayDefinition)
UDTDefinition(org.jooq.meta.UDTDefinition)
AbstractTypedElementDefinition.getDataType(org.jooq.meta.AbstractTypedElementDefinition.getDataType)
JavaTypeResolver(org.jooq.meta.JavaTypeResolver)
ParameterDefinition(org.jooq.meta.ParameterDefinition)
UDTRecordImpl(org.jooq.impl.UDTRecordImpl)
ConstraintDefinition(org.jooq.meta.ConstraintDefinition)
LazySchema(org.jooq.impl.LazySchema)
DESC(org.jooq.SortOrder.DESC)
SchemaImpl(org.jooq.impl.SchemaImpl)
Date(java.sql.Date)
ReflectException(org.jooq.tools.reflect.ReflectException)
AbstractTypedElementDefinition(org.jooq.meta.AbstractTypedElementDefinition)
Configuration(org.jooq.Configuration)
Collectors.toList(java.util.stream.Collectors.toList)
Catalog(org.jooq.Catalog)
UDTImpl(org.jooq.impl.UDTImpl)
StopWatch(org.jooq.tools.StopWatch)
OrderField(org.jooq.OrderField)
TableOptions(org.jooq.TableOptions)
EnumDefinition(org.jooq.meta.EnumDefinition)
ForeignKeyDefinition(org.jooq.meta.ForeignKeyDefinition)
Collections(java.util.Collections)
CheckConstraintDefinition(org.jooq.meta.CheckConstraintDefinition)
Internal(org.jooq.impl.Internal)
Schema(org.jooq.Schema)
LazySchema(org.jooq.impl.LazySchema)
TableImpl(org.jooq.impl.TableImpl)
ArrayList(java.util.ArrayList)
Check(org.jooq.Check)
Index(org.jooq.Index)
Name(org.jooq.Name)
ForeignKeyDefinition(org.jooq.meta.ForeignKeyDefinition)
SelectField(org.jooq.SelectField)
TableField(org.jooq.TableField)
Field(org.jooq.Field)
UDTField(org.jooq.UDTField)
OrderField(org.jooq.OrderField)
TableOptions(org.jooq.TableOptions)
IndexDefinition(org.jooq.meta.IndexDefinition)
TableDefinition(org.jooq.meta.TableDefinition)
Record(org.jooq.Record)
Arrays.asList(java.util.Arrays.asList)
ArrayList(java.util.ArrayList)
List(java.util.List)
Collectors.toList(java.util.stream.Collectors.toList)
DSL(org.jooq.impl.DSL)
Identity(org.jooq.Identity)
Records(org.jooq.Records)
Pattern(java.util.regex.Pattern)
SchemaDefinition(org.jooq.meta.SchemaDefinition)
Table(org.jooq.Table)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
SelectField(org.jooq.SelectField)
IdentityDefinition(org.jooq.meta.IdentityDefinition)
EmbeddableDefinition(org.jooq.meta.EmbeddableDefinition)
AttributeDefinition(org.jooq.meta.AttributeDefinition)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
RoutineDefinition(org.jooq.meta.RoutineDefinition)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
IdentityDefinition(org.jooq.meta.IdentityDefinition)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
CheckConstraintDefinition(org.jooq.meta.CheckConstraintDefinition)
SequenceDefinition(org.jooq.meta.SequenceDefinition)
TableDefinition(org.jooq.meta.TableDefinition)
IndexDefinition(org.jooq.meta.IndexDefinition)
SchemaDefinition(org.jooq.meta.SchemaDefinition)
CatalogDefinition(org.jooq.meta.CatalogDefinition)
DataTypeDefinition(org.jooq.meta.DataTypeDefinition)
PackageDefinition(org.jooq.meta.PackageDefinition)
PostgresRoutineDefinition(org.jooq.meta.postgres.PostgresRoutineDefinition)
TypedElementDefinition(org.jooq.meta.TypedElementDefinition)
DomainDefinition(org.jooq.meta.DomainDefinition)
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
DefaultDataTypeDefinition(org.jooq.meta.DefaultDataTypeDefinition)
Definition(org.jooq.meta.Definition)
EmbeddableDefinition(org.jooq.meta.EmbeddableDefinition)
ArrayDefinition(org.jooq.meta.ArrayDefinition)
UDTDefinition(org.jooq.meta.UDTDefinition)
ParameterDefinition(org.jooq.meta.ParameterDefinition)
ConstraintDefinition(org.jooq.meta.ConstraintDefinition)
AbstractTypedElementDefinition(org.jooq.meta.AbstractTypedElementDefinition)
EnumDefinition(org.jooq.meta.EnumDefinition)
ForeignKeyDefinition(org.jooq.meta.ForeignKeyDefinition)
ForeignKey(org.jooq.ForeignKey)
TableField(org.jooq.TableField)
ColumnDefinition(org.jooq.meta.ColumnDefinition)
IndexColumnDefinition(org.jooq.meta.IndexColumnDefinition)
EmbeddableColumnDefinition(org.jooq.meta.EmbeddableColumnDefinition)
UniqueKey(org.jooq.UniqueKey)
Arrays(java.util.Arrays)
ParameterDefinition(org.jooq.meta.ParameterDefinition)
Example 4 with UniqueKeyDefinition
use of org.jooq.meta.UniqueKeyDefinition in project jOOQ by jOOQ.
the class JavaGenerator method generateRelations.
protected void generateRelations(SchemaDefinition schema) {
log.info("Generating Keys");
boolean empty = true;
JavaWriter out = newJavaWriter(getStrategy().getGlobalReferencesFile(schema, ConstraintDefinition.class));
out.refConflicts(getStrategy().getJavaIdentifiers(database.getKeys(schema)));
out.refConflicts(getStrategy().getJavaIdentifiers(database.getForeignKeys(schema)));
printGlobalReferencesPackage(out, schema, ConstraintDefinition.class);
if (!kotlin) {
printClassJavadoc(out, "A class modelling foreign key relationships and constraints of tables in " + schemaNameOrDefault(schema) + ".");
printClassAnnotations(out, schema, Mode.DEFAULT);
}
final String referencesClassName = getStrategy().getGlobalReferencesJavaClassName(schema, ConstraintDefinition.class);
if (scala)
out.println("%sobject %s {", visibility(), referencesClassName);
else if (kotlin) {
} else
out.println("%sclass %s {", visibility(), referencesClassName);
// [#1459] [#10554] [#10653] Distribute keys to nested classes only if necessary
boolean distributeUniqueKeys = database.getKeys(schema).size() > maxMembersPerInitialiser();
boolean distributeForeignKeys = database.getForeignKeys(schema).size() > maxMembersPerInitialiser();
List<UniqueKeyDefinition> allUniqueKeys = new ArrayList<>();
List<ForeignKeyDefinition> allForeignKeys = new ArrayList<>();
// Unique keys
try {
for (UniqueKeyDefinition uniqueKey : database.getKeys(schema)) {
empty = false;
final String keyType = out.ref(getStrategy().getFullJavaClassName(uniqueKey.getTable(), Mode.RECORD));
final String keyId = getStrategy().getJavaIdentifier(uniqueKey);
final int block = allUniqueKeys.size() / maxMembersPerInitialiser();
// [#10480] Print header before first key
if (allUniqueKeys.isEmpty()) {
out.header("UNIQUE and PRIMARY KEY definitions");
out.println();
}
if (distributeUniqueKeys)
if (scala)
out.println("%sval %s = UniqueKeys%s.%s", visibility(), keyId, block, keyId);
else if (kotlin)
out.println("%sval %s: %s<%s> = UniqueKeys%s.%s", visibility(), keyId, UniqueKey.class, keyType, block, keyId);
else
out.println("%sstatic final %s<%s> %s = UniqueKeys%s.%s;", visibility(), UniqueKey.class, keyType, keyId, block, keyId);
else
printUniqueKey(out, -1, uniqueKey, distributeUniqueKeys);
allUniqueKeys.add(uniqueKey);
}
} catch (Exception e) {
log.error("Error while generating unique keys for schema " + schema, e);
}
// Foreign keys
try {
for (ForeignKeyDefinition foreignKey : database.getForeignKeys(schema)) {
empty = false;
final String keyType = out.ref(getStrategy().getFullJavaClassName(foreignKey.getKeyTable(), Mode.RECORD));
final String referencedType = out.ref(getStrategy().getFullJavaClassName(foreignKey.getReferencedTable(), Mode.RECORD));
final String keyId = getStrategy().getJavaIdentifier(foreignKey);
final int block = allForeignKeys.size() / maxMembersPerInitialiser();
// [#10480] Print header before first key
if (allForeignKeys.isEmpty()) {
out.header("FOREIGN KEY definitions");
out.println();
}
if (distributeForeignKeys)
if (scala)
out.println("%sval %s = ForeignKeys%s.%s", visibility(), keyId, block, keyId);
else if (kotlin)
out.println("%sval %s: %s<%s, %s> = ForeignKeys%s.%s", visibility(), keyId, ForeignKey.class, keyType, referencedType, block, keyId);
else
out.println("%sstatic final %s<%s, %s> %s = ForeignKeys%s.%s;", visibility(), ForeignKey.class, keyType, referencedType, keyId, block, keyId);
else
printForeignKey(out, -1, foreignKey, distributeForeignKeys);
allForeignKeys.add(foreignKey);
}
} catch (Exception e) {
log.error("Error while generating foreign keys for schema " + schema, e);
}
// [#1459] [#10554] [#10653] Print nested classes for actual static field initialisations
// keeping top-level initialiser small
int uniqueKeyCounter = 0;
int foreignKeyCounter = 0;
if (distributeUniqueKeys || distributeForeignKeys) {
out.header("[#1459] [#10554] [#10653] distribute members to avoid static initialisers > 64kb");
if (distributeUniqueKeys) {
for (UniqueKeyDefinition uniqueKey : allUniqueKeys) printUniqueKey(out, uniqueKeyCounter++, uniqueKey, distributeUniqueKeys);
if (uniqueKeyCounter > 0)
out.println("}");
}
if (distributeForeignKeys) {
for (ForeignKeyDefinition foreignKey : allForeignKeys) printForeignKey(out, foreignKeyCounter++, foreignKey, distributeForeignKeys);
if (foreignKeyCounter > 0)
out.println("}");
}
}
generateRelationsClassFooter(schema, out);
if (!kotlin)
out.println("}");
if (empty) {
log.info("Skipping empty keys");
} else {
closeJavaWriter(out);
watch.splitInfo("Keys generated");
}
}
Also used :
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
ArrayList(java.util.ArrayList)
ForeignKey(org.jooq.ForeignKey)
IOException(java.io.IOException)
SQLDialectNotSupportedException(org.jooq.exception.SQLDialectNotSupportedException)
ReflectException(org.jooq.tools.reflect.ReflectException)
CheckConstraintDefinition(org.jooq.meta.CheckConstraintDefinition)
ConstraintDefinition(org.jooq.meta.ConstraintDefinition)
ForeignKeyDefinition(org.jooq.meta.ForeignKeyDefinition)
UniqueKey(org.jooq.UniqueKey)
Example 5 with UniqueKeyDefinition
use of org.jooq.meta.UniqueKeyDefinition in project jOOQ by jOOQ.
the class JavaGenerator method printColumnJPAAnnotation.
protected void printColumnJPAAnnotation(JavaWriter out, ColumnDefinition column) {
int indent = out.indent();
if (generateJPAAnnotations()) {
String prefix = kotlin ? "get:" : "";
UniqueKeyDefinition pk = column.getPrimaryKey();
if (pk != null) {
if (pk.getKeyColumns().size() == 1) {
// Since JPA 1.0
out.println("@%s%s", prefix, out.ref("jakarta.persistence.Id"));
// Since JPA 1.0
if (pk.getKeyColumns().get(0).isIdentity())
out.println("@%s%s(strategy = %s.IDENTITY)", prefix, out.ref("jakarta.persistence.GeneratedValue"), out.ref("jakarta.persistence.GenerationType"));
}
}
String nullable = "";
if (!column.getType(resolver(out)).isNullable())
nullable = ", nullable = false";
String length = "";
String precision = "";
String scale = "";
if (column.getType(resolver(out)).getLength() > 0) {
length = ", length = " + column.getType(resolver(out)).getLength();
} else if (column.getType(resolver(out)).getPrecision() > 0) {
precision = ", precision = " + column.getType(resolver(out)).getPrecision();
if (column.getType(resolver(out)).getScale() > 0) {
scale = ", scale = " + column.getType(resolver(out)).getScale();
}
}
// [#8535] The unique flag is not set on the column, but only on
// the table's @UniqueConstraint section.
// Since JPA 1.0
out.print("@%s%s(name = \"", prefix, out.ref("jakarta.persistence.Column"));
out.print(escapeString(column.getName()));
out.print("\"");
out.print(nullable);
out.print(length);
out.print(precision);
out.print(scale);
out.println(")");
}
// [#10196] The above logic triggers an indent level of -1, incorrectly
out.indent(indent);
}
Also used :
UniqueKeyDefinition(org.jooq.meta.UniqueKeyDefinition)
Aggregations
UniqueKeyDefinition (org.jooq.meta.UniqueKeyDefinition)7
CheckConstraintDefinition (org.jooq.meta.CheckConstraintDefinition)5
ColumnDefinition (org.jooq.meta.ColumnDefinition)5
ForeignKeyDefinition (org.jooq.meta.ForeignKeyDefinition)5
IndexColumnDefinition (org.jooq.meta.IndexColumnDefinition)5
IndexDefinition (org.jooq.meta.IndexDefinition)5
SchemaDefinition (org.jooq.meta.SchemaDefinition)5
IOException (java.io.IOException)4
ArrayList (java.util.ArrayList)4
SQLDialectNotSupportedException (org.jooq.exception.SQLDialectNotSupportedException)4
ConstraintDefinition (org.jooq.meta.ConstraintDefinition)4
EmbeddableColumnDefinition (org.jooq.meta.EmbeddableColumnDefinition)4
BigInteger (java.math.BigInteger)3
ForeignKey (org.jooq.ForeignKey)3
UniqueKey (org.jooq.UniqueKey)3
ReflectException (org.jooq.tools.reflect.ReflectException)3
DatatypeConverter (jakarta.xml.bind.DatatypeConverter)2
File (java.io.File)2
RandomAccessFile (java.io.RandomAccessFile)2
Modifier (java.lang.reflect.Modifier)2
