Redis 统计实时在线人数

zset

Redis 中的 sorted set (有序集合) 也称为 zset. 它提供了两个参数, 一个为 score, 一个为 member.

其中, score 为排序的分数, 它可以是双精度或者整数. 其结构类似

          | (score) member1

key => | (score) member2

          | (score) member3

如何实现

我们可以简单的通过 zadd 方法向一个有序集合内添加成员.

为了方便演示, 在这里使用 Interval 的方式, 定时请求指定接口. 将 score 设置为请求时的 unix时间戳

zset 提供了一个 zcount 方法, 可以让我们快速获取指定 socre 范围内的成员数量.

我们假设用户5秒钟内无操作, 即掉线. 这样便可以通过 zcount 获取实时在线人数

func main() {
    // 使用 gin 提供 web 服务
    r := gin.Default()

    r.GET("/", func(c *gin.Context) {
        _redis := redisutil.R.Get() // redis pool
        defer redis.Close()

        // 用户标识
        userId := c.ClientIp()
        _, _ = _redis.Do("ZADD", "online_user", time.Now().Unix(), userId)

        // 获取在线人数
        count, _ := redis.Int(_redis.Do("ZCOUNT", "online_user", time.Now().Unix()-5, "+inf"))

        c.JSON(200, gin.H{
            "count": count,
        })
    })

    r.Run()
}

当然, 为了防止 数据一直增加, 对于 5秒前的数据, 应该定时将其移出.

我们可以通过 zremrangebyscore 方法, 来移除 指定范围内的数据.

func AutoRemover() {
    _redis := redisutil.R.Get() // redis pool
    defer redis.Close()

    c := Cron.New()

    c.AddFunc("@every 5s", func() {
        _, _ = _redis.Do("ZREMRANGEBYSCORE", "online_user", "-inf", time.Now().Unix()-5)
    })
    c.Start()
}

zset 的其他方法

除了上述提到的方法, zset 还提供了很多其他的方法, 如:

  • zcard key 直接获取 指定 key 内的成员数量
  • zscore key member 获取指定成员的分数
  • zincrby key incr member 增加指定成员的分数
  • zrange key start end [WITHSCORES] 获取指定范围内的成员 并按照分数从小到大排序
  • zrevrange key start end [WITHSCORES] 与上一个相反
  • etc.

其他

当然, 也可以通过 WebSocket 来实现类似的功能, 这样会更加准确, 但是可能会带来一定的性能损失.

© 版权声明
THE END
喜欢就支持一下吧
点赞15 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容