Files
redka/internal/command/server/dbsize.go
Anton 2e507f8319 fix: command - always return non-nil command when parsing (fix #27)
To avoid panic when calling pcmd.Error in handlers.parse.
2024-06-09 00:14:12 +05:00

29 lines
567 B
Go

package server
import "github.com/nalgeon/redka/internal/redis"
// Returns the number of keys in the database.
// DBSIZE
// https://redis.io/commands/dbsize
type DBSize struct {
redis.BaseCmd
}
func ParseDBSize(b redis.BaseCmd) (DBSize, error) {
cmd := DBSize{BaseCmd: b}
if len(cmd.Args()) != 0 {
return DBSize{}, redis.ErrInvalidArgNum
}
return cmd, nil
}
func (cmd DBSize) Run(w redis.Writer, red redis.Redka) (any, error) {
n, err := red.Key().Len()
if err != nil {
w.WriteError(cmd.Error(err))
return nil, err
}
w.WriteInt(n)
return n, nil
}