voldemort-rb 0.1.1 → 0.1.2

data/lib/connection/connection.rb

@@ -1,4 +1,5 @@
 require 'rexml/document'
+require 'json'
 
 class Connection
   include REXML
@@ -29,9 +30,13 @@ class Connection
   end
 
   def bootstrap
-    response = self.get_from("metadata", "cluster.xml", false)
-    xml = response[1][0][1]
-    self.nodes = self.parse_nodes_from(xml)
+    cluster_response = self.get_from("metadata", "cluster.xml", false)
+    cluster_xml = cluster_response[1][0][1]
+    self.nodes = self.parse_nodes_from(cluster_xml)
+    
+    stores_response = self.get_from("metadata", "stores.xml", false)
+    stores_xml = stores_response[1][0][1]
+    
     self.connect_to_random_node
   rescue StandardError => e
     raise("There was an error trying to bootstrap from the specified servers: #{e}")

data/lib/connection/tcp_connection.rb

@@ -128,7 +128,17 @@ class TCPConnection < Connection
   def receive
     raw_size = self.socket.recv(4)
     size = raw_size.unpack('N')
-    self.socket.recv(size[0])
+    
+    # Read until we get to size
+    read = 0
+    buffer = ""
+    
+    while read < size[0] and size[0] > 0
+      data = self.socket.recv(size[0] - read)
+      buffer << data
+      read += data.length
+    end
+    return buffer
   rescue
     self.reconnect!
   end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 1
-  version: 0.1.1
+  - 2
+  version: 0.1.2
 platform: ruby
 authors: 
 - Alejandro Crosa
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-06-18 00:00:00 -07:00
+date: 2010-07-13 00:00:00 -07:00
 default_executable: 
 dependencies: []