vertica 0.7.4

data/lib/vertica/messages/data_row.rb

@@ -0,0 +1,23 @@
+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)
+          }
+        end
+      end
+    end
+  end
+end

data/lib/vertica/messages/describe.rb

@@ -0,0 +1,30 @@
+module Vertica
+  module Messages
+    class Describe < FrontendMessage
+      message_id ?D
+
+      def initialize(describe_type, describe_name)
+        if describe_type == :portal
+          @describe_type = ?P
+        elsif describe_type == :prepared_statement
+          @describe_type = ?S
+        else
+          raise ArgumentError.new("#{describe_type} is not a valid describe_type.  Must be either :portal or :prepared_statement.")
+        end
+        @describe_name = describe_name
+      end
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE
+        size += 1
+        size += @describe_name.length + 1
+        
+        stream.write_byte(message_id)
+        stream.write_network_int32(size) # size
+        stream.write_byte(@describe_type)
+        stream.write_cstring(@describe_name)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/empty_query_response.rb

@@ -0,0 +1,8 @@
+module Vertica
+  module Messages
+    class EmptyQueryResponse < BackendMessage
+      message_id ?I
+      
+    end
+  end
+end

data/lib/vertica/messages/error_response.rb

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

data/lib/vertica/messages/execute.rb

@@ -0,0 +1,24 @@
+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(stream)
+        size = LENGTH_SIZE
+        size += @portal_name.length + 1
+        size += 4
+        
+        stream.write_byte(message_id)
+        stream.write_network_int32(size) # size
+        stream.write_cstring(@portal_name)
+        stream.write_network_int32(@max_rows)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/flush.rb

@@ -0,0 +1,15 @@
+module Vertica
+  module Messages
+    class Flush < FrontendMessage
+      message_id ?H
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE
+        
+        stream.write_byte(message_id)
+        stream.write_network_int32(size) # size
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/message.rb

@@ -0,0 +1,85 @@
+module Vertica
+  module Messages
+    class Message
+      LENGTH_SIZE   = 4
+
+      class << self
+
+        def message_id(message_id)
+          self.const_set(:MESSAGE_ID, message_id)
+          class_eval "def message_id; MESSAGE_ID end"
+        end
+
+      end
+    end
+
+    class BackendMessage < Message
+      MessageIdMap = {}
+
+      attr_reader :size
+
+      class << self
+        def message_id(message_id)
+          super
+          MessageIdMap[message_id.chr] = self
+        end
+
+        def read(stream)
+          type = stream.read_byte
+          size = stream.read_network_int32
+
+          raise Vertica::Error::MessageError.new("Bad message size: #{size}") unless size >= 4
+
+          message_klass = MessageIdMap[type.chr]
+          if message_klass.nil?
+            Messages::Unknown.new(type)
+          else
+            message_klass.new(stream, size)
+          end
+        end
+
+      end
+
+      def initialize(stream, size)
+        @size = size
+      end
+
+    end
+
+    class FrontendMessage < Message
+    end
+
+  end
+end
+
+require 'vertica/messages/unknown'
+require 'vertica/messages/error_response'
+require 'vertica/messages/startup'
+require 'vertica/messages/authentication'
+require 'vertica/messages/password'
+require 'vertica/messages/parameter_status'
+require 'vertica/messages/backend_key_data'
+require 'vertica/messages/ready_for_query'
+require 'vertica/messages/terminate'
+require 'vertica/messages/notification_response'
+require 'vertica/messages/query'
+require 'vertica/messages/notice_response'
+require 'vertica/messages/row_description'
+require 'vertica/messages/command_complete'
+require 'vertica/messages/data_row'
+require 'vertica/messages/empty_query_response'
+require 'vertica/messages/sync'
+require 'vertica/messages/ssl_request'
+require 'vertica/messages/parse'
+require 'vertica/messages/parse_complete'
+require 'vertica/messages/bind'
+require 'vertica/messages/bind_complete'
+require 'vertica/messages/describe'
+require 'vertica/messages/flush'
+require 'vertica/messages/parameter_description'
+require 'vertica/messages/no_data'
+require 'vertica/messages/execute'
+require 'vertica/messages/close'
+require 'vertica/messages/close_complete'
+require 'vertica/messages/portal_suspended'
+require 'vertica/messages/cancel_request'

data/lib/vertica/messages/no_data.rb

@@ -0,0 +1,8 @@
+module Vertica
+  module Messages
+    class NoData < BackendMessage
+      message_id ?n
+      
+    end
+  end
+end

data/lib/vertica/messages/notice_response.rb

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

data/lib/vertica/messages/notification_response.rb

@@ -0,0 +1,18 @@
+module Vertica
+  module Messages
+    class NotificationResponse < BackendMessage
+      message_id ?A
+      
+      attr_reader :pid
+      attr_reader :condition
+      attr_reader :addition_info
+      
+      def initialize(stream, size)
+        super
+        @pid           = stream.read_network_int32
+        @condition     = stream.read_cstring
+        @addition_info = stream.read_cstring
+      end
+    end
+  end
+end

data/lib/vertica/messages/parameter_description.rb

@@ -0,0 +1,19 @@
+module Vertica
+  module Messages
+    class ParameterDescription < BackendMessage
+      message_id ?t
+
+      attr_reader :parameter_count
+      attr_reader :parameter_types
+      
+      def initialize(stream, size)
+        super
+        @parameter_types = []
+        @parameter_count = stream.read_network_int16
+        @parameter_count.times do
+          @parameter_types << Vertica::Column::DATA_TYPES[stream.read_network_int32]
+        end
+      end
+    end
+  end
+end

data/lib/vertica/messages/parameter_status.rb

@@ -0,0 +1,17 @@
+module Vertica
+  module Messages
+    class ParameterStatus < BackendMessage
+      message_id ?S
+      
+      attr_reader :name
+      attr_reader :value
+      
+      def initialize(stream, size)
+        super
+        @name  = stream.read_cstring
+        @value = stream.read_cstring
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/parse.rb

@@ -0,0 +1,31 @@
+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(stream)
+        size = LENGTH_SIZE
+        size += @name.length + 1
+        size += @query.length + 1
+        size += 2
+        size += (@param_types.length * 4)
+        
+        stream.write_byte(message_id)
+        stream.write_network_int32(size) # size
+        stream.write_cstring(@name)
+        stream.write_cstring(@query)
+        stream.write_network_int16(@param_types.length)
+        @param_types.each do |param_type|
+          stream.write_network_int32(param_type)
+        end
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/parse_complete.rb

@@ -0,0 +1,8 @@
+module Vertica
+  module Messages
+    class ParseComplete < BackendMessage
+      message_id ?1
+      
+    end
+  end
+end

data/lib/vertica/messages/password.rb

@@ -0,0 +1,33 @@
+require 'digest/md5'
+          
+module Vertica
+  module Messages
+    class Password < FrontendMessage
+      message_id ?p
+
+      def initialize(password, authentication_method = Messages::Authentication::CLEARTEXT_PASSWORD, options = {})
+        case authentication_method
+        when Messages::Authentication::CLEARTEXT_PASSWORD
+          @password = password
+        when Messages::Authentication::CRYPT_PASSWORD
+          @password = password.crypt(options[:salt])
+        when Messages::Authentication::MD5_PASSWORD
+          @password = Digest::MD5.hexdigest(password + options[:user])
+          @password = Digest::MD5.hexdigest(m + options[:salt])
+          @password = 'md5' + @password
+        else
+          raise ArgumentError.new("unsupported authentication method: #{authentication_method}")
+        end
+      end
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE
+        size += @password.length + 1
+        stream.write_byte(message_id)
+        stream.write_network_int32(size) # size
+        stream.write_cstring(@password)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/portal_suspended.rb

@@ -0,0 +1,8 @@
+module Vertica
+  module Messages
+    class PortalSuspended < BackendMessage
+      message_id ?s
+
+    end
+  end
+end

data/lib/vertica/messages/query.rb

@@ -0,0 +1,20 @@
+module Vertica
+  module Messages
+    class Query < FrontendMessage
+      message_id ?Q
+
+      def initialize(query_string)
+        @query_string = query_string
+      end
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE
+        size += @query_string.length + 1
+        stream.write_byte(message_id.bytes.first)
+        stream.write_network_int32(size) # size
+        stream.write_cstring(@query_string)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/ready_for_query.rb

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

data/lib/vertica/messages/row_description.rb

@@ -0,0 +1,29 @@
+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 << {
+            :name             => stream.read_cstring,
+            :table_oid        => stream.read_network_int32,
+            :attribute_number => stream.read_network_int16,
+            :data_type_oid    => stream.read_network_int32,
+            :data_type_size   => stream.read_network_int16,
+            :type_modifier    => stream.read_network_int32,
+            :format_code      => stream.read_network_int16
+          }
+        end
+      end
+    end
+  end
+end

data/lib/vertica/messages/ssl_request.rb

@@ -0,0 +1,14 @@
+module Vertica
+  module Messages
+    class SslRequest < FrontendMessage
+      message_id nil
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE + 4
+        stream.write_network_int32(size) # size
+        stream.write_network_int32(80877103) # size
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/startup.rb

@@ -0,0 +1,38 @@
+module Vertica
+  module Messages
+    class Startup < FrontendMessage
+      message_id nil
+
+      def initialize(user, database, options = nil)
+        @user     = user
+        @database = database
+        @options  = options
+      end
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE + 4 # length + protocol
+        size += @user.length     + 4 + 2 if @user
+        size += @database.length + 8 + 2 if @database
+        size += @options.length  + 7 + 2 if @options
+        size += 1 # ending zero
+
+        stream.write_network_int32(size) # size
+        stream.write_network_int32(Vertica::PROTOCOL_VERSION) # proto version
+        if @user
+          stream.write_cstring('user')
+          stream.write_cstring(@user)
+        end
+        if @database
+          stream.write_cstring('database')
+          stream.write_cstring(@database)
+        end
+        if @options
+          stream.write_cstring('options')
+          stream.write_cstring(@options)
+        end
+        stream.write_byte(0)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/sync.rb

@@ -0,0 +1,15 @@
+module Vertica
+  module Messages
+    class Sync < FrontendMessage
+      message_id ?S
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE
+        
+        stream.write_byte(message_id)
+        stream.write_network_int32(size)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/terminate.rb

@@ -0,0 +1,14 @@
+module Vertica
+  module Messages
+    class Terminate < FrontendMessage
+      message_id ?X
+
+      def to_bytes(stream)
+        size = LENGTH_SIZE
+        stream.write_byte(message_id.bytes.first)
+        stream.write_network_int32(size)
+      end
+
+    end
+  end
+end

data/lib/vertica/messages/unknown.rb

@@ -0,0 +1,11 @@
+module Vertica
+  module Messages
+    class Unknown < BackendMessage
+      attr_reader :message_id
+      
+      def initialize(message_id)
+        @message_id = message_id
+      end
+    end
+  end
+end

data/lib/vertica/notice.rb

@@ -0,0 +1,11 @@
+module Vertica
+  class Notice
+    attr_reader :field_type
+    attr_reader :field_value
+    
+    def initialize(field_type, field_value)
+      @field_type  = field_type
+      @field_value = field_value
+    end
+  end
+end

data/lib/vertica/notification.rb

@@ -0,0 +1,13 @@
+module Vertica
+  class Notification
+    attr_reader :process_pid
+    attr_reader :condition
+    attr_reader :additional_info
+    
+    def initialize(process_pid, condition, additional_info)
+      @process_pid     = process_pid
+      @condition       = condition
+      @additional_info = additional_info
+    end
+  end
+end

data/lib/vertica/result.rb

@@ -0,0 +1,28 @@
+module Vertica
+  class Result
+    
+    def initialize(field_descriptions, field_values)
+      @field_descriptions = field_descriptions
+      @field_values = field_values
+    end
+    
+    def row_count
+      @row_count ||= @field_values.length
+    end
+    
+    def columns
+      @columns ||= @field_descriptions.map { |fd| Column.new(fd[:type_modifier], fd[:format_code], fd[:table_oid], fd[:name], fd[:attribute_number], fd[:data_type_oid], fd[:data_type_size]) }
+    end
+    
+    def rows
+      @field_values.map do |fv|
+        index = 0
+        fv.map do |f|
+          index += 1
+          self.columns[index-1].convert(f[:value])
+        end
+      end
+    end
+    
+  end
+end

data/lib/vertica/vertica_socket.rb

@@ -0,0 +1,8 @@
+require 'socket'
+require 'vertica/bit_helper'
+
+module Vertica
+  class VerticaSocket < TCPSocket
+    include BitHelper
+  end
+end

data/lib/vertica.rb

@@ -0,0 +1,19 @@
+module Vertica
+  
+  class Error < StandardError
+    
+    class ConnectionError < Error; end
+    
+    class MessageError < Error; end
+    
+  end
+  
+  PROTOCOL_VERSION = 3 << 16
+  
+  VERSION = "0.7.3"
+  
+end
+
+require 'vertica/column'
+require 'vertica/result'
+require 'vertica/connection'