Let Result.getRowsUpdated() return Long.

[resolves #484] Signed-off-by: Mark Paluch <mpaluch@vmware.com>
pgjdbc · Jan 27, 2022 · 1277a52 · 1277a52
1 parent 775b1aa
commit 1277a52
Show file tree

Hide file tree

Showing 12 changed files with 25 additions and 25 deletions.
diff --git a/pom.xml b/pom.xml
@@ -47,7 +47,7 @@
         <postgresql.version>42.3.1</postgresql.version>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
         <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
-        <r2dbc-spi.version>0.9.0.RELEASE</r2dbc-spi.version>
+        <r2dbc-spi.version>1.0.0.BUILD-SNAPSHOT</r2dbc-spi.version>
         <reactor.version>2020.0.15</reactor.version>
         <scram-client.version>2.1</scram-client.version>
         <spring-framework.version>5.3.14</spring-framework.version>

diff --git a/src/main/java/io/r2dbc/postgresql/PostgresqlConnection.java b/src/main/java/io/r2dbc/postgresql/PostgresqlConnection.java
@@ -71,7 +71,7 @@ final class PostgresqlConnection implements io.r2dbc.postgresql.api.PostgresqlCo
 
     private final Codecs codecs;
 
-    private final Flux<Integer> validationQuery;
+    private final Flux<Long> validationQuery;
 
     private final AtomicReference<NotificationAdapter> notificationAdapter = new AtomicReference<>();
 
@@ -380,15 +380,15 @@ public Mono<Boolean> validate(ValidationDepth depth) {
                 return;
             }
 
-            this.validationQuery.subscribe(new CoreSubscriber<Integer>() {
+            this.validationQuery.subscribe(new CoreSubscriber<Long>() {
 
                 @Override
                 public void onSubscribe(Subscription s) {
                     s.request(Integer.MAX_VALUE);
                 }
 
                 @Override
-                public void onNext(Integer integer) {
+                public void onNext(Long integer) {
 
                 }
 

diff --git a/src/main/java/io/r2dbc/postgresql/PostgresqlResult.java b/src/main/java/io/r2dbc/postgresql/PostgresqlResult.java
@@ -61,7 +61,7 @@ final class PostgresqlResult extends AbstractReferenceCounted implements io.r2db
 
     @Override
     @SuppressWarnings({"rawtypes", "unchecked"})
-    public Mono<Integer> getRowsUpdated() {
+    public Mono<Long> getRowsUpdated() {
 
         return this.messages
             .<Integer>handle((message, sink) -> {
@@ -88,7 +88,7 @@ public Mono<Integer> getRowsUpdated() {
                     return;
                 }
 
-                int sum = 0;
+                long sum = 0;
 
                 for (Integer integer : list) {
                     sum += integer;

diff --git a/src/main/java/io/r2dbc/postgresql/PostgresqlSegmentResult.java b/src/main/java/io/r2dbc/postgresql/PostgresqlSegmentResult.java
@@ -114,7 +114,7 @@ private PostgresqlSegmentResult(Flux<Segment> segments) {
     }
 
     @Override
-    public Mono<Integer> getRowsUpdated() {
+    public Mono<Long> getRowsUpdated() {
         return this.segments
             .<Integer>handle((segment, sink) -> {
 
@@ -137,7 +137,7 @@ public Mono<Integer> getRowsUpdated() {
                     return;
                 }
 
-                int sum = 0;
+                long sum = 0;
 
                 for (Integer integer : list) {
                     sum += integer;

diff --git a/src/main/java/io/r2dbc/postgresql/api/PostgresqlResult.java b/src/main/java/io/r2dbc/postgresql/api/PostgresqlResult.java
@@ -33,7 +33,7 @@ public interface PostgresqlResult extends Result {
      * {@inheritDoc}
      */
     @Override
-    Mono<Integer> getRowsUpdated();
+    Mono<Long> getRowsUpdated();
 
     /**
      * {@inheritDoc}

diff --git a/src/test/java/io/r2dbc/postgresql/AbstractCodecIntegrationTests.java b/src/test/java/io/r2dbc/postgresql/AbstractCodecIntegrationTests.java
@@ -738,7 +738,7 @@ private <IN, OUT> void testCodec(Class<IN> javaType, IN value, Class<OUT> outTyp
 
                         .concatWith(close(connection)))
                     .as(StepVerifier::create)
-                    .expectNext(1)
+                    .expectNext(1L)
                     .verifyComplete();
 
                 SERVER.getJdbcOperations().execute("DELETE FROM test");
@@ -754,7 +754,7 @@ private <IN, OUT> void testCodec(Class<IN> javaType, IN value, Class<OUT> outTyp
 
                         .concatWith(close(connection)))
                     .as(StepVerifier::create)
-                    .expectNext(1)
+                    .expectNext(1L)
                     .verifyComplete();
             } else {
 
@@ -769,7 +769,7 @@ private <IN, OUT> void testCodec(Class<IN> javaType, IN value, Class<OUT> outTyp
 
                         .concatWith(close(connection)))
                     .as(StepVerifier::create)
-                    .expectNext(1)
+                    .expectNext(1L)
                     .verifyComplete();
 
                 SERVER.getJdbcOperations().execute("DELETE FROM test");
@@ -785,7 +785,7 @@ private <IN, OUT> void testCodec(Class<IN> javaType, IN value, Class<OUT> outTyp
 
                         .concatWith(close(connection)))
                     .as(StepVerifier::create)
-                    .expectNext(1)
+                    .expectNext(1L)
                     .verifyComplete();
             }
 
@@ -856,7 +856,7 @@ private <W, R> void testCodecReadAs(W toWrite, Class<R> javaTypeToRead, Consumer
 
                     .concatWith(close(connection)))
                 .as(StepVerifier::create)
-                .expectNext(1)
+                .expectNext(1L)
                 .verifyComplete();
 
             this.connectionFactory.create()

diff --git a/src/test/java/io/r2dbc/postgresql/PostgresqlResultUnitTests.java b/src/test/java/io/r2dbc/postgresql/PostgresqlResultUnitTests.java
@@ -43,7 +43,7 @@ void toResultCommandComplete() {
 
         result.getRowsUpdated()
             .as(StepVerifier::create)
-            .expectNext(1)
+            .expectNext(1L)
             .verifyComplete();
     }
 
@@ -57,7 +57,7 @@ void toResultCommandCompleteUsingSegments() {
 
         result.getRowsUpdated()
             .as(StepVerifier::create)
-            .expectNext(1)
+            .expectNext(1L)
             .verifyComplete();
     }
 

diff --git a/src/test/java/io/r2dbc/postgresql/PostgresqlSegmentResultUnitTests.java b/src/test/java/io/r2dbc/postgresql/PostgresqlSegmentResultUnitTests.java
@@ -93,7 +93,7 @@ void shouldConsumeRowsUpdated() {
 
         result.getRowsUpdated()
             .as(StepVerifier::create)
-            .expectNext(42)
+            .expectNext(42L)
             .verifyComplete();
     }
 
@@ -105,7 +105,7 @@ void filterShouldRetainUpdateCount() {
 
         result.filter(Result.UpdateCount.class::isInstance).getRowsUpdated()
             .as(StepVerifier::create)
-            .expectNext(42)
+            .expectNext(42L)
             .verifyComplete();
     }
 

diff --git a/src/test/java/io/r2dbc/postgresql/PostgresqlStatementErrorsIntegrationTests.java b/src/test/java/io/r2dbc/postgresql/PostgresqlStatementErrorsIntegrationTests.java
@@ -47,7 +47,7 @@ void shouldReportDataIntegrityViolationUsingSimpleFlow() {
 
         SERVER.getJdbcOperations().execute("CREATE TABLE test (id SERIAL PRIMARY KEY)");
 
-        Flux<Integer> insert = Flux.from(this.connection.createStatement("INSERT INTO test (id) VALUES (1) RETURNING *").execute()).flatMap(Result::getRowsUpdated);
+        Flux<?> insert = Flux.from(this.connection.createStatement("INSERT INTO test (id) VALUES (1) RETURNING *").execute()).flatMap(Result::getRowsUpdated);
 
         insert.thenMany(insert).as(StepVerifier::create).verifyError(R2dbcDataIntegrityViolationException.class);
     }
@@ -57,7 +57,7 @@ void shouldReportDataIntegrityViolationUsingExtendedFlow() {
 
         SERVER.getJdbcOperations().execute("CREATE TABLE test (id SERIAL PRIMARY KEY)");
 
-        Flux<Integer> insert = Flux.from(this.connection.createStatement("INSERT INTO test (id) VALUES ($1) RETURNING *").bind("$1", 1).execute()).flatMap(Result::getRowsUpdated);
+        Flux<?> insert = Flux.from(this.connection.createStatement("INSERT INTO test (id) VALUES ($1) RETURNING *").bind("$1", 1).execute()).flatMap(Result::getRowsUpdated);
 
         insert.thenMany(insert).as(StepVerifier::create).verifyError(R2dbcDataIntegrityViolationException.class);
     }

diff --git a/src/test/java/io/r2dbc/postgresql/api/MockPostgresqlResult.java b/src/test/java/io/r2dbc/postgresql/api/MockPostgresqlResult.java
@@ -42,7 +42,7 @@ public static Builder builder() {
     }
 
     @Override
-    public Mono<Integer> getRowsUpdated() {
+    public Mono<Long> getRowsUpdated() {
         return Mono.empty();
     }
 

diff --git a/src/test/java/io/r2dbc/postgresql/codec/EnumCodecIntegrationTests.java b/src/test/java/io/r2dbc/postgresql/codec/EnumCodecIntegrationTests.java
@@ -86,7 +86,7 @@ void shouldBindEnumTypeAsString() {
             .execute()
             .flatMap(PostgresqlResult::getRowsUpdated)
             .as(StepVerifier::create)
-            .expectNext(1)
+            .expectNext(1L)
             .verifyComplete();
 
         String result = SERVER.getJdbcOperations().queryForObject("SELECT the_value FROM enum_test", String.class);
@@ -107,7 +107,7 @@ void shouldBindEnumArrayTypeAsString() {
             .execute()
             .flatMap(PostgresqlResult::getRowsUpdated)
             .as(StepVerifier::create)
-            .expectNext(1)
+            .expectNext(1L)
             .verifyComplete();
 
         String result = SERVER.getJdbcOperations().queryForObject("SELECT the_value FROM enum_test", String.class);
@@ -125,7 +125,7 @@ void shouldBindEnumArrayType() {
             .execute()
             .flatMap(PostgresqlResult::getRowsUpdated)
             .as(StepVerifier::create)
-            .expectNext(1)
+            .expectNext(1L)
             .verifyComplete();
 
         this.connection.createStatement("SELECT * FROM enum_test")

diff --git a/src/test/java/io/r2dbc/postgresql/replication/LogicalDecodeIntegrationTests.java b/src/test/java/io/r2dbc/postgresql/replication/LogicalDecodeIntegrationTests.java
@@ -102,7 +102,7 @@ void shouldReceiveReplication() {
 
         ReplicationStream replicationStream = replicationConnection.startReplication(replicationRequest).block(Duration.ofSeconds(10));
 
-        connection.createStatement("INSERT INTO logical_decode_test VALUES('Hello World')").execute().flatMap(PostgresqlResult::getRowsUpdated).as(StepVerifier::create).expectNext(1).verifyComplete();
+        connection.createStatement("INSERT INTO logical_decode_test VALUES('Hello World')").execute().flatMap(PostgresqlResult::getRowsUpdated).as(StepVerifier::create).expectNext(1L).verifyComplete();
 
         replicationStream.map(byteBuf -> byteBuf.toString(StandardCharsets.UTF_8))
             .as(StepVerifier::create)