Check filenames for func migrations. (#32)

stimtech · Jun 28, 2023 · da6d853 · da6d853
1 parent 6d163bd
commit da6d853
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/migration.go b/migration.go
@@ -83,6 +83,14 @@ func (s *Service) Migrate() error {
 			if err != nil {
 				return fmt.Errorf("failed to get checksum for file %s: %w", mig, err)
 			}
+
+			if strings.HasSuffix(mig, ".go") {
+				c, err = s.checksum(bytes.NewReader([]byte(mig)))
+				if err != nil {
+					return fmt.Errorf("failed to create checksum for migration: %w", err)
+				}
+			}
+
 			if c != chkSum {
 				return fmt.Errorf("file %s has been updated since it was migrated", mig)
 			}