From 0793b2ab383c8d8afa2ab77e34a148ab6e3e1b77 Mon Sep 17 00:00:00 2001
From: wangchuxiao <wangchuxiao97@outlook.com>
Date: Thu, 28 Apr 2022 16:29:40 +0800
Subject: [PATCH] cache

---
 pkg/common/db/redisModel.go | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/pkg/common/db/redisModel.go b/pkg/common/db/redisModel.go
index 8b3cdb4c7..c6defb8bd 100644
--- a/pkg/common/db/redisModel.go
+++ b/pkg/common/db/redisModel.go
@@ -6,7 +6,6 @@ import (
 	pbCommon "Open_IM/pkg/proto/sdk_ws"
 	"encoding/json"
 	"github.com/garyburd/redigo/redis"
-	"strings"
 )
 
 const (
@@ -193,8 +192,12 @@ func (d *DataBases) ReduceFriendToCache(userID string, friendIDList ...interface
 }
 
 func (d *DataBases) GetFriendIDListFromCache(userID string) ([]string, error) {
-	result, err := redis.String(d.Exec("SMEMBERS", friendRelationCache+userID))
-	return strings.Split(result, " "), err
+	result, err := redis.Values(d.Exec("SMEMBERS", friendRelationCache+userID))
+	var userIDList []string
+	for _, v := range result {
+		userIDList = append(userIDList, v.(string))
+	}
+	return userIDList, err
 }
 
 func (d *DataBases) AddBlackUserToCache(userID string, blackList ...interface{}) error {
@@ -208,6 +211,10 @@ func (d *DataBases) ReduceBlackUserFromCache(userID string, blackList ...interfa
 }
 
 func (d *DataBases) GetBlackListFromCache(userID string) ([]string, error) {
-	result, err := redis.String(d.Exec("SMEMBERS", blackListCache+userID))
-	return strings.Split(result, " "), err
+	result, err := redis.Values(d.Exec("SMEMBERS", blackListCache+userID))
+	var userIDList []string
+	for _, v := range result {
+		userIDList = append(userIDList, v.(string))
+	}
+	return userIDList, err
 }