看起来你在使用 go-redis
库来从 Redis 数据库中获取数据。如果你的目标是从 Redis 中获取存储的 JSON 数据,并且你希望对该部分代码进行优化或修改,我可以为你提供帮助。
更新代码示例
以下是如何从 Redis 中获取 hashKey
下的 "data"
字段,并处理可能出现的错误:
package main
import (
"context"
"encoding/json"
"fmt"
"github.com/go-redis/redis/v8" // 确保导入 go-redis 包
)
var ctx = context.Background()
type Subscription struct {
NfStatusNotificationUri string `json:"nfStatusNotificationUri"`
SubscriptionId string `json:"subscriptionId"`
ReqNfType string `json:"reqNfType"`
}
type UpdateResponse struct {
StatusCode int `json:"status_code"`
Data interface{} `json:"data,omitempty"` // 使用 interface{} 可以灵活处理不同数据类型
}
// 更新订阅信息的函数
func UpdateSubscriptionProcedure(redisClient *redis.Client, subscriptionID string, patchJSON []byte) UpdateResponse {
// 定义哈希键名称
hashKey := fmt.Sprintf("Subscriptions:%s", subscriptionID)
// 更新订阅信息
if err := redisClient.HSet(ctx, hashKey, "data", patchJSON).Err(); err != nil {
return UpdateResponse{StatusCode: 500} // 返回错误状态码
}
// 获取更新后的文档
result, err := redisClient.HGet(ctx, hashKey, "data").Result()
// 错误处理,如果未找到该字段则返回404状态码。
if err == redis.Nil {
return UpdateResponse{StatusCode: 404} // 找不到资源返回404状态码
} else if err != nil {
return UpdateResponse{StatusCode: 500} // 返回其他错误时,返回500状态码
}
var subscription Subscription
// 将结果解析为 Subscription 对象
if err := json.Unmarshal([]byte(result), &subscription); err != nil {
return UpdateResponse{StatusCode: 500} // JSON 解码失败,返回500状态码
}
// 返回成功响应及更新后的数据
return UpdateResponse{
StatusCode: 200,
Data: subscription,
}
}
func main() {
redisClient := redis.NewClient(&redis.Options{
Addr: "localhost:6379", // 使用适当的地址和端口连接到你的Redis服务器
})
subscriptionID := "68"
patchData := Subscription{
NfStatusNotificationUri: "https://example.com/callback",
SubscriptionId: subscriptionID,
ReqNfType: "AMF",
}
patchJSON, _ := json.Marshal(patchData)
response := UpdateSubscriptionProcedure(redisClient, subscriptionID, patchJSON)
if response.StatusCode != 200 {
fmt.Println("Error updating subscription:", response.StatusCode)
return
}
fmt.Printf("Updated Subscription Response: %+v\n", response.Data)
}
功能说明:
HGet
方法:现在包含了更详细的错误处理。我们检查是否存在值 (redis.Nil
) 和其他潜在错误。- 404 状态:如果尝试访问不存在的数据,将返回一个 404 状态代码。
- JSON 解码:如果无法解码 JSON,则会返回一个 500 状态代码。
这种方式使得代码在执行过程中更加稳健,对各种情况都有清晰的应对策略。请根据你的实际需求进行相应调整!如有进一步问题,请随时问我!
内容由零声教学AI助手提供,问题来源于学员提问