forked from mtcontrib/colddb
120 lines
2.7 KiB
Lua
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()
|