class ChefZero::Endpoints::SearchEndpoint
/search/INDEX
Public Instance Methods
get(request)
click to toggle source
# File lib/chef_zero/endpoints/search_endpoint.rb, line 12 def get(request) results = search(request) results['rows'] = results['rows'].map { |name,uri,value,search_value| value } json_response(200, results) end
post(request)
click to toggle source
# File lib/chef_zero/endpoints/search_endpoint.rb, line 18 def post(request) full_results = search(request) keys = JSON.parse(request.body, :create_additions => false) partial_results = full_results['rows'].map do |name, uri, doc, search_value| data = {} keys.each_pair do |key, path| if path.size > 0 value = search_value path.each do |path_part| value = value[path_part] if !value.nil? end data[key] = value else data[key] = nil end end { 'url' => uri, 'data' => data } end json_response(200, { 'rows' => partial_results, 'start' => full_results['start'], 'total' => full_results['total'] }) end
Private Instance Methods
deep_merge!(source, dest)
click to toggle source
Deep Merge core documentation. deep_merge! method permits merging of arbitrary child elements. The two top level elements must be hashes. These hashes can contain unlimited (to stack limit) levels of child elements. These child elements to not have to be of the same types. Where child elements are of the same type, deep_merge will attempt to merge them together. Where child elements are not of the same type, deep_merge will skip or optionally overwrite the destination element with the contents of the source element at that level. So if you have two hashes like this:
source = {:x => [1,2,3], :y => 2} dest = {:x => [4,5,'6'], :y => [7,8,9]} dest.deep_merge!(source) Results: {:x => [1,2,3,4,5,'6'], :y => 2}
By default, “deep_merge!” will overwrite any unmergeables and merge everything else. To avoid this, use “deep_merge” (no bang/exclamation mark)
# File lib/chef_zero/endpoints/search_endpoint.rb, line 155 def deep_merge!(source, dest) # if dest doesn't exist, then simply copy source to it if dest.nil? dest = source; return dest end case source when nil dest when Hash source.each do |src_key, src_value| if dest.kind_of?(Hash) if dest[src_key] dest[src_key] = deep_merge!(src_value, dest[src_key]) else # dest[src_key] doesn't exist so we take whatever source has dest[src_key] = src_value end else # dest isn't a hash, so we overwrite it completely dest = source end end when Array if dest.kind_of?(Array) dest = dest | source else dest = source end when String dest = source else # src_hash is not an array or hash, so we'll have to overwrite dest dest = source end dest end
expand_for_indexing(value, index, id)
click to toggle source
# File lib/chef_zero/endpoints/search_endpoint.rb, line 67 def expand_for_indexing(value, index, id) if index == 'node' result = {} deep_merge!(value['default'] || {}, result) deep_merge!(value['normal'] || {}, result) deep_merge!(value['override'] || {}, result) deep_merge!(value['automatic'] || {}, result) result['recipe'] = [] result['role'] = [] if value['run_list'] value['run_list'].each do |run_list_entry| if run_list_entry =~ /^(recipe|role)\[(.*)\]/ result[$1] << $2 end end end value.each_pair do |key, value| result[key] = value unless %w(default normal override automatic).include?(key) end result elsif !%w(client environment role).include?(index) DataNormalizer.normalize_data_bag_item(value, index, id, 'GET') else value end end
search(request)
click to toggle source
# File lib/chef_zero/endpoints/search_endpoint.rb, line 95 def search(request) # Extract parameters index = request.rest_path[3] query_string = request.query_params['q'] || '*:*' solr_query = ChefZero::Solr::SolrParser.new(query_string).parse sort_string = request.query_params['sort'] start = request.query_params['start'] start = start.to_i if start rows = request.query_params['rows'] rows = rows.to_i if rows # Get the search container container, expander = search_container(request, index) # Search! result = [] list_data(request, container).each do |name| value = get_data(request, container + [name]) expanded = expander.call(JSON.parse(value, :create_additions => false), name) result << [ name, build_uri(request.base_uri, container + [name]), expanded, expand_for_indexing(expanded, index, name) ] end result = result.select do |name, uri, value, search_value| solr_query.matches_doc?(ChefZero::Solr::SolrDoc.new(search_value, name)) end total = result.size # Sort if sort_string sort_key, sort_order = sort_string.split(/\s+/, 2) result = result.sort_by { |name,uri,value,search_value| ChefZero::Solr::SolrDoc.new(search_value, name)[sort_key] } result = result.reverse if sort_order == "DESC" end # Paginate if start result = result[start..start+(rows||-1)] end { 'rows' => result, 'start' => start || 0, 'total' => total } end
search_container(request, index)
click to toggle source
# File lib/chef_zero/endpoints/search_endpoint.rb, line 48 def search_container(request, index) relative_parts, normalize_proc = case index when 'client' [ ['clients'], Proc.new { |client, name| DataNormalizer.normalize_client(client, name) } ] when 'node' [ ['nodes'], Proc.new { |node, name| DataNormalizer.normalize_node(node, name) } ] when 'environment' [ ['environments'], Proc.new { |environment, name| DataNormalizer.normalize_environment(environment, name) } ] when 'role' [ ['roles'], Proc.new { |role, name| DataNormalizer.normalize_role(role, name) } ] else [ ['data', index], Proc.new { |data_bag_item, id| DataNormalizer.normalize_data_bag_item(data_bag_item, index, id, 'DELETE') } ] end [ request.rest_path[0..1] + relative_parts, normalize_proc ] end