default key ascending

kuafu · Jun 13, 2016 · fd2c814 · fd2c814
1 parent bd3d9b7
commit fd2c814
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lualib/mongo.lua b/lualib/mongo.lua
@@ -344,7 +344,13 @@ local function IndexModel(option)
 	local keys = {}
 	local name
 	for _, kv in ipairs(option) do
-		local k,v = next(kv)
+		local k,v
+		if type(kv) == "string" then
+			k = kv
+			v = 1
+		else
+			k,v = next(kv)
+		end
 		table.insert(keys, k)
 		table.insert(keys, v)
 		name = (name == nil) and k or (name .. "_" .. k)
@@ -358,6 +364,8 @@ local function IndexModel(option)
 end
 
 -- collection:createIndex { { key1 = 1}, { key2 = 1 },  unique = true }
+-- or collection:createIndex { "key1", "key2",  unique = true }
+-- or collection:createIndex( { key1 = 1} , { unique = true } )	-- For compatibility
 function mongo_collection:createIndex(arg1 , arg2)
 	if arg2 then
 		return createIndex_onekey(self, arg1, arg2)