From a071c9d1693f52f7356a592c825e5a2fd4d92739 Mon Sep 17 00:00:00 2001 From: Hieu Nguyen Date: Mon, 8 Jan 2024 20:20:56 +0800 Subject: [PATCH] Update type definition --- lib/sequel_data/migrate/migrator.rb | 6 +++--- sig/sequel_data/migrate.rbs | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/sequel_data/migrate/migrator.rb b/lib/sequel_data/migrate/migrator.rb index 7e253a7..ea2a737 100644 --- a/lib/sequel_data/migrate/migrator.rb +++ b/lib/sequel_data/migrate/migrator.rb @@ -15,7 +15,7 @@ def initialize(config) end def migrate - connect_to_db do |db| + connect_database do |db| dataset = ensure_table_exists(db) already_migrated = dataset.select_map(column).to_set @@ -32,7 +32,7 @@ def migrate end def rollback(step = 1) - connect_to_db do |db| + connect_database do |db| dataset = ensure_table_exists(db) already_migrated = dataset.select_map(column).to_set @@ -65,7 +65,7 @@ def table :data_migrations end - def connect_to_db(&block) + def connect_database(&block) raise ConfigurationError, "db_configuration is not set" if config.db_configuration.host.nil? Sequel.connect(config.db_configuration.host, &block) diff --git a/sig/sequel_data/migrate.rbs b/sig/sequel_data/migrate.rbs index fa21048..72ab5ca 100644 --- a/sig/sequel_data/migrate.rbs +++ b/sig/sequel_data/migrate.rbs @@ -54,7 +54,7 @@ module SequelData attr_reader config: untyped def column: -> :version def table: -> :data_migrations - def connect_database: -> untyped + def connect_database: () { (untyped db) -> void } -> untyped def ensure_table_exists: (untyped db) -> untyped def fetch_migration_files: -> Array[String] def load_migration_file: (String file) -> untyped