jruby-elasticsearch 0.0.1

data/lib/elasticsearch/actionlistener.rb

@@ -0,0 +1,46 @@
+require "java"
+require "elasticsearch/namespace"
+
+class ElasticSearch::ActionListener
+  include org.elasticsearch.action.ActionListener
+
+  def initialize
+    @failure_callbacks = []
+    @success_callbacks = []
+  end # def initialize
+
+  # Helper for registering callbacks.
+  # 'what' should be either :failure or :success
+  #
+  # You can register multiple callbacks if you wish.
+  # Callbacks are invoked in order of addition.
+  def on(what, &block)
+    case what
+    when :failure
+      @failure_callbacks << block
+    when :success
+      @success_callbacks << block
+    end
+    return self
+  end # def on
+ 
+  # Conforming to Interface org.elasticsearch.action.ActionListener
+  def onFailure(exception)
+    if !@failure_callbacks.empty?
+      @failure_callbacks.each { |c| c.call(exception) }
+    else
+      # Default is no failure callbacks
+      raise exception
+    end
+  end # def onFailure
+
+  # Conforming to Interface org.elasticsearch.action.ActionListener
+  def onResponse(response)
+    if !@success_callbacks.empty?
+      @success_callbacks.each { |c| c.call(response) }
+    else
+      # Default if no success callbacks
+      puts "#{self.class.name}#onResponse => #{response.inspect}"
+    end
+  end # def onResponse
+end # class ElasticSearch::ActionListener

data/lib/elasticsearch/bulkrequest.rb

@@ -0,0 +1,31 @@
+require "elasticsearch/namespace"
+require "elasticsearch/request"
+
+class ElasticSearch::BulkRequest < ElasticSearch::Request
+  # Create a new index request.
+  def initialize(client)
+    @client = client
+
+    @prep = @client.prepareBulk(index, type, id)
+    super
+  end
+
+  # Execute this index request.
+  # This call is asynchronous.
+  #
+  # If a block is given, register it for both failure and success.
+  def execute(&block)
+    use_callback(&block) if block_given
+    action = @prep.execute(@handler)
+    return action
+  end
+
+  # Execute this index request synchronously
+  def execute!
+    return @prep.execute.actionGet()
+  end
+
+  def <<(request)
+    @prep.add(request)
+  end # def <<
+end # def ElasticSearch::BulkRequest

data/lib/elasticsearch/bulkstream.rb

@@ -0,0 +1,7 @@
+require "elasticsearch/namespace"
+
+class ElasticSearch::BulkStream
+  def initialize
+    @bulkthread = Thread.new
+  end
+end

data/lib/elasticsearch/client.rb

@@ -0,0 +1,54 @@
+require "java"
+require "elasticsearch/namespace"
+require "elasticsearch/indexrequest"
+
+class ElasticSearch::Client
+  def initialize
+    @node = org.elasticsearch.node.NodeBuilder.nodeBuilder.client(true).node
+    @client = @node.client
+  end # def initialize
+
+  # Index a new document
+  #
+  # args:
+  #   index: the index name
+  #   type: the type name
+  #   id: (optional) the id of the document
+  #   data: (optional) the data for this document
+  #   &block: (optional) optional block for using the DSL to add data
+  #
+  # Returns an ElasticSearch::IndexRequest instance.
+  #
+  # Example w/ DSL:
+  #
+  #     request = client.index("foo", "logs") do
+  #       filename "/var/log/message"
+  #       mesage "hello world"
+  #       timestamp 123456
+  #     end
+  #
+  #     request.execute!
+  def index(index, type, id=nil, data={}, &block)
+    # Permit 'id' being omitted entirely.
+    # Thus a call call: index("foo", "bar", somehash) is valid.
+    if id.is_a?(Hash)
+      data = id
+      id = nil
+    end
+
+    indexreq = ElasticSearch::IndexRequest.new(@client, index, type, id, data)
+    if block_given?
+      indexreq.instance_eval(&block)
+    end
+    return indexreq
+  end # def index
+
+  def cluster
+    return @client.admin.cluster
+  end
+
+  def node
+    return @client.admin.cluster
+  end
+end # class ElasticSearch::Client
+

data/lib/elasticsearch/indexrequest.rb

@@ -0,0 +1,42 @@
+require "elasticsearch/namespace"
+require "elasticsearch/request"
+
+class ElasticSearch::IndexRequest < ElasticSearch::Request
+  # Create a new index request.
+  def initialize(client, index, type, id=nil, data={})
+    @client = client
+    @index = index
+    @type = type
+    @id = id
+    @data = data
+
+    @prep = @client.prepareIndex(index, type, id)
+    super
+  end
+
+  # Execute this index request.
+  # This call is asynchronous.
+  #
+  # If a block is given, register it for both failure and success.
+  def execute(&block)
+    @prep.setSource(@data)
+    use_callback(&block) if block_given?
+
+    action = @prep.execute(@handler)
+    return action
+  end
+
+  # Execute this index request synchronously
+  def execute!
+    @prep.setSource(@data)
+    return @prep.execute.actionGet()
+  end
+
+  # DSL helper.
+  # TODO(sissel): Move this away to a DSL module.
+  def method_missing(*args)
+    key, value = args
+    puts "Adding: #{key}: #{value.inspect}"
+    @data[key.to_s] = value
+  end
+end

data/lib/elasticsearch/namespace.rb

@@ -0,0 +1,4 @@
+
+module ElasticSearch
+# empty for now
+end

data/lib/elasticsearch/request.rb

@@ -0,0 +1,25 @@
+require "elasticsearch/namespace"
+require "elasticsearch/actionlistener"
+
+class ElasticSearch::Request
+  # Create a new index request.
+  def initialize()
+    @handler = ElasticSearch::ActionListener.new
+  end
+
+  # See ElasticSearch::ActionListener#on
+  def on(event, &block)
+    return @handler.on(*args, &block)
+  end
+
+  # Execute this index request.
+  # This call is asynchronous.
+  #
+  # If a block is given, register it for both failure and success.
+  def use_callback(&block)
+    if block_given?
+      on(:failure, &block)
+      on(:success, &block)
+    end
+  end
+end

data/lib/elasticsearch.rb

@@ -0,0 +1,8 @@
+require "elasticsearch/client"
+
+#class Proc
+  #include java.lang.Runnable
+  #alias_method :run, :call
+#end
+
+module ElasticSearch; end;

metadata

@@ -0,0 +1,73 @@
+--- !ruby/object:Gem::Specification 
+name: jruby-elasticsearch
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+    - 0
+    - 0
+    - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+  - Jordan Sissel
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-02-27 00:00:00 -08:00
+default_executable: 
+dependencies: []
+
+description: ...
+email: 
+  - jls@semicomplete.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+  - lib/elasticsearch.rb
+  - lib/elasticsearch/namespace.rb
+  - lib/elasticsearch/bulkrequest.rb
+  - lib/elasticsearch/actionlistener.rb
+  - lib/elasticsearch/request.rb
+  - lib/elasticsearch/indexrequest.rb
+  - lib/elasticsearch/client.rb
+  - lib/elasticsearch/bulkstream.rb
+has_rdoc: true
+homepage: https://github.com/jordansissel/jruby-elasticsearch
+licenses: 
+  - Apache License (2.0)
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+  - lib
+  - lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+          - 0
+        version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+          - 0
+        version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: JRuby API for ElasticSearch using the native ES Java API
+test_files: []
+