auto_create_week_schedules_service.go 26KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613
  1. package service
  2. import (
  3. "fmt"
  4. "strings"
  5. "time"
  6. "XT_New/models"
  7. "XT_New/utils"
  8. "github.com/jinzhu/gorm"
  9. "github.com/robfig/cron"
  10. )
  11. // cron表达式 https://www.cnblogs.com/zuxingyu/p/6023919.html
  12. var createWeekSchedulesCronJob *cron.Cron
  13. func init() {
  14. utils.InfoLog("开启自动排班定时任务")
  15. createWeekSchedulesCronJob = cron.New()
  16. spec := "0 0 0 ? * MON" // 每周一凌晨0点生成排班
  17. //spec := "0 */1 * * * ?"
  18. createWeekSchedulesCronJob.AddFunc(spec, func() {
  19. AutoCreateWeekSchedules()
  20. NewAutoCreateWeekSchedules()
  21. })
  22. }
  23. func BeginAutoCreateWeekSchedulesJob() {
  24. createWeekSchedulesCronJob.Start()
  25. }
  26. func NewAutoCreateWeekSchedules() {
  27. utils.TraceLog("自动排班任务开始执行")
  28. var modes []*models.PatientScheduleTemplateMode
  29. getModesErr := readDb.Model(&models.PatientScheduleTemplateMode{}).Where("mode <> 0 AND status = 1").Find(&modes).Error
  30. if getModesErr != nil {
  31. utils.ErrorLog("自动排班-获取启用模板模式列表失败:%v", getModesErr)
  32. CreateDBErrorRecord(0, "自动排班-获取启用模板模式列表失败:%v", getModesErr)
  33. return
  34. }
  35. for _, mode := range modes {
  36. templates, getTemplatesErr := GetOrgPatientScheduleTemplateIDs(mode.OrgID)
  37. if getTemplatesErr != nil {
  38. utils.ErrorLog("自动排班-org_id: %v :获取模板 ID 失败:%v", mode.OrgID, getTemplatesErr)
  39. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取模板 ID 失败:%v", mode.OrgID, getTemplatesErr)
  40. continue
  41. } else if len(templates) == 0 {
  42. utils.WarningLog("自动排班-org_id: %v :没有模板", mode.OrgID)
  43. continue
  44. }
  45. now := time.Now()
  46. nextSecWeek := now.AddDate(0, 0, 14)
  47. //nextThridWeek := now.AddDate(0, 0, 21)
  48. //nextFourWeek := now.AddDate(0, 0, 28)
  49. tx := writeDb.Begin()
  50. thisTime := time.Now()
  51. _, theWeek := thisTime.ISOWeek()
  52. //mode.Week == 0 为了兼容开发新功能前的老数据
  53. //新功能加了,预约排班生成的,int64(theWeek) >= mode.Week 当判断当前周数是否小大于或者等于模版预约第几周生成的周数
  54. if mode.Week == 0 || int64(theWeek) >= mode.Week {
  55. if mode.Mode == 1 { // 用第一周模板生成下第二周的排班
  56. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  57. if getItemsErr != nil {
  58. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  59. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  60. continue
  61. }
  62. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  63. if insertErr != nil {
  64. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  65. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  66. tx.Rollback()
  67. continue
  68. } else {
  69. //更新周次
  70. templates[0].Week = count
  71. UpdateTemplateItemWeek(templates[0].OrgID, templates[0].ID, int8(templates[0].Week))
  72. continue
  73. }
  74. } else if mode.Mode == 2 { // mode.ExecuteTimes % 2 的结果,如果为0,用第一周模板生成下第二周的排班,否则用第二周的模板生成排班
  75. if mode.ExecuteTimes%2 == 0 {
  76. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  77. if getItemsErr != nil {
  78. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  79. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  80. continue
  81. }
  82. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  83. if insertErr != nil {
  84. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  85. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  86. tx.Rollback()
  87. continue
  88. } else {
  89. //更新周次
  90. templates[0].Week = count
  91. UpdateTemplateItemWeek(templates[0].OrgID, templates[0].ID, int8(templates[0].Week))
  92. continue
  93. }
  94. } else {
  95. if len(templates) < 2 {
  96. utils.WarningLog("自动排班-org_id: %v :没有第二周的模板")
  97. continue
  98. }
  99. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[1].ID)
  100. if getItemsErr != nil {
  101. utils.ErrorLog("自动排班-org_id: %v :获取第二周模板Item失败:%v", mode.OrgID, getItemsErr)
  102. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第二周模板Item失败:%v", mode.OrgID, getItemsErr)
  103. continue
  104. }
  105. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  106. if insertErr != nil {
  107. utils.ErrorLog("自动排班-org_id: %v :使用第二周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  108. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第二周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  109. tx.Rollback()
  110. continue
  111. } else {
  112. //更新周次
  113. templates[1].Week = count
  114. UpdateTemplateItemWeek(templates[1].OrgID, templates[1].ID, int8(templates[1].Week))
  115. continue
  116. }
  117. }
  118. } else if mode.Mode == 3 { // mode.ExecuteTimes % 2 的结果,如果为0,用第一周模板生成下第二周的排班,否则用第二周的模板生成排班
  119. if mode.ExecuteTimes%3 == 0 {
  120. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  121. if getItemsErr != nil {
  122. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  123. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  124. continue
  125. }
  126. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  127. if insertErr != nil {
  128. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  129. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  130. tx.Rollback()
  131. continue
  132. } else {
  133. templates[0].Week = count
  134. UpdateTemplateItemWeek(templates[0].OrgID, templates[0].ID, int8(templates[0].Week))
  135. continue
  136. }
  137. } else if mode.ExecuteTimes%3 == 1 {
  138. if len(templates) < 2 {
  139. utils.WarningLog("自动排班-org_id: %v :没有第二周的模板")
  140. continue
  141. }
  142. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[1].ID)
  143. if getItemsErr != nil {
  144. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  145. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  146. continue
  147. }
  148. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  149. if insertErr != nil {
  150. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  151. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  152. tx.Rollback()
  153. continue
  154. } else {
  155. //更新周次
  156. templates[1].Week = count
  157. UpdateTemplateItemWeek(templates[1].OrgID, templates[1].ID, int8(templates[1].Week))
  158. continue
  159. }
  160. } else {
  161. if len(templates) < 3 {
  162. utils.WarningLog("自动排班-org_id: %v :没有第三周的模板")
  163. continue
  164. }
  165. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[2].ID)
  166. if getItemsErr != nil {
  167. utils.ErrorLog("自动排班-org_id: %v :获取第三周模板Item失败:%v", mode.OrgID, getItemsErr)
  168. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第三周模板Item失败:%v", mode.OrgID, getItemsErr)
  169. continue
  170. }
  171. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  172. if insertErr != nil {
  173. utils.ErrorLog("自动排班-org_id: %v :使用第三周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  174. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第三周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  175. tx.Rollback()
  176. continue
  177. } else {
  178. //更新周次
  179. templates[2].Week = count
  180. UpdateTemplateItemWeek(templates[2].OrgID, templates[2].ID, int8(templates[2].Week))
  181. continue
  182. }
  183. }
  184. } else if mode.Mode == 4 { // mode.ExecuteTimes % 2 的结果,如果为0,用第一周模板生成下第二周的排班,否则用第二周的模板生成排班
  185. if mode.ExecuteTimes%4 == 0 {
  186. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  187. if getItemsErr != nil {
  188. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  189. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  190. continue
  191. }
  192. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  193. if insertErr != nil {
  194. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  195. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  196. tx.Rollback()
  197. continue
  198. } else {
  199. //更新周次
  200. templates[0].Week = count
  201. UpdateTemplateItemWeek(templates[0].OrgID, templates[0].ID, int8(templates[0].Week))
  202. continue
  203. }
  204. } else if mode.ExecuteTimes%4 == 1 {
  205. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[1].ID)
  206. if getItemsErr != nil {
  207. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  208. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  209. continue
  210. }
  211. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  212. if insertErr != nil {
  213. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  214. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  215. tx.Rollback()
  216. continue
  217. } else {
  218. //更新周次
  219. templates[1].Week = count
  220. UpdateTemplateItemWeek(templates[1].OrgID, templates[1].ID, int8(templates[1].Week))
  221. continue
  222. }
  223. } else if mode.ExecuteTimes%4 == 2 {
  224. if len(templates) < 2 {
  225. utils.WarningLog("自动排班-org_id: %v :没有第二周的模板")
  226. continue
  227. }
  228. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[2].ID)
  229. if getItemsErr != nil {
  230. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  231. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  232. continue
  233. }
  234. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  235. if insertErr != nil {
  236. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  237. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  238. tx.Rollback()
  239. continue
  240. } else {
  241. //更新周次
  242. templates[2].Week = count
  243. UpdateTemplateItemWeek(templates[2].OrgID, templates[2].ID, int8(templates[2].Week))
  244. continue
  245. }
  246. } else if mode.ExecuteTimes%4 == 3 {
  247. if len(templates) < 3 {
  248. utils.WarningLog("自动排班-org_id: %v :没有第三周的模板")
  249. continue
  250. }
  251. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[3].ID)
  252. if getItemsErr != nil {
  253. utils.ErrorLog("自动排班-org_id: %v :获取第三周模板Item失败:%v", mode.OrgID, getItemsErr)
  254. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第三周模板Item失败:%v", mode.OrgID, getItemsErr)
  255. continue
  256. }
  257. insertErr, count := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  258. if insertErr != nil {
  259. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第四周排班失败:%v", mode.OrgID, insertErr)
  260. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第三周排班失败:%v", mode.OrgID, insertErr)
  261. tx.Rollback()
  262. continue
  263. } else {
  264. //更新周次
  265. templates[3].Week = count
  266. UpdateTemplateItemWeek(templates[3].OrgID, templates[3].ID, int8(templates[3].Week))
  267. continue
  268. }
  269. }
  270. }
  271. mode.ExecuteTimes += 1
  272. mode.ModifyTime = now.Unix()
  273. updateModeErr := tx.Save(mode).Error
  274. if updateModeErr != nil {
  275. utils.ErrorLog("自动排班-org_id:%v :更新启用模板模式的执行次数失败:%v", mode.OrgID, updateModeErr)
  276. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id:%v :更新启用模板模式的执行次数失败:%v", mode.OrgID, updateModeErr)
  277. tx.Rollback()
  278. continue
  279. }
  280. tx.Commit()
  281. utils.SuccessLog("自动排班-插入排班成功:org_id: %v", mode.OrgID)
  282. time.Sleep(time.Second * 30)
  283. }
  284. utils.SuccessLog("自动排班任务完成")
  285. }
  286. }
  287. func AutoCreateWeekSchedules() {
  288. utils.TraceLog("自动排班任务开始执行")
  289. var modes []*models.PatientScheduleTemplateMode
  290. getModesErr := readDb.Model(&models.PatientScheduleTemplateMode{}).Where("mode <> 0 AND status = 1").Find(&modes).Error
  291. if getModesErr != nil {
  292. utils.ErrorLog("自动排班-获取启用模板模式列表失败:%v", getModesErr)
  293. CreateDBErrorRecord(0, "自动排班-获取启用模板模式列表失败:%v", getModesErr)
  294. return
  295. }
  296. for _, mode := range modes {
  297. templates, getTemplatesErr := GetOrgPatientScheduleTemplateIDs(mode.OrgID)
  298. if getTemplatesErr != nil {
  299. utils.ErrorLog("自动排班-org_id: %v :获取模板 ID 失败:%v", mode.OrgID, getTemplatesErr)
  300. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取模板 ID 失败:%v", mode.OrgID, getTemplatesErr)
  301. continue
  302. } else if len(templates) == 0 {
  303. utils.WarningLog("自动排班-org_id: %v :没有模板", mode.OrgID)
  304. continue
  305. }
  306. now := time.Now()
  307. nextSecWeek := now.AddDate(0, 0, 14)
  308. tx := writeDb.Begin()
  309. if mode.Mode == 1 { // 用第一周模板生成下第二周的排班
  310. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  311. if getItemsErr != nil {
  312. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  313. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  314. continue
  315. }
  316. insertErr, _ := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  317. if insertErr != nil {
  318. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  319. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  320. tx.Rollback()
  321. continue
  322. }
  323. } else if mode.Mode == 2 { // mode.ExecuteTimes % 2 的结果,如果为0,用第一周模板生成下第二周的排班,否则用第二周的模板生成排班
  324. if mode.ExecuteTimes%2 == 0 {
  325. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  326. if getItemsErr != nil {
  327. utils.ErrorLog("自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  328. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第一周模板Item失败:%v", mode.OrgID, getItemsErr)
  329. continue
  330. }
  331. insertErr, _ := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  332. if insertErr != nil {
  333. utils.ErrorLog("自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  334. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  335. tx.Rollback()
  336. continue
  337. }
  338. } else {
  339. if len(templates) < 2 {
  340. utils.WarningLog("自动排班-org_id: %v :没有第二周的模板")
  341. continue
  342. }
  343. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[1].ID)
  344. if getItemsErr != nil {
  345. utils.ErrorLog("自动排班-org_id: %v :获取第二周模板Item失败:%v", mode.OrgID, getItemsErr)
  346. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :获取第二周模板Item失败:%v", mode.OrgID, getItemsErr)
  347. continue
  348. }
  349. insertErr, _ := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, nextSecWeek, items, nil)
  350. if insertErr != nil {
  351. utils.ErrorLog("自动排班-org_id: %v :使用第二周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  352. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id: %v :使用第二周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  353. tx.Rollback()
  354. continue
  355. } else {
  356. }
  357. }
  358. }
  359. mode.ExecuteTimes += 1
  360. mode.ModifyTime = now.Unix()
  361. updateModeErr := tx.Save(mode).Error
  362. if updateModeErr != nil {
  363. utils.ErrorLog("自动排班-org_id:%v :更新启用模板模式的执行次数失败:%v", mode.OrgID, updateModeErr)
  364. CreateDBErrorRecord(mode.OrgID, "自动排班-org_id:%v :更新启用模板模式的执行次数失败:%v", mode.OrgID, updateModeErr)
  365. tx.Rollback()
  366. continue
  367. }
  368. tx.Commit()
  369. utils.SuccessLog("自动排班-插入排班成功:org_id: %v", mode.OrgID)
  370. time.Sleep(time.Second * 30)
  371. }
  372. utils.SuccessLog("自动排班任务完成")
  373. }
  374. func _refreshWeekSchedulesWithTemplateItems(tx *gorm.DB, orgID int64, week time.Time, templateItems []*models.PatientScheduleTemplateItem, deviceNumberMap map[int64]*DeviceNumberViewModel) (error, int) {
  375. if deviceNumberMap == nil {
  376. deviceNumbers, getDeviceNumbersErr := GetAllValidDeviceNumbers(orgID)
  377. //fmt.Println("getDeviceNumbersErr3232322332323232323323223232323",getDeviceNumbersErr)
  378. if getDeviceNumbersErr != nil {
  379. return getDeviceNumbersErr, 0
  380. }
  381. deviceNumberMap = make(map[int64]*DeviceNumberViewModel, 0)
  382. for _, deviceNumber := range deviceNumbers {
  383. deviceNumberMap[deviceNumber.ID] = deviceNumber
  384. }
  385. }
  386. now := time.Now()
  387. monday, sunday := utils.GetMondayAndSundayOfWeekDate(&week)
  388. disableErr := tx.Model(&models.Schedule{}).Where("user_org_id = ? AND status = 1 AND schedule_date >= ? AND schedule_date <= ?", orgID, monday.Unix(), sunday.Unix()).Updates(map[string]interface{}{"status": 0, "updated_time": now.Unix()}).Error
  389. //fmt.Println("disableErr323232332323232233222332232332",disableErr)
  390. if disableErr != nil {
  391. return disableErr, 0
  392. }
  393. var week_count int
  394. // 批量插入排班
  395. if len(templateItems) > 0 {
  396. timeLayout := "20060102"
  397. loc, _ := time.LoadLocation("Local")
  398. tmp, _ := time.ParseInLocation(timeLayout, time.Unix(monday.Unix(), 0).Format("2006-01-02"), loc)
  399. _, weeks := tmp.ISOWeek()
  400. week_count = weeks
  401. valueStrs := make([]string, 0, len(templateItems))
  402. values := make([]interface{}, 0, len(templateItems)*11)
  403. for _, item := range templateItems {
  404. deviceNumber := deviceNumberMap[item.DeviceNumberID]
  405. if deviceNumber == nil {
  406. continue
  407. }
  408. so, _ := GetDialysisSolutionTwo(item.OrgID, item.PatientID, item.TreatMode)
  409. filedRecordOne, _ := FindFiledBy(item.OrgID, "透析器")
  410. filedRecordTwo, _ := FindFiledBy(item.OrgID, "灌流器")
  411. filedRecordThree, _ := FindFiledBy(item.OrgID, "透析器/灌流器")
  412. var DialysisMachineName string
  413. if filedRecordOne.IsShow == 1 {
  414. DialysisMachineName = so.DialysisDialyszers
  415. //service.UpdateSchDName(schedule)
  416. }
  417. if filedRecordTwo.IsShow == 1 {
  418. DialysisMachineName = so.DialysisIrrigation
  419. //service.UpdateSchDName(schedule)
  420. }
  421. if filedRecordThree.IsShow == 1 {
  422. DialysisMachineName = so.DialyzerPerfusionApparatus
  423. //service.UpdateSchDName(schedule)
  424. }
  425. valueStrs = append(valueStrs, "(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?,?)")
  426. values = append(values, orgID)
  427. values = append(values, deviceNumber.ZoneID)
  428. values = append(values, item.DeviceNumberID)
  429. values = append(values, item.PatientID)
  430. values = append(values, monday.AddDate(0, 0, int(item.Weekday-1)).Unix())
  431. values = append(values, item.Weekday)
  432. values = append(values, item.TimeType)
  433. values = append(values, item.TreatMode)
  434. values = append(values, 1)
  435. values = append(values, now.Unix())
  436. values = append(values, now.Unix())
  437. values = append(values, DialysisMachineName)
  438. }
  439. sql := fmt.Sprintf("INSERT INTO xt_schedule (user_org_id, partition_id, bed_id, patient_id, schedule_date, schedule_week, schedule_type, mode_id, status, created_time, updated_time,dialysis_machine_name) VALUES %v;", strings.Join(valueStrs, ", "))
  440. if insertErr := tx.Exec(sql, values...).Error; insertErr != nil {
  441. //fmt.Println("insertErr232332233223332232323222222www",insertErr)
  442. return insertErr, 0
  443. }
  444. }
  445. return nil, week_count
  446. }
  447. func FakeAutoCreateNextTwoWeekSchedules(orgID int64) {
  448. mode, getModeErr := GetOrgPatientScheduleTemplateMode(orgID)
  449. if getModeErr != nil {
  450. utils.ErrorLog("%v", getModeErr)
  451. return
  452. }
  453. templates, getTemplatesErr := GetOrgPatientScheduleTemplateIDs(mode.OrgID)
  454. if getTemplatesErr != nil {
  455. utils.ErrorLog("获取模板 ID 失败:%v", getTemplatesErr)
  456. return
  457. } else if len(templates) == 0 {
  458. utils.WarningLog("没有模板")
  459. return
  460. }
  461. deviceNumbers, getDeviceNumbersErr := GetAllValidDeviceNumbers(orgID)
  462. if getDeviceNumbersErr != nil {
  463. utils.ErrorLog("获取床位列表失败:%v", getDeviceNumbersErr)
  464. return
  465. }
  466. deviceNumberMap := make(map[int64]*DeviceNumberViewModel, 0)
  467. for _, deviceNumber := range deviceNumbers {
  468. deviceNumberMap[deviceNumber.ID] = deviceNumber
  469. }
  470. now := time.Now()
  471. tx := writeDb.Begin()
  472. if mode.Mode == 1 { // 用第一周模板生成下两周的排班
  473. items, getItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  474. if getItemsErr != nil {
  475. utils.ErrorLog("获取第一周模板Item失败:%v", getItemsErr)
  476. return
  477. }
  478. insertErr, _ := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, now.AddDate(0, 0, 7), items, deviceNumberMap)
  479. if insertErr != nil {
  480. utils.ErrorLog("使用第一周模板插入下第一周排班失败:%v", insertErr)
  481. tx.Rollback()
  482. return
  483. }
  484. insertErr, _ = _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, now.AddDate(0, 0, 14), items, deviceNumberMap)
  485. if insertErr != nil {
  486. utils.ErrorLog("使用第一周模板插入下第二周排班失败:%v", insertErr)
  487. tx.Rollback()
  488. return
  489. }
  490. } else if mode.Mode == 2 {
  491. firstItems, getFirstItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[0].ID)
  492. if getFirstItemsErr != nil {
  493. utils.ErrorLog("获取第一周模板Item失败:%v", getFirstItemsErr)
  494. return
  495. }
  496. secItems, getSecItemsErr := GetOrgPatientScheduleTemplateItemsWithoutPatientByTemplateID(mode.OrgID, templates[1].ID)
  497. if getSecItemsErr != nil {
  498. utils.ErrorLog("获取第一周模板Item失败:%v", getSecItemsErr)
  499. return
  500. }
  501. if mode.ExecuteTimes%2 == 0 {
  502. insertErr, _ := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, now.AddDate(0, 0, 7), firstItems, deviceNumberMap)
  503. if insertErr != nil {
  504. utils.ErrorLog("使用第一周模板插入下第一周排班失败:%v", insertErr)
  505. tx.Rollback()
  506. return
  507. }
  508. insertErr, _ = _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, now.AddDate(0, 0, 14), secItems, deviceNumberMap)
  509. if insertErr != nil {
  510. utils.ErrorLog("使用第二周模板插入下第二周排班失败:%v", insertErr)
  511. tx.Rollback()
  512. return
  513. }
  514. } else { // mode.ExecuteTimes%2 == 1
  515. insertErr, _ := _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, now.AddDate(0, 0, 7), secItems, deviceNumberMap)
  516. if insertErr != nil {
  517. utils.ErrorLog("使用第二周模板插入下第一周排班失败:%v", mode.OrgID, insertErr)
  518. tx.Rollback()
  519. return
  520. }
  521. insertErr, _ = _refreshWeekSchedulesWithTemplateItems(tx, mode.OrgID, now.AddDate(0, 0, 14), firstItems, deviceNumberMap)
  522. if insertErr != nil {
  523. utils.ErrorLog("使用第一周模板插入下第二周排班失败:%v", mode.OrgID, insertErr)
  524. tx.Rollback()
  525. return
  526. }
  527. }
  528. }
  529. mode.ExecuteTimes += 2
  530. mode.ModifyTime = now.Unix()
  531. updateModeErr := tx.Save(mode).Error
  532. if updateModeErr != nil {
  533. utils.ErrorLog("更新启用模板模式的执行次数失败:%v", mode.OrgID, updateModeErr)
  534. tx.Rollback()
  535. return
  536. }
  537. tx.Commit()
  538. utils.SuccessLog("模拟自动更新下两周排班成功")
  539. }