-- lua mongo test script
-- utils
function string:split(sep)
local sep, fields = sep or ":", {}
local pattern = ("([^%s]+)", sep)
self:gsub(pattern, function(c) fields[#fields + 1] = c end)
return fields
end
-- Constant
HOST = "127.0.0.1"
PORT = 27017
KEEPALIVE_TIMEOUT = 60000
KEEPALIVE_SIZE = 100
CONN_TIMEOUT = 3000
DB_USER = "user"
DB_PASSWD = "password"
DB_NAME = "blog"
DB_COLLECTION = "article"
-- Quote
mongo = require("mongo")
cjson = require("")
cbson = require("bson")
-- state
local status_msg = "error"
local status_code = 500
local message = "unknown error"
local mongo_query = {["category_id"] = {["$in"] = {1,2,3,4}}, ["status"] = {["$ne"] = 2}, ["create_time"] = {["$lte"] = 1427102260}}
local mongo_sort = {["create_time"] = 1}
local mongo_limit = 100
local mongo_skip = 0
local mongo_fields = { ["_id"] = false }
-- For fields involving time, you need to use bson to convert it
if mongo_query["create_time"] then
local create_time = mongo_query["create_time"]
local t = type(create_time)
if t == "table" then
for key, value in pairs(create_time) do
mongo_query["create_time"][key] = (value)
end
else
mongo_query["create_time"] = (create_time)
end
end
local conn = ({ host = HOST, port = PORT })
conn:set_timeout(CONN_TIMEOUT)
local db = conn:getDB(DB_NAME)
local reused_times = conn:get_reused_times()
if reused_times == 0 then
db:auth(DB_USER, DB_PASSWD)
end
local col = db:getCollection(DB_COLLECTION)
local result = {}
-- count
local count, err = col:count(mongo_query)
local ok, err = conn:set_keepalive(KEEPALIVE_TIMEOUT, KEEPALIVE_SIZE)
if count ~= nil then
result = count
status_code = 200
status_msg = "ok"
message = "success"
end
-- query
local bson_obj
if mongo_sort then
bson_obj = cbson.encode_order("$query", mongo_query, "$orderby", mongo_sort)
else
bson_obj = ({ ["$query"] = mongo_query })
end
local results = col:query(bson_obj, mongo_fields, mongo_skip, mongo_limit)
local ok, err = conn:set_keepalive(KEEPALIVE_TIMEOUT, KEEPALIVE_SIZE)
if results then
for _, object in pairs(results) do
for key, value in pairs(object) do
if value == then
object[key] =
else
local type_name, value = (value)
object[key] = value
end
end
end
result = results
status_code = 200
status_msg = "ok"
message = "success"
end
-- findOne
local results = col:findOne({["id"] = 14 })
local ok, err = conn:set_keepalive(KEEPALIVE_TIMEOUT, KEEPALIVE_SIZE)
if results then
for key, value in pairs(results) do
if value == then
results[key] =
else
local type_name, value = (value)
results[key] = value
end
end
result = results
status_code = 200
status_msg = "ok"
message = "success"
end
= status_code
json_out = ({ status = status_msg, message = message, data = result })
["Content-Length"] = json_out:len()
(json_out)