vertica 0.7.4 → 0.8.1

data/lib/vertica/core_ext/numeric.rb

@@ -0,0 +1,13 @@
+class Numeric
+  def to_network_int16
+    [self].pack('n')
+  end
+
+  def to_network_int32
+    [self].pack('N')
+  end
+
+  def to_byte
+    [self].pack('C')
+  end
+end

data/lib/vertica/core_ext/string.rb

@@ -0,0 +1,22 @@
+class String
+  def to_cstring
+    raise ArgumentError, "Invalid cstring" if self.include?("\000")
+    "#{self}\000"
+  end
+
+  def from_cstring
+    self[0..-2]
+  end
+
+  def to_byte
+    unpack('C').first
+  end
+
+  def to_network_int16
+    unpack('n').first
+  end
+
+  def to_network_int32
+    unpack('l').first
+  end
+end

data/lib/vertica/messages/{authentication.rb → backend_messages/authentication.rb}

@@ -12,20 +12,17 @@ module Vertica
       GSS                 = 7
       GSS_CONTINUE        = 8
       SSPI                = 9
-      
+
       attr_reader :code
       attr_reader :salt
       attr_reader :auth_data
-      
+
       def initialize(stream, size)
         super
-        @code = stream.read_network_int32
-        if @code == CRYPT_PASSWORD
-          @salt = stream.readn(2)
-        elsif @code == MD5_PASSWORD
-          @salt = stream.readn(4)
-        elsif @code == GSS_CONTINUE
-          @auth_data = stream.readn(size - 9)
+        case @code = stream.read_network_int32
+        when CRYPT_PASSWORD then @salt = stream.readn(2)
+        when MD5_PASSWORD   then @salt = stream.readn(4)
+        when GSS_CONTINUE   then @auth_data = stream.readn(size - 9)
         end
       end
     end

data/lib/vertica/messages/{backend_key_data.rb → backend_messages/backend_key_data.rb}

@@ -2,10 +2,10 @@ module Vertica
   module Messages
     class BackendKeyData < BackendMessage
       message_id ?K
-      
+
       attr_reader :pid
       attr_reader :key
-      
+
       def initialize(stream, size)
         super
         @pid = stream.read_network_int32

data/lib/vertica/messages/{bind_complete.rb → backend_messages/bind_complete.rb}

@@ -2,7 +2,6 @@ module Vertica
   module Messages
     class BindComplete < BackendMessage
       message_id ?2
-      
     end
   end
 end

data/lib/vertica/messages/{close_complete.rb → backend_messages/close_complete.rb}

@@ -2,7 +2,6 @@ module Vertica
   module Messages
     class CloseComplete < BackendMessage
       message_id ?3
-
     end
   end
 end

data/lib/vertica/messages/{command_complete.rb → backend_messages/command_complete.rb}

@@ -2,15 +2,15 @@ module Vertica
   module Messages
     class CommandComplete < BackendMessage
       message_id ?C
-      
+
       attr_reader :tag
       attr_reader :rows
-      
+
       def initialize(stream, size)
         @tag = stream.read_cstring
         @rows = @tag.split[-1].to_i
       end
-      
+
     end
   end
 end

data/lib/vertica/messages/{data_row.rb → backend_messages/data_row.rb}

@@ -2,20 +2,17 @@ module Vertica
   module Messages
     class DataRow < BackendMessage
       message_id ?D
-      
+
       attr_reader :field_count
       attr_reader :fields
-      
+
       def initialize(stream, size)
         @fields = []
 
         @field_count = stream.read_network_int16
         @field_count.times do |field_index|
           size = stream.read_network_int32
-          @fields << {
-            :size   => size,
-            :value  => size == -1 ? nil : stream.readn(size)
-          }
+          @fields << (size == -1 ? nil : stream.readn(size))
         end
       end
     end

data/lib/vertica/messages/{empty_query_response.rb → backend_messages/empty_query_response.rb}

@@ -2,7 +2,6 @@ module Vertica
   module Messages
     class EmptyQueryResponse < BackendMessage
       message_id ?I
-      
     end
   end
 end

data/lib/vertica/messages/backend_messages/error_response.rb

@@ -0,0 +1,35 @@
+module Vertica
+  module Messages
+    class ErrorResponse < BackendMessage
+      message_id ?E
+
+      ERRORS = {
+        ?q => [0, "Internal Query"],
+        ?S => [1, "Severity"],
+        ?M => [2, "Message"],
+        ?C => [3, "Sqlstate"],
+        ?D => [4, "Detail"],
+        ?H => [5, "Hint"],
+        ?P => [6, "Position"],
+        ?W => [7, "Where"],
+        ?p => [8, "Internal Position"],
+        ?R => [10, "Routine"],
+        ?F => [11, "File"],
+        ?L => [12, "Line"],
+      }
+
+      def initialize(stream, size)
+        super
+        @errors, type = {}, nil
+        @errors[type] = stream.read_cstring while (type = stream.read_byte) != 0
+      end
+
+      def error
+        @errors.map { |type, msg| [(ERRORS[type] || [ERRORS.size, type.to_s]), msg].flatten }.
+                sort_by { |e| e.first }.
+                map { |e| "#{e[1]}: #{e[2]}" }.
+                join(', ')
+      end
+    end
+  end
+end

data/lib/vertica/messages/{no_data.rb → backend_messages/no_data.rb}

@@ -2,7 +2,6 @@ module Vertica
   module Messages
     class NoData < BackendMessage
       message_id ?n
-      
     end
   end
 end

data/lib/vertica/messages/backend_messages/notice_response.rb

@@ -0,0 +1,16 @@
+module Vertica
+  module Messages
+    class NoticeResponse < BackendMessage
+      message_id ?N
+
+      attr_reader :notices
+
+      def initialize(stream, size)
+        super
+        @notices, type = [], nil
+        @notices << [type, stream.read_cstring] while (type = stream.read_byte) != 0
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/{parse_complete.rb → backend_messages/parse_complete.rb}

@@ -2,7 +2,6 @@ module Vertica
   module Messages
     class ParseComplete < BackendMessage
       message_id ?1
-      
     end
   end
 end

data/lib/vertica/messages/{portal_suspended.rb → backend_messages/portal_suspended.rb}

@@ -2,7 +2,6 @@ module Vertica
   module Messages
     class PortalSuspended < BackendMessage
       message_id ?s
-
     end
   end
 end

data/lib/vertica/messages/{ready_for_query.rb → backend_messages/ready_for_query.rb}

@@ -2,9 +2,9 @@ module Vertica
   module Messages
     class ReadyForQuery < BackendMessage
       message_id ?Z
-      
+
       attr_reader :transaction_status
-      
+
       def initialize(stream, size)
         super
         @transaction_status = stream.read_byte

data/lib/vertica/messages/{row_description.rb → backend_messages/row_description.rb}

@@ -2,15 +2,15 @@ module Vertica
   module Messages
     class RowDescription < BackendMessage
       message_id ?T
-      
+
       attr_reader :field_count
       attr_reader :fields
-      
+
       def initialize(stream, size)
         super
 
         @fields = []
-        
+
         @field_count = stream.read_network_int16
         @field_count.times do |field_index|
           @fields << {

data/lib/vertica/messages/{unknown.rb → backend_messages/unknown.rb}

@@ -2,8 +2,8 @@ module Vertica
   module Messages
     class Unknown < BackendMessage
       attr_reader :message_id
-      
-      def initialize(message_id)
+
+      def initialize(stream, size)
         @message_id = message_id
       end
     end

data/lib/vertica/messages/frontend_messages/bind.rb

@@ -0,0 +1,28 @@
+module Vertica
+  module Messages
+    class Bind < FrontendMessage
+      message_id ?B
+
+      def initialize(portal_name, prepared_statement_name, parameter_values)
+        @portal_name = portal_name
+        @prepared_statement_name = prepared_statement_name
+        @parameter_values = parameter_values.map(&:to_s)
+      end
+
+      def to_bytes
+        bytes = [
+          @portal_name.to_cstring,                    # portal name ("")
+          @prepared_statement_name.to_cstring,        # prep
+          0.to_network_int16,                         # format codes (0 - default text format)
+          @parameter_values.length.to_network_int16,  # number of parameters
+        ]
+        @parameter_values.each do |parameter_value|
+          bytes << parameter_value.length.to_network_int32  # parameter value (which is represented as a string) length
+          bytes << parameter_value                          # parameter value written out in text representation
+        end
+        message_string bytes
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/frontend_messages/cancel_request.rb

@@ -0,0 +1,21 @@
+module Vertica
+  module Messages
+    class CancelRequest < FrontendMessage
+      message_id nil
+
+      def initialize(backend_pid, backend_key)
+        @backend_pid = backend_pid
+        @backend_key = backend_key
+      end
+
+      def to_bytes
+        message_string([
+          80877102.to_network_int32,
+          @backend_pid.to_network_int32,
+          @backend_key.to_network_int32
+        ])
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/frontend_messages/close.rb

@@ -0,0 +1,24 @@
+module Vertica
+  module Messages
+    class Close < FrontendMessage
+      message_id ?C
+
+      def initialize(close_type, close_name)
+        @close_name = close_name
+        @close_type = case close_type
+        when :portal              then ?P
+        when :prepared_statement  then ?S
+        else raise ArgumentError.new("#{close_type} is not a valid close_type.  Must be either :portal or :prepared_statement.")
+        end
+      end
+
+      def to_bytes
+        message_string([
+          @close_type.to_byte,
+          @close_name.to_cstring
+        ])
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/frontend_messages/describe.rb

@@ -0,0 +1,24 @@
+module Vertica
+  module Messages
+    class Describe < FrontendMessage
+      message_id ?D
+
+      def initialize(describe_type, describe_name)
+        @describe_name = describe_name
+        @describe_type = case describe_type
+        when :portal              then ?P
+        when :prepared_statement  then ?S
+        else raise ArgumentError.new("#{describe_type} is not a valid describe_type.  Must be either :portal or :prepared_statement.")
+        end
+      end
+
+      def to_bytes
+        message_string([
+          @describe_type.to_byte,
+          @describe_name.to_cstring
+        ])
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/frontend_messages/execute.rb

@@ -0,0 +1,20 @@
+module Vertica
+  module Messages
+    class Execute < FrontendMessage
+      message_id ?E
+
+      def initialize(portal_name, max_rows)
+        @portal_name = portal_name
+        @max_rows    = max_rows
+      end
+
+      def to_bytes
+        message_string([ 
+          @portal_name.to_cstring,
+          @max_rows.to_network_int32
+        ])
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/frontend_messages/flush.rb

@@ -0,0 +1,7 @@
+module Vertica
+  module Messages
+    class Flush < FrontendMessage
+      message_id ?H
+    end
+  end
+end

data/lib/vertica/messages/frontend_messages/parse.rb

@@ -0,0 +1,23 @@
+module Vertica
+  module Messages
+    class Parse < FrontendMessage
+      message_id ?P
+
+      def initialize(name, query, param_types)
+        @name         = name
+        @query        = query
+        @param_types  = param_types
+      end
+
+      def to_bytes
+        message_string([ 
+          @name.to_cstring,
+          @query.to_cstring,
+          @param_types.length.to_network_int16,
+          @param_types.map { |type| type.to_network_int32 }
+        ].flatten)
+      end
+
+    end
+  end
+end