Golang操作数据库Redis

Posted 2021-04-07 Go键盘侠

在项目开发中redis的使用也比较频繁，本文介绍了Go语言中go-redis库的基本使用。

Redis介绍

Redis是一个开源的内存数据库，Redis提供了多种不同类型的数据结构，很多业务场景下的问题都可以很自然地映射到这些数据结构上。除此之外，通过复制、持久化和客户端分片等特性，我们可以很方便地将Redis扩展成一个能够包含数百GB数据、每秒处理上百万次请求的系统。

Redis支持的数据结构

Redis支持诸如字符串（strings）、哈希（hashes）、列表（lists）、集合（sets）、带范围查询的排序集合（sorted sets）、位图（bitmaps）、hyperloglogs、带半径查询和流的地理空间索引等数据结构（geospatial indexes）。

Redis应用场景

• 缓存系统，减轻主数据库（mysql）的压力。

• 计数场景，比如微博、抖音中的关注数和粉丝数。

• 热门排行榜，需要排序的场景特别适合使用ZSET。

• 利用LIST可以实现队列的功能。

准备Redis环境

这里直接使用Docker启动一个redis环境，方便学习使用。

docker启动一个名为redis507的5.0.7版本的redis server示例：

docker run --name redis507 -p 6379:6379 -d redis:5.0.7

注意：此处的版本、容器名和端口号请根据自己需要设置。

启动一个redis-cli连接上面的redis server：

docker run -it --network host --rm redis:5.0.7 redis-cli

go-redis库

安装

区别于另一个比较常用的Go语言redis client库：redigo，我们这里采用https://github.com/go-redis/redis 连接Redis数据库并进行操作，因为go-redis支持连接哨兵及集群模式的Redis。

使用以下命令下载并安装：

go get -u github.com/go-redis/redis

连接

普通连接

package main
import "github.com/go-redis/redis"
// 声明一个全局的rdb变量var rdb *redis.Client
// 初始化连接func initClient() (err error) { rdb = redis.NewClient(&redis.Options{ Addr: "localhost:6379", Password: "", // no password set DB: 0, // use default DB })
 _, err = rdb.Ping().Result() if err != nil { return err } return nil}
func main() { err := initClient() if err != nil { panic(err) }}

V8新版本相关

最新版本的go-redis库的相关命令都需要传递context.Context参数，例如：

package main
import ( "context" "fmt" "github.com/go-redis/redis" "time"
 "github.com/go-redis/redis/v8" // 注意导入的是新版本)
var ( rdb *redis.Client)
// 初始化连接func initClient() (err error) { rdb = redis.NewClient(&redis.Options{ Addr: "localhost:16379", Password: "", // no password set DB: 0, // use default DB PoolSize: 100, // 连接池大小 })
 ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel()
 _, err = rdb.Ping(ctx).Result() return err}
func V8Example() { ctx := context.Background() if err := initClient(); err != nil { return }
 err := rdb.Set(ctx, "key", "value", 0).Err() if err != nil { panic(err) }
 val, err := rdb.Get(ctx, "key").Result() if err != nil { panic(err) } fmt.Println("key", val)
 val2, err := rdb.Get(ctx, "key2").Result() if err == redis.Nil { fmt.Println("key2 does not exist") } else if err != nil { panic(err) } else { fmt.Println("key2", val2) } // Output: key value // key2 does not exist}

连接Redis哨兵模式

func initClient()(err error){ rdb := redis.NewFailoverClient(&redis.FailoverOptions{ MasterName: "master", SentinelAddrs: []string{"x.x.x.x:26379", "xx.xx.xx.xx:26379", "xxx.xxx.xxx.xxx:26379"}, }) _, err = rdb.Ping().Result() if err != nil { return err } return nil}

连接Redis集群

func initClient()(err error){ rdb := redis.NewClusterClient(&redis.ClusterOptions{ Addrs: []string{":7000", ":7001", ":7002", ":7003", ":7004", ":7005"}, }) _, err = rdb.Ping().Result() if err != nil { return err } return nil}

基本使用

set/get示例

func redisExample() { err := rdb.Set("score", 100, 0).Err() if err != nil { fmt.Printf("set score failed, err:%v\n", err) return }
 val, err := rdb.Get("score").Result() if err != nil { fmt.Printf("get score failed, err:%v\n", err) return } fmt.Println("score", val)
 val2, err := rdb.Get("name").Result() if err == redis.Nil { fmt.Println("name does not exist") } else if err != nil { fmt.Printf("get name failed, err:%v\n", err) return } else { fmt.Println("name", val2) }}

zset示例

func redisExample2() { zsetKey := "language_rank" languages := []redis.Z{ redis.Z{Score: 90.0, Member: "Golang"}, redis.Z{Score: 98.0, Member: "Java"}, redis.Z{Score: 95.0, Member: "Python"}, redis.Z{Score: 97.0, Member: "javascript"}, redis.Z{Score: 99.0, Member: "C/C++"}, } // ZADD num, err := rdb.ZAdd(zsetKey, languages...).Result() if err != nil { fmt.Printf("zadd failed, err:%v\n", err) return } fmt.Printf("zadd %d succ.\n", num)
 // 把Golang的分数加10 newScore, err := rdb.ZIncrBy(zsetKey, 10.0, "Golang").Result() if err != nil { fmt.Printf("zincrby failed, err:%v\n", err) return } fmt.Printf("Golang's score is %f now.\n", newScore)
 // 取分数最高的3个 ret, err := rdb.ZRevRangeWithScores(zsetKey, 0, 2).Result() if err != nil { fmt.Printf("zrevrange failed, err:%v\n", err) return } for _, z := range ret { fmt.Println(z.Member, z.Score) }
 // 取95~100分的 op := redis.ZRangeBy{ Min: "95", Max: "100", } ret, err = rdb.ZRangeByScoreWithScores(zsetKey, op).Result() if err != nil { fmt.Printf("zrangebyscore failed, err:%v\n", err) return } for _, z := range ret { fmt.Println(z.Member, z.Score) }}

输出结果如下：

$ ./06redis_demo zadd 0 succ.Golang's score is 100.000000 now.Golang 100C/C++ 99Java 98JavaScript 97Java 98C/C++ 99Golang 100

根据前缀获取Key

vals, err := rdb.Keys(ctx, "prefix*").Result()

执行自定义命令

res, err := rdb.Do(ctx, "set", "key", "value").Result()

按通配符删除key

当通配符匹配的key的数量不多时，可以使用Keys()得到所有的key在使用Del命令删除。如果key的数量非常多的时候，我们可以搭配使用Scan命令和Del命令完成删除。

func exampleTongPeiFu() { ctx := context.Background() iter := rdb.Scan(ctx, 0, "prefix*", 0).Iterator() for iter.Next(ctx) { err := rdb.Del(ctx, iter.Val()).Err() if err != nil { panic(err) } } if err := iter.Err(); err != nil { panic(err) }}

Pipeline

Pipeline 主要是一种网络优化。它本质上意味着客户端缓冲一堆命令并一次性将它们发送到服务器。这些命令不能保证在事务中执行。这样做的好处是节省了每个命令的网络往返时间（RTT）。

Pipeline 基本示例如下：

pipe := rdb.Pipeline()
incr := pipe.Incr("pipeline_counter")pipe.Expire("pipeline_counter", time.Hour)
_, err := pipe.Exec()fmt.Println(incr.Val(), err)

上面的代码相当于将以下两个命令一次发给redis server端执行，与不使用Pipeline相比能减少一次RTT。

INCR pipeline_counterEXPIRE pipeline_counts 3600

也可以使用Pipelined：

var incr *redis.IntCmd_, err := rdb.Pipelined(func(pipe redis.Pipeliner) error { incr = pipe.Incr("pipelined_counter") pipe.Expire("pipelined_counter", time.Hour) return nil})fmt.Println(incr.Val(), err)

在某些场景下，当我们有多条命令要执行时，就可以考虑使用pipeline来优化。

https://github.com/Scoefield/gokeyboardman/tree/main/goredis

以上是关于Go语言操作Redis的介绍，感谢您的阅读，如有疑问或意见请及时反馈给我们。

---

上一篇文章：