1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
|
# Licensed to Elasticsearch B.V. under one or more contributor
# license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright
# ownership. Elasticsearch B.V. licenses this file to you 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 Elasticsearch
module DSL
module Search
# Contains the classes for Elasticsearch aggregations
#
module Aggregations;end
class AggregationsCollection < Hash
def to_hash
@hash ||= Hash[map { |k,v| [k, v.to_hash] }]
end
end
# Wraps the `aggregations` part of a search definition
#
# @see https://www.elastic.co/guide/en/elasticsearch/reference/current/search-aggregations.html
#
class Aggregation
def initialize(*args, &block)
@block = block
end
# Looks up the corresponding class for a method being invoked, and initializes it
#
# @raise [NoMethodError] When the corresponding class cannot be found
#
def method_missing(name, *args, &block)
klass = Utils.__camelize(name)
if Aggregations.const_defined? klass
@value = Aggregations.const_get(klass).new *args, &block
else
raise NoMethodError, "undefined method '#{name}' for #{self}"
end
end
# Defines an aggregation nested in another one
#
def aggregation(*args, &block)
call
@value.__send__ :aggregation, *args, &block
end
# Returns the aggregations
#
def aggregations
call
@value.__send__ :aggregations
end
# Evaluates the block passed to initializer, ensuring it is called just once
#
# @return [self]
#
# @api private
#
def call
@block.arity < 1 ? self.instance_eval(&@block) : @block.call(self) if @block && ! @_block_called
@_block_called = true
self
end
# Converts the object to a Hash
#
# @return [Hash]
#
def to_hash(options={})
call
if @value
case
when @value.respond_to?(:to_hash)
@value.to_hash
else
@value
end
else
{}
end
end
end
end
end
end
|