Support additional flags for ACL in CommandsInfo

This commit is contained in:
y_uuki 2020-06-05 03:27:34 +09:00
parent a999d1ecd8
commit 949ccaed3a
1 changed files with 21 additions and 2 deletions

View File

@ -2019,6 +2019,7 @@ type CommandInfo struct {
Name string
Arity int8
Flags []string
ACLFlags []string
FirstKeyPos int8
LastKeyPos int8
StepCount int8
@ -2071,8 +2072,8 @@ func (cmd *CommandsInfoCmd) readReply(rd *proto.Reader) error {
}
func commandInfoParser(rd *proto.Reader, n int64) (interface{}, error) {
if n != 6 {
return nil, fmt.Errorf("redis: got %d elements in COMMAND reply, wanted 6", n)
if n != 7 {
return nil, fmt.Errorf("redis: got %d elements in COMMAND reply, wanted 7", n)
}
var cmd CommandInfo
@ -2125,6 +2126,24 @@ func commandInfoParser(rd *proto.Reader, n int64) (interface{}, error) {
}
cmd.StepCount = int8(stepCount)
_, err = rd.ReadReply(func(rd *proto.Reader, n int64) (interface{}, error) {
cmd.ACLFlags = make([]string, n)
for i := 0; i < len(cmd.Flags); i++ {
switch s, err := rd.ReadString(); {
case err == Nil:
cmd.ACLFlags[i] = ""
case err != nil:
return nil, err
default:
cmd.ACLFlags[i] = s
}
}
return nil, nil
})
if err != nil {
return nil, err
}
for _, flag := range cmd.Flags {
if flag == "readonly" {
cmd.ReadOnly = true