forked from mirror/redis
Refactor NewXInfoGroupsCmd
This commit is contained in:
parent
a32502b1d9
commit
86326c4924
88
command.go
88
command.go
|
@ -1395,10 +1395,10 @@ func (cmd *XPendingExtCmd) readReply(rd *proto.Reader) error {
|
||||||
|
|
||||||
type XInfoGroupsCmd struct {
|
type XInfoGroupsCmd struct {
|
||||||
baseCmd
|
baseCmd
|
||||||
val []XInfoGroups
|
val []XInfoGroup
|
||||||
}
|
}
|
||||||
|
|
||||||
type XInfoGroups struct {
|
type XInfoGroup struct {
|
||||||
Name string
|
Name string
|
||||||
Consumers int64
|
Consumers int64
|
||||||
Pending int64
|
Pending int64
|
||||||
|
@ -1416,11 +1416,11 @@ func NewXInfoGroupsCmd(ctx context.Context, stream string) *XInfoGroupsCmd {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cmd *XInfoGroupsCmd) Val() []XInfoGroups {
|
func (cmd *XInfoGroupsCmd) Val() []XInfoGroup {
|
||||||
return cmd.val
|
return cmd.val
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cmd *XInfoGroupsCmd) Result() ([]XInfoGroups, error) {
|
func (cmd *XInfoGroupsCmd) Result() ([]XInfoGroup, error) {
|
||||||
return cmd.val, cmd.err
|
return cmd.val, cmd.err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1429,58 +1429,66 @@ func (cmd *XInfoGroupsCmd) String() string {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cmd *XInfoGroupsCmd) readReply(rd *proto.Reader) error {
|
func (cmd *XInfoGroupsCmd) readReply(rd *proto.Reader) error {
|
||||||
_, err := rd.ReadArrayReply(func(rd *proto.Reader, n int64) (interface{}, error) {
|
n, err := rd.ReadArrayLen()
|
||||||
for i := int64(0); i < n; i++ {
|
if err != nil {
|
||||||
v, err := rd.ReadReply(xGroupInfoParser)
|
return err
|
||||||
if err != nil {
|
}
|
||||||
return nil, err
|
|
||||||
}
|
cmd.val = make([]XInfoGroup, n)
|
||||||
cmd.val = append(cmd.val, v.(XInfoGroups))
|
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
|
cmd.val[i], err = readXGroupInfo(rd)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
return nil, nil
|
}
|
||||||
})
|
|
||||||
return err
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func xGroupInfoParser(rd *proto.Reader, n int64) (interface{}, error) {
|
func readXGroupInfo(rd *proto.Reader) (XInfoGroup, error) {
|
||||||
if n != 8 {
|
var group XInfoGroup
|
||||||
return nil, fmt.Errorf("redis: got %d elements in XINFO GROUPS reply,"+
|
|
||||||
"wanted 8", n)
|
n, err := rd.ReadArrayLen()
|
||||||
|
if err != nil {
|
||||||
|
return group, err
|
||||||
|
}
|
||||||
|
if n != 8 {
|
||||||
|
return group, fmt.Errorf("redis: got %d elements in XINFO GROUPS reply, wanted 8", n)
|
||||||
}
|
}
|
||||||
var (
|
|
||||||
err error
|
|
||||||
grp XInfoGroups
|
|
||||||
key string
|
|
||||||
val string
|
|
||||||
)
|
|
||||||
|
|
||||||
for i := 0; i < 4; i++ {
|
for i := 0; i < 4; i++ {
|
||||||
key, err = rd.ReadString()
|
key, err := rd.ReadString()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return group, err
|
||||||
}
|
}
|
||||||
val, err = rd.ReadString()
|
|
||||||
|
val, err := rd.ReadString()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return group, err
|
||||||
}
|
}
|
||||||
|
|
||||||
switch key {
|
switch key {
|
||||||
case "name":
|
case "name":
|
||||||
grp.Name = val
|
group.Name = val
|
||||||
case "consumers":
|
case "consumers":
|
||||||
grp.Consumers, err = strconv.ParseInt(val, 0, 64)
|
group.Consumers, err = strconv.ParseInt(val, 0, 64)
|
||||||
|
if err != nil {
|
||||||
|
return group, err
|
||||||
|
}
|
||||||
case "pending":
|
case "pending":
|
||||||
grp.Pending, err = strconv.ParseInt(val, 0, 64)
|
group.Pending, err = strconv.ParseInt(val, 0, 64)
|
||||||
|
if err != nil {
|
||||||
|
return group, err
|
||||||
|
}
|
||||||
case "last-delivered-id":
|
case "last-delivered-id":
|
||||||
grp.LastDeliveredID = val
|
group.LastDeliveredID = val
|
||||||
default:
|
default:
|
||||||
return nil, fmt.Errorf("redis: unexpected content %s "+
|
return group, fmt.Errorf("redis: unexpected content %s in XINFO GROUPS reply", key)
|
||||||
"in XINFO GROUPS reply", key)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return grp, err
|
|
||||||
|
return group, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
|
@ -2257,7 +2265,7 @@ func (cmd *SlowLogCmd) readReply(rd *proto.Reader) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
cmdString := make([]string, cmdLen)
|
cmdString := make([]string, cmdLen)
|
||||||
for i := 0; i < int(cmdLen); i++ {
|
for i := 0; i < cmdLen; i++ {
|
||||||
cmdString[i], err = rd.ReadString()
|
cmdString[i], err = rd.ReadString()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -2265,7 +2273,7 @@ func (cmd *SlowLogCmd) readReply(rd *proto.Reader) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
var address, name string
|
var address, name string
|
||||||
for i := 4; i < int(n); i++ {
|
for i := 4; i < n; i++ {
|
||||||
str, err := rd.ReadString()
|
str, err := rd.ReadString()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
|
@ -1517,7 +1517,7 @@ func (c cmdable) XRead(ctx context.Context, a *XReadArgs) *XStreamSliceCmd {
|
||||||
keyPos += 2
|
keyPos += 2
|
||||||
}
|
}
|
||||||
args = append(args, "streams")
|
args = append(args, "streams")
|
||||||
keyPos += 1
|
keyPos++
|
||||||
for _, s := range a.Streams {
|
for _, s := range a.Streams {
|
||||||
args = append(args, s)
|
args = append(args, s)
|
||||||
}
|
}
|
||||||
|
@ -1592,10 +1592,10 @@ func (c cmdable) XReadGroup(ctx context.Context, a *XReadGroupArgs) *XStreamSlic
|
||||||
}
|
}
|
||||||
if a.NoAck {
|
if a.NoAck {
|
||||||
args = append(args, "noack")
|
args = append(args, "noack")
|
||||||
keyPos += 1
|
keyPos++
|
||||||
}
|
}
|
||||||
args = append(args, "streams")
|
args = append(args, "streams")
|
||||||
keyPos += 1
|
keyPos++
|
||||||
for _, s := range a.Streams {
|
for _, s := range a.Streams {
|
||||||
args = append(args, s)
|
args = append(args, s)
|
||||||
}
|
}
|
||||||
|
|
|
@ -181,7 +181,7 @@ func (r *Reader) ReadArrayReply(m MultiBulkParse) (interface{}, error) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Reader) ReadArrayLen() (int64, error) {
|
func (r *Reader) ReadArrayLen() (int, error) {
|
||||||
line, err := r.ReadLine()
|
line, err := r.ReadLine()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
|
@ -190,7 +190,11 @@ func (r *Reader) ReadArrayLen() (int64, error) {
|
||||||
case ErrorReply:
|
case ErrorReply:
|
||||||
return 0, ParseErrorReply(line)
|
return 0, ParseErrorReply(line)
|
||||||
case ArrayReply:
|
case ArrayReply:
|
||||||
return parseArrayLen(line)
|
n, err := parseArrayLen(line)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return int(n), nil
|
||||||
default:
|
default:
|
||||||
return 0, fmt.Errorf("redis: can't parse array reply: %.100q", line)
|
return 0, fmt.Errorf("redis: can't parse array reply: %.100q", line)
|
||||||
}
|
}
|
||||||
|
@ -216,7 +220,8 @@ func (r *Reader) ReadScanReply() ([]string, uint64, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
keys := make([]string, n)
|
keys := make([]string, n)
|
||||||
for i := int64(0); i < n; i++ {
|
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
key, err := r.ReadString()
|
key, err := r.ReadString()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, 0, err
|
return nil, 0, err
|
||||||
|
|
Loading…
Reference in New Issue