Project

General

Profile

Download (3.66 KB) Statistics
| Branch: | Tag: | Revision:
#
# Copyright 2014 Red Hat, Inc.
#
# This software is licensed to you under the GNU General Public
# License as published by the Free Software Foundation; either version
# 2 of the License (GPLv2) or (at your option) any later version.
# There is NO WARRANTY for this software, express or implied,
# including the implied warranties of MERCHANTABILITY,
# NON-INFRINGEMENT, or FITNESS FOR A PARTICULAR PURPOSE. You should
# have received a copy of GPLv2 along with this software; if not, see
# http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.

module Katello
module Glue::ElasticSearch::ContentViewPuppetEnvironment

# TODO: break this up into modules
# rubocop:disable MethodLength
def self.included(base)
base.send :include, Ext::IndexedModel

base.class_eval do
index_options :extended_json => :extended_index_attrs,
:json => {:except => [:pulp_repo_facts, :feed_cert]}

mapping do
indexes :name, :type => 'string', :analyzer => :kt_name_analyzer
indexes :name_sort, :type => 'string', :index => :not_analyzed
indexes :labels, :type => 'string', :index => :not_analyzed
end

before_destroy :clear_content_indices
end

def extended_index_attrs
{
:environment => self.environment.try(:name),
:archive => self.archive?,
:environment_id => self.environment.try(:id),
:name_sort => self.name
}
end

def index_puppet_modules
Tire.index Katello::PuppetModule.index do
create :settings => Katello::PuppetModule.index_settings, :mappings => Katello::PuppetModule.index_mapping
end
puppet_modules = self.puppet_modules.collect{|puppet_module| puppet_module.as_json.merge(puppet_module.index_options)}
puppet_modules.each_slice(Katello.config.pulp.bulk_load_size) do |sublist|
Tire.index Katello::PuppetModule.index do
import sublist
end if !sublist.empty?
end
end

def update_puppet_modules_index
# for each of the puppet_modules in the repo, unassociate the repo from the module
puppet_modules = self.puppet_modules.collect{|puppet_module| puppet_module.as_json.merge(puppet_module.index_options)}
pulp_id = self.pulp_id

unless puppet_modules.empty?
Tire.index Katello::PuppetModule.index do
create :settings => Katello::PuppetModule.index_settings, :mappings => Katello::PuppetModule.index_mapping
end unless Tire.index(Katello::PuppetModule.index).exists?

Tire.index Katello::PuppetModule.index do
import puppet_modules do |documents|
documents.each do |document|
if !document["repoids"].nil? && document["repoids"].length > 1
# if there is more than 1 repo associated w/ the pkg, remove this repo
document["repoids"].delete(pulp_id)
end
end
end
end
end

# now, for any module that only had this repo asscociated with it, remove the module from the index
repoids = "repoids:#{pulp_id}"
Tire::Configuration.client.delete "#{Tire::Configuration.url}/katello_puppet_module/_query?q=#{repoids}"
Tire.index('katello_puppet_module').refresh
end

def puppet_module_count
results = Katello::PuppetModule.legacy_search('', :page_size => 1, :repoids => [self.pulp_id])
results.empty? ? 0 : results.total
end

def index_content
self.index_puppet_modules
true
end

def clear_content_indices
update_puppet_modules_index
end
end
end
end
(9-9/33)