72 lines
1.5 KiB
Ruby
72 lines
1.5 KiB
Ruby
|
# DataMapper and Elasticsearch
|
||
|
|
# ============================
|
||
|
|
#
|
||
|
|
# https://github.com/datamapper/dm-core
|
||
|
|
# https://github.com/datamapper/dm-active_model
|
||
|
|
|
||
|
|
|
||
|
|
$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
|
||
|
|
|
||
|
|
require 'pry'
|
||
|
|
Pry.config.history.file = File.expand_path('../../tmp/elasticsearch_development.pry', __FILE__)
|
||
|
|
|
||
|
|
require 'logger'
|
||
|
|
require 'ansi/core'
|
||
|
|
|
||
|
|
require 'data_mapper'
|
||
|
|
require 'dm-active_model'
|
||
|
|
|
||
|
|
require 'active_support/all'
|
||
|
|
|
||
|
|
require 'elasticsearch/model'
|
||
|
|
|
||
|
|
DataMapper::Logger.new(STDOUT, :debug)
|
||
|
|
DataMapper.setup(:default, 'sqlite::memory:')
|
||
|
|
|
||
|
|
class Article
|
||
|
|
include DataMapper::Resource
|
||
|
|
|
||
|
|
property :id, Serial
|
||
|
|
property :title, String
|
||
|
|
property :published_at, DateTime
|
||
|
|
end
|
||
|
|
|
||
|
|
DataMapper.auto_migrate!
|
||
|
|
DataMapper.finalize
|
||
|
|
|
||
|
|
Article.create title: 'Foo'
|
||
|
|
Article.create title: 'Bar'
|
||
|
|
Article.create title: 'Foo Foo'
|
||
|
|
|
||
|
|
# Extend the model with Elasticsearch support
|
||
|
|
#
|
||
|
|
Article.__send__ :include, Elasticsearch::Model
|
||
|
|
|
||
|
|
# The DataMapper adapter
|
||
|
|
#
|
||
|
|
module DataMapperAdapter
|
||
|
|
|
||
|
|
# Implement the interface for fetching records
|
||
|
|
#
|
||
|
|
module Records
|
||
|
|
def records
|
||
|
|
klass.all(id: @ids)
|
||
|
|
end
|
||
|
|
|
||
|
|
# ...
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
# Register the adapter
|
||
|
|
#
|
||
|
|
Elasticsearch::Model::Adapter.register(
|
||
|
|
DataMapperAdapter,
|
||
|
|
lambda { |klass| defined?(::DataMapper::Resource) and klass.ancestors.include?(::DataMapper::Resource) }
|
||
|
|
)
|
||
|
|
|
||
|
|
response = Article.search 'foo';
|
||
|
|
|
||
|
|
Pry.start(binding, prompt: lambda { |obj, nest_level, _| '> ' },
|
||
|
|
input: StringIO.new('response.records.to_a'),
|
||
|
|
quiet: true)
|