go使用redis位运算做用户点赞统计

使用redis的位运算,统计点赞数.
使用"0"字符作为初始化的二进制字符, 二进制为: 00110000
使用redis的命令:
set: set key value
setbit: setbit key offset value(0或1)
getbit: getbit key offset(下标)
bitcount: bitcount key [start, end](start:为开始字符下标, end:为字符的最大长度减1)
使用实例: <a href="https://github.com/yunziyuan/cntech-go" traget='_blank'>https://github.com/yunziyuan/cntech-go</a>

初始化redis连接

  var RedisConn *redis.Client
  var ctx = context.Background()
  func InitRedis()  {
    RedisConn = redis.NewClient(&redis.Options{
        Addr:"127.0.0.1:7109",
        Password:"",
        DB:0,
    })
    //defer RedisConn.Close()
  }

初始化存储

  // 初始化点赞数据 key: 存在业务 total: 存储总数量
  // 使用0字符串的二进制作为二进制
  func initClick(key string, total int64) (bool,error) {
    // 拼接字符长度
    if total <= 0 {
        return false, errors.New("总长度不正确")
    }
    str := "0"
    var l int
    if total > 8 {
        l = int(math.Ceil(float64(total) / 8.00))
        for i := 1; i < l; i++ {
            str +="0"
        }
    }
    // 计算需要置0的位置
    var sitZeros []int64
    for i := 0; i < l * 8; i++ {
        fSit := int64(2 + 8 * i)
        if fSit >= int64(l * 8) {
            break
        }
        sits := []int64{
            fSit, fSit + 1,
        }
        sitZeros = append(sitZeros, sits...)
    }
    // 字符:      0000
    // 对应二进制: 00110000 00110000 00110000 00110000
    // 对应下标:   01234567 8.....15 16....23 24....31
    err := RedisConn.Set(ctx, key, str,0).Err()
    if err != nil {
        fmt.Println("设置失败:", err)
    }
    // 将对应1的位置修改为0
    // 需要置0的位置 2,3,10,11,18,19,26,27
    for _,v := range sitZeros {
        err = RedisConn.SetBit(ctx, key, v, 0).Err()
        if err != nil {
            fmt.Println("置0失败:", err)
            return false, err
        }
    }
    return true, nil
  }

按下标进行点赞

  // 获取点赞数量 使用32个二进制位表示
  func clickNum(key string) (int64,error) {
    // 统计所有的位置的1的数量
    // 字符:        0000
    // 对应位置下标: 0123
    l,err := RedisConn.StrLen(ctx, key).Result()
    if err != nil {
        return 0, err
    }
    sit := redis.BitCount{Start: 0,End: l-1}
    num, err := RedisConn.BitCount(ctx, key, &sit).Result()
    if err != nil {
        return 0, err
    }
    return num, nil
  }

统计点赞数

  // 点赞设置 sit:点赞下标  value 设置值 只能是 0或者1
  func clickSet(key string, sit int64, value int) (bool,error) {
    err:= RedisConn.SetBit(ctx, key, sit, value).Err()
    if err != nil {
        fmt.Println("置0失败:", err)
        return false, err
    }
    return true, nil
  }

使用实例

  func main()  {
    InitRedis()
    // 初始化点赞
    key := "click2"
    //bol, err := initClick(key, 33)
    //if !bol || err != nil {
    //  log.Fatal("初始化数据失败:",err)
    //}
  
    // 点赞
    bol, err := clickSet(key,87, 1)
    if !bol || err != nil {
        log.Fatal("点赞失败:",err)
    }
    fmt.Println("点赞成功")
  
    // 统计点赞
    num, err := clickNum(key)
    if err != nil {
        log.Fatal("点赞获取失败:", err)
    }
    fmt.Println("点赞人数:", num)
  }