使用go net 實現簡單的redis通信協議
圖解redis通信協議(轉自:https://www.jianshu.com/p/f670dfc9409b)
請求協議:
請求協議一般格式:
*<參數數量> CR LF
$<參數 1 的字節數量> CR LF
<參數 1 的數據> CR LF
...
$<參數 N 的字節數量> CR LF
<參數 N 的數據> CR LF
例如,一個登錄命令:
*2 2-> 參數數量
$4 4-> 字節數量
AUTH
$13
password@2018
返回結果:
+OK
實際上,發送的命令爲“*2\r\n$4\r\nAUTH\r\n$13\r\npassword@2018\r\n”
測試代碼:
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