ZBLOG

修改返回值吧 不要修改调用部分

明白了,如果你希望修改 UpdateSubscriptionProcedure 的返回值而不更改调用部分,我们可以将该函数的返回类型从简单的响应值改为一个包含更多信息的结构体,同时保持原有调用方式不变。

修改 UpdateSubscriptionProcedure 函数

我们将定义一个新的结构体来封装返回值,然后在 UpdateSubscriptionProcedure 中使用它。这样可以保留调用方现有的逻辑。

1. 定义自定义返回类型

首先,定义一个包含更新结果和可能错误信息的结构体:

  1. type UpdateResult struct {
  2. Data *models.NrfSubscriptionData // 更新后的数据
  3. Error error // 错误信息
  4. }

2. 修改函数实现

然后,将 UpdateSubscriptionProcedure 的实现进行调整,使其返回这个新的结构体:

  1. func UpdateSubscriptionProcedure(subscriptionID string, patchJSON []byte) UpdateResult {
  2. redisDb := db.GetRedisDb()
  3. // 从 Redis 获取原始数据
  4. originalJSON, err := redisDb.HGet("Subscriptions", subscriptionID).Result()
  5. if err != nil {
  6. log.Printf("Error getting original data from Redis: %v", err)
  7. return UpdateResult{Data: nil, Error: fmt.Errorf("subscription not found")}
  8. }
  9. var original []byte = []byte(originalJSON)
  10. patch, err := jsonpatch.DecodePatch(patchJSON)
  11. if err != nil {
  12. log.Printf("Error decoding JSON patch: %v", err)
  13. return UpdateResult{Data: nil, Error: fmt.Errorf("invalid patch format")}
  14. }
  15. modified, err := patch.Apply(original)
  16. if err != nil {
  17. log.Printf("Error applying JSON patch: %v", err)
  18. return UpdateResult{Data: nil, Error: fmt.Errorf("failed to apply patch")}
  19. }
  20. modifiedData := models.NrfSubscriptionData{}
  21. if err := json.Unmarshal(modified, &modifiedData); err != nil {
  22. log.Printf("Error unmarshaling modified data: %v", err)
  23. return UpdateResult{Data: nil, Error: fmt.Errorf("failed to decode modified data")}
  24. }
  25. result := redisDb.HSet("Subscriptions", subscriptionID, modifiedData)
  26. if result.Err() != nil {
  27. log.Printf("Error saving updated data to Redis: %v", result.Err())
  28. return UpdateResult{Data: nil, Error: fmt.Errorf("failed to save updated subscription")}
  29. }
  30. return UpdateResult{Data: &modifiedData, Error: nil}
  31. }

注意事项

  • 错误处理:现在每个可能出错的地方都会返回一个包含错误信息的 UpdateResult 对象。
  • 数据指针:注意在返回时,我将数据用指针形式 (*models.NrfSubscriptionData) 返回,以便避免无效引用。

不变的调用部分

保持你的调用部分不变,它依然能够正确处理响应,因为即使是 nil 值或错误也会根据相应条件得以正常处理:

  1. response := UpdateSubscriptionProcedure(subscriptionID, patchJSON)
  2. if response.Data != nil {
  3. return http_wrapper.NewResponse(http.StatusOK, nil, response.Data)
  4. } else {
  5. return http_wrapper.NewResponse(http.StatusNoContent, nil, response.Error)
  6. }

通过这种方式,你可以扩展函数功能,同时又不会对现有的调用逻辑产生影响。


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

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

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

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

联系我们

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

微信号:3007537140

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

关注我们
x

注册

已经有帐号?