1599190683670_migrate.go 546 B

1234567891011121314151617181920212223242526272829
  1. package version
  2. import (
  3. common "device-manage/common/models"
  4. "runtime"
  5. "gorm.io/gorm"
  6. "device-manage/app/admin/models"
  7. "device-manage/cmd/migrate/migration"
  8. )
  9. func init() {
  10. _, fileName, _, _ := runtime.Caller(0)
  11. migration.Migrate.SetVersion(migration.GetFilename(fileName), _1599190683670Test)
  12. }
  13. func _1599190683670Test(db *gorm.DB, version string) error {
  14. return db.Transaction(func(tx *gorm.DB) error {
  15. if err := models.InitDb(tx); err != nil {
  16. }
  17. return tx.Create(&common.Migration{
  18. Version: version,
  19. }).Error
  20. })
  21. }