cassandra-driver 1.0.0.beta.2-java

data/lib/cassandra/execution/info.rb

@@ -0,0 +1,51 @@
+# encoding: utf-8
+
+#--
+# Copyright 2013-2014 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 Execution
+    class Info
+      # @return [String] keyspace used for the query
+      attr_reader :keyspace
+      # @return [Cassandra::Statement] original statement
+      attr_reader :statement
+      # @return [Cassandra::Execution::Options] original execution options
+      attr_reader :options
+      # @return [Array<Cassandra::Host>] a list of attempted hosts
+      attr_reader :hosts
+      # Actual consistency used, it can differ from consistency in {Cassandra::Execution::Info#options} if a retry policy modified it.
+      # @return [Symbol] one of {Cassandra::CONSISTENCIES}
+      attr_reader :consistency
+      # @return [Integer] number of retries
+      attr_reader :retries
+      # Returns {Cassandra::Execution::Trace} if `trace: true` was passed to {Cassandra::Session#execute} or {Cassandra::Session#execute_async}
+      # @return [Cassandra::Execution::Trace, nil] a Trace if it has been enabled for request
+      attr_reader :trace
+
+      # @private
+      def initialize(keyspace, statement, options, hosts, consistency, retries, trace)
+        @keyspace    = keyspace
+        @statement   = statement
+        @options     = options
+        @hosts       = hosts
+        @consistency = consistency
+        @retries     = retries
+        @trace       = trace
+      end
+    end
+  end
+end

data/lib/cassandra/execution/options.rb

@@ -0,0 +1,80 @@
+# encoding: utf-8
+
+#--
+# Copyright 2013-2014 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 Execution
+    class Options
+      # @return [Symbol] consistency for request. Must be one of
+      #   {Cassandra::CONSISTENCIES}
+      attr_reader :consistency
+      # @return [Symbol] consistency for request with conditional updates
+      #   (lightweight - compare-and-set, CAS - transactions). Must be one of
+      #   {Cassandra::SERIAL_CONSISTENCIES}
+      attr_reader :serial_consistency
+      # @return [Integer] requested page size
+      attr_reader :page_size
+      # @return [Numeric] request timeout interval
+      attr_reader :timeout
+
+      # @private
+      def initialize(options)
+        consistency        = options[:consistency]
+        page_size          = options[:page_size]
+        trace              = options[:trace]
+        timeout            = options[:timeout]
+        serial_consistency = options[:serial_consistency]
+
+        raise ::ArgumentError, ":consistency must be one of #{CONSISTENCIES.inspect}, #{consistency.inspect} given" unless CONSISTENCIES.include?(consistency)
+        raise ::ArgumentError, ":serial_consistency must be one of #{SERIAL_CONSISTENCIES.inspect}, #{serial_consistency.inspect} given" if serial_consistency && !SERIAL_CONSISTENCIES.include?(serial_consistency)
+
+        page_size = page_size && Integer(page_size)
+        timeout   = timeout   && Integer(timeout)
+
+        raise ::ArgumentError, ":page_size must be greater than 0, #{page_size.inspect} given" if page_size && page_size <= 0
+        raise ::ArgumentError, ":timeout must be greater than 0, #{timeout.inspect} given" if timeout && timeout <= 0
+
+        @consistency        = consistency
+        @page_size          = page_size
+        @trace              = !!trace
+        @timeout            = timeout
+        @serial_consistency = serial_consistency
+      end
+
+      # @return [Boolean] whether request tracing was enabled
+      def trace?
+        @trace
+      end
+
+      # @private
+      def override(options)
+        Options.new(to_h.merge!(options))
+      end
+
+      # @private
+      def to_h
+        {
+          :consistency        => @consistency,
+          :page_size          => @page_size,
+          :trace              => @trace,
+          :timeout            => @timeout,
+          :serial_consistency => @serial_consistency
+        }
+      end
+    end
+  end
+end

data/lib/cassandra/execution/trace.rb

@@ -0,0 +1,152 @@
+# encoding: utf-8
+
+#--
+# Copyright 2013-2014 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 Execution
+    class Trace
+      class Event
+        # @return [Cassandra::Uuid] event uuid
+        attr_reader :id
+
+        # @return [String] description of activity
+        attr_reader :activity
+
+        attr_reader :source
+        attr_reader :source_elapsed
+        attr_reader :thread
+
+        # @private
+        def initialize(id, activity, source, source_elapsed, thread)
+          @id             = id
+          @activity       = activity
+          @source         = source
+          @source_elapsed = source_elapsed
+          @thread         = thread
+        end
+
+        def ==(other)
+          other == @id
+        end
+
+        alias :eql? :==
+      end
+
+      include MonitorMixin
+
+      # @return [Cassandra::Uuid] trace id
+      attr_reader :id
+
+      # @private
+      def initialize(id, client)
+        @id     = id
+        @client = client
+
+        mon_initialize
+      end
+
+      # Returns the ip of coordinator node. Typically the same as {Cassandra::Execution::Info#hosts}`.last`
+      #
+      # @return [IPAddr] ip of the coordinator node
+      def coordinator
+        load unless @coordinator
+
+        @coordinator
+      end
+
+      def duration
+        load unless @duration
+
+        @duration
+      end
+
+      def parameters
+        load unless @parameters
+
+        @parameters
+      end
+
+      def request
+        load unless @request
+
+        @request
+      end
+
+      def started_at
+        load unless @started_at
+
+        @started_at
+      end
+
+      # Returns all trace events
+      #
+      # @return [Array<Cassandra::Execution::Trace::Event>] events
+      def events
+        load_events unless @events
+
+        @events
+      end
+
+      def inspect
+        "#<#{self.class.name}:0x#{self.object_id.to_s(16)} @id=#{@id.inspect}>"
+      end
+
+      private
+
+      # @private
+      SELECT_SESSION = "SELECT * FROM system_traces.sessions WHERE session_id = ?"
+      # @private
+      SELECT_EVENTS  = "SELECT * FROM system_traces.events WHERE session_id = ?"
+
+      # @private
+      def load
+        synchronize do
+          return if @loaded
+
+          data = @client.query(Statements::Simple.new(SELECT_SESSION, @id), VOID_OPTIONS).get.first
+          raise ::RuntimeError, "unable to load trace #{@id}" if data.nil?
+
+          @coordinator = data['coordinator']
+          @duration    = data['duration']
+          @parameters  = data['parameters']
+          @request     = data['request']
+          @started_at  = data['started_at']
+          @loaded      = true
+        end
+
+        nil
+      end
+
+      # @private
+      def load_events
+        synchronize do
+          return if @loaded_events
+
+          @events = []
+
+          @client.query(Statements::Simple.new(SELECT_EVENTS, @id), VOID_OPTIONS).get.each do |row|
+            @events << Event.new(row['event_id'], row['activity'], row['source'], row['source_elapsed'], row['thread'])
+          end
+
+          @events.freeze
+
+          @loaded_events = true
+        end
+      end
+    end
+  end
+end