yugabyte-ycql-driver 3.2.3.1

data/lib/cassandra/protocol/request.rb

@@ -0,0 +1,41 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class Request
+      extend AttrBoolean
+
+      attr_reader :opcode
+      attr_boolean :trace
+
+      def initialize(opcode, trace = false)
+        @opcode = opcode
+        @trace = trace
+      end
+
+      def compressable?
+        true
+      end
+
+      def payload?
+        false
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/requests/auth_response_request.rb

@@ -0,0 +1,51 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class AuthResponseRequest < Request
+      attr_reader :token
+
+      def initialize(token)
+        super(0x0f)
+        @token = token
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_bytes(@token)
+      end
+
+      def to_s
+        %(AUTH_RESPONSE #{@token.bytesize})
+      end
+
+      def eql?(other)
+        token == other.token
+      end
+      alias == eql?
+
+      def hash
+        @h ||= begin
+          h = 17
+          h = 31 * h + @token.hash
+          h
+        end
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/requests/batch_request.rb

@@ -0,0 +1,117 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class BatchRequest < Request
+      LOGGED_TYPE = 0
+      UNLOGGED_TYPE = 1
+      COUNTER_TYPE = 2
+
+      attr_reader :type, :timestamp, :payload
+      attr_accessor :consistency, :retries
+
+      def initialize(type,
+                     consistency,
+                     trace = false,
+                     serial_consistency = nil,
+                     timestamp = nil,
+                     payload = nil)
+        super(0x0D, trace)
+        @type  = type
+        @parts = []
+        @consistency = consistency
+        @serial_consistency = serial_consistency
+        @timestamp = timestamp
+        @payload = payload
+      end
+
+      def payload?
+        !!@payload
+      end
+
+      def clear
+        @parts.clear
+        nil
+      end
+
+      def add_query(cql, values, types)
+        @parts << [:simple, cql, values, types]
+        nil
+      end
+
+      def add_prepared(id, values, types)
+        @parts << [:prepared, id, values, types]
+        nil
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append(@type.chr)
+        buffer.append_short(@parts.size)
+
+        @parts.each do |(statement_kind, *arguments)|
+          __send__(:"write_#{statement_kind}",
+                   buffer,
+                   protocol_version,
+                   encoder,
+                   *arguments)
+        end
+
+        buffer.append_consistency(@consistency)
+
+        if protocol_version > 2
+          flags  = 0
+          flags |= 0x10 if @serial_consistency
+          flags |= 0x20 if @timestamp
+
+          buffer.append(flags.chr)
+          buffer.append_consistency(@serial_consistency) if @serial_consistency
+          buffer.append_long(@timestamp) if @timestamp
+        end
+
+        buffer
+      end
+
+      def to_s
+        type_str = case @type
+                   when LOGGED_TYPE then 'LOGGED'
+                   when UNLOGGED_TYPE then 'UNLOGGED'
+                   when COUNTER_TYPE then 'COUNTER'
+                   end
+        %(BATCH #{type_str} #{@parts.size} #{@consistency.to_s.upcase})
+      end
+
+      private
+
+      QUERY_KIND = "\x00".freeze
+      PREPARED_KIND = "\x01".freeze
+
+      def write_simple(buffer, protocol_version, encoder, cql, values, types)
+        buffer.append(QUERY_KIND)
+        buffer.append_long_string(cql)
+        encoder.write_parameters(buffer, values, types)
+      end
+
+      def write_prepared(buffer, protocol_version, encoder, id, values, types)
+        buffer.append(PREPARED_KIND)
+        buffer.append_short_bytes(id)
+        encoder.write_parameters(buffer, values, types)
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/requests/credentials_request.rb

@@ -0,0 +1,51 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class CredentialsRequest < Request
+      attr_reader :credentials
+
+      def initialize(credentials)
+        super(4)
+        @credentials = credentials.dup.freeze
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_string_map(@credentials)
+      end
+
+      def to_s
+        %(CREDENTIALS #{@credentials})
+      end
+
+      def eql?(rq)
+        rq.is_a?(self.class) && rq.credentials.eql?(@credentials)
+      end
+      alias == eql?
+
+      def hash
+        @h ||= begin
+          h = 17
+          h = 31 * h + @credentials.hash
+          h
+        end
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/requests/execute_request.rb

@@ -0,0 +1,122 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class ExecuteRequest < Request
+      attr_reader :metadata, :request_metadata, :page_size, :paging_state, :payload,
+                  :serial_consistency, :timestamp, :values
+      attr_accessor :consistency, :id, :retries
+
+      def initialize(id,
+                     metadata,
+                     values,
+                     request_metadata,
+                     consistency,
+                     serial_consistency = nil,
+                     page_size = nil,
+                     paging_state = nil,
+                     trace = false,
+                     timestamp = nil,
+                     payload = nil)
+        if metadata.size != values.size
+          raise ArgumentError, "Metadata for #{metadata.size} columns, but " \
+              "#{values.size} values given"
+        end
+        if consistency.nil? || !CONSISTENCIES.include?(consistency)
+          raise ArgumentError, %(No such consistency: #{consistency.inspect})
+        end
+        unless serial_consistency.nil? || CONSISTENCIES.include?(serial_consistency)
+          raise ArgumentError, %(No such consistency: #{serial_consistency.inspect})
+        end
+        raise ArgumentError, %(Paging state given but no page size) if paging_state && !page_size
+        super(10, trace)
+        @id = id
+        @metadata = metadata
+        @values = values
+        @request_metadata = request_metadata
+        @consistency = consistency
+        @serial_consistency = serial_consistency
+        @page_size = page_size
+        @paging_state = paging_state
+        @timestamp = timestamp
+        @payload = payload
+      end
+
+      def payload?
+        !!@payload
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_short_bytes(@id)
+        if protocol_version > 1
+          buffer.append_consistency(@consistency)
+          flags  = 0
+          flags |= 0x01 unless @values.empty?
+          flags |= 0x02 unless @request_metadata
+          flags |= 0x04 if @page_size
+          flags |= 0x08 if @paging_state
+          flags |= 0x10 if @serial_consistency
+          flags |= 0x20 if protocol_version > 2 && @timestamp
+          buffer.append(flags.chr)
+          encoder.write_parameters(buffer, @values, @metadata) unless @values.empty?
+          buffer.append_int(@page_size) if @page_size
+          buffer.append_bytes(@paging_state) if @paging_state
+          buffer.append_consistency(@serial_consistency) if @serial_consistency
+          buffer.append_long(@timestamp) if protocol_version > 2 && @timestamp
+        else
+          encoder.write_parameters(buffer, @values, @metadata)
+          buffer.append_consistency(@consistency)
+        end
+        buffer
+      end
+
+      def to_s
+        id = @id.each_byte.map { |x| x.to_s(16) }.join('')
+        %(EXECUTE #{id} #{@values} #{@consistency.to_s.upcase})
+      end
+
+      def eql?(rq)
+        rq.is_a?(self.class) &&
+          rq.id == id &&
+          rq.metadata == metadata &&
+          rq.values == values &&
+          rq.consistency == consistency &&
+          rq.serial_consistency == serial_consistency &&
+          rq.page_size == page_size &&
+          rq.paging_state == paging_state
+      end
+      alias == eql?
+
+      def hash
+        @h ||= begin
+          h = 17
+          h = 31 * h + @id.hash
+          h = 31 * h + @metadata.hash
+          h = 31 * h + @values.hash
+          h = 31 * h + @consistency.hash
+          h = 31 * h + @serial_consistency.hash
+          h = 31 * h + @page_size.hash
+          h = 31 * h + @paging_state.hash
+          h = 31 * h + @timestamp.hash
+          h
+        end
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/requests/options_request.rb

@@ -0,0 +1,39 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class OptionsRequest < Request
+      def initialize
+        super(5)
+      end
+
+      def compressable?
+        false
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer
+      end
+
+      def to_s
+        %(OPTIONS)
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/requests/prepare_request.rb

@@ -0,0 +1,59 @@
+# encoding: utf-8
+
+#--
+# Copyright DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+module Cassandra
+  module Protocol
+    class PrepareRequest < Request
+      attr_reader :cql, :payload
+      attr_accessor :consistency, :retries
+
+      def initialize(cql, trace = false, payload = nil)
+        raise ArgumentError, 'No CQL given!' unless cql
+        super(9, trace)
+        @cql = cql
+        @consistency = :quorum
+        @payload = payload
+      end
+
+      def payload?
+        !!@payload
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_long_string(@cql)
+      end
+
+      def to_s
+        %(PREPARE "#{@cql}")
+      end
+
+      def eql?(rq)
+        rq.is_a?(self.class) && rq.cql == cql
+      end
+      alias == eql?
+
+      def hash
+        @h ||= begin
+          h = 17
+          h = 31 * h + @cql.hash
+          h
+        end
+      end
+    end
+  end
+end