vertica 0.12.0 → 1.0.0.rc1

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Authentication < BackendMessage
       message_id 'R'
 

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class BackendKeyData < BackendMessage
       message_id 'K'
 

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class BindComplete < BackendMessage
       message_id '2'
     end

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class CloseComplete < BackendMessage
       message_id '3'
     end

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class CommandComplete < BackendMessage
       message_id 'C'
 

data/lib/vertica/{messages/backend_messages → protocol/backend}/copy_in_response.rb

@@ -1,8 +1,8 @@
 module Vertica
-  module Messages
+  module Protocol
     class CopyInResponse < BackendMessage
       message_id 'G'
-      
+
       def initialize(data)
         values = data.unpack('Cn*')
         @format = values[0]

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class DataRow < BackendMessage
       message_id 'D'
 

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class EmptyQueryResponse < BackendMessage
       message_id 'I'
     end

data/lib/vertica/{messages/backend_messages → protocol/backend}/error_response.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class ErrorResponse < NoticeResponse
       message_id 'E'
     end

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class NoData < BackendMessage
       message_id 'n'
     end

data/lib/vertica/{messages/backend_messages → protocol/backend}/notice_response.rb

@@ -1,8 +1,8 @@
 module Vertica
-  module Messages
+  module Protocol
     class NoticeResponse < BackendMessage
       message_id 'N'
-      
+
       FIELDS_DEFINITIONS = [
         { :type => 'q', :name => "Internal Query", :method => :internal_query },
         { :type => 'S', :name => "Severity", :method => :severity },
@@ -17,9 +17,9 @@ module Vertica
         { :type => 'F', :name => "File", :method => :file },
         { :type => 'L', :name => "Line", :method => :line }
       ]
-      
+
       FIELDS = Hash[*FIELDS_DEFINITIONS.map { |f| [f[:type], f[:name]] }.flatten]
-      
+
       attr_reader :values
 
       def initialize(data)
@@ -32,12 +32,12 @@ module Vertica
       end
 
       def error_message
-        ordered_values = FIELDS_DEFINITIONS.map do |field| 
+        ordered_values = FIELDS_DEFINITIONS.map do |field|
           "#{field[:name]}: #{@values[field[:name]]}" if @values[field[:name]]
         end
         ordered_values.compact.join(', ')
       end
-      
+
       FIELDS_DEFINITIONS.each do |field_def|
         define_method(field_def[:method]) do
           @values[field_def[:name]]

data/lib/vertica/{messages/backend_messages → protocol/backend}/parameter_description.rb

@@ -1,10 +1,10 @@
 module Vertica
-  module Messages
+  module Protocol
     class ParameterDescription < BackendMessage
       message_id 't'
 
       attr_reader :parameter_types
-      
+
       def initialize(data)
         parameter_count    = data.unpack('n').first
         parameter_type_ids = data.unpack("@2N#{parameter_count}")

data/lib/vertica/{messages/backend_messages → protocol/backend}/parameter_status.rb

@@ -1,10 +1,10 @@
 module Vertica
-  module Messages
+  module Protocol
     class ParameterStatus < BackendMessage
       message_id 'S'
-      
+
       attr_reader :name, :value
-      
+
       def initialize(data)
         @name, @value = data.unpack('Z*Z*')
       end

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class ParseComplete < BackendMessage
       message_id '1'
     end

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class PortalSuspended < BackendMessage
       message_id 's'
     end

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

@@ -1,7 +1,7 @@
 module Vertica
-  module Messages
+  module Protocol
     class ReadyForQuery < BackendMessage
-      
+
       STATUSES = {
         'I' => :no_transaction,
         'T' => :in_transaction,

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class RowDescription < BackendMessage
       message_id 'T'
 
@@ -12,15 +12,15 @@ module Vertica
         field_count.times do |field_index|
           field_info = data.unpack("@#{pos}Z*NnNnNn")
           @fields << {
-            :name             => field_info[0].force_encoding('UTF-8'),
-            :table_oid        => field_info[1],
-            :attribute_number => field_info[2],
-            :data_type_oid    => field_info[3],
-            :data_type_size   => field_info[4],
-            :type_modifier    => field_info[5],
-            :format_code      => field_info[6],
+            :name               => field_info[0].force_encoding('UTF-8'),
+            :table_oid          => field_info[1],
+            :attribute_number   => field_info[2],
+            :data_type_oid      => field_info[3],
+            :data_type_size     => field_info[4],
+            :data_type_modifier => field_info[5],
+            :format_code        => field_info[6],
           }
-          
+
           pos += 19 + field_info[0].size
         end
       end

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

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Unknown < BackendMessage
       attr_reader :message_id
 

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/bind.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Bind < FrontendMessage
       message_id 'B'
 
@@ -9,11 +9,10 @@ module Vertica
         @parameter_values = parameter_values
       end
 
-      def to_bytes
+      def message_body
         bytes = [@portal_name, @prepared_statement_name, 0, @parameter_values.length].pack('Z*Z*nn')
         bytes << @parameter_values.map { |val| val.nil? ? [-1].pack('N') : [val.length, val].pack('Na*') }.join('')
         bytes << [0].pack('n')
-        message_string bytes
       end
     end
   end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/cancel_request.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class CancelRequest < FrontendMessage
       message_id nil
 
@@ -8,10 +8,9 @@ module Vertica
         @backend_key = backend_key
       end
 
-      def to_bytes
-        message_string [80877102, @backend_pid, @backend_key].pack('N3')
+      def message_body
+        [80877102, @backend_pid, @backend_key].pack('N3')
       end
-
     end
   end
 end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/close.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Close < FrontendMessage
       message_id 'C'
 
@@ -12,8 +12,8 @@ module Vertica
         end
       end
 
-      def to_bytes
-        message_string [@close_type, @close_name].pack('AZ*')
+      def message_body
+        [@close_type, @close_name].pack('AZ*')
       end
     end
   end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/copy_data.rb

@@ -1,15 +1,15 @@
 module Vertica
-  module Messages
+  module Protocol
     class CopyData < FrontendMessage
       message_id 'd'
-      
+
       def initialize(data)
         @data = data
       end
-      
-      def to_bytes
-        message_string [@data].pack('a*')
-      end      
+
+      def message_body
+        @data
+      end
     end
   end
 end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/copy_done.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class CopyDone < FrontendMessage
       message_id 'c'
     end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/copy_fail.rb

@@ -1,14 +1,14 @@
 module Vertica
-  module Messages
+  module Protocol
     class CopyFail < FrontendMessage
       message_id 'f'
-      
+
       def initialize(error_message)
         @error_message = error_message
       end
-      
-      def to_bytes
-        message_string [@error_message].pack('Z*')
+
+      def message_body
+        [@error_message].pack('Z*')
       end
     end
   end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/describe.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Describe < FrontendMessage
       message_id 'D'
 
@@ -12,8 +12,8 @@ module Vertica
         end
       end
 
-      def to_bytes
-        message_string [@describe_type, @describe_name].pack('AZ*')
+      def message_body
+        [@describe_type, @describe_name].pack('AZ*')
       end
     end
   end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/execute.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Execute < FrontendMessage
       message_id 'E'
 
@@ -8,8 +8,8 @@ module Vertica
         @max_rows    = max_rows
       end
 
-      def to_bytes
-        message_string [@portal_name, @max_rows].pack('Z*N')
+      def message_body
+        [@portal_name, @max_rows].pack('Z*N')
       end
     end
   end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/flush.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Flush < FrontendMessage
       message_id 'H'
     end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/parse.rb

@@ -1,5 +1,5 @@
 module Vertica
-  module Messages
+  module Protocol
     class Parse < FrontendMessage
       message_id 'P'
 
@@ -9,8 +9,8 @@ module Vertica
         @param_types  = param_types
       end
 
-      def to_bytes
-        message_string([@name, @query, @param_types.length, *@param_types].pack('Z*Z*nN*'))
+      def message_body
+        [@name, @query, @param_types.length, *@param_types].pack('Z*Z*nN*')
       end
     end
   end

data/lib/vertica/protocol/frontend/password.rb

@@ -0,0 +1,32 @@
+module Vertica
+  module Protocol
+    class Password < FrontendMessage
+      message_id 'p'
+
+      def initialize(password, auth_method: Vertica::Protocol::Authentication::CLEARTEXT_PASSWORD, salt: nil, user: nil)
+        @password = password
+        @auth_method, @salt, @user = auth_method, salt, user
+      end
+
+      def encoded_password
+        case @auth_method
+        when Vertica::Protocol::Authentication::CLEARTEXT_PASSWORD
+          @password
+        when Vertica::Protocol::Authentication::CRYPT_PASSWORD
+          @password.crypt(@salt)
+        when Vertica::Protocol::Authentication::MD5_PASSWORD
+          require 'digest/md5'
+          @password = Digest::MD5.hexdigest("#{@password}#{@user}")
+          @password = Digest::MD5.hexdigest("#{@password}#{@salt}")
+          @password = "md5#{@password}"
+        else
+          raise ArgumentError.new("unsupported authentication method: #{@auth_method}")
+        end
+      end
+
+      def message_body
+        [encoded_password].pack('Z*')
+      end
+    end
+  end
+end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/query.rb

@@ -1,6 +1,5 @@
 module Vertica
-  module Messages
-
+  module Protocol
     class Query < FrontendMessage
       message_id 'Q'
 
@@ -8,8 +7,8 @@ module Vertica
         @query_string = query_string
       end
 
-      def to_bytes
-        message_string [@query_string].pack('Z*')
+      def message_body
+        [@query_string].pack('Z*')
       end
     end
   end

data/lib/vertica/{messages/frontend_messages → protocol/frontend}/ssl_request.rb

@@ -1,10 +1,10 @@
 module Vertica
-  module Messages
+  module Protocol
     class SslRequest < FrontendMessage
       message_id nil
 
       def to_bytes
-        message_string [80877103].pack('N')
+        [80877103].pack('N')
       end
     end
   end