cakedb 0.0.1

data/lib/cakedb.rb

@@ -0,0 +1,89 @@
+#!/usr/bin/ruby
+
+require 'socket'
+
+class CakeDB
+  def initialize( server="127.0.0.1", port=8888 )
+    puts "Connection Initiated"
+    @server = TCPSocket.new(server, port)
+    @sids = {}
+  end
+  def get_sid(stream)
+    if @sids[stream].nil?
+      @server.write [stream.bytesize,5].pack("L>S>")  
+      @server.write stream  
+      @server.flush
+      sid = @server.recv(2).unpack("S>")
+      @sids[stream] = sid[0]
+      return sid[0]
+    else
+      return @sids[stream]
+    end
+  end
+
+  def write(stream,payload)
+    sid = get_sid(stream)
+    puts sid
+    @server.write [(payload.bytesize + 2),2,sid].pack("L>S>S>")
+    @server.write payload
+    @server.flush
+  end
+
+  def get_tasking(msg)
+    details = Hash.new
+    details["database"] = msg["region"]
+    details["task"]     = msg["tasking"]
+    details["date"]     = msg["date"]
+    details["rediskey"] = details["database"] + "-" + details["task"]
+    return details
+  end
+
+  def read(stream, ts)
+    #Issue the query
+    sid = @sids[stream]
+    puts sid
+    @server.write [10,4,sid,ts].pack("L>S>S>Q>") 
+    @server.flush
+
+    #How much data we gotta get?
+    recv_total = @server.recv(4).unpack("L>")[0]
+    if recv_total > 0
+      counter = 0
+      #Recive the result
+      while recv_total > 0
+        counter+=1 
+	#Get the header - 12 bytes
+	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"
+	end
+	if header_whole.bytesize !=12
+	  puts "Incomplete header"
+	  puts "Length: #{header[1]}"
+	  puts "TS: #{header[0]}"
+	end
+
+	#recieve the data, header says how much
+        result = @server.recv(header[1])
+	
+	#Keep reading till we have everything we expect
+        while(result.bytesize < header[1])
+          result += @server.recv(header[1] - result.bytesize)
+          puts "Incomplete read: #{result.bytesize} / #{header[1]}"
+	end
+	if result.bytesize != header[1]
+          puts "Incomplete read on payload receive"
+	  puts "Raw Data: #{result}"  
+	  puts "Length: #{header[1]} bytes"
+	  puts "Raw Data length: #{result.bytesize - 12}"
+	  puts "Total Length: #{recv_total}"
+	  exit(1)
+        end
+	return result
+      end
+    else
+      # I SEE NOTHING!
+    end
+  end
+end

metadata

@@ -0,0 +1,45 @@
+--- !ruby/object:Gem::Specification
+name: cakedb
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- David Hows
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-09 00:00:00.000000000 Z
+dependencies: []
+description: CakeDB Ruby Driver
+email: david@hows.id.au
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/cakedb.rb
+homepage: http://rubygems.org/gems/cakedb
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: CakeDB Ruby Driver
+test_files: []