Polishing

[fixes #341][#373]

Author: mp911de

src/main/java/io/r2dbc/postgresql/ConnectionResources.java

@@ -22,9 +22,6 @@
 import io.r2dbc.postgresql.codec.Codecs;
 import io.r2dbc.postgresql.util.Assert;
 
-/**
- * @author Mark Paluch
- */
 final class ConnectionResources {
 
     private final Client client;

src/main/java/io/r2dbc/postgresql/ExtendedQueryPostgresqlStatement.java

@@ -59,7 +59,7 @@ final class ExtendedQueryPostgresqlStatement implements PostgresqlStatement {
     private String[] generatedColumns;
 
     ExtendedQueryPostgresqlStatement(ConnectionResources resources, String sql) {
-        this.resources = Assert.requireNonNull(resources, "context must not be null");
+        this.resources = Assert.requireNonNull(resources, "resources must not be null");
         this.connectionContext = resources.getClient().getContext();
         this.sql = Assert.requireNonNull(sql, "sql must not be null");
         this.bindings = new Bindings(expectedSize(sql));

src/main/java/io/r2dbc/postgresql/PostgresqlResult.java

@@ -43,9 +43,7 @@
  */
 final class PostgresqlResult extends AbstractReferenceCounted implements io.r2dbc.postgresql.api.PostgresqlResult {
 
-    private static final Predicate<BackendMessage> TAKE_UNTIL = or(CommandComplete.class::isInstance, EmptyQueryResponse.class::isInstance);
-
-    private final ConnectionResources context;
+    private final ConnectionResources resources;
 
     private final Flux<BackendMessage> messages;
 
@@ -55,8 +53,8 @@ final class PostgresqlResult extends AbstractReferenceCounted implements io.r2db
 
     private volatile RowDescription rowDescription;
 
-    PostgresqlResult(ConnectionResources context, Flux<BackendMessage> messages, ExceptionFactory factory) {
-        this.context = Assert.requireNonNull(context, "context must not be null");
+    PostgresqlResult(ConnectionResources resources, Flux<BackendMessage> messages, ExceptionFactory factory) {
+        this.resources = Assert.requireNonNull(resources, "resources must not be null");
         this.messages = Assert.requireNonNull(messages, "messages must not be null");
         this.factory = Assert.requireNonNull(factory, "factory must not be null");
     }
@@ -92,7 +90,7 @@ public Mono<Integer> getRowsUpdated() {
     public <T> Flux<T> map(BiFunction<Row, RowMetadata, ? extends T> f) {
         Assert.requireNonNull(f, "f must not be null");
 
-        return this.messages.takeUntil(TAKE_UNTIL)
+        return this.messages
             .handle((message, sink) -> {
 
                 try {
@@ -103,12 +101,12 @@ public <T> Flux<T> map(BiFunction<Row, RowMetadata, ? extends T> f) {
 
                     if (message instanceof RowDescription) {
                         this.rowDescription = (RowDescription) message;
-                        this.metadata = PostgresqlRowMetadata.toRowMetadata(this.context.getCodecs(), (RowDescription) message);
+                        this.metadata = PostgresqlRowMetadata.toRowMetadata(this.resources.getCodecs(), (RowDescription) message);
                         return;
                     }
 
                     if (message instanceof DataRow) {
-                        PostgresqlRow row = PostgresqlRow.toRow(this.context, (DataRow) message, this.rowDescription);
+                        PostgresqlRow row = PostgresqlRow.toRow(this.resources, (DataRow) message, this.rowDescription);
 
                         sink.next(f.apply(row, this.metadata));
                     }
@@ -134,13 +132,13 @@ public ReferenceCounted touch(Object hint) {
     @Override
     public String toString() {
         return "PostgresqlResult{" +
-            "context=" + this.context +
+            "context=" + this.resources +
             ", messages=" + this.messages +
             '}';
     }
 
-    static PostgresqlResult toResult(ConnectionResources context, Flux<BackendMessage> messages, ExceptionFactory factory) {
-        return new PostgresqlResult(context, messages, factory);
+    static PostgresqlResult toResult(ConnectionResources resources, Flux<BackendMessage> messages, ExceptionFactory factory) {
+        return new PostgresqlResult(resources, messages, factory);
     }
 
 }

src/main/java/io/r2dbc/postgresql/SimpleQueryPostgresqlStatement.java

@@ -55,7 +55,7 @@ final class SimpleQueryPostgresqlStatement implements PostgresqlStatement {
     private int fetchSize;
 
     SimpleQueryPostgresqlStatement(ConnectionResources resources, String sql) {
-        this.resources = Assert.requireNonNull(resources, "context must not be null");
+        this.resources = Assert.requireNonNull(resources, "resources must not be null");
         this.sql = Assert.requireNonNull(sql, "sql must not be null");
         fetchSize(isBatch() ? NO_LIMIT : this.resources.getConfiguration().getFetchSize(sql));
     }

src/test/java/io/r2dbc/postgresql/ExtendedQueryPostgresqlStatementUnitTests.java

@@ -142,9 +142,9 @@ void bindWrongIdentifierFormat() {
     }
 
     @Test
-    void constructorNoContext() {
+    void constructorNoResources() {
         assertThatIllegalArgumentException().isThrownBy(() -> new ExtendedQueryPostgresqlStatement(null, "test-query"))
-            .withMessage("context must not be null");
+            .withMessage("resources must not be null");
     }
 
     @Test

src/test/java/io/r2dbc/postgresql/PostgresqlResultUnitTests.java

@@ -69,7 +69,7 @@ void toResultEmptyQueryResponse() {
     @Test
     void toResultNoContext() {
         assertThatIllegalArgumentException().isThrownBy(() -> PostgresqlResult.toResult(null, Flux.empty(), ExceptionFactory.INSTANCE))
-            .withMessage("context must not be null");
+            .withMessage("resources must not be null");
     }
 
     @Test

src/test/java/io/r2dbc/postgresql/SimpleQueryPostgresqlStatementUnitTests.java

@@ -106,9 +106,9 @@ void fetchSizeWithOneQuery() {
     }
 
     @Test
-    void constructorNoContext() {
+    void constructorNoResources() {
         assertThatIllegalArgumentException().isThrownBy(() -> new SimpleQueryPostgresqlStatement(null, "test-query"))
-            .withMessage("context must not be null");
+            .withMessage("resources must not be null");
     }
 
     @Test