ZBLOG

注册 原创投稿

func GetUserHealthDiary(userId int) ai_common.UserHealthDiary { resp := ai_common.UserHealthDiary{} date := time.Now().Format("2026-01-02") dbData, isExist := dbrespository.GetUserHealthDiaryByUserId(userId, date) if !isExist { // create tes...

作者：golang云原生栏目：公开2025-03-29 10:4637

你可以使用 Go 的协程（goroutine）和通道（channel）来优化 GetUserHealthDiary 函数，以实现并发处理。下面是一个重构后的版本，该版本将一些耗时的调用放入协程中，从而提升性能。

  1. func GetUserHealthDiary(userId int) ai_common.UserHealthDiary {
  2. 	resp := ai_common.UserHealthDiary{}
  3. 	date := time.Now().Format("2006-01-02") // 使用正确的日期格式
  4. 	dbData, isExist := dbrespository.GetUserHealthDiaryByUserId(userId, date)
  5. 	if !isExist {
  6. 		// create test data
  7. 		dbData.UserID = userId
  8. 		resp.RecordDate = dbData.RecordDate // 记录健康日记的日期，默认是当天
  9. 		dbrespository.CreateUserHealthDiary(&dbData)
  10. 	}
  11. 	resp.RecordDate = dbData.RecordDate // 记录健康日记的日期，默认是当天
  13. 	var wg sync.WaitGroup
  14. 	var mu sync.Mutex // 用于保护对 resp 的写操作
  16. 	wg.Add(1)
  17. 	go func() {
  18. 		defer wg.Done()
  19. 		resp.HappinessIndex = getHappinessIndex(dbData.HappinessIndexID)
  20. 		mu.Lock()
  21. 		dbData.HappinessIndexID = resp.HappinessIndex.ID
  22. 		mu.Unlock()
  23. 	}()
  25. 	wg.Add(1)
  26. 	go func() {
  27. 		defer wg.Done()
  28. 		resp.PsychologicalPressure = getPsychologicalPressure(dbData.PsychologicalPressureID)
  29.         mu.Lock()
  30.         dbData.PsychologicalPressureID = resp.PsychologicalPressure.ID 
  31.         mu.Unlock()
  32.     }()
  33. 	
  34. 	wg.Add(1)
  35. 	go func() {
  36.         defer wg.Done()
  37.         resp.WorkStudyIntensity = getWorkStudyIntensity(dbData.WorkStudyIntensityID)
  38.         mu.Lock()
  39.         dbData.WorkStudyIntensityID = resp.WorkStudyIntensity.ID 
  40.         mu.Unlock()
  41.     }()
  43. 	wg.Add(1)
  44. 	go func() {
  45. 	    defer wg.Done()
  46. 	    resp.OvertimeSituation = getOvertimeSituation(dbData.OvertimeSituationID)
  47. 	    mu.Lock()
  48. 	    dbData.OvertimeSituationID = resp.OvertimeSituation.ID 
  49. 	    mu.Unlock()
  50.     }()
  52. 	wg.Add(1)
  53. 	go func() {
  54. 	    defer wg.Done()
  55. 	    resp.ElectronicDeviceTime = getElectronicDeviceTime(dbData.ElectronicDeviceTimeID)
  56. 	    mu.Lock()
  57. 	    dbData.ElectronicDeviceTimeID = resp.ElectronicDeviceTime.ID 
  58. 	    mu.Unlock()
  59.     }()
  61. 	wg.Add(1)
  62. 	go func() {
  63. 	    defer wg.Done()
  64. 	    resp.ExerciseIntensity = getExerciseIntensity(dbData.ExerciseIntensityID)
  65. 	    mu.Lock()
  66. 	    dbData.ExerciseIntensityID = resp.ExerciseIntensity.ID 
  67. 	    mu.Unlock()
  68.     }()
  70. 	for _, healthDiaryBadLiving := range dbData.HealthDiaryBadLiving {
  71.     	resp.BadLivingBehaviors = append(resp.BadLivingBehaviors, healthDiaryBadLiving.BadLiving)
  72.    }
  74.    if len(dbData.HealthDiaryBadLiving) == 0 {
  75.      // create test data
  76.    }
  78.    wg.Add(1)
  79.    go func() {
  80.        defer wg.Done()
  81.        resp.BadSocialEmotions = getBadSocialEmotions(dbData.BadSocialEmotionsID)  
  82.        mu.Lock() 
  83.        dbData.BadSocialEmotionsID = resp.BadSocialEmotions.ID 
  84.        mu.Unlock()     
  85.    }())
  87.    //其他数据项继续添加协程进行处理...
  88.    
  89.    // 在这里也可以添加所有其余需要通过函数获取的数据项
  90.    
  91.    // 等待所有协程完成
  92.    wg.Wait()
  94. 	return resp
  95. }

在这个例子中，我使用了 sync.WaitGroup 来等待所有协程完成，并使用 sync.Mutex 来确保对共享资源 resp 和 dbData 的安全访问。这种方式可以提高函数的性能，特别是在有很多独立的数据项需要获取时。

请注意，确保被调用的每个函数 (getHappinessIndex, getPsychologicalPressure, 等等) 是线程安全的，以避免竞态条件。如果它们不是线程安全的，你可能需要进一步改进或考虑如何更好地设计这些函数。


内容由零声教学AI助手提供，问题来源于学员提问

本站部分文章来源于网络，版权归原作者所有，如有侵权请联系站长删除。
转载请注明出处：https://golang.0voice.com/?id=22672

golang云原生 golang云原生 管理员
分享：
扫描分享到社交APP
上一篇
下一篇

相关推荐

发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论，快来说点什么吧~

golang云原生

golang云原生

  • 22721文章
  • 0评论
  • 2粉丝
  • 0被赞
最近发布
热门排行
热评文章
猜您喜欢
热门标签
Copyright ©2022    湖南零声信息科技有限公司    版权所有     湘ICP备18009867号-1          证照中心 

联系我们

在线咨询： 点击这里给我发消息

微信号：3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号？
简单注册快人一步