使用go net实现简单的redis通信协议

时间:2021-08-13 09:05:06

 图解redis通信协议

使用go net实现简单的redis通信协议

请求协议:

请求协议一般格式:

?
1
2
3
4
5
6
*<参数数量> CR LF
$<参数 1 的字节数量> CR LF
<参数 1 的数据> CR LF
...
$<参数 N 的字节数量> CR LF
<参数 N 的数据> CR LF

例如,一个登录命令:

?
1
2
3
4
5
*2                                      2-> 参数数量
$4                                  4-> 字节数量
AUTH
$13
password@2018

返回结果:

+OK

实际上,发送的命令为“*2\r\n$4\r\nAUTH\r\n$13\r\npassword@2018\r\n”

测试代码:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package main
 
import (
    "bufio"
    "fmt"
    "net"
    "strconv"
    "testing"
    "time"
)
type Conn struct {
    // Shared
    conn    net.Conn
    // Read
    br          *bufio.Reader
    // Write
    bw           *bufio.Writer
}
 
 
func (c *Conn) Write(cmd string, args ...string) error {
    _, err := c.bw.WriteString("*" + strconv.Itoa(1 + len(args)) + "\r\n")
    if err != nil {
        return err
    }
    _, err =c.bw.WriteString("$" + strconv.Itoa(len(cmd)) + "\r\n" + cmd + "\r\n")
    if err != nil {
        return err
    }
    for _, v := range args {
        _, err :=c.bw.WriteString("$" + strconv.Itoa(len(v)) + "\r\n" + v + "\r\n")
        if err != nil {
            return err
        }
    }
    err = c.bw.Flush()
    if err != nil {
        return err
    }
    return nil
}
 
func TestDail(t *testing.T) {
    conn,err  := net.Dial("tcp","127.0.0.1:6379")
    if err != nil {
        panic(err)
    }
 
    var c = &Conn{
        conn: conn,
        br : bufio.NewReader(conn),
        bw: bufio.NewWriter(conn),
    }
    err = c.Write("AUTH","password@2018")
    if err != nil {
        panic(err)
    }
    p, err := c.br.ReadSlice('\n')
    if err != nil {
        panic(err)
    }
    fmt.Println(string(p))
    err = c.Write("SELECT", "5")
    if err != nil {
        panic(err)
    }
    p, err = c.br.ReadSlice('\n')
    if err != nil {
        panic(err)
    }
    fmt.Println(string(p))
    err = c.Write("keys","*")
    if err != nil {
        panic(err)
    }
    buffer := make([]byte, 4096)
    n, err := c.br.Read(buffer)
    if err != nil {
        panic(err)
    }
    fmt.Println(string(buffer[:n]))
}

运行结果

+OK

+OK

*9
$4
set2
$1
s
$4
set3
$7
string3
$4
int1
$4
test
$7
string2
$7
string1
$7
string4

到此这篇关于使用go net实现简单的redis通信协议的文章就介绍到这了,更多相关go redis通信协议内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://blog.csdn.net/luslin1711/article/details/106095061