Contents

require 'elasticsearch'

module Sequel
  module Plugins
    module Elasticsearch
      def self.apply(model, _opts = OPTS)
        model.instance_variable_set(:@elasticsearch_opts, {})
        model.instance_variable_set(:@elasticsearch_index, nil)
        model.instance_variable_set(:@elasticsearch_type, 'sync')
        model
      end

      def self.configure(model, opts = OPTS)
        model.elasticsearch_opts = opts[:elasticsearch] || {}
        model.elasticsearch_index = (opts[:index] || model.table_name).to_sym
        model.elasticsearch_type = (opts[:type] || :sync).to_sym
        model
      end

      module ClassMethods
        attr_accessor :elasticsearch_opts, :elasticsearch_index, :elasticsearch_type
      end

      module InstanceMethods
        def after_create
          super
          index_document
        end

        def after_destroy
          super
          destroy_document
        end

        def after_update
          super
          index_document
        end

        def es_client
          @es_client = ::Elasticsearch::Client.new self.class.elasticsearch_opts
        end

        private

        def document_id
          doc_id = pk
          doc_id = doc_id.join('_') if doc_id.is_a? Array
          doc_id
        end

        def document_path
          {
            index: self.class.elasticsearch_index,
            type: self.class.elasticsearch_type,
            id: document_id
          }
        end

        def index_document
          params = document_path
          params[:body] = values
          es_client.index params
        end

        def destroy_document
          es_client.delete document_path
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
sequel-elasticsearch-0.1.0	lib/sequel/plugins/elasticsearch.rb