colddb/cli.lua
2015-04-07 15:37:11 +08:00

120 lines
2.7 KiB
Lua

local flatdb = require("flatdb")
local pp = require("pp")
local function split(s, sep, maxsplit, plain)
assert(sep and sep ~= "")
maxsplit = maxsplit or 1/0
local items = {}
if #s > 0 then
local init = 1
for i = 1, maxsplit do
local m, n = s:find(sep, init, plain)
if m and m <= n then
table.insert(items, s:sub(init, m - 1))
init = n + 1
else
break
end
end
table.insert(items, s:sub(init))
end
return items
end
local db = assert(flatdb("./db"))
local page = "0"
local function handle(input)
local c = split(input, " ", 2)
local cmd, key, value = string.upper(c[1]), c[2], c[3]
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 == "SELECT" then
if tonumber(key) and tonumber(key) < 16 then
if not db[key] then
db[key] = {}
end
end
if db[key] then
page = key
print("OK")
else
print("ERROR")
end
elseif cmd == "GET" then
print(pp.format(db[page][key]))
elseif cmd == "SET" then
local ok, tmp = pcall(load("return "..tostring(value), "=(load)", "t", db))
db[page][key] = ok and tmp or value
print("OK")
elseif cmd == "INCR" then
if not db[page][key] then
db[page][key] = 0
end
db[page][key] = db[page][key] + 1
print("OK")
elseif cmd == "DECR" then
if not db[page][key] then
db[page][key] = 0
end
db[page][key] = db[page][key] - 1
print("OK")
elseif cmd == "RPUSH" then
if not db[page][key] then
db[page][key] = {}
end
table.insert(db[page][key], value)
print(#db[page][key])
elseif cmd == "LPUSH" then
if not db[page][key] then
db[page][key] = {}
end
table.insert(db[page][key], 1, value)
print(#db[page][key])
elseif cmd == "RPOP" then
if not db[page][key] then
db[page][key] = {}
end
print(table.remove(db[page][key], value))
elseif cmd == "LPOP" then
if not db[page][key] then
db[page][key] = {}
end
print(table.remove(db[page][key], 1))
elseif cmd == "LLEN" then
print(db[page][key] and #db[page][key] or 0)
elseif cmd == "LINDEX" then
if db[page][key] then
local i = tonumber(value)
if i < 0 then i = i + #db[page][key] end
i = i + 1
print(db[page][key][i])
else
print("nil")
end
end
end
local function main()
if not db["0"] then
db["0"] = {}
end
io.stdout:setvbuf("no")
while true do
local prefix = page == "0" and "" or ("["..page.."]")
io.write("flatdb"..prefix.."> ")
local input = io.read("*l")
if string.upper(input) == "EXIT" or string.upper(input) == "QUIT" then
break
end
handle(input)
end
end
main()