ItGo.me - 专注IT技术分享

首页 > Redis > golang sortedset 实现(基于redis skiplist)

golang sortedset 实现(基于redis skiplist)

时间:2016-08-28来源:网友分享 点击:

在游戏中排行榜是很常见的需求,之前一直使用的是redis的sortedset,刚好看到一个lua的实现,于是就移植到golang版本,github地址:,使用方法和redis 的cli命令基本一致。

     使用了cgo封装,skiplist的实现基本和redis的一致,虽然cgo用起来很方便,但涉及到C中回调go时比较费事,

感觉还不如lua来得清爽。

        go-zet并没有提供持久化的方法,我在中提供了持久化。主要是没有好的持久化方案,不像redis使用rdb格式。在cham中,和redis一样,fork后写文件,

只不过我使用的是json格式。

//use fork snapshot to savefunc (z *zset) BgStore(name string) error {f, err := helper.LockFile(name, true)if err != nil {return err}pid, errno := helper.Fork()if errno != 0 {return errors.New("fork error," + errno.Error())}//childif pid == 0 {buf := bufio.NewWriter(f)encoder := json.NewEncoder(buf)encoder.Encode(z.tbl)buf.Flush()f.Close()os.Exit(0)}return nil}

helper代码如下:

package helperimport ("os""runtime""syscall")func Fork() (int, syscall.Errno) {r1, r2, err := syscall.RawSyscall(syscall.SYS_FORK, 0, 0, 0)if err != 0 {return 0, err}if runtime.GOOS == "darwin" && r2 == 1 {r1 = 1}return int(r1), 0}// refererfunc LockFile(name string, truncate bool) (*os.File, error) {flag := os.O_RDWR | os.O_CREATEif truncate {flag |= os.O_TRUNC}f, err := os.OpenFile(name, flag, 0666)if err != nil {return nil, err}if err := syscall.Flock(int(f.Fd()), syscall.LOCK_EX|syscall.LOCK_NB); err != nil {f.Close()return nil, err}return f, nil}

看看test基本就知道用法了

package zsetimport ("fmt""testing""time")func equal(a []string, b []string) bool {if len(a) != len(b) {return false}for i := 0; i < len(a); i++ {if a[i] != b[i] {return false}}return true}func assert(t *testing.T, ok bool, s string) {if !ok {t.Error(s)}}func TestBase(t *testing.T) {z := New()assert(t, z.Count() == 0, "empty Count error")z.Add(1, "12")z.Add(1, "32")assert(t, z.Count() == 2, "not empty Count error")var score float64var ex boolscore, ex = z.Score("12")assert(t, score == 1, "Score error")z.Add(2, "12")assert(t, z.Count() == 2, "after add duplicate Count error")score, ex = z.Score("12")assert(t, score == 2, "after add Score error")z.Rem("12")assert(t, z.Count() == 1, "after rem Count error")score, ex = z.Score("12")assert(t, ex == false, "not exist Score error")fmt.Println("")}func TestRangeByScore(t *testing.T) {z := New()z.Add(2, "22")z.Add(1, "11")z.Add(3, "33")s := "TestRangeByScore error"assert(t, equal(z.RangeByScore(2, 3), []string{"22", "33"}), s)assert(t, equal(z.RangeByScore(0, 5), []string{"11", "22", "33"}), s)assert(t, equal(z.RangeByScore(10, 5), []string{}), s)assert(t, equal(z.RangeByScore(10, 0), []string{"33", "22", "11"}), s)}func TestRange(t *testing.T) {z := New()z.Add(100.1, "1")z.Add(100.9, "9")z.Add(100.5, "5")assert(t, equal(z.Range(1, 3), []string{"1", "5", "9"}), "Range1 error")assert(t, equal(z.Range(3, 1), []string{"9", "5", "1"}), "Range2 error")assert(t, equal(z.RevRange(1, 2), []string{"9", "5"}), "RevRange1 error")assert(t, equal(z.RevRange(3, 2), []string{"1", "5"}), "RevRange2 error")}func TestRank(t *testing.T) {z := New()assert(t, z.Rank("kehan") == 0, "Rank empty error")z.Add(1111.1111, "kehan")assert(t, z.Rank("kehan") == 1, "Rank error")z.Add(222.2222, "lwy")assert(t, z.Rank("kehan") == 2, "Rank 2 error")assert(t, z.RevRank("kehan") == 1, "RevRank error")}func TestLimit(t *testing.T) {z := New()z.Add(1, "1")z.Add(2, "2")z.Add(3, "3")z.Limit(1)assert(t, z.Count() == 1, "Limit error")assert(t, z.Rank("3") == 0, "Limit Rank error")z.Add(4.4, "4")z.Add(5.5, "5")z.Add(0.5, "0.5")z.Dump()assert(t, z.RevLimit(4) == 0, "RevLimit error")assert(t, z.RevLimit(0) == 4, "RevLimit2 error")}func TestStore(t *testing.T) {z := New()z.Add(1, "1")z.Add(2, "2")z.Add(3, "3")z.BgStore("zdb.json")go func() {time.Sleep(time.Second * 2)fmt.Println("continue add")z.Add(4, "4")z.Add(5, "5")}()}func TestReStore(t *testing.T) {z := New()assert(t, len(z.tbl) == 0, "restore error")z.ReStore("zdb.json")fmt.Println(z.tbl)assert(t, len(z.tbl) != 0, "restore2 error")assert(t, z.Count() == len(z.tbl), "restore3 error")}

golang sortedset 实现(基于redis skiplist)

golang sortedset 实现(基于redis skiplist)  讨论


redis集群方案

根据一些测试整理出来的一份方案: 1. Redis 性能 对于redis 的一些简单测试,仅供参考: 测试环境:Redhat6.2 , Xeon E5520(4核)*2/8G,1000M网卡 Redis 版本:2.6.9 客户端机器使用redis-benchmark 简单GET、...

Redis 到 WebSocket 中继

RedistowebsocketrelayRelaysaRedischanneltoawebsocket,ormultiplechannelstomultiplesockets.标签: 源码与演示: 代码片段 (2) 1. [文件] index.html~1KB(7) 跳至 htmlheadtitleRedis Listener/titlescript type="text/javascript" src=""/scriptscr...

Redis集群创建报错

Redis集群环境:och163/och164/och165 在执行如下脚本时报错: ./src/ create 10.1.253.163:6379 10.1.253.164:6379 10.1.253.165:6379 /usr/lib/ruby/gems/1.8/gems/redis-3.0.7/lib/redis/:97:in `call': ERR Slot 16011 is already busy (Redis::Com...

在游戏中排行榜是很常见的需求,之前一直使用的是redis的sortedset,刚好看到一个lua的实现,于是就移植到golang版本,github地址:,使用方法和redis 的cli命令基本一致。 使用了cgo封装,skiplis
------分隔线----------------------------