本篇文章給大家分享的是有關怎么在golang中實現(xiàn)一個redis-cli方法，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

10年建站經(jīng)驗， 網(wǎng)站設計、成都網(wǎng)站建設客戶的見證與正確選擇。創(chuàng)新互聯(lián)提供完善的營銷型網(wǎng)頁建站明細報價表。后期開發(fā)更加便捷高效，我們致力于追求更美、更快、更規(guī)范。

golang的優(yōu)點

golang是一種編譯語言，可以將代碼編譯為機器代碼，編譯后的二進制文件可以直接部署到目標機器而無需額外的依賴，所以golang的性能優(yōu)于其他的解釋性語言，且可以在golang中使用goroutine來實現(xiàn)并發(fā)性，它提供了一個非常優(yōu)雅的goroutine調(diào)度程序系統(tǒng)，可以很容易地生成數(shù)百萬個goroutine。

0. redis通信協(xié)議

redis的客戶端(redis-cli)和服務端(redis-server)的通信是建立在tcp連接之上， 兩者之間數(shù)據(jù)傳輸?shù)木幋a解碼方式就是所謂的redis通信協(xié)議。所以，只要我們的redis-cli實現(xiàn)了這個協(xié)議的解析和編碼，那么我們就可以完成所有的redis操作。

redis 協(xié)議設計的非常易讀，也易于實現(xiàn)，關于具體的redis通信協(xié)議請參考：通信協(xié)議（protocol）。后面我們在實現(xiàn)這個協(xié)議的過程中也會簡單重復介紹一下具體實現(xiàn)

1. 建立tcp連接

redis客戶端和服務端的通信是建立tcp連接之上，所以第一步自然是先建立連接

package main

import (
 "flag"
 "log"
 "net"
)

var host string
var port string

func init() {
 flag.StringVar(&host, "h", "localhost", "hsot")
 flag.StringVar(&port, "p", "6379", "port")
}

func main() {
 flag.Parse()

 tcpAddr := &net.TCPAddr{IP: net.ParseIP(host), Port: port}
 conn, err := net.DialTCP("tcp", nil, tcpAddr)
 if err != nil {
 log.Println(err)
  }
  defer conn.Close()

 // to be continue
}

后續(xù)我們發(fā)送和接受數(shù)據(jù)便都可以使用conn.Read()和conn.Write()來進行了

2. 發(fā)送請求

發(fā)送請求第一個第一個字節(jié)是"*"，中間是包含命令本身的參數(shù)個數(shù)，后面跟著"\r\n" 。之后使用"$"加參數(shù)字節(jié)數(shù)量并使用"\r\n"結尾，然后緊跟參數(shù)內(nèi)容同時也使用"\r\n"結尾。如執(zhí)行 SET key liangwt 客戶端發(fā)送的請求為"*3\r\n$3\r\nSET\r\n$3\r\nkey\r\n$7\r\nliangwt\r\n"

注意：

1. 命令本身也作為協(xié)議的其中一個參數(shù)來發(fā)送

2. \r\n 對應byte的十進制為 13 10

我們可以使用telnet測試下

wentao@bj:～/github.com/liangwt/redis-cli$ telnet 127.0.0.1 6379
Trying 127.0.0.1...
Connected to 127.0.0.1.
Escape character is '^]'.
*3
$3
SET
$3
key
$7
liangwt
+OK

先暫時忽略服務端的回復，通過telnet我們可以看出請求協(xié)議非常簡單，所以對于請求協(xié)議的實現(xiàn)不做過多的介紹了，直接放代碼(如下使用基于字符串拼接，只是為了更直觀的演示，效率并不高，實際代碼中我們使用bytes.Buffer來實現(xiàn))

func MultiBulkMarshal(args ...string) string {
 var s string
 s = "*"
 s += strconv.Itoa(len(args))
 s += "\r\n"

 // 命令所有參數(shù)
 for _, v := range args {
 s += "$"
 s += strconv.Itoa(len(v))
 s += "\r\n"
 s += v
 s += "\r\n"
 }

 return s
}

在實現(xiàn)了對命令和參數(shù)進行編碼之后，我們便可以通過conn.Write()把數(shù)據(jù)推送到服務端

func main() {
  // ....
 req := MultiBulkMarshal("SET", "key", "liangwt")
 _, err = conn.Write([]byte(req))
 if err != nil {
 log.Fatal(err)
 }
 // to be continue
}

3. 獲取回復

我們首先實現(xiàn)通過tcp獲取服務端返回值，就是上面提到過的conn.Read()。

func main() {
  // ....
 p := make([]byte, 1024)
 _, err = conn.Read(p)
 if err != nil {
 log.Fatal(err)
 }
 // to be continue
}

4. 解析回復

我們拿到p之后我們就可以解析返回值了，redis服務端的回復是分為幾種情況的

• 狀態(tài)回復

• 錯誤回復

• 整數(shù)回復

• 批量回復

• 多條批量回復

我們把前四種單獨看作一組，因為他們都是單一類型的返回值

我們把最后的多條批量回復看成單獨的一組，因為它是包含前面幾種類型的混合類型。而且你可以發(fā)現(xiàn)它和我們的請求協(xié)議是一樣的

也正是基于以上的考慮我們創(chuàng)建兩個函數(shù)來分別解析單一類型和混合類型，這樣在解析混合類型中的某一類型時就只需要調(diào)用單一類型解析的函數(shù)即可

在解析具體協(xié)議前我們先實現(xiàn)一個是讀取到\r\n為止的函數(shù)

func ReadLine(p []byte) ([]byte, error) {
 for i := 0; i < len(p); i++ {
 if p[i] == '\r' {
  if p[i+1] != '\n' {
  return []byte{}, errors.New("format error")
  }
  return p[0:i], nil
 }
 }
 return []byte{}, errors.New("format error")
}

第一種狀態(tài)回復：

狀態(tài)回復是一段以 "+" 開始， "\r\n" 結尾的單行字符串。如 SET 命令成功的返回值："+OK\r\n"

所以我們判斷第一個字符是否等于 '+' 如果相等，則讀取到\r

func SingleUnMarshal(p []byte) ([]byte, int, error) {
 var (
 result []byte
 err  error
 length int
 )
 switch p[0] {
 case '+':
 result, err = ReadLine(p[1:])
 length = len(result) + 3
 }

 return result, length, err
}

注：我們在返回實際回復內(nèi)容的同時也返回了整個回復的長度，方便后面解析多條批量回復時定位下一次的解析位置

第二種錯誤回復：

錯誤回復的第一個字節(jié)是 "-"， "\r\n" 結尾的單行字符串。如執(zhí)行 SET key缺少參數(shù)時返回值："-ERR wrong number of arguments for 'set' command\r\n"

錯誤回復和狀態(tài)回復非常相似，解析方式也是一樣到。所以我們只需添加一個case即可

func SingleUnMarshal(p []byte) ([]byte, int, error) {
 var (
 result []byte
 err  error
 length int
 )
 switch p[0] {
 case '+', '-':
 result, err = ReadLine(p[1:])
 length = len(result) + 3
 }
 return result, length, err
}

第三種整數(shù)回復：

整數(shù)回復的第一個字節(jié)是":"，中間是字符串表示的整數(shù)，"\r\n" 結尾的單行字符串。如執(zhí)行LLEN mylist命令時返回 ":10\r\n"

整數(shù)回復也和上面兩種是一樣的，只不過返回的是字符串表示的十進制整數(shù)

func SingleUnMarshal(p []byte) ([]byte, int, error) {
 var (
 result []byte
 err  error
 length int
 )
 switch p[0] {
 case '+', '-', ':':
 result, err = ReadLine(p[1:])
 length = len(result) + 3
 }
 return result, length, err
}

第四種批量回復：

批量回復的第一個字節(jié)為 "$"，接下來是字符串表示的整數(shù)，它表示實際回復的長度，之后跟著一個 "\r\n"，再后面跟著的是實際回復數(shù)據(jù)，最末尾是另一個 "\r\n"。如GET key 命令的返回值："$7\r\nliangwt\r\n"

所以批量回復解析的實現(xiàn)：

• 讀取第一行得到實際回復的長度

• 把字符串類型的長度轉換成對應十進制整數(shù)

• 從第二行開始位置往下讀對應長度

但是對于某些不存在的key，批量回復會將特殊值 -1 用作回復的長度值, 此時我們不需要繼續(xù)往下讀取實際回復。例如GET NOT_EXIST_KEY 返回值："$-1", 所以我們需要對此特殊情況判斷，讓函數(shù)返回一個空對象(nil)而不是空值("")

func SingleUnMarshal(p []byte) ([]byte, int, error) {
 // ....
 case '$':
 n, err := ReadLine(p[1:])
 if err != nil {
  return []byte{}, 0, err
 }
 l, err := strconv.Atoi(string(n))
 if err != nil {
  return []byte{}, 0, err
 }
 if l == -1 {
  return nil, 0, nil
 }
 // +3 的原因 $ \r \n 三個字符
 result = p[len(n)+3 : len(n)+3+l]
 length = len(n) + 5 + l
 }
 return result, length, err
}

思考：

為什么redis要使用提前告知字節(jié)數(shù)，然后往下讀取指定長度的方式，而不是直接讀取第二行到\r\n為止？

答案很明顯：此方式可以讓redis讀取返回值時不受具體的返回內(nèi)容影響，在按行讀取的情況下，無論使用任何分割符都有可能導致redis在解析具體內(nèi)容時把內(nèi)容中的分割符當作時結尾，導致解析錯誤。

思考一下這種情況：我們SET key "liang\r\nwt" ，那么當我們GET key時，服務端返回值為"$9\r\nliang\r\nwt\r\n" 完全規(guī)避了value中的\r\n影響

第五種多條批量回復：

多條批量回復是由多個回復組成的數(shù)組，它的第一個字節(jié)為"*"， 后跟一個字符串表示的整數(shù)值， 這個值記錄了多條批量回復所包含的回復數(shù)量， 再后面是一個"\r\n"。如LRANGE mylist 0 -1的返回值："*3\r\n$1\r\n3\r\n$1\r\n2\r\n$1\r\n1"。

所以多條批量回復解析的實現(xiàn)：

• 解析第一行數(shù)據(jù)獲得字符串類型的回復數(shù)量

• 把字符串類型的長度轉換成對應十進制整數(shù)

• 按照單條回復依次逐個解析，一共解析成上面得到的數(shù)量

在這里我們用到了單條解析時返回的字節(jié)長度length，通過這個長度我們可以很方便的知道下次單條解析的開始位置為上一次位置+length

在解析多條批量回復時需要注意兩點：

第一，多條批量回復也可以是空白的（empty）。例如執(zhí)行LRANGE NOT_EXIST_KEY 0 -1 服務端返回值"*0\r\n"。此時客戶端返回的應該空數(shù)組[][]byte

第二，多條批量回復也可以是無內(nèi)容的（null multi bulk reply)。例如執(zhí)行BLPOP key 1 服務端返回值"*-1\r\n"。此時客戶端返回的應該是nil

func MultiUnMarsh(p []byte) ([][]byte, error) {
 if p[0] != '*' {
 return [][]byte{}, errors.New("format error")
 }
 n, err := ReadLine(p[1:])
 if err != nil {
 return [][]byte{}, err
 }
 l, err := strconv.Atoi(string(n))
 if err != nil {
 return [][]byte{}, err
 }
 // 多條批量回復也可以是空白的（empty)
 if l == 0 {
 return [][]byte{}, nil
 }

 // 無內(nèi)容的多條批量回復（null multi bulk reply）也是存在的,
 // 客戶端庫應該返回一個 null 對象, 而不是一個空數(shù)組。
 if l == -1 {
 return nil, nil
 }
 result := make([][]byte, l)
 t := len(n) + 3
 for i := 0; i < l; i++ {
 ret, length, err := SingleUnMarshal(p[t:])
 if err != nil {
  return [][]byte{}, errors.New("format error")
 }
 result[i] = ret
 t += length
 }

 return result, nil
}

5. 命令行模式

一個可用的redis-cli自然是一個交互式的，用戶輸入指令然后輸出返回值。在go中我們可以使用以下代碼即可獲得一個類似的交互式命令行

func main() {
 // ....
 for {
 fmt.Printf("%s:%d>", host, port)

 bio := bufio.NewReader(os.Stdin)
 input, _, err := bio.ReadLine()
 if err != nil {
  log.Fatal(err)
 }
 fmt.Printf("%s\n", input)
 }
}

我們運行以上代碼就可以實現(xiàn)

localhost:6379>set key liang
set key liang
localhost:6379>get key
get key
localhost:6379>

結合上我們的redis發(fā)送請求和解析請求即可完成整個redis-cli

func main() {
  // ....
 for {
 fmt.Printf("%s:%d>", host, port)

 // 獲取輸入命令和參數(shù)
 bio := bufio.NewReader(os.Stdin)
 input, err := bio.ReadString('\n')
 if err != nil {
  log.Fatal(err)
 }
 fields := strings.Fields(input)

 // 編碼發(fā)送請求
 req := MultiBulkMarshal(fields...)

 // 發(fā)送請求
 _, err = conn.Write([]byte(req))
 if err != nil {
  log.Fatal(err)
 }

 // 讀取返回內(nèi)容
 p := make([]byte, 1024)
 _, err = conn.Read(p)
 if err != nil {
  log.Fatal(err)
 }

 // 解析返回內(nèi)容
 if p[0] == '*' {
  result, err := MultiUnMarsh(p)
 } else {
  result, _, err := SingleUnMarshal(p)
 }

  }
  // ....
}

以上就是怎么在golang中實現(xiàn)一個redis-cli方法，小編相信有部分知識點可能是我們?nèi)粘９ぷ鲿姷交蛴玫降?。希望你能通過這篇文章學到更多知識。更多詳情敬請關注創(chuàng)新互聯(lián)成都網(wǎng)站設計公司行業(yè)資訊頻道。

另外有需要云服務器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。

網(wǎng)頁名稱：怎么在golang中實現(xiàn)一個redis-cli方法-創(chuàng)新互聯(lián)
標題鏈接：http://jinyejixie.com/article14/djchde.html

成都網(wǎng)站建設公司_創(chuàng)新互聯(lián)，為您提供App開發(fā)、云服務器、網(wǎng)站改版、靜態(tài)網(wǎng)站、外貿(mào)網(wǎng)站建設、網(wǎng)站策劃

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉載內(nèi)容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉載，或轉載時需注明來源： 創(chuàng)新互聯(lián)