mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 08:46:56 +08:00
bf7f628810
wip: raft does not care about migrating wip: optimize code wip: raft election wip wip: fix raft leader missing log entries wip fix a dead lock batch set slot route wip: raft persist wip refactor cluster suite remove relay rename relay2 refactor: allow customizing client factory test raft refactor re-balance avoid errors caused by inconsistent status on follower nodes during raft commits test raft election
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package cluster
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"github.com/hdt3213/godis/redis/parser"
|
|
"github.com/hdt3213/godis/redis/protocol"
|
|
)
|
|
|
|
func encodeCmdLine(cmdLines []CmdLine) [][]byte {
|
|
var result [][]byte
|
|
for _, line := range cmdLines {
|
|
raw := protocol.MakeMultiBulkReply(line).ToBytes()
|
|
encoded := make([]byte, base64.StdEncoding.EncodedLen(len(raw)))
|
|
base64.StdEncoding.Encode(encoded, raw)
|
|
result = append(result, encoded)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func parseEncodedMultiRawReply(args [][]byte) (*protocol.MultiRawReply, error) {
|
|
cmdBuf := new(bytes.Buffer)
|
|
for _, arg := range args {
|
|
dbuf := make([]byte, base64.StdEncoding.DecodedLen(len(arg)))
|
|
n, err := base64.StdEncoding.Decode(dbuf, arg)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
cmdBuf.Write(dbuf[:n])
|
|
}
|
|
cmds, err := parser.ParseBytes(cmdBuf.Bytes())
|
|
if err != nil {
|
|
return nil, protocol.MakeErrReply(err.Error())
|
|
}
|
|
return protocol.MakeMultiRawReply(cmds), nil
|
|
}
|
|
|
|
// todo: use multi raw reply instead of base64
|
|
func encodeMultiRawReply(src *protocol.MultiRawReply) *protocol.MultiBulkReply {
|
|
args := make([][]byte, 0, len(src.Replies))
|
|
for _, rep := range src.Replies {
|
|
raw := rep.ToBytes()
|
|
encoded := make([]byte, base64.StdEncoding.EncodedLen(len(raw)))
|
|
base64.StdEncoding.Encode(encoded, raw)
|
|
args = append(args, encoded)
|
|
}
|
|
return protocol.MakeMultiBulkReply(args)
|
|
}
|