Alias and force name

lib/waistband/index.rb

@@ -29,6 +29,10 @@ def initialize(index_name, options = {})
         @subs = @subs.flatten     if @subs.is_a?(Array)
       end
 
+      if options['force_name'].present?
+        @force_name = options['force_name']
+      end
+
     end
 
     def exists?
@@ -261,6 +265,18 @@ def alias_exists?(alias_name)
       )
     end
 
+    def parent_index_names
+      client.indices.get_alias(name: config_name).map { |k, v| k }
+    rescue Elasticsearch::Transport::Transport::Errors::NotFound
+      nil
+    end
+
+    def get_first_parent_index_reference
+      index_name = parent_index_names&.first
+      return nil if index_name.nil?
+      self.class.new(@index_name, force_name: index_name)
+    end
+
     def config
       ::Waistband.config.index @index_name
     end
@@ -345,7 +361,7 @@ def full_alias_name(alias_name)
       end
 
       def custom_name?
-        !!config['name']
+        !!config['name'] || !!(defined? @force_name)
       end
 
       def stringify_all(data)
@@ -381,6 +397,7 @@ def settings
       end
 
       def config_name
+        return @force_name if defined? @force_name
         @subs ? "#{base_config_name}__#{@subs.join('_')}" : base_config_name
       end
 

spec/lib/index/index_spec.rb

@@ -301,6 +301,26 @@
 
   end
 
+  describe 'force_name' do
+
+    let(:forced_name_index) { Waistband::Index.new 'events_no_name', force_name: 'new_forced_name' }
+
+    around(:each) do |example|
+      forced_name_index.delete
+      example.run
+      forced_name_index.delete
+    end
+
+    it "creates the index" do
+      expect(forced_name_index.exists?).to be_falsey
+
+      forced_name_index.create!
+      expect(forced_name_index.exists?).to be_truthy
+      expect(forced_name_index.send(:config_name)).to eql 'new_forced_name'
+    end
+
+  end
+
   describe '#base_config_name' do
 
     it "gets a default name that makes sense for the index when not defined" do
@@ -348,6 +368,58 @@
 
     end
 
+    it 'returns the name of the parent index' do
+      index.alias('events_alias')
+      alias_ref = Waistband::Index.new 'events', force_name: index.send(:full_alias_name, 'events_alias')
+
+      parent_index_names = alias_ref.parent_index_names
+      expect(parent_index_names.length).to eq(1)
+      expect(parent_index_names[0]).to eq(index.send(:config_name))
+    end
+
+    it 'can return a reference to the parent index' do
+      index.alias('events_alias')
+      alias_ref = Waistband::Index.new 'events', force_name: index.send(:full_alias_name, 'events_alias')
+      parent_index = alias_ref.get_first_parent_index_reference
+
+      expect(parent_index.exists?).to be_truthy
+      expect(parent_index.send(:config_name)).to eq(index.send(:config_name))
+    end
+
+    it "can save read and update documents from an alias" do
+      index.alias('events_alias')
+      alias_ref = Waistband::Index.new 'events', force_name: index.send(:full_alias_name, 'events_alias')
+
+      expect(alias_ref.save('__test_write', {'ok' => 'yeah', 'not_ok' => 'yeah'})).to be_present
+      expect(alias_ref.update('__test_write', {'not_ok' => 'no'})).to be_present
+      expect(alias_ref.read('__test_write')).to eql({
+        '_id' => '__test_write',
+        '_index' => 'events_test',
+        '_source' => {'ok' => 'yeah', 'not_ok' => 'no'},
+        '_type' => 'event',
+        '_version' => 2,
+        'found' => true
+      })
+    end
+
+    it "updates mappings from an alias" do
+      index = Waistband::Index.new 'geo', force_name: 'geo_index'
+      index.delete
+      index.create!
+      index.alias('geo_index_alias')
+
+      alias_ref = Waistband::Index.new 'geo', force_name: 'geo_index_alias'
+
+      responses = alias_ref.update_all_mappings
+      index.delete
+
+      expect(responses).to be_an Array
+
+      response = responses.first
+      expect(response['acknowledged']).to be_truthy
+      expect(response['_type']).to eq('geo')
+    end
+
   end
 
   describe 'logging' do

spec/spec_helper.rb

@@ -8,6 +8,7 @@
 require 'rspec'
 require 'timecop'
 require 'active_support/core_ext/integer/time'
+require 'json'
 
 unless ENV['ON_TRAVIS']
   require 'byebug'