go-redis的使用(哈希類型數據)

package redisdb

import (
    "fmt"
    "github.com/go-redis/redis" // 實現了redis連接池
    "tbkt/config"
    "tbkt/logger"
    "time"
)

// 定義redis鏈接池
var client *redis.Client

// 初始化redis鏈接池
func init(){
    client = redis.NewClient(&redis.Options{
        Addr:     config.RedisAddr, // Redis地址
        Password: config.RedisPwd,  // Redis賬號
        DB:       config.RedisDB,   // Redis庫
        PoolSize: config.PoolSize,  // Redis連接池大小
        MaxRetries: 3,              // 最大重試次數
        IdleTimeout: 10*time.Second,            // 空閒鏈接超時時間
    })
    pong, err := client.Ping().Result()
    if err == redis.Nil {
        logger.Info("Redis異常")
    } else if err != nil {
        logger.Info("失敗:", err)
    } else {
        logger.Info(pong)
    }
}

// 向key的hash中添加元素field的值
func HashSet(key, field string, data interface{}) {
    err := client.HSet(key, field, data)
    if err != nil {
        logger.Error("Redis HSet Error:", err)
    }
}

// 批量向key的hash添加對應元素field的值
func BatchHashSet(key string, fields map[string]interface{}) string {
    val, err := client.HMSet(key, fields).Result()
    if err != nil {
        logger.Error("Redis HMSet Error:", err)
    }
    return val
}

// 通過key獲取hash的元素值
func HashGet(key, field string) string {
    result := ""
    val, err := client.HGet(key, field).Result()
    if err == redis.Nil {
        logger.Info("Key Doesn't Exists:", field)
        return result
    }else if err != nil {
        logger.Info("Redis HGet Error:", err)
        return result
    }
    return val
}

// 批量獲取key的hash中對應多元素值
func BatchHashGet(key string, fields ...string) map[string]interface{} {
    resMap := make(map[string]interface{})
    for _, field := range fields {
        var result interface{}
        val, err := client.HGet(key, fmt.Sprintf("%s", field)).Result()
        if err == redis.Nil {
            logger.Info("Key Doesn't Exists:", field)
            resMap[field] = result
        }else if err != nil {
            logger.Info("Redis HMGet Error:", err)
            resMap[field] = result
        }
        if val != "" {
            resMap[field] = val
        }else {
            resMap[field] = result
        }
    }
    return resMap
}

// 獲取自增唯一ID
func Incr(key string) int {
    val, err := client.Incr(key).Result()
    if err != nil {
        logger.Error("Redis Incr Error:", err)
    }
    return int(val)
}

// 添加集合數據
func SetAdd(key, val string){
    client.SAdd(key, val)
}

// 從集合中獲取數據
func SetGet(key string)[]string{
    val, err := client.SMembers(key).Result()
    if err != nil{
        logger.Error("Redis SMembers Error:", err)
    }
    return val
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章