Expose query metadata in PostgresRowSequence
#504
Conversation
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## main #504 +/- ##
==========================================
- Coverage 55.12% 55.04% -0.08%
==========================================
Files 127 127
Lines 10174 10188 +14
==========================================
Hits 5608 5608
- Misses 4566 4580 +14
|
|
|
||
| /// Collect and return all rows, alongside the query metadata. | ||
| /// - Returns: The query metadata and the rows. | ||
| public func collectWithMetadata() async throws -> (metadata: PostgresQueryMetadata, rows: [PostgresRow]) { |
There was a problem hiding this comment.
I'm open to better names.
| public func consume( | ||
| onRow: @Sendable (PostgresRow) throws -> () | ||
| ) async throws -> PostgresQueryMetadata { |
There was a problem hiding this comment.
I'm open to better names.
There was a problem hiding this comment.
Also the closure is intentionally not async since it's generally not a good idea to be doing async things on each row.
It's still possible to accumulate the rows and do whatever you want after.
I could change that if you think with async it would be better.
| let countQuery = PostgresQuery(unsafeSQL: "SELECT COUNT(*) FROM table1") | ||
| let countRows = try await connection.query(countQuery, logger: .psqlTest) | ||
| var countIterator = countRows.makeAsyncIterator() | ||
| let insertedRowsCount = try await countIterator.next()?.decode(Int.self, context: .default) | ||
| XCTAssertEqual(rowsCount, insertedRowsCount) | ||
| let result = try await connection.query(insertionQuery, logger: .psqlTest) | ||
| let metadata = try await result.collectWithMetadata().metadata | ||
| XCTAssertEqual(metadata.rows, rowsCount) |
There was a problem hiding this comment.
The metadata contains the count of the inserted rows so no need to do another query.
|
@fabianfett I've thought a bit more about this and possibly we might want to have another type that wraps |
Currently the functions that return
PostgresRowSequencehave no way of reporting back themetadataof the query.This PR exposes the
metadatathrough thecollectWithMetadata()andconsume(onRow:)methods.