elasticated 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8dc9235eb397f995d6bd706184dde4d43a3a85dc
+  data.tar.gz: 10cd46beddee5efa0aaa3bf301d534880413f968
+SHA512:
+  metadata.gz: 27039469aef119a66796daaa93ab1713845a91cb61345bd3592a55e41ccd84827cc72ec9e4ebeaa450d24979f98a5f09feef6eedf1ed003be86a69ba8c0d242d
+  data.tar.gz: 1998fbde3e419a39fcf6090fc827ffce2a498237056130efb5fd7b2b152d87983026ac203fcbed5e5f9c7e739a9eb92ebb0b6f74c1b57867dd38b93832bbcda4

data/.gitignore

@@ -0,0 +1,35 @@
+*.gem
+*.rbc
+/.config
+/coverage/
+/InstalledFiles
+/pkg/
+/spec/reports/
+/test/tmp/
+/test/version_tmp/
+/tmp/
+
+## Specific to RubyMotion:
+.dat*
+.repl_history
+build/
+
+## Documentation cache and generated files:
+/.yardoc/
+/_yardoc/
+/doc/
+/rdoc/
+
+## Environment normalisation:
+/.bundle/
+/vendor/bundle
+/lib/bundler/man/
+
+# for a library or gem, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+Gemfile.lock
+.ruby-version
+.ruby-gemset
+
+# unless supporting rvm < 1.11.0 or doing something fancy, ignore this:
+.rvmrc

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in elasticated.gemspec
+gemspec

data/README.md

@@ -0,0 +1,3 @@
+# elasticated
+
+Elasticsearch Wrapper, with Query & Mapping Builders

data/Rakefile

@@ -0,0 +1,6 @@
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/elasticated.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'elasticated'
+  spec.version       = Elasticated::VERSION
+  spec.authors       = ['Pablo Fernandez']
+  spec.email         = ['psfutn@gmail.com']
+  spec.summary       = 'Elasticsearch Wrapper, with Query & Mapping Builders'
+  spec.description   = ''
+  spec.homepage      = 'http://github.com/pablo31/elasticated'
+  spec.license       = ''
+
+  spec.files         = `git ls-files`.split($\)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+
+  spec.add_development_dependency 'bundler', '~> 1.5'
+  spec.add_development_dependency 'rspec', '3.0.0'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'pry-byebug'
+
+  spec.add_runtime_dependency 'elasticsearch'
+  spec.add_runtime_dependency 'hash_ext', '~> 0.1.1'
+
+end

data/lib/elasticated.rb

@@ -0,0 +1,102 @@
+require 'json'
+require 'hash_ext'
+require 'elasticsearch'
+
+require_relative 'elasticated/helpers'
+require_relative 'elasticated/block_evaluation'
+require_relative 'elasticated/clonable'
+require_relative 'elasticated/inspectionable'
+require_relative 'elasticated/configuration'
+require_relative 'elasticated/configurable'
+require_relative 'elasticated/default_logger'
+
+# query conditions
+
+require_relative 'elasticated/conditions/standard_condition'
+require_relative 'elasticated/conditions/terms_condition'
+require_relative 'elasticated/conditions/wildcard_condition'
+require_relative 'elasticated/conditions/exists_condition'
+require_relative 'elasticated/conditions/missing_condition'
+require_relative 'elasticated/conditions/range_condition'
+require_relative 'elasticated/conditions/nested_condition'
+require_relative 'elasticated/conditions/script_condition'
+require_relative 'elasticated/conditions/custom_condition'
+
+# delimiters
+
+require_relative 'elasticated/delimiters/standard_field_delimiter'
+require_relative 'elasticated/delimiters/term_field_delimiter'
+require_relative 'elasticated/delimiters/date_field_delimiter'
+
+# query components
+
+require_relative 'elasticated/conditions_builder'
+require_relative 'elasticated/boolean_clause'
+require_relative 'elasticated/query_conditions'
+require_relative 'elasticated/query_aggregations'
+
+# query
+
+require_relative 'elasticated/query'
+
+# aggregations
+
+require_relative 'elasticated/aggregation'
+require_relative 'elasticated/aggregations/subaggregated'
+
+require_relative 'elasticated/aggregations/terms_aggregation'
+
+require_relative 'elasticated/aggregations/histogram_aggregation'
+require_relative 'elasticated/aggregations/date_histogram_aggregation'
+
+require_relative 'elasticated/aggregations/single_value_aggregation'
+require_relative 'elasticated/aggregations/cardinality_aggregation'
+require_relative 'elasticated/aggregations/count_distinct_aggregation'
+
+require_relative 'elasticated/aggregations/group_aggregation'
+require_relative 'elasticated/aggregations/count_aggregation'
+require_relative 'elasticated/aggregations/sum_distinct_aggregation'
+
+require_relative 'elasticated/aggregations/count_filtered_aggregation'
+require_relative 'elasticated/aggregations/filter_aggregation_evaluator'
+require_relative 'elasticated/aggregations/filter_aggregation'
+
+require_relative 'elasticated/aggregations/nested_aggregation'
+
+require_relative 'elasticated/aggregations/ranges_builder'
+require_relative 'elasticated/aggregations/range_aggregation_evaluator'
+require_relative 'elasticated/aggregations/range_aggregation'
+
+require_relative 'elasticated/aggregations/top_hits_aggregation'
+
+require_relative 'elasticated/aggregations/custom_aggregation'
+
+# repository
+
+require_relative 'elasticated/document'
+require_relative 'elasticated/results'
+
+require_relative 'elasticated/client'
+require_relative 'elasticated/repository'
+
+require_relative 'elasticated/mapping'
+require_relative 'elasticated/mapping/partial'
+require_relative 'elasticated/mapping/fields_builder'
+require_relative 'elasticated/mapping/object_builder'
+require_relative 'elasticated/mapping/nested_builder'
+require_relative 'elasticated/mapping/type_builder'
+require_relative 'elasticated/mapping/builder'
+
+require_relative 'elasticated/index_selector'
+require_relative 'elasticated/partitioned_repository'
+
+require_relative 'elasticated/repositories/monthly_partitioned_repository'
+
+
+module Elasticated
+
+  def self.configure(&block)
+    block.call Configuration
+  end
+
+end

data/lib/elasticated/aggregation.rb

@@ -0,0 +1,36 @@
+module Elasticated
+  class Aggregation
+
+    # abstract class
+    # child must implement 'default_name'
+    # child must implement 'build'
+    # child must implement 'parse(response)'
+
+    include BlockEvaluation
+    include Clonable
+
+    attr_accessor :field, :alias_name, :extra_params
+
+    def name
+      return Helpers.string_to_agg_name(alias_name).to_sym if alias_name
+      Helpers.string_to_agg_name(default_name).to_sym
+    end
+
+    def original_name
+      return alias_name.to_sym if alias_name
+      default_name.to_s.to_sym
+    end
+
+    def initialize(field, *args, &block)
+      self.field = field
+      opts = args.last
+      if opts.is_a? Hash
+        self.extra_params = Helpers.hash_deep_dup opts
+        self.alias_name = extra_params.delete :as
+      else
+        self.extra_params = Hash.new
+      end
+    end
+
+  end
+end

data/lib/elasticated/aggregations/cardinality_aggregation.rb

@@ -0,0 +1,15 @@
+module Elasticated
+  class CardinalityAggregation < SingleValueAggregation
+
+    # override
+    def default_name
+      "#{field}_cardinality"
+    end
+
+    # implementation
+    def operation
+      :cardinality
+    end
+
+  end
+end

data/lib/elasticated/aggregations/count_aggregation.rb

@@ -0,0 +1,15 @@
+module Elasticated
+  class CountAggregation < TermsAggregation
+
+    def default_name
+      "count_by_#{field}"
+    end
+
+    def parse(response)
+      response['buckets'].inject({}) do |hash, bucket|
+        hash.merge bucket['key'] => bucket['doc_count']
+      end
+    end
+
+  end
+end

data/lib/elasticated/aggregations/count_distinct_aggregation.rb

@@ -0,0 +1,15 @@
+module Elasticated
+  class CountDistinctAggregation < SingleValueAggregation
+
+    # override
+    def default_name
+      "#{field}_value_count"
+    end
+
+    # implementation
+    def operation
+      :value_count
+    end
+
+  end
+end

data/lib/elasticated/aggregations/count_filtered_aggregation.rb

@@ -0,0 +1,29 @@
+module Elasticated
+  class CountFilteredAggregation < Aggregation
+
+    attr_accessor :_conditions, :_filter_name
+
+    def initialize(filter_name, *args, &block)
+      self._conditions = QueryConditions.new
+      self._filter_name = filter_name
+      _conditions.evaluate block
+    end
+
+    def default_name
+      _filter_name
+    end
+
+    def conditions(&block)
+      _conditions.instance_exec &block
+    end
+
+    def build
+      { filter: _conditions.build }
+    end
+
+    def parse(response)
+      response['doc_count']
+    end
+
+  end
+end

data/lib/elasticated/aggregations/custom_aggregation.rb

@@ -0,0 +1,25 @@
+module Elasticated
+  class CustomAggregation < Aggregation
+
+    attr_accessor :_name, :_hash
+
+    def initialize(name, &block)
+      self._name = name
+      super
+      self._hash = block.call
+    end
+
+    def default_name
+      _name
+    end
+
+    def build
+      _hash
+    end
+
+    def parse(response)
+      response
+    end
+
+  end
+end

data/lib/elasticated/aggregations/date_histogram_aggregation.rb

@@ -0,0 +1,35 @@
+module Elasticated
+  class DateHistogramAggregation < HistogramAggregation
+    include Subaggregated
+
+    attr_accessor :format, :offset
+
+    def initialize(field, opts={}, &block)
+      self.format = opts.delete(:format) || 'yyyy-MM-dd'
+      interval = opts.delete(:interval) || 'day'
+      self.offset = opts.delete(:offset)
+      super field, interval, opts, &block
+    end
+
+    def build
+      terms = { field: field, interval: interval, format: format }
+      if offset # '1.4 style'
+        terms.merge! pre_offset: offset
+        terms.merge! post_offset: offset
+      end
+      terms.merge! extra_params
+      aggregation_struct = { date_histogram: terms }
+      aggregation_struct.merge! build_subaggregations
+      aggregation_struct
+    end
+
+    def parse(response)
+      response['buckets'].inject({}) do |hash, bucket|
+        bucket_hash = { 'count' => bucket['doc_count'] }
+        bucket_hash.merge! parse_subaggregations(bucket)
+        hash.merge bucket['key_as_string'] => bucket_hash
+      end
+    end
+
+  end
+end

data/lib/elasticated/aggregations/filter_aggregation.rb

@@ -0,0 +1,33 @@
+module Elasticated
+  class FilterAggregation < Aggregation
+    include Subaggregated
+
+    attr_accessor :_evaluator, :_filter_name
+
+    def initialize(filter_name, *args, &block)
+      self._filter_name = filter_name
+      super
+      initialize_subaggregations FilterAggregationEvaluator.new, &block
+    end
+
+    def default_name
+      _filter_name
+    end
+
+    def build
+      # _subaggregations is a FilterAggregationEvaluator, so...
+      conditions = _subaggregations.build_conditions
+      body = { filter: conditions }
+      body.merge! build_subaggregations
+      body
+    end
+
+    def parse(response)
+      response_body = response
+      ret = { 'count' => response_body['doc_count'] }
+      ret.merge! parse_subaggregations(response_body)
+      ret
+    end
+
+  end
+end

data/lib/elasticated/aggregations/filter_aggregation_evaluator.rb

@@ -0,0 +1,22 @@
+module Elasticated
+  class FilterAggregationEvaluator < QueryAggregations
+
+    include BlockEvaluation
+
+    attr_accessor :_conditions
+
+    def initialize
+      super
+      self._conditions = QueryConditions.new
+    end
+
+    def conditions(&block)
+      _conditions.evaluate block
+    end
+
+    def build_conditions
+      _conditions.build
+    end
+
+  end
+end

data/lib/elasticated/aggregations/group_aggregation.rb

@@ -0,0 +1,29 @@
+module Elasticated
+  class GroupAggregation < TermsAggregation
+    include Subaggregated
+
+    def initialize(field, *args, &block)
+      super
+      initialize_subaggregations &block
+    end
+
+    def default_name
+      "group_by_#{field}"
+    end
+
+    def build
+      aggregation_struct = super
+      aggregation_struct.merge! build_subaggregations
+      aggregation_struct
+    end
+
+    def parse(response)
+      response['buckets'].inject({}) do |hash, bucket|
+        bucket_hash = { 'count' => bucket['doc_count'] }
+        bucket_hash.merge! parse_subaggregations(bucket)
+        hash.merge bucket['key'] => bucket_hash
+      end
+    end
+
+  end
+end