Merge pull request #192 from datastax/RUBY-187

Ruby 187

Author: stamhankar999

integration/indexes/indexes_test.rb

@@ -54,6 +54,8 @@ def test_can_create_index
 
     assert @cluster.keyspace('simplex').table('test').has_index?('b_index')
     index = @cluster.keyspace('simplex').table('test').index('b_index')
+    assert @cluster.keyspace('simplex').has_index?('b_index')
+    assert_same(index, @cluster.keyspace('simplex').index('b_index'))
     assert_equal 'b_index', index.name
     assert_equal 'test', index.table.name
     assert_equal :composites, index.kind

integration/indexes/materialized_view_test.rb

@@ -70,6 +70,10 @@ def test_can_retrieve_materialized_view_metadata
     assert @cluster.keyspace('simplex').has_materialized_view?('monthlyhigh')
     mv_meta = @cluster.keyspace('simplex').materialized_view('monthlyhigh')
 
+    table = @cluster.keyspace('simplex').table('scores')
+    assert table.has_materialized_view?('monthlyhigh')
+    assert_same(mv_meta, table.materialized_view('monthlyhigh'))
+
     assert_equal 'monthlyhigh', mv_meta.name
     refute_nil mv_meta.id
     assert_equal 'scores', mv_meta.base_table.name

integration/session_test.rb

@@ -843,6 +843,8 @@ def test_can_set_protocol_version_explicitly
   # @test_category connection
   #
   def test_cluster_session_inspect
+    setup_schema
+
     cluster = Cassandra.cluster(hosts: ['127.0.0.1'], consistency: :quorum, page_size: 10)
     session = cluster.connect('simplex')
 

lib/cassandra.rb

@@ -820,6 +820,7 @@ def self.validate_and_massage_options(options)
 require 'cassandra/materialized_view'
 require 'cassandra/keyspace'
 require 'cassandra/index'
+require 'cassandra/trigger'
 
 require 'cassandra/execution/info'
 require 'cassandra/execution/options'

lib/cassandra/cluster/schema/fetchers.rb

@@ -62,7 +62,7 @@ def custom_class_name
       @options['class_name']
     end
 
-    # @return [String] a cql representation of this table
+    # @return [String] a cql representation of this index
     def to_cql
       keyspace_name = Util.escape_name(@table.keyspace.name)
       table_name = Util.escape_name(@table.name)

lib/cassandra/index.rb

@@ -69,11 +69,18 @@ def initialize(name,
       @views          = views
 
       # Set the keyspace attribute on the tables and views.
+      # Also set up the index collection on the keyspace and the view-collection on each table.
+      @indexes_hash = {}
       @tables.each_value do |t|
         t.set_keyspace(self)
+        t.each_index do |index|
+          @indexes_hash[index.name] = index
+        end
       end
       @views.each_value do |v|
         v.set_keyspace(self)
+        table = v.base_table
+        table.add_view(v) if table
       end
     end
 
@@ -110,6 +117,34 @@ def each_table(&block)
     end
     alias tables each_table
 
+    # @return [Boolean] whether this keyspace has an index with the given name
+    # @param name [String] index name
+    def has_index?(name)
+      @indexes_hash.key?(name)
+    end
+
+    # @return [Cassandra::Index, nil] an index or nil
+    # @param name [String] index name
+    def index(name)
+      @indexes_hash[name]
+    end
+
+    # Yield or enumerate each index defined in this keyspace
+    # @overload each_index
+    #   @yieldparam index [Cassandra::Index] current index
+    #   @return [Cassandra::Keyspace] self
+    # @overload each_index
+    #   @return [Array<Cassandra::Index>] a list of indexes
+    def each_index(&block)
+      if block_given?
+        @indexes_hash.each_value(&block)
+        self
+      else
+        @indexes_hash.values
+      end
+    end
+    alias indexes each_index
+
     # @return [Boolean] whether this keyspace has a materialized view with the given name
     # @param name [String] materialized view name
     def has_materialized_view?(name)

lib/cassandra/keyspace.rb

@@ -38,6 +38,10 @@ def initialize(keyspace,
       @clustering_order = clustering_order.freeze
       @indexes = []
       @indexes_hash = {}
+      @materialized_views = []
+      @materialized_views_hash = {}
+      @triggers = []
+      @triggers_hash = {}
     end
 
     # @param name [String] index name
@@ -68,6 +72,62 @@ def each_index(&block)
     end
     alias indexes each_index
 
+    # @param name [String] trigger name
+    # @return [Boolean] whether this table has a given trigger
+    def has_trigger?(name)
+      @triggers_hash.key?(name)
+    end
+
+    # @param name [String] trigger name
+    # @return [Cassandra::Trigger, nil] a trigger or nil
+    def trigger(name)
+      @triggers_hash[name]
+    end
+
+    # Yield or enumerate each trigger bound to this table
+    # @overload each_trigger
+    #   @yieldparam trigger [Cassandra::Index] current trigger
+    #   @return [Cassandra::Table] self
+    # @overload each_trigger
+    #   @return [Array<Cassandra::Trigger>] a list of triggers
+    def each_trigger(&block)
+      if block_given?
+        @triggers.each(&block)
+        self
+      else
+        @triggers.freeze
+      end
+    end
+    alias triggers each_trigger
+
+    # @param name [String] materialized view name
+    # @return [Boolean] whether this table has a given materialized view
+    def has_materialized_view?(name)
+      @materialized_views_hash.key?(name)
+    end
+
+    # @param name [String] materialized view name
+    # @return [Cassandra::MaterializedView, nil] a materialized view or nil
+    def materialized_view(name)
+      @materialized_views_hash[name]
+    end
+
+    # Yield or enumerate each materialized view bound to this table
+    # @overload each_materialized_view
+    #   @yieldparam materialized_view [Cassandra::MaterializedView] current materialized view
+    #   @return [Cassandra::Table] self
+    # @overload each_materialized_view
+    #   @return [Array<Cassandra::MaterializedView>] a list of materialized views
+    def each_materialized_view(&block)
+      if block_given?
+        @materialized_views.each(&block)
+        self
+      else
+        @materialized_views.freeze
+      end
+    end
+    alias materialized_views each_materialized_view
+
     # @return [String] a cql representation of this table
     def to_cql
       cql = "CREATE TABLE #{Util.escape_name(@keyspace.name)}.#{Util.escape_name(@name)} (\n"
@@ -134,6 +194,18 @@ def add_index(index)
       @indexes_hash[index.name] = index
     end
 
+    # @private
+    def add_view(view)
+      @materialized_views << view
+      @materialized_views_hash[view.name] = view
+    end
+
+    # @private
+    def add_trigger(trigger)
+      @triggers << trigger
+      @triggers_hash[trigger.name] = trigger
+    end
+
     # @private
     def eql?(other)
       other.is_a?(Table) &&

lib/cassandra/table.rb

@@ -0,0 +1,67 @@
+# encoding: utf-8
+
+#--
+# Copyright 2013-2016 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
+  # Represents a trigger on a cassandra table
+  class Trigger
+    # @return [Cassandra::Table] table that the trigger applies to.
+    attr_reader :table
+    # @return [String] name of the trigger.
+    attr_reader :name
+    # @return [Hash] options of the trigger.
+    attr_reader :options
+
+    # @private
+    def initialize(table,
+                   name,
+                   options)
+      @table = table
+      @name = name.freeze
+      @options = options.freeze
+    end
+
+    # @return [String] name of the trigger class
+    def custom_class_name
+      @options['class']
+    end
+
+    # @return [String] a cql representation of this trigger
+    def to_cql
+      keyspace_name = Util.escape_name(@table.keyspace.name)
+      table_name = Util.escape_name(@table.name)
+      trigger_name = Util.escape_name(@name)
+
+      "CREATE TRIGGER #{trigger_name} ON #{keyspace_name}.#{table_name} USING '#{@options['class']}';"
+    end
+
+    # @private
+    def eql?(other)
+      other.is_a?(Trigger) &&
+        @table == other.table &&
+        @name == other.name &&
+        @options == other.options
+    end
+    alias == eql?
+
+    # @private
+    def inspect
+      "#<#{self.class.name}:0x#{object_id.to_s(16)} " \
+          "@name=#{@name.inspect} @table=#{@table.inspect} @options=#{@options.inspect}>"
+    end
+  end
+end

lib/cassandra/trigger.rb

@@ -392,4 +392,4 @@ WITH bloom_filter_fp_chance = 0.01
  AND gc_grace_seconds = 2
  AND populate_io_cache_on_flush = 'false'
  AND read_repair_chance = 0.1
- AND replicate_on_write = 'true';
+ AND replicate_on_write = 'true';