nalc/colddb
1
0
Fork 0
forked from mtcontrib/colddb
Code Issues Pull Requests Releases Wiki Activity

make cli more redis-like

Browse Source
This commit is contained in:
uleelx 2015-04-06 23:17:43 +08:00
parent ae95bae681
commit ec225bb6d7
1 changed files with 52 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
cli.lua
View File

@ -22,31 +22,36 @@ local function split(s, sep, maxsplit, plain)
end
local db = assert(flatdb("./db"))
local book = "default"
local book = "0"
local function handle(input)
local c = split(input, " ", 2)
local cmd, key, value = string.upper(c[1]), c[2], c[3]
if cmd == "GET" then
print(pp.format(db[book][key]))
elseif cmd == "SET" then
local ok, tmp = pcall(load("return "..tostring(value), "=(load)", "t", db))
db[book][key] = ok and tmp or value
print("OK")
elseif cmd == "SAVE" then
if cmd == "SAVE" then
print(db:save(key) and "OK" or "ERROR")
elseif cmd == "TOUCH" then
if not db[key] then
db[key] = {}
end
print("OK")
elseif cmd == "OPEN" then
elseif cmd == "SELECT" then
if tonumber(key) and tonumber(key) < 16 then
if not db[key] then
db[key] = {}
end
end
if db[key] then
book = key
print("OK")
else
print("ERROR")
end
elseif cmd == "GET" then
print(pp.format(db[book][key]))
elseif cmd == "SET" then
local ok, tmp = pcall(load("return "..tostring(value), "=(load)", "t", db))
db[book][key] = ok and tmp or value
print("OK")
elseif cmd == "INCR" then
if not db[book][key] then
db[book][key] = 0
@ -59,18 +64,52 @@ local function handle(input)
end
db[book][key] = db[book][key] - 1
print("OK")
elseif cmd == "RPUSH" then
if not db[book][key] then
db[book][key] = {}
end
table.insert(db[book][key], value)
print(#db[book][key])
elseif cmd == "LPUSH" then
if not db[book][key] then
db[book][key] = {}
end
table.insert(db[book][key], 1, value)
print(#db[book][key])
elseif cmd == "RPOP" then
if not db[book][key] then
db[book][key] = {}
end
print(table.remove(db[book][key], value))
elseif cmd == "LPOP" then
if not db[book][key] then
db[book][key] = {}
end
print(table.remove(db[book][key], 1))
elseif cmd == "LLEN" then
print(db[book][key] and #db[book][key] or 0)
elseif cmd == "LINDEX" then
if db[book][key] then
local i = tonumber(value)
if i < 0 then i = i + #db[book][key] end
i = i + 1
print(db[book][key][i])
else
print("nil")
end
end
end
local function main()
if not db.default then
db.default = {}
if not db["0"] then
db["0"] = {}
end
io.stdout:setvbuf("no")
while true do
io.write("> ")
local prefix = book == "0" and "" or ("["..book.."]")
io.write("flatdb"..prefix.."> ")
local input = io.read("*l")
if string.upper(input) == "EXIT" then
if string.upper(input) == "EXIT" or string.upper(input) == "QUIT" then
break
end
handle(input)