improvements based on noah's CR

ooyala · Apr 5, 2012 · 949958a · 949958a
1 parent 7466810
commit 949958a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/cassandra-queue.rb b/lib/cassandra-queue.rb
@@ -100,7 +100,7 @@ def initialize(qid, string_queue, keyspace, servers)
 
       @key = qid_to_rowkey qid
       # Set cassandra client if it has not already been set
-      @client ||= create_client(keyspace, servers)
+      @client = create_client(keyspace, servers)
       @queue_cf = string_queue ? DEFAULT_STRING_QUEUE : DEFAULT_BYTES_QUEUE
     end
 
@@ -109,7 +109,7 @@ def qid_to_rowkey(qid)
     end
 
     def create_client(keyspace, servers)
-      ::Cassandra.new(keyspace, servers.flatten)
+      ::Cassandra.new(keyspace, [servers].flatten)
     end
   end
 end