yugabyte-ycql-driver 3.2.3.1

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

@@ -0,0 +1,112 @@
+# 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 QueryRequest < Request
+      attr_reader :cql, :page_size, :paging_state, :payload, :serial_consistency,
+                  :timestamp, :type_hints, :values
+      attr_accessor :consistency, :retries
+
+      def initialize(cql,
+                     values,
+                     type_hints,
+                     consistency,
+                     serial_consistency = nil,
+                     page_size = nil,
+                     paging_state = nil,
+                     trace = false,
+                     names = EMPTY_LIST,
+                     timestamp = nil,
+                     payload = nil)
+        super(7, trace)
+        @cql = cql
+        @values = values
+        @type_hints = type_hints
+        @consistency = consistency
+        @serial_consistency = serial_consistency
+        @page_size = page_size
+        @paging_state = paging_state
+        @names = names
+        @timestamp = timestamp
+        @payload = payload
+      end
+
+      def payload?
+        !!@payload
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_long_string(@cql)
+        buffer.append_consistency(@consistency)
+        if protocol_version > 1
+          flags  = 0
+          flags |= 0x04 if @page_size
+          flags |= 0x08 if @paging_state
+          flags |= 0x10 if @serial_consistency
+          flags |= 0x20 if protocol_version > 2 && @timestamp
+          if @values && !@values.empty?
+            flags |= 0x01
+            flags |= 0x40 if protocol_version > 2 && !@names.empty?
+            buffer.append(flags.chr)
+            encoder.write_parameters(buffer, @values, @type_hints, @names)
+          else
+            buffer.append(flags.chr)
+          end
+          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
+        end
+        buffer
+      end
+
+      def to_s
+        %(QUERY "#{@cql}" #{@consistency.to_s.upcase})
+      end
+
+      def eql?(rq)
+        rq.is_a?(self.class) &&
+          rq.cql == cql &&
+          rq.values == values &&
+          rq.type_hints == type_hints &&
+          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 + @cql.hash
+          h = 31 * h + @values.hash
+          h = 31 * h + @type_hints.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 + @names.hash
+          h = 31 * h + @timestamp.hash
+          h
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,38 @@
+# 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 RegisterRequest < Request
+      attr_reader :events
+
+      def initialize(*events)
+        super(11)
+        @events = events
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_string_list(@events)
+      end
+
+      def to_s
+        %(REGISTER #{@events})
+      end
+    end
+  end
+end

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

@@ -0,0 +1,49 @@
+# 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 StartupRequest < Request
+      # @private
+      CQL_VERSION = 'CQL_VERSION'.freeze
+      # @private
+      COMPRESSION = 'COMPRESSION'.freeze
+
+      attr_reader :options
+
+      def initialize(cql_version, compression = nil)
+        super(1)
+        raise ArgumentError, "Invalid CQL version: #{cql_version.inspect}" unless cql_version
+        @options = {CQL_VERSION => cql_version}
+        @options[COMPRESSION] = compression if compression
+      end
+
+      def compressable?
+        false
+      end
+
+      def write(buffer, protocol_version, encoder)
+        buffer.append_string_map(@options)
+      end
+
+      def to_s
+        %(STARTUP #{@options})
+      end
+    end
+  end
+end

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

@@ -0,0 +1,24 @@
+# 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 VoidQueryRequest
+    end
+  end
+end

data/lib/cassandra/protocol/response.rb

@@ -0,0 +1,28 @@
+# 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 Response
+      # @private
+      RESPONSE_TYPES = [
+        # populated by subclasses
+      ]
+    end
+  end
+end

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

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

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

@@ -0,0 +1,36 @@
+# 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 AuthChallengeResponse < Response
+      # @private
+      RESPONSE_TYPES[0x0e] = self
+
+      attr_reader :token
+
+      def initialize(token)
+        @token = token
+      end
+
+      def to_s
+        %(AUTH_CHALLENGE #{@token.bytesize})
+      end
+    end
+  end
+end

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

@@ -0,0 +1,36 @@
+# 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 AuthSuccessResponse < Response
+      # @private
+      RESPONSE_TYPES[0x10] = self
+
+      attr_reader :token
+
+      def initialize(token)
+        @token = token
+      end
+
+      def to_s
+        %(AUTH_SUCCESS #{@token && @token.bytesize})
+      end
+    end
+  end
+end

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

@@ -0,0 +1,36 @@
+# 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 AuthenticateResponse < Response
+      # @private
+      RESPONSE_TYPES[0x03] = self
+
+      attr_reader :authentication_class
+
+      def initialize(authentication_class)
+        @authentication_class = authentication_class
+      end
+
+      def to_s
+        %(AUTHENTICATE #{authentication_class})
+      end
+    end
+  end
+end

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

@@ -0,0 +1,142 @@
+# 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 ErrorResponse < Response
+      # @private
+      RESPONSE_TYPES[0x00] = self
+
+      attr_reader :code, :message, :custom_payload, :warnings
+
+      def initialize(*args)
+        @custom_payload, @warnings, @code, @message = args
+      end
+
+      def to_s
+        hex_code = @code.to_s(16).rjust(4, '0').upcase
+        %(ERROR 0x#{hex_code} "#{@message}")
+      end
+
+      def to_error(keyspace, statement, options, hosts, consistency, retries)
+        case @code
+        when 0x0000 then Errors::ServerError.new(@message,
+                                                 @custom_payload,
+                                                 @warnings,
+                                                 keyspace,
+                                                 statement,
+                                                 options,
+                                                 hosts,
+                                                 consistency,
+                                                 retries)
+        when 0x000A then Errors::ProtocolError.new(@message,
+                                                   @custom_payload,
+                                                   @warnings,
+                                                   keyspace,
+                                                   statement,
+                                                   options,
+                                                   hosts,
+                                                   consistency,
+                                                   retries)
+        when 0x0100 then Errors::AuthenticationError.new(@message,
+                                                         @custom_payload,
+                                                         @warnings,
+                                                         keyspace,
+                                                         statement,
+                                                         options,
+                                                         hosts,
+                                                         consistency,
+                                                         retries)
+        when 0x1001 then Errors::OverloadedError.new(@message,
+                                                     @custom_payload,
+                                                     @warnings,
+                                                     keyspace,
+                                                     statement,
+                                                     options,
+                                                     hosts,
+                                                     consistency,
+                                                     retries)
+        when 0x1002 then Errors::IsBootstrappingError.new(@message,
+                                                          @custom_payload,
+                                                          @warnings,
+                                                          keyspace,
+                                                          statement,
+                                                          options,
+                                                          hosts,
+                                                          consistency,
+                                                          retries)
+        when 0x1003 then Errors::TruncateError.new(@message,
+                                                   @custom_payload,
+                                                   @warnings,
+                                                   keyspace,
+                                                   statement,
+                                                   options,
+                                                   hosts,
+                                                   consistency,
+                                                   retries)
+        when 0x2000 then Errors::SyntaxError.new(@message,
+                                                 @custom_payload,
+                                                 @warnings,
+                                                 keyspace,
+                                                 statement,
+                                                 options,
+                                                 hosts,
+                                                 consistency,
+                                                 retries)
+        when 0x2100 then Errors::UnauthorizedError.new(@message,
+                                                       @custom_payload,
+                                                       @warnings,
+                                                       keyspace,
+                                                       statement,
+                                                       options,
+                                                       hosts,
+                                                       consistency,
+                                                       retries)
+        when 0x2200 then Errors::InvalidError.new(@message,
+                                                  @custom_payload,
+                                                  @warnings,
+                                                  keyspace,
+                                                  statement,
+                                                  options,
+                                                  hosts,
+                                                  consistency,
+                                                  retries)
+        when 0x2300 then Errors::ConfigurationError.new(@message,
+                                                        @custom_payload,
+                                                        @warnings,
+                                                        keyspace,
+                                                        statement,
+                                                        options,
+                                                        hosts,
+                                                        consistency,
+                                                        retries)
+        else
+          Errors::ServerError.new(@message,
+                                  @custom_payload,
+                                  @warnings,
+                                  keyspace,
+                                  statement,
+                                  options,
+                                  hosts,
+                                  consistency,
+                                  retries)
+        end
+      end
+    end
+  end
+end