fix srandmember (OpenAtomFoundation#224)

* fix srandmember * bugfix
qyy123123 · Dec 21, 2017 · 45659bd · 45659bd
1 parent 8cc011d
commit 45659bd
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/pika_set.cc b/src/pika_set.cc
@@ -420,11 +420,16 @@ void SRandmemberCmd::Do() {
   std::vector<std::string> members;
   nemo::Status s = g_pika_server->db()->SRandMember(key_, members, count_);
   if (s.ok() || s.IsNotFound()) {
-    res_.AppendArrayLen(members.size());
-    std::vector<std::string>::const_iterator iter = members.begin();
-    for (; iter != members.end(); iter++) {
-      res_.AppendStringLen(iter->size());
-      res_.AppendContent(*iter);
+    if (members.size() == 1) {
+      res_.AppendStringLen(members[0].size());
+      res_.AppendContent(members[0]);
+    } else {
+      res_.AppendArrayLen(members.size());
+      std::vector<std::string>::const_iterator iter = members.begin();
+      for (; iter != members.end(); iter++) {
+        res_.AppendStringLen(iter->size());
+        res_.AppendContent(*iter);
+      }
     }
   } else {
     res_.SetRes(CmdRes::kErrOther, s.ToString());