Bugfix to group by index without missing versions with same timestamp.

amberframework · May 13, 2018 · 314847d · 314847d
1 parent 3c60d8a
commit 314847d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/micrate.cr b/src/micrate.cr
@@ -147,7 +147,8 @@ module Micrate
     Dir.entries(migrations_dir)
        .select { |name| File.file? File.join("db/migrations", name) }
        .select { |name| /^\d+_.+\.sql$/ =~ name }
-       .map { |name| Migration.from_file(name) }
+       .sort
+       .map_with_index { |name, index| Migration.from_file(name, index) }
        .index_by { |migration| migration.version }
   end
 

diff --git a/src/micrate/migration.cr b/src/micrate/migration.cr
@@ -67,9 +67,9 @@ module Micrate
       s.split("--")[0].strip.ends_with? ";"
     end
 
-    def self.from_file(file_name)
+    def self.from_file(file_name, index = 0)
       full_path = File.join(Micrate.migrations_dir, file_name)
-      version = file_name.split("_")[0].to_i64
+      version = file_name.split("_")[0].to_i64 + index
       new(version, file_name, File.read(full_path))
     end