Fix bug with too much rows being buffered by Migration Tool

SwissBorg · Oct 19, 2020 · e1b330f · e1b330f
1 parent d12a28c
commit e1b330f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.../src/main/scala/akka/persistence/postgres/migration/v2/V2__Extract_journal_metadata.scala b/.../src/main/scala/akka/persistence/postgres/migration/v2/V2__Extract_journal_metadata.scala
@@ -62,7 +62,7 @@ private[migration] class V2__Extract_journal_metadata(
     snapshotTableConfig.schemaName.map(_ + ".").getOrElse("") + snapshotTableConfig.tableName
 
   private val migrationConf: Config = globalConfig.getConfig("akka-persistence-postgres.migration")
-  private val migrationBatchSize: Long = migrationConf.getLong("v2.batchSize")
+  private val migrationBatchSize: Int = migrationConf.getInt("v2.batchSize")
   private val conversionParallelism: Int = migrationConf.getInt("v2.parallelism")
 
   @throws[Exception]
@@ -101,7 +101,7 @@ private[migration] class V2__Extract_journal_metadata(
           .withStatementParameters(
             rsType = ResultSetType.ForwardOnly,
             rsConcurrency = ResultSetConcurrency.ReadOnly,
-            fetchSize = migrationBatchSize.max(Int.MaxValue).toInt)
+            fetchSize = migrationBatchSize)
           .transactionally)
     }
 
@@ -162,7 +162,7 @@ private[migration] class V2__Extract_journal_metadata(
           .withStatementParameters(
             rsType = ResultSetType.ForwardOnly,
             rsConcurrency = ResultSetConcurrency.ReadOnly,
-            fetchSize = migrationBatchSize.max(Int.MaxValue).toInt)
+            fetchSize = migrationBatchSize)
           .transactionally
       }
     }