init.go 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. package models
  2. import (
  3. db "Medical_ERP/common/initialize"
  4. "fmt"
  5. _ "github.com/go-sql-driver/mysql"
  6. "log"
  7. "sync"
  8. )
  9. func init() {
  10. AutoMigrateDB()
  11. }
  12. func AutoMigrateDB() {
  13. //自动迁移模式
  14. err := db.DB.Set("gorm:table_options", "charset=utf8mb4").
  15. AutoMigrate(
  16. &OperationLog{},
  17. &RoleApi{},
  18. &Product{},
  19. &Enterprise{},
  20. &Spec{},
  21. &DosageForm{},
  22. &Unit{},
  23. &MedicineTemplate{},
  24. &StockIn{},
  25. &StockOut{},
  26. &MedicineInventory{},
  27. &Barcode{},
  28. )
  29. if err != nil {
  30. log.Fatalf("migrate db fail: %v", err)
  31. }
  32. }
  33. var Spec_list *sync.Map
  34. var DosageForm_list *sync.Map
  35. var Enterprise_list *sync.Map
  36. var Product_list *sync.Map
  37. var Unit_list *sync.Map
  38. // 初始化基础数据
  39. func InitBasicData(deptId int) {
  40. Spec_list = new(sync.Map)
  41. DosageForm_list = new(sync.Map)
  42. Enterprise_list = new(sync.Map)
  43. Product_list = new(sync.Map)
  44. Unit_list = new(sync.Map)
  45. Read_Spec_All_Map(deptId)
  46. Read_Product_All_Map(deptId)
  47. Read_DosageForm_All_Map(deptId)
  48. Read_Enterprise_All_Map(deptId)
  49. Read_Unit_All_Map(deptId)
  50. }
  51. // 获取全部规格
  52. func Read_Spec_All_Map(deptId int) {
  53. var spec []Spec
  54. if err := db.DB.Where("dept_id = ?", deptId).Find(&spec).Error; err != nil {
  55. fmt.Println("Error loading Specs:", err)
  56. return
  57. }
  58. for _, v := range spec {
  59. Spec_list.Store(v.Id, v.Name)
  60. }
  61. }
  62. func Read_Spec_Get(Id int) string {
  63. v, ok := Spec_list.Load(Id)
  64. if ok {
  65. return v.(string)
  66. } else {
  67. return ""
  68. }
  69. }
  70. // 获取全部剂型
  71. func Read_DosageForm_All_Map(deptId int) {
  72. var dosageForm []DosageForm
  73. if err := db.DB.Where("dept_id = ?", deptId).Find(&dosageForm).Error; err != nil {
  74. fmt.Println("Error loading DosageForm:", err)
  75. return
  76. }
  77. for _, v := range dosageForm {
  78. DosageForm_list.Store(v.Id, v.Name)
  79. }
  80. }
  81. func Read_DosageForm_Get(Id int) string {
  82. v, ok := DosageForm_list.Load(Id)
  83. if ok {
  84. return v.(string)
  85. } else {
  86. return ""
  87. }
  88. }
  89. // 获取全部生产企业
  90. func Read_Enterprise_All_Map(deptId int) {
  91. var enterprise []Enterprise
  92. if err := db.DB.Where("dept_id = ?", deptId).Find(&enterprise).Error; err != nil {
  93. fmt.Println("Error loading Enterprise:", err)
  94. return
  95. }
  96. for _, v := range enterprise {
  97. Enterprise_list.Store(v.Id, v.Name)
  98. }
  99. }
  100. func Read_Enterprise_Get(Id int) string {
  101. v, ok := Enterprise_list.Load(Id)
  102. if ok {
  103. return v.(string)
  104. } else {
  105. return ""
  106. }
  107. }
  108. // 获取全部品名
  109. func Read_Product_All_Map(deptId int) {
  110. var product []Product
  111. if err := db.DB.Where("dept_id = ?", deptId).Find(&product).Error; err != nil {
  112. fmt.Println("Error loading Product:", err)
  113. return
  114. }
  115. for _, v := range product {
  116. Product_list.Store(v.Id, v.Name)
  117. }
  118. }
  119. func Read_Product_Get(Id int) string {
  120. v, ok := Product_list.Load(Id)
  121. if ok {
  122. return v.(string)
  123. } else {
  124. return ""
  125. }
  126. }
  127. // 获取全部单位
  128. func Read_Unit_All_Map(deptId int) {
  129. var unit []Unit
  130. if err := db.DB.Where("dept_id = ?", deptId).Find(&unit).Error; err != nil {
  131. fmt.Println("Error loading Unit:", err)
  132. return
  133. }
  134. for _, v := range unit {
  135. Unit_list.Store(v.Id, v.Name)
  136. }
  137. }
  138. func Read_Unit_Get(Id int) string {
  139. v, ok := Unit_list.Load(Id)
  140. if ok {
  141. return v.(string)
  142. } else {
  143. return ""
  144. }
  145. }