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

发布时间:2022-04-19 发布网站:脚本宝典
脚本宝典收集整理的这篇文章主要介绍了使用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 = &amp;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通信协议内容请搜索脚本宝典以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本宝典!

脚本宝典总结

以上是脚本宝典为你收集整理的使用go net实现简单的redis通信协议全部内容,希望文章能够帮你解决使用go net实现简单的redis通信协议所遇到的问题。

如果觉得脚本宝典网站内容还不错,欢迎将脚本宝典推荐好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。
标签: