使用go net實現(xiàn)簡單的redis通信協(xié)議

更新時間：2021年12月03日 11:48:18 作者：luslin1711

本文主要介紹了go net實現(xiàn)簡單的redis通信協(xié)議，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

?圖解redis通信協(xié)議

請求協(xié)議：

請求協(xié)議一般格式：

*<參數(shù)數(shù)量> CR LF
$<參數(shù) 1 的字節(jié)數(shù)量> CR LF
<參數(shù) 1 的數(shù)據(jù)> CR LF
...
$<參數(shù) N 的字節(jié)數(shù)量> CR LF
<參數(shù) N 的數(shù)據(jù)> CR LF

例如，一個登錄命令：

*2    									2-> 參數(shù)數(shù)量
$4    								4-> 字節(jié)數(shù)量
AUTH
$13
password@2018

返回結果：

+OK

實際上，發(fā)送的命令為“*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]))
}

運行結果