cakedb 0.0.6 → 0.0.7

data/lib/cakedb.rb

@@ -3,7 +3,7 @@
 require 'socket'
 
 class CakeDB
-  def initialize( server="127.0.0.1", port=8888 )
+  def initialize( server="127.0.0.1", port=8888, loggingLevel=0 )
     #puts "Connection Initiated"
     @server = TCPSocket.new(server, port)
     @sids = {}
@@ -51,6 +51,7 @@ class CakeDB
     result = Array.new
 
     #How much data we gotta get?
+
     recv_total = @server.recv(4).unpack("L>")[0]
     #puts "total is #{recv_total}"
     if recv_total > 0
@@ -61,12 +62,18 @@ class CakeDB
 	header_whole = @server.recv(12)
 	header = header_whole.unpack("Q>L>")
 	if header[0] == nil || header[1] == nil
-	  puts "You have been eaten by a Grue"
+	  if loggingLevel == 1
+	    puts "You have been eaten by a Grue"
+	  end
+	  raise "You have been eaten by a Grue"
 	end
 	if header_whole.bytesize !=12
-	  puts "Incomplete header"
-	  puts "Length: #{header[1]}"
-	  puts "TS: #{header[0]}"
+	  if loggingLevel == 1
+	    puts "Incomplete header"
+	    puts "Length: #{header[1]}"
+	    puts "TS: #{header[0]}"
+	  end
+	  raise "Header Incomplete"
 	end
 
 	#recieve the data, header says how much
@@ -80,12 +87,13 @@ class CakeDB
           puts "Incomplete read: #{result[counter]["data"].bytesize} / #{header[1]}"
 	end
 	if result[counter]["data"].bytesize != header[1]
-          puts "Incomplete read on payload receive"
-	  puts "Raw Data: #{resultt[counter]}"  
-	  puts "Length: #{header[1]} bytes"
-	  puts "Raw Data length: #{resultt[counter].bytesize - 12}"
-	  puts "Total Length: #{recv_total}"
-	  exit(1)
+	  if loggingLevel == 1
+            puts "Incomplete read on payload receive"
+	    puts "Raw Data: #{resultt[counter]}"  
+	    puts "Length: #{header[1]} bytes"
+	    puts "Raw Data length: #{resultt[counter].bytesize - 12}"
+	    puts "Total Length: #{recv_total}"
+	  end
         end
 	recv_total = recv_total - header[1] - 12
 	counter+=1

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cakedb
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
   prerelease: 
 platform: ruby
 authors: