minor improvments

Author: ttt161

apps/bender/priv/migrations/sequence/1770994589112-copy-sequence.erl

@@ -3,8 +3,9 @@
 -export([perform/2]).
 
 -spec perform(_, _) -> _.
-perform(Connection, _MigrationOpts) ->
-    perform_batch(Connection, 0, 100).
+perform(Connection, MigrationOpts) ->
+    BatchSize = proplists:get_value(batch_size, MigrationOpts, 5000),
+    perform_batch(Connection, 0, BatchSize).
 
 perform_batch(Connection, Offset, Limit) ->
     SQL = "SELECT process_id, aux_state FROM bender_sequence_processes ORDER BY created_at OFFSET $1 LIMIT $2",

apps/bender/src/bender.erl

@@ -21,6 +21,7 @@
 -export_type([schema/0]).
 
 -define(DB_REF, bender).
+-define(DEFAULT_MIGRATION_OPTS, [{batch_size, 5000}]).
 
 %% API
 
@@ -163,6 +164,11 @@ db_init(machinery) ->
     ok.
 
 db_init(MigrationsDir, DbOpts) ->
-    {ok, _} = epg_migrator:perform("generator", DbOpts, [], filename:join([MigrationsDir, "generator"])),
-    {ok, _} = epg_migrator:perform("sequence", DbOpts, [], filename:join([MigrationsDir, "sequence"])),
+    MigrationOpts = application:get_env(bender, migration_opts, ?DEFAULT_MIGRATION_OPTS),
+    logger:info("migrations for generator start"),
+    {ok, _} = epg_migrator:perform("generator", DbOpts, MigrationOpts, filename:join([MigrationsDir, "generator"])),
+    logger:info("migrations for generator success"),
+    logger:info("migrations for sequence start"),
+    {ok, _} = epg_migrator:perform("sequence", DbOpts, MigrationOpts, filename:join([MigrationsDir, "sequence"])),
+    logger:info("migrations for sequence success"),
     ok.