1. 程式人生 > 資料庫 >使用go操作redis的有序集合(zset)

使用go操作redis的有序集合(zset)

我就廢話不多說了,大家還是直接看程式碼吧~

package main
import (
 "fmt"
 "github.com/garyburd/redigo/redis"
)
func main() {
 // 連線redis資料庫,指定資料庫的IP和埠
 conn,err := redis.Dial("tcp","36.99.16.197:6379")
 if err != nil {
 fmt.Println("Connect to redis error",err)
 return
 } else {
 fmt.Println("Connect to redis ok.")
 }
 // 函式退出時關閉連線
 defer conn.Close()
 // 執行一個有序zset插入
 _,err = conn.Do("ZADD","mykey","INCR",1,"robot1")
 if err != nil {
 fmt.Println("redis set failed:",err)
 }
 // 再執行一個有序zset插入
 _,"robot2")
 if err != nil {
 fmt.Println("redis set failed:",err)
 }
 // 讀取指定zset
 user_map,err := redis.StringMap(conn.Do("ZRANGE",10,"withscores"))
 if err != nil {
 fmt.Println("redis get failed:",err)
 } else {
 fmt.Printf("Get mykey: %v \n",user_map)
 }
 for user := range user_map {
 fmt.Printf("user name: %v %v\n",user,user_map[user])
 }
}

輸出:

Connect to redis ok.
Get mykey: map[robot1:1 robot2:1]
user name: robot1 1
user name: robot2 1

補充:Redis中zset的golang實現

zset

初衷是在不用Redis的前提下實現排行榜

專案地址https://github.com/liyiheng/zset

Installation

go get -u github.com/liyiheng/zset

Usage

Removed RWLock in the SortedSet.

Just implement it yourself if you need.

s := zset.New()
// add data
s.Set(66,1001,"test1")
s.Set(77,1002,"test2")
s.Set(88,1003,"test3")
s.Set(100,1004,"liyiheng")
s.Set(99,1005,"test4")
s.Set(44,1006,"test5")
// update data
s.Set(44,"test1")
// get rank by id
rank,score,extra := s.GetRank(1004,false)
// get data by rank
id,extra := s.GetDataByRank(0,true)
// get data by id
dat,ok := s.GetData(1001)
// delete data by id
s.Delete(1001)

Benchmark

go test -test.bench=".*"
BenchmarkSortedSet_Add-4   1000000  4121 ns/op
BenchmarkSortedSet_GetRank-4  500000  3592 ns/op
BenchmarkSortedSet_GetDataByRank-4 2000000  667 ns/op
PASS
ok zset 11.365s

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援我們。如有錯誤或未考慮完全的地方,望不吝賜教。