Polishing.

Guard long adding against overflows. [#597][#598] Signed-off-by: Mark Paluch <mpaluch@vmware.com>
pgjdbc · Jun 16, 2023 · 9b656c9 · 9b656c9
1 parent cb8b5ec
commit 9b656c9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/io/r2dbc/postgresql/PostgresqlResult.java b/src/main/java/io/r2dbc/postgresql/PostgresqlResult.java
@@ -92,7 +92,12 @@ public Mono<Long> getRowsUpdated() {
                 long sum = 0;
 
                 for (Long value : list) {
-                    sum += value;
+                    try {
+                        sum = Math.addExact(sum, value);
+                    } catch (ArithmeticException e) {
+                        sum = Long.MAX_VALUE;
+                        break;
+                    }
                 }
 
                 sink.next(sum);