Update type definition

Kaligo · Jan 8, 2024 · a071c9d · a071c9d
1 parent 0061ca6
commit a071c9d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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