IsComplete helper to track when migrations are all done

dnnyjns · Dec 22, 2019 · cf08dee · cf08dee
1 parent 5d54484
commit cf08dee
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/migration.go b/migration.go
@@ -46,6 +46,10 @@ func Create(dir, name string) {
 	}
 }
 
+func IsComplete(db *gorm.DB) bool {
+	return internalMigrations.IsComplete(db)
+}
+
 func Migrate(db *gorm.DB) error {
 	db.AutoMigrate(&Migration{})
 	return internalMigrations.Migrate(db)

diff --git a/migrations.go b/migrations.go
@@ -8,6 +8,21 @@ import (
 
 type migrations []*Migration
 
+func (m *migrations) IsComplete(db *gorm.DB) bool {
+	var (
+		count int
+		total = len(*m)
+		ids   = make([]string, total)
+	)
+
+	for i, migration := range *m {
+		ids[i] = migration.Version
+	}
+	db.Model(&Migration{}).Where(ids).Count(&count)
+
+	return total == count
+}
+
 func (m *migrations) Migrate(db *gorm.DB) error {
 	var (
 		ids       = make([]string, len(*m))