github/jooq
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[jOOQ/jOOQ#8950] Better FOR ORDINALITY emulation in PostgreSQL

Browse Source
This commit is contained in:
Lukas Eder 2020-04-18 12:04:05 +02:00
parent 2f28c8e304
commit f190e1784e
6 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jOOQ/src/main/java/org/jooq/SelectForStep.java
View File

@ -40,10 +40,12 @@ package org.jooq;
// ...
// ...
// ...
// ...
import static org.jooq.SQLDialect.H2;
import static org.jooq.SQLDialect.MARIADB;
import static org.jooq.SQLDialect.MYSQL;
// ...
// ...
import static org.jooq.SQLDialect.POSTGRES;
// ...

3
jOOQ/src/main/java/org/jooq/SelectForXMLCommonDirectivesStep.java
View File

@ -116,6 +116,9 @@ package org.jooq;

3
jOOQ/src/main/java/org/jooq/SelectForXMLPathDirectivesStep.java
View File

@ -110,6 +110,9 @@ package org.jooq;

3
jOOQ/src/main/java/org/jooq/SelectForXMLRawDirectivesStep.java
View File

@ -110,6 +110,9 @@ package org.jooq;

3
jOOQ/src/main/java/org/jooq/SelectForXMLStep.java
View File

@ -138,6 +138,9 @@ package org.jooq;

18
jOOQ/src/main/java/org/jooq/impl/JSONTable.java
View File

@ -43,7 +43,6 @@ import static org.jooq.SQLDialect.MYSQL;
import static org.jooq.conf.ParamType.INLINED;
import static org.jooq.impl.DSL.inline;
import static org.jooq.impl.DSL.keyword;
import static org.jooq.impl.DSL.rowNumber;
import static org.jooq.impl.DSL.select;
import static org.jooq.impl.DSL.val;
import static org.jooq.impl.Keywords.K_COLUMNS;
@ -209,23 +208,30 @@ implements
private final void acceptPostgres(Context<?> ctx) {
List<SelectField<?>> cols = new ArrayList<>();
for (JSONTableColumn col : columns)
if (col.forOrdinality)
cols.add(rowNumber().over().as(col.field));
else
boolean requireOrdinality = false;
for (JSONTableColumn col : columns) {
if (col.forOrdinality) {
requireOrdinality = true;
cols.add(DSL.field("o").as(col.field));
}
else {
cols.add(
DSL.field("(jsonb_path_query_first(j, {0}::jsonpath)->>0)::{1}",
col.path != null ? val(col.path) : inline("$." + col.field.getName()),
keyword(col.type.getCastTypeName(ctx.configuration()))
).as(col.field)
);
}
}
ctx.sql('(')
.formatIndentStart()
.formatNewLine()
.subquery(true)
.visit(
select(cols).from("jsonb_path_query({0}, {1}::jsonpath) as t(j)",
select(cols).from(requireOrdinality
? "jsonb_path_query({0}, {1}::jsonpath) {with} {ordinality} {as} t(j, o)"
: "jsonb_path_query({0}, {1}::jsonpath) {as} t(j)",
json.getType() == JSONB.class ? json : json.cast(JSONB),
path
)