From 99ea335783d0445d69ac381e5dc3ca774088e9d7 Mon Sep 17 00:00:00 2001 From: Lukas Eder Date: Thu, 27 Aug 2020 15:38:29 +0200 Subject: [PATCH] [jOOQ/jOOQ#10540] [jOOQ/jOOQ#10527] Don't apply this feature for embeddables yet --- .../java/org/jooq/impl/AliasedSelect.java | 120 ++++++++++++++++++ jOOQ/src/main/java/org/jooq/impl/Tools.java | 7 +- 2 files changed, 124 insertions(+), 3 deletions(-) create mode 100644 jOOQ/src/main/java/org/jooq/impl/AliasedSelect.java diff --git a/jOOQ/src/main/java/org/jooq/impl/AliasedSelect.java b/jOOQ/src/main/java/org/jooq/impl/AliasedSelect.java new file mode 100644 index 0000000000..c0b1bea7cf --- /dev/null +++ b/jOOQ/src/main/java/org/jooq/impl/AliasedSelect.java @@ -0,0 +1,120 @@ +/* + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + * Other licenses: + * ----------------------------------------------------------------------------- + * Commercial licenses for this work are available. These replace the above + * ASL 2.0 and offer limited warranties, support, maintenance, and commercial + * database integrations. + * + * For more information, please visit: http://www.jooq.org/licenses + * + * + * + * + * + * + * + * + * + * + * + * + * + * + * + * + */ +package org.jooq.impl; + +import static org.jooq.impl.Names.N_SELECT; +import static org.jooq.impl.Tools.selectQueryImpl; +import static org.jooq.impl.Tools.visitSubquery; +import static org.jooq.impl.Tools.DataKey.DATA_SELECT_ALIASES; + +import org.jooq.Clause; +import org.jooq.Context; +import org.jooq.Name; +import org.jooq.Record; +import org.jooq.Select; +import org.jooq.Table; +import org.jooq.TableOptions; + +/** + * A {@link Select} query that re-aliases its projection, or produces a derived + * table if re-aliasing is not possible (e.g. in the presence of + * ORDER BY). + * + * @author Lukas Eder + */ +final class AliasedSelect extends AbstractTable { + + private static final long serialVersionUID = 6763689261249123076L; + + private final Select query; + private final Name[] aliases; + + AliasedSelect(Select query) { + this(query, Tools.fieldNames(query.getSelect().size())); + } + + AliasedSelect(Select query, Name... aliases) { + super(TableOptions.expression(), N_SELECT); + + this.query = query; + this.aliases = aliases; + } + + final Select query() { + return query; + } + + @Override + public final Table as(Name alias) { + SelectQueryImpl q = selectQueryImpl(query); + + if (q != null && (!q.getOrderBy().isEmpty() || Tools.hasEmbeddedFields(q.getSelect()))) + return query.asTable(alias, aliases); + else + return new TableAlias<>(this, alias, true); + } + + @Override + public final Table as(Name alias, Name... fieldAliases) { + return new TableAlias<>(this, alias, fieldAliases, true); + } + + @Override + final Fields fields0() { + return new Fields<>(query.asTable(DSL.name("t"), aliases).fields()); + } + + @Override + public final Class getRecordType() { + return query.getRecordType(); + } + + @Override + public final void accept(Context ctx) { + Object previous = ctx.data(DATA_SELECT_ALIASES); + + ctx.data(DATA_SELECT_ALIASES, aliases); + visitSubquery(ctx, query); + ctx.data(DATA_SELECT_ALIASES, previous); + } + + @Override // Avoid AbstractTable implementation + public final Clause[] clauses(Context ctx) { + return null; + } +} diff --git a/jOOQ/src/main/java/org/jooq/impl/Tools.java b/jOOQ/src/main/java/org/jooq/impl/Tools.java index d880834471..8a7961ab4d 100644 --- a/jOOQ/src/main/java/org/jooq/impl/Tools.java +++ b/jOOQ/src/main/java/org/jooq/impl/Tools.java @@ -5403,10 +5403,11 @@ final class Tools { for (Field f : flattenCollection(select, false)) result.add(f); - String[] fieldNames = fieldNameStrings(result.size()); - Table t = field.query.asTable("t", fieldNames); + Name tableName = name("t"); + Name[] fieldNames = fieldNames(result.size()); + Table t = new AliasedSelect<>(field.query, fieldNames).as("t"); for (int i = 0; i < result.size(); i++) - result.set(i, DSL.field(DSL.select(DSL.field(name("t", fieldNames[i]), result.get(i).getDataType())).from(t))); + result.set(i, DSL.field(DSL.select(DSL.field(tableName.append(fieldNames[i]), result.get(i).getDataType())).from(t))); return result.toArray(EMPTY_FIELD); }