基于GO的即时聊天后端项目关系模块之好友模块设计与实现

package main
 
import (
    "fmt"
    "github.com/go-redis/redis"
    "github.com/satori/go.uuid"
)
 
// 假设已经有了Redis客户端和用户模型的定义
var redisClient *redis.Client
 
// 用户模型
type User struct {
    ID       string
    Username string
}
 
// 好友模块的数据结构
type FriendService struct {
    client *redis.Client
}
 
// 创建好友服务的实例
func NewFriendService(client *redis.Client) *FriendService {
    return &FriendService{
        client: client,
    }
}
 
// 添加好友
func (s *FriendService) AddFriend(userId string, friendId string) error {
    // 将好友ID添加到用户的好友集合中
    key := fmt.Sprintf("user:%s:friends", userId)
    return s.client.SAdd(key, friendId).Err()
}
 
// 移除好友
func (s *FriendService) RemoveFriend(userId string, friendId string) error {
    // 从用户的好友集合中移除好友ID
    key := fmt.Sprintf("user:%s:friends", userId)
    return s.client.SRem(key, friendId).Err()
}
 
// 获取用户的所有好友ID
func (s *FriendService) GetFriends(userId string) ([]string, error) {
    // 获取用户的好友集合
    key := fmt.Sprintf("user:%s:friends", userId)
    return s.client.SMembers(key).Result()
}
 
func main() {
    // 假设已经初始化了redisClient
    redisClient = redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "", // 无密码
        DB:       0,  // 默认数据库
    })
 
    // 创建好友服务实例
    friendService := NewFriendService(redisClient)
 
    // 示例用户
    user1Id := uuid.NewV4().String()
    user2Id := uuid.NewV4().String()
 
    // 添加好友
    err := friendService.AddFriend(user1Id, user2Id)
    if err != nil {
        panic(err)
    }
 
    // 获取好友列表
    friends, err := friendService.GetFriends(user1Id)
    if err != nil {
        panic(err)
    }
    fmt.Println("Friends:", friends)
 
    // 移除好友
    err = friendService.RemoveFriend(user1Id, user2Id)
    if err != nil {
        panic(err)
    }
 
    // 获取好友列表
    friends, err = friendService.GetFriends(user1Id)
    if err != nil {
        panic(err)
    }
    fmt.Println("Friends after removal:", friends)
}

这个代码实例演示了如何使用Go语言和Redis来实现一个简单的好友模块。代码中定义了一个FriendService结构体，它有添加好友、移除好友和获取好友列表的方法。这些方法使用Redis的集合（Set）数据结构来存储用户的好友信息。在main函数中，我们创建了FriendService的实例，并演示了如何添加、获取和移除好友。这个例子提供了一个清晰的教学示例，展示了如何将Redis用作存储解决方案，并且如何通过Go代码与之交互。