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

[jOOQ/jOOQ#11700] Add support for ResultQuery.getRecordType()

Browse Source 
This allows for using selectFrom() and get UpdatableRecords (which can't be stored yet, but at least they have getters / setters).
This commit is contained in:
Lukas Eder 2021-03-24 10:01:13 +01:00
parent 74569c5989
commit 9db3cf002f
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
jOOQ/src/main/java/org/jooq/impl/Flow.java
View File

@ -37,9 +37,12 @@
*/
package org.jooq.impl;
import static org.jooq.impl.Tools.recordFactory;
import java.sql.SQLException;
import java.util.ArrayDeque;
import java.util.concurrent.atomic.AtomicLong;
import java.util.function.Supplier;
import org.jooq.Cursor;
import org.jooq.Field;
@ -112,26 +115,38 @@ final class Flow {
s.request(1);
}
@SuppressWarnings("unchecked")
@Override
public void onNext(io.r2dbc.spi.Result r) {
r.map((row, m) -> {
// TODO: Cache this getFields() call
Field<?>[] fields = null;
Field<?>[] fields;
try {
fields = query.getFields(null);
}
// TODO: Can this happen? If so, trigger onError() correctly
catch (SQLException ignore) {
throw new DataAccessException("", ignore);
}
R record = (R) query.configuration().dsl().newRecord(fields);
// TODO: This call is duplicated from CursorImpl and related classes.
// Refactor this call to make sure code is re-used, especially when
// ExecuteListener lifecycle management is implemented
RecordDelegate<AbstractRecord> delegate = Tools.newRecord(true, (Supplier<AbstractRecord>) recordFactory(query.getRecordType(), Tools.row0(fields)), query.configuration());
for (int j = 0; j < fields.length; j++) {
Field<?> f = fields[j];
record.set((Field) f, row.get(j, f.getType()));
}
return (R) delegate.operate(record -> {
// TODO: Go through Field.getBinding()
// TODO: Make sure all the embeddable records, and other types of nested records are supported
for (int i = 0; i < fields.length; i++) {
Field<?> f = fields[i];
Object value = row.get(i, f.getType());
record.values[i] = value;
record.originals[i] = value;
}
return record;
return record;
});
}).subscribe(new RowSubscriber<R>(this));
}