tcjlj/V2bX
1
0
Fork 0
mirror of https://github.com/wyx2685/V2bX.git synced 2025-02-02 06:48:14 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimization redis recorder

Browse Source
This commit is contained in:
yuzuki999 2022-11-02 14:54:24 +08:00
parent e76ec30c92
commit b946283696
2 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/iprecoder/redis.go
View File

@ -29,19 +29,17 @@ func NewRedis(c *conf.RedisConfig) *Redis {
func (r *Redis) SyncOnlineIp(Ips []dispatcher.UserIpList) ([]dispatcher.UserIpList, error) { func (r *Redis) SyncOnlineIp(Ips []dispatcher.UserIpList) ([]dispatcher.UserIpList, error) {
ctx := context.Background() ctx := context.Background()
for i := range Ips { for i := range Ips {
if !r.client.SIsMember(ctx, "UserList", strconv.Itoa(Ips[i].Uid)).Val() { err := r.client.SAdd(ctx, "UserList", Ips[i].Uid).Err()
err := r.client.SAdd(ctx, "UserList", Ips[i].Uid).Err() if err != nil {
if err != nil { return nil, fmt.Errorf("add user failed: %s", err)
return nil, fmt.Errorf("add user failed: %s", err)
}
} }
r.client.Expire(ctx, "UserList", time.Duration(2)*time.Minute) r.client.Expire(ctx, "UserList", 2*time.Minute)
for _, ip := range Ips[i].IpList { for _, ip := range Ips[i].IpList {
err := r.client.SAdd(ctx, strconv.Itoa(Ips[i].Uid), ip).Err() err := r.client.SAdd(ctx, strconv.Itoa(Ips[i].Uid), ip).Err()
if err != nil { if err != nil {
return nil, fmt.Errorf("add ip failed: %s", err) return nil, fmt.Errorf("add ip failed: %s", err)
} }
r.client.Expire(ctx, strconv.Itoa(Ips[i].Uid), time.Duration(2)*time.Minute) r.client.Expire(ctx, strconv.Itoa(Ips[i].Uid), 2*time.Minute)
} }
} }
c := r.client.SMembers(ctx, "UserList") c := r.client.SMembers(ctx, "UserList")

2
api/iprecoder/redis_test.go
View File

@ -14,7 +14,7 @@ func TestRedis_SyncOnlineIp(t *testing.T) {
Db: 0, Db: 0,
}) })
users, err := r.SyncOnlineIp([]dispatcher.UserIpList{ users, err := r.SyncOnlineIp([]dispatcher.UserIpList{
{2, []string{"3.3.3.3", "4.4.4.4"}}, {1, []string{"5.5.5.5", "4.4.4.4"}},
}) })
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)