使用go net实现简单的redis通信协议
作者：luslin1711
本文主要介绍了go net实现简单的redis通信协议，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
图解redis通信协议
请求协议：
请求协议一般格式：
*<参数数量> 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
到此这篇关于使用go net实现简单的redis通信协议的文章就介绍到这了,更多相关go redis通信协议内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！