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