b55300b729
* Start on lua scripting * Implement evalsha, script load, script exists, and script flush * Type conversions from lua to resp/json. Refactor to make luastate and luascripts persistent in the controller. * Change controller.command and all underlying commands to return resp.Value. Serialize only during the ouput. * First stab at tile38 call from lua * Change tile38 into tile38.call in Lua * Property return errors from scripts * Minor refactoring. No locking on script run * Cleanup/refactoring * Create a pool of 5 lua states, allow for more as needed. Refactor. * Use safe map for scripts. Add a limit for max number of lua states. Refactor. * Refactor * Refactor script commands into atomic, read-only, and non-atomic classes. Proper locking for all three classes. Add tests for scripts * More tests for scripts * Properly escape newlines in lua-produced errors * Better test for readonly failure * Correctly convert ok/err messages between lua and resp. Add pcall, sha1hex, error_reply, status_reply functions to tile38 namespace in lua. * Add pcall test. Change writeErr to work with string argument * Make sure eval/evalsha never attempt to write AOF * Add eval-set and eval-get to benchmarks * Fix eval benchmark tests, add more * Improve benchmarks * Optimizations and refactoring. * Add lua memtest * Typo * Add dependency * golint fixes * gofmt fixes * Add scripting commands to the core/commands.json * Use ARGV for args inside lua
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package controller
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/tidwall/btree"
|
|
"github.com/tidwall/resp"
|
|
"github.com/tidwall/tile38/controller/glob"
|
|
"github.com/tidwall/tile38/controller/server"
|
|
)
|
|
|
|
func (c *Controller) cmdKeys(msg *server.Message) (res resp.Value, err error) {
|
|
var start = time.Now()
|
|
vs := msg.Values[1:]
|
|
|
|
var pattern string
|
|
var ok bool
|
|
if vs, pattern, ok = tokenval(vs); !ok || pattern == "" {
|
|
return server.NOMessage, errInvalidNumberOfArguments
|
|
}
|
|
if len(vs) != 0 {
|
|
return server.NOMessage, errInvalidNumberOfArguments
|
|
}
|
|
|
|
var wr = &bytes.Buffer{}
|
|
var once bool
|
|
if msg.OutputType == server.JSON {
|
|
wr.WriteString(`{"ok":true,"keys":[`)
|
|
}
|
|
var everything bool
|
|
var greater bool
|
|
var greaterPivot string
|
|
var vals []resp.Value
|
|
|
|
iterator := func(item btree.Item) bool {
|
|
key := item.(*collectionT).Key
|
|
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 == server.JSON {
|
|
wr.WriteByte(',')
|
|
}
|
|
} else {
|
|
once = true
|
|
}
|
|
switch msg.OutputType {
|
|
case server.JSON:
|
|
wr.WriteString(jsonString(key))
|
|
case server.RESP:
|
|
vals = append(vals, resp.StringValue(key))
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
if pattern == "*" {
|
|
everything = true
|
|
c.cols.Ascend(iterator)
|
|
} else {
|
|
if strings.HasSuffix(pattern, "*") {
|
|
greaterPivot = pattern[:len(pattern)-1]
|
|
if glob.IsGlob(greaterPivot) {
|
|
greater = false
|
|
c.cols.Ascend(iterator)
|
|
} else {
|
|
greater = true
|
|
c.cols.AscendGreaterOrEqual(&collectionT{Key: greaterPivot}, iterator)
|
|
}
|
|
} else if glob.IsGlob(pattern) {
|
|
greater = false
|
|
c.cols.Ascend(iterator)
|
|
} else {
|
|
greater = true
|
|
greaterPivot = pattern
|
|
c.cols.AscendGreaterOrEqual(&collectionT{Key: greaterPivot}, iterator)
|
|
}
|
|
}
|
|
if msg.OutputType == server.JSON {
|
|
wr.WriteString(`],"elapsed":"` + time.Now().Sub(start).String() + "\"}")
|
|
return resp.StringValue(wr.String()), nil
|
|
}
|
|
return resp.ArrayValue(vals), nil
|
|
}
|