init.go 763 B

123456789101112131415161718192021222324252627282930313233
  1. package models
  2. import (
  3. db "ERP_storage/initialize"
  4. "ERP_storage/models/ContractReview"
  5. "ERP_storage/models/IOTNetworkCard"
  6. "ERP_storage/models/Percentage"
  7. "ERP_storage/models/validationtool"
  8. _ "github.com/go-sql-driver/mysql"
  9. "log"
  10. )
  11. func init() {
  12. AutoMigrateDB()
  13. }
  14. func AutoMigrateDB() {
  15. //自动迁移模式
  16. err := db.DB.Set("gorm:table_options", "charset=utf8mb4").
  17. AutoMigrate(
  18. &ContractReview.ServiceType{},
  19. &ContractReview.ServiceItem{},
  20. &ContractReview.ContractReview{},
  21. &ContractReview.ContractReviewServiceItem{},
  22. &validationtool.ValidationTool{},
  23. &validationtool.ValidationToolClass{},
  24. &IOTNetworkCard.IOTNetworkCard{},
  25. &Percentage.Percentage{},
  26. )
  27. if err != nil {
  28. log.Fatalf("migrate db fail: %v", err)
  29. }
  30. }