prepare code for multidb

widefix · May 1, 2024 · 848b048 · 848b048
1 parent 8a671e0
commit 848b048
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/lib/actual_db_schema.rb b/lib/actual_db_schema.rb
@@ -29,7 +29,24 @@ class << self
   }
 
   def self.migrated_folder
-    Rails.root.join("tmp", "migrated").tap { |folder| FileUtils.mkdir_p(folder) }
+    migrated_folders.first
+  end
+
+  def self.migrated_folders
+    return [Rails.root.join("tmp", "migrated")] unless migrations_paths
+
+    Array(migrations_paths).map do |path|
+      if path.end_with?("db/migrate")
+        Rails.root.join("tmp", "migrated")
+      else
+        postfix = path.split("/").last
+        Rails.root.join("tmp", "migrated_#{postfix}")
+      end
+    end
+  end
+
+  def self.migrations_paths
+    ActiveRecord::Base.connection_db_config.migrations_paths
   end
 
   def self.migration_filename(fullpath)

diff --git a/lib/actual_db_schema/store.rb b/lib/actual_db_schema/store.rb
@@ -9,6 +9,7 @@ class Store
 
     def write(filename)
       basename = File.basename(filename)
+      FileUtils.mkdir_p(folder)
       FileUtils.copy(filename, folder.join(basename))
       record_metadata(filename)
     end