init.go 972 B

123456789101112131415161718192021222324252627282930313233343536373839
  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/Purchase"
  8. "ERP_storage/models/RepairWorkOrder"
  9. "ERP_storage/models/validationtool"
  10. _ "github.com/go-sql-driver/mysql"
  11. "log"
  12. )
  13. func init() {
  14. AutoMigrateDB()
  15. }
  16. func AutoMigrateDB() {
  17. //自动迁移模式
  18. err := db.DB.Set("gorm:table_options", "charset=utf8mb4").
  19. AutoMigrate(
  20. &ContractReview.ServiceType{},
  21. &ContractReview.ServiceItem{},
  22. &ContractReview.ContractReview{},
  23. &ContractReview.ContractReviewServiceItem{},
  24. &validationtool.ValidationTool{},
  25. &validationtool.ValidationToolClass{},
  26. &IOTNetworkCard.IOTNetworkCard{},
  27. &Percentage.VerifyCompany{},
  28. &Percentage.Percentage{},
  29. &RepairWorkOrder.RepairWorkOrder{},
  30. &Purchase.Purchase{},
  31. &validationtool.ValidationToolHistory{},
  32. )
  33. if err != nil {
  34. log.Fatalf("migrate db fail: %v", err)
  35. }
  36. }