go知名第三方包ssdb

李海彬

发布于 2019-01-08 17:44:54

7650

发布于 2019-01-08 17:44:54

文章被收录于专栏：Golang语言社区

原文作者:laijh 来源：简书

因为beego中cache模块中使用了ssdb，所以准备学习下ssdb （１）ssdb简介（２）ssdb的基本操作（３）gossdb怎么使用?

１、ssdb简介 SSDB 是一个 C/C++ 语言开发的高性能 NoSQL 数据库, 支持 KV, list, map(hash), zset(sorted set),qlist(队列) 等数据结构, 用来替代或者与 Redis 配合存储十亿级别列表的数据. SSDB 是稳定的, 生产环境使用的, 已经在许多互联网公司得到广泛使用, 如奇虎 360, TOPGAME. 特性替代 Redis 数据库, Redis 的 100 倍容量 LevelDB 网络支持, 使用 C/C++ 开发 Redis API 兼容, 支持 Redis 客户端适合存储集合数据, 如kv, list, hashtable, zset,hset,qlist... 客户端 API 支持的语言包括: C++, PHP, Python, Java, Go 持久化的队列服务主从复制, 负载均衡

既然有redis这种NoSQL数据库，为什么需要ssdb？我在网上得到的结论是Redis是内存型，内存成本太高，SSDB针对这个弱点，使用硬盘存储，使用Google高性能的存储引擎LevelDB。

２、ssdb的基本操作 命令 ３、gossdb包怎么使用? 源码，打开源码文件，代码大概在190行左右，不是很多。这个包也不复杂。结构体Client

1type Client struct {
2    sock     *net.TCPConn
3    recv_buf bytes.Buffer
4}

相关方法：

1）创建Client func Connect(ip string, port int) (*Client, error)

 1func Connect(ip string, port int) (*Client, error) {
 2    addr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("%s:%d", ip, port))
 3    if err != nil {
 4        return nil, err
 5    }
 6    sock, err := net.DialTCP("tcp", nil, addr)
 7    if err != nil {
 8        return nil, err
 9    }
10    var c Client
11    c.sock = sock
12    return &c, nil
13}

2）发送命令到ssdb服务器 func (c *Client) Do(args ...interface{}) ([]string, error)

1func (c *Client) Do(args ...interface{}) ([]string, error) {
2    err := c.send(args)
3    if err != nil {
4        return nil, err
5    }
6    resp, err := c.recv()
7    return resp, err
8}

3）设置值 func (c *Client) Set(key string, val string) (interface{}, error)

 1func (c *Client) Set(key string, val string) (interface{}, error) {
 2    resp, err := c.Do("set", key, val)
 3    if err != nil {
 4        return nil, err
 5    }
 6    if len(resp) == 2 && resp[0] == "ok" {
 7        return true, nil
 8    }
 9    return nil, fmt.Errorf("bad response")
10}

4）获取值

 1func (c *Client) Get(key string) (interface{}, error) {
 2    resp, err := c.Do("get", key)
 3    if err != nil {
 4        return nil, err
 5    }
 6    if len(resp) == 2 && resp[0] == "ok" {
 7        return resp[1], nil
 8    }
 9    if resp[0] == "not_found" {
10        return nil, nil
11    }
12    return nil, fmt.Errorf("bad response")
13}

5）func (c *Client) Del(key string) (interface{}, error) 删除值

 1func (c *Client) Del(key string) (interface{}, error) {
 2    resp, err := c.Do("del", key)
 3    if err != nil {
 4        return nil, err
 5    }
 6
 7    //response looks like this: [ok 1]
 8    if len(resp) > 0 && resp[0] == "ok" {
 9        return true, nil
10    }
11    return nil, fmt.Errorf("bad response:resp:%v:", resp)
12}

6）func (c *Client) Send(args ...interface{}) error 发送命令到ssdb服务器

 1func (c *Client) Send(args ...interface{}) error {
 2    return c.send(args);
 3}
 4
 5func (c *Client) send(args []interface{}) error {
 6    var buf bytes.Buffer
 7    for _, arg := range args {
 8        var s string
 9        switch arg := arg.(type) {
10        case string:
11            s = arg
12        case []byte:
13            s = string(arg)
14        case []string:
15            for _, s := range arg {
16                buf.WriteString(fmt.Sprintf("%d", len(s)))
17                buf.WriteByte('\n')
18                buf.WriteString(s)
19                buf.WriteByte('\n')
20            }
21            continue
22        case int:
23            s = fmt.Sprintf("%d", arg)
24        case int64:
25            s = fmt.Sprintf("%d", arg)
26        case float64:
27            s = fmt.Sprintf("%f", arg)
28        case bool:
29            if arg {
30                s = "1"
31            } else {
32                s = "0"
33            }
34        case nil:
35            s = ""
36        default:
37            return fmt.Errorf("bad arguments")
38        }
39        buf.WriteString(fmt.Sprintf("%d", len(s)))
40        buf.WriteByte('\n')
41        buf.WriteString(s)
42        buf.WriteByte('\n')
43    }
44    buf.WriteByte('\n')
45    _, err := c.sock.Write(buf.Bytes())
46    return err
47}

7）func (c *Client) Recv() ([]string, error) 接收ssdb的信息

 1func (c *Client) Recv() ([]string, error) {
 2    return c.recv();
 3}
 4
 5func (c *Client) recv() ([]string, error) {
 6    var tmp [8192]byte
 7    for {
 8        resp := c.parse()
 9        if resp == nil || len(resp) > 0 {
10            return resp, nil
11        }
12        n, err := c.sock.Read(tmp[0:])
13        if err != nil {
14            return nil, err
15        }
16        c.recv_buf.Write(tmp[0:n])
17    }
18}

8）func (c *Client) Close() error 关闭连接

1func (c *Client) Close() error {
2    return c.sock.Close()
3}

解析命令

 1func (c *Client) parse() []string {
 2    resp := []string{}
 3    buf := c.recv_buf.Bytes()
 4    var idx, offset int
 5    idx = 0
 6    offset = 0
 7
 8    for {
 9        idx = bytes.IndexByte(buf[offset:], '\n')
10        if idx == -1 {
11            break
12        }
13        p := buf[offset : offset+idx]
14        offset += idx + 1
15        //fmt.Printf("> [%s]\n", p);
16        if len(p) == 0 || (len(p) == 1 && p[0] == '\r') {
17            if len(resp) == 0 {
18                continue
19            } else {
20                var new_buf bytes.Buffer
21                new_buf.Write(buf[offset:])
22                c.recv_buf = new_buf
23                return resp
24            }
25        }
26
27        size, err := strconv.Atoi(string(p))
28        if err != nil || size < 0 {
29            return nil
30        }
31        if offset+size >= c.recv_buf.Len() {
32            break
33        }
34
35        v := buf[offset : offset+size]
36        resp = append(resp, string(v))
37        offset += size + 1
38    }
39
40    //fmt.Printf("buf.size: %d packet not ready...\n", len(buf))
41    return []string{}
42}

参考链接： 1、ssdb学习： https://www.w3cschool.cn/ssdb/ http://ssdb.io/docs/zh_cn/commands/index.html 2、gossdb源码： https://github.com/ssdb/gossdb

版权申明：内容来源网络，版权归原创者所有。除非无法确认，我们都会标明作者及出处，如有侵权烦请告知，我们会立即删除并表示歉意。谢谢。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-01-06，如有侵权请联系 cloudcommunity@tencent.com 删除

存储