yugabyte-ycql-driver 3.2.3.1

data/lib/cassandra/protocol/responses/event_response.rb

@@ -0,0 +1,30 @@
+# 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 EventResponse < ResultResponse
+      # @private
+      RESPONSE_TYPES[0x0c] = self
+      # @private
+      EVENT_TYPES = {
+        # populated by subclasses
+      }
+    end
+  end
+end

data/lib/cassandra/protocol/responses/function_failure_error_response.rb

@@ -0,0 +1,52 @@
+# 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 FunctionFailureErrorResponse < ErrorResponse
+      attr_reader :keyspace, :name, :signature
+
+      def initialize(custom_payload, warnings, code, message, keyspace, name, signature)
+        super(custom_payload, warnings, code, message)
+
+        @keyspace  = keyspace
+        @name      = name
+        @signature = signature
+      end
+
+      def to_error(keyspace, statement, options, hosts, consistency, retries)
+        Errors::FunctionCallError.new(@message,
+                                      @custom_payload,
+                                      @warnings,
+                                      keyspace,
+                                      statement,
+                                      options,
+                                      hosts,
+                                      consistency,
+                                      retries,
+                                      @keyspace,
+                                      @name,
+                                      @signature)
+      end
+
+      def to_s
+        "#{super} #{@keyspace} #{@name} #{@signature}"
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/prepared_result_response.rb

@@ -0,0 +1,62 @@
+# 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 PreparedResultResponse < ResultResponse
+      # @private
+      RESULT_TYPES[0x04] = self
+
+      attr_reader :id, :metadata, :result_metadata, :pk_idx
+
+      def initialize(custom_payload,
+                     warnings,
+                     id,
+                     metadata,
+                     result_metadata,
+                     pk_idx,
+                     trace_id)
+        super(custom_payload, warnings, trace_id)
+        @id              = id
+        @metadata        = metadata
+        @result_metadata = result_metadata
+        @pk_idx          = pk_idx
+      end
+
+      def eql?(other)
+        id == other.id && metadata == other.metadata && trace_id == other.trace_id
+      end
+      alias == eql?
+
+      def hash
+        @h ||= begin
+          h = 17
+          h = 31 * h + @id.hash
+          h = 31 * h + @metadata.hash
+          h = 31 * h + @trace_id.hash
+          h
+        end
+      end
+
+      def to_s
+        hex_id = @id.each_byte.map { |x| x.to_s(16).rjust(2, '0') }.join('')
+        %(RESULT PREPARED #{hex_id} #{@metadata})
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/raw_rows_result_response.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 RawRowsResultResponse < RowsResultResponse
+      def initialize(custom_payload,
+                     warnings,
+                     protocol_version,
+                     raw_rows,
+                     paging_state,
+                     trace_id,
+                     custom_type_handlers = nil)
+        super(custom_payload, warnings, nil, nil, paging_state, trace_id)
+        @protocol_version = protocol_version
+        @raw_rows = raw_rows
+        @custom_type_handlers = custom_type_handlers
+      end
+
+      def materialize(metadata)
+        @metadata = metadata
+
+        @rows = if @protocol_version == 4
+                  Coder.read_values_v4(@raw_rows, @metadata, @custom_type_handlers)
+                elsif @protocol_version == 3
+                  Coder.read_values_v3(@raw_rows, @metadata)
+                else
+                  Coder.read_values_v1(@raw_rows, @metadata)
+                end
+
+        @rows
+      end
+
+      def rows
+        raise Errors::DecodingError, 'Not materialized!' unless @rows
+        @rows
+      end
+
+      def to_s
+        %(RESULT ROWS (raw))
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/read_failure_error_response.rb

@@ -0,0 +1,71 @@
+# 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 ReadFailureErrorResponse < ErrorResponse
+      attr_reader :consistency, :received, :blockfor, :numfailures, :data_present, :failures_by_node
+
+      def initialize(custom_payload,
+                     warnings,
+                     code,
+                     message,
+                     consistency,
+                     received,
+                     blockfor,
+                     numfailures,
+                     data_present,
+                     failures_by_node)
+        super(custom_payload, warnings, code, message)
+
+        @consistency  = consistency
+        @received     = received
+        @blockfor     = blockfor
+        @data_present = data_present
+        @failures_by_node = failures_by_node
+
+        # If failures_by_node is set, numfailures isn't, and v.v. Set @numfailures to the size of the failure-map
+        # if numfailures is nil.
+        @numfailures = numfailures || @failures_by_node.size
+      end
+
+      def to_error(keyspace, statement, options, hosts, consistency, retries)
+        Errors::ReadError.new(@message,
+                              @custom_payload,
+                              @warnings,
+                              keyspace,
+                              statement,
+                              options,
+                              hosts,
+                              consistency,
+                              retries,
+                              @data_present,
+                              @consistency,
+                              @blockfor,
+                              @numfailures,
+                              @received,
+                              @failures_by_node)
+      end
+
+      def to_s
+        "#{super} #{@consistency} #{@received} #{@blockfor} #{@numfailures} " \
+            "#{@data_present}"
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/read_timeout_error_response.rb

@@ -0,0 +1,61 @@
+# 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 ReadTimeoutErrorResponse < ErrorResponse
+      attr_reader :consistency, :received, :blockfor, :data_present
+
+      def initialize(custom_payload,
+                     warnings,
+                     code,
+                     message,
+                     consistency,
+                     received,
+                     blockfor,
+                     data_present)
+        super(custom_payload, warnings, code, message)
+
+        @consistency  = consistency
+        @received     = received
+        @blockfor     = blockfor
+        @data_present = data_present
+      end
+
+      def to_error(keyspace, statement, options, hosts, consistency, retries)
+        Errors::ReadTimeoutError.new(@message,
+                                     @custom_payload,
+                                     @warnings,
+                                     keyspace,
+                                     statement,
+                                     options,
+                                     hosts,
+                                     consistency,
+                                     retries,
+                                     @data_present,
+                                     @consistency,
+                                     @blockfor,
+                                     @received)
+      end
+
+      def to_s
+        "#{super} #{@consistency} #{@received} #{@blockfor} #{@data_present}"
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/ready_response.rb

@@ -0,0 +1,43 @@
+# 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 ReadyResponse < Response
+      # @private
+      RESPONSE_TYPES[0x02] = self
+
+      def eql?(rs)
+        rs.is_a?(self.class)
+      end
+      alias == eql?
+
+      def hash
+        @h ||= begin
+          h = 17
+          h = 31 * h + 'READY'.hash
+          h
+        end
+      end
+
+      def to_s
+        'READY'
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/result_response.rb

@@ -0,0 +1,42 @@
+# 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 ResultResponse < Response
+      # @private
+      RESPONSE_TYPES[0x08] = self
+      # @private
+      RESULT_TYPES = [
+        # populated by subclasses
+      ]
+
+      attr_reader :custom_payload, :warnings, :trace_id
+
+      def initialize(custom_payload, warnings, trace_id)
+        @custom_payload = custom_payload
+        @warnings = warnings
+        @trace_id = trace_id
+      end
+
+      def void?
+        false
+      end
+    end
+  end
+end

data/lib/cassandra/protocol/responses/rows_result_response.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 RowsResultResponse < ResultResponse
+      # @private
+      RESULT_TYPES[0x02] = self
+
+      attr_reader :rows, :metadata, :paging_state
+
+      def initialize(custom_payload, warnings, rows, metadata, paging_state, trace_id)
+        super(custom_payload, warnings, trace_id)
+        @rows = rows
+        @metadata = metadata
+        @paging_state = paging_state
+      end
+
+      def to_s
+        %(RESULT ROWS #{@metadata} #{@rows})
+      end
+    end
+  end
+end