mirror of https://github.com/tidwall/tile38.git
96 lines
1.9 KiB
Go
96 lines
1.9 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/tidwall/resp"
|
|
"github.com/tidwall/tile38/internal/collection"
|
|
"github.com/tidwall/tile38/internal/glob"
|
|
)
|
|
|
|
func (s *Server) cmdKeys(msg *Message) (res resp.Value, err error) {
|
|
var start = time.Now()
|
|
vs := msg.Args[1:]
|
|
|
|
var pattern string
|
|
var ok bool
|
|
if vs, pattern, ok = tokenval(vs); !ok || pattern == "" {
|
|
return NOMessage, errInvalidNumberOfArguments
|
|
}
|
|
if len(vs) != 0 {
|
|
return NOMessage, errInvalidNumberOfArguments
|
|
}
|
|
|
|
var wr = &bytes.Buffer{}
|
|
var once bool
|
|
if msg.OutputType == JSON {
|
|
wr.WriteString(`{"ok":true,"keys":[`)
|
|
}
|
|
var wild bool
|
|
if strings.Contains(pattern, "*") {
|
|
wild = true
|
|
}
|
|
var everything bool
|
|
var greater bool
|
|
var greaterPivot string
|
|
var vals []resp.Value
|
|
|
|
iter := func(key string, col *collection.Collection) bool {
|
|
var match bool
|
|
if everything {
|
|
match = true
|
|
} else if greater {
|
|
if !strings.HasPrefix(key, greaterPivot) {
|
|
return false
|
|
}
|
|
match = true
|
|
} else {
|
|
match, _ = glob.Match(pattern, key)
|
|
}
|
|
if match {
|
|
if once {
|
|
if msg.OutputType == JSON {
|
|
wr.WriteByte(',')
|
|
}
|
|
} else {
|
|
once = true
|
|
}
|
|
switch msg.OutputType {
|
|
case JSON:
|
|
wr.WriteString(jsonString(key))
|
|
case RESP:
|
|
vals = append(vals, resp.StringValue(key))
|
|
}
|
|
|
|
// If no more than one match is expected, stop searching
|
|
if !wild {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// TODO: This can be further optimized by using glob.Parse and limits
|
|
if pattern == "*" {
|
|
everything = true
|
|
s.cols.Scan(iter)
|
|
} else if strings.HasSuffix(pattern, "*") {
|
|
greaterPivot = pattern[:len(pattern)-1]
|
|
if glob.IsGlob(greaterPivot) {
|
|
s.cols.Scan(iter)
|
|
} else {
|
|
greater = true
|
|
s.cols.Ascend(greaterPivot, iter)
|
|
}
|
|
} else {
|
|
s.cols.Scan(iter)
|
|
}
|
|
if msg.OutputType == JSON {
|
|
wr.WriteString(`],"elapsed":"` + time.Since(start).String() + "\"}")
|
|
return resp.StringValue(wr.String()), nil
|
|
}
|
|
return resp.ArrayValue(vals), nil
|
|
}
|