elastic_search_thrift 0.0.1

data/.gitignore

@@ -0,0 +1 @@
+Gemfile.lock

data/CHANGELOG

@@ -0,0 +1,3 @@
+== 0.0.1 2013-12-11
+
+ * Hi.

data/Gemfile

@@ -0,0 +1,6 @@
+source :rubygems
+gemspec
+
+gem "byebug", platform: :ruby_20
+gem 'debugger', '~> 1.6.0', platform: :ruby_19
+gem 'ritual', '~> 0.4.0'

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) George Ogata
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.markdown

@@ -0,0 +1,39 @@
+## Elastic Search Thrift
+
+Ruby ElasticSearch client that uses the thrift transport.
+
+## Install
+
+    gem install elastic_search_thrift
+
+You will of course also need the
+[thrift transport plugin for ElasticSearch][plugin].
+
+[plugin]: https://github.com/elasticsearch/elasticsearch-transport-thrift
+
+## Usage
+
+    require 'elasticsearch_thrift'
+
+    # This is the default configuration.
+    ElasticSearchThrift.configure(host: '127.0.0.1', port: 9500)
+
+    ElasticSearchThrift.client.open do |client|
+      client.get('/')
+    end
+
+## Tire Integration
+
+    ElasticSearchThrift.configure_tire
+
+## Contributing
+
+ * [Bug reports](https://github.com/howaboutwe/elastic_search_thrift/issues)
+ * [Source](https://github.com/howaboutwe/elastic_search_thrift)
+ * Patches: Fork on Github, send pull request.
+   * Include tests where practical.
+   * Leave the version alone, or bump it in a separate commit.
+
+## Copyright
+
+Copyright (c) George Ogata. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,16 @@
+require 'ritual'
+
+task :configure do
+  FileUtils.mkdir_p 'config'
+  system "curl https://raw.github.com/elasticsearch/elasticsearch-transport-thrift/master/elasticsearch.thrift > config/elasticsearch.thrift"
+  system "sed -i '' -e 's|namespace rb ElasticSearch.Thrift|namespace rb ElasticSearchThrift|' config/elasticsearch.thrift"
+end
+
+task :thrift do
+  system "thrift --gen rb -out lib/elastic_search_thrift config/elasticsearch.thrift"
+
+  Dir['lib/elastic_search_thrift/{elasticsearch_constants,elasticsearch_types,rest}.rb'].each do |path|
+    system "sed -i '' -e \"s|require 'thrift'|require 'elastic_search_thrift/mini_thrift'|\" #{path}"
+    system "sed -i '' -e \"s|require 'elasticsearch_types'|require 'elastic_search_thrift/elasticsearch_types'|\" #{path}"
+  end
+end

data/config/elasticsearch.thrift

@@ -0,0 +1,82 @@
+namespace java org.elasticsearch.thrift
+namespace csharp ElasticSearch.Thrift
+namespace cpp  elasticsearch.thrift
+namespace rb ElasticSearchThrift
+namespace py elasticsearch
+namespace perl Elasticsearch
+namespace php Elasticsearch
+
+enum Method {
+    GET = 0,
+    PUT = 1,
+    POST = 2,
+    DELETE = 3,
+    HEAD = 4,
+    OPTIONS = 5
+}
+
+struct RestRequest {
+    1: required Method method,
+    2: required string uri
+    3: optional map<string, string> parameters
+    4: optional map<string, string> headers
+    5: optional binary body
+}
+
+enum Status {
+    CONT = 100,
+    SWITCHING_PROTOCOLS = 101,
+    OK = 200,
+    CREATED = 201,
+    ACCEPTED = 202,
+    NON_AUTHORITATIVE_INFORMATION = 203,
+    NO_CONTENT = 204,
+    RESET_CONTENT = 205,
+    PARTIAL_CONTENT = 206,
+    MULTI_STATUS = 207,
+    MULTIPLE_CHOICES = 300,
+    MOVED_PERMANENTLY = 301,
+    FOUND = 302,
+    SEE_OTHER = 303,
+    NOT_MODIFIED = 304,
+    USE_PROXY = 305,
+    TEMPORARY_REDIRECT = 307,
+    BAD_REQUEST = 400,
+    UNAUTHORIZED = 401,
+    PAYMENT_REQUIRED = 402,
+    FORBIDDEN = 403,
+    NOT_FOUND = 404,
+    METHOD_NOT_ALLOWED = 405,
+    NOT_ACCEPTABLE = 406,
+    PROXY_AUTHENTICATION = 407,
+    REQUEST_TIMEOUT = 408,
+    CONFLICT = 409,
+    GONE = 410,
+    LENGTH_REQUIRED = 411,
+    PRECONDITION_FAILED = 412,
+    REQUEST_ENTITY_TOO_LARGE = 413,
+    REQUEST_URI_TOO_LONG = 414,
+    UNSUPPORTED_MEDIA_TYPE = 415,
+    REQUESTED_RANGE_NOT_SATISFIED = 416,
+    EXPECTATION_FAILED = 417,
+    UNPROCESSABLE_ENTITY = 422,
+    LOCKED = 423,
+    FAILED_DEPENDENCY = 424,
+    INTERNAL_SERVER_ERROR = 500,
+    NOT_IMPLEMENTED = 501,
+    BAD_GATEWAY = 502,
+    SERVICE_UNAVAILABLE = 503,
+    GATEWAY_TIMEOUT = 504,
+    INSUFFICIENT_STORAGE = 506
+}
+
+struct RestResponse {
+    1: required Status status,
+    2: optional map<string, string> headers,
+    3: optional binary body
+}
+
+service Rest {
+    RestResponse execute(1:required RestRequest request)        
+}
+

data/elastic_search_thrift.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+$:.unshift File.expand_path('lib', File.dirname(__FILE__))
+require 'elastic_search_thrift/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = 'elastic_search_thrift'
+  gem.version       = ElasticSearchThrift::VERSION
+  gem.authors       = ['George Ogata']
+  gem.email         = ['george.ogata@gmail.com']
+  gem.description   = "Thrift client for ElasticSearch"
+  gem.summary       = "Thrift client for ElasticSearch"
+  gem.homepage      = 'https://github.com/howaboutwe/elastic_search_thrift'
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+
+  gem.add_dependency 'thrift', '~> 0.9.1'
+  gem.add_dependency 'json', '>= 1.6.0', '< 1.9.0'
+  gem.add_development_dependency 'bundler'
+end

data/lib/elastic_search_thrift.rb

@@ -0,0 +1,23 @@
+module ElasticSearchThrift
+  autoload :Client, 'elastic_search_thrift/client'
+  autoload :TireClient, 'elastic_search_thrift/tire_client'
+  autoload :VERSION, 'elastic_search_thrift/version'
+
+  class << self
+    attr_accessor :configuration
+
+    def configure(attributes)
+      self.configuration = attributes
+    end
+
+    def client
+      @client ||= Client.new(configuration)
+    end
+
+    def configure_tire
+      TireClient.install
+    end
+  end
+
+  configure host: '127.0.0.1', port: 9500
+end

data/lib/elastic_search_thrift/client.rb

@@ -0,0 +1,86 @@
+require 'json'
+
+require 'elastic_search_thrift/mini_thrift'
+require 'elastic_search_thrift/elasticsearch_constants'
+require 'elastic_search_thrift/elasticsearch_types'
+require 'elastic_search_thrift/rest'
+require 'elastic_search_thrift/thrift_extensions'
+
+module ElasticSearchThrift
+  class Client
+    def initialize(options = {})
+      host = options[:host] || options['host'] || '127.0.0.1'
+      port = options[:port] || options['port'] || 9500
+      @socket = Thrift::Socket.new(host, port)
+      @transport = Thrift::BufferedTransport.new(socket)
+      @protocol = Thrift::BinaryProtocol.new(transport)
+      @client = ElasticSearchThrift::Rest::Client.new(protocol)
+
+      # Create finalizer proc in separate scope so it doesn't prevent GC.
+      ObjectSpace.define_finalizer(self, self.class.finalizer_for(self))
+    end
+
+    def self.finalizer_for(instance)
+      -> { instance.close }
+    end
+
+    attr_reader :socket, :transport, :protocol, :client
+
+    def get(path, parameters = {}, body = {})
+      request(ElasticSearchThrift::Method::GET, path, parameters, body)
+    end
+
+    def put(path, parameters = {}, body = {})
+      request(ElasticSearchThrift::Method::PUT, path, parameters, body)
+    end
+
+    def post(path, parameters = {}, body = {})
+      request(ElasticSearchThrift::Method::POST, path, parameters, body)
+    end
+
+    def delete(path, parameters = {}, body = {})
+      request(ElasticSearchThrift::Method::DELETE, path, parameters, body)
+    end
+
+    def request(method, uri, parameters = {}, body = {})
+      request = RestRequest.new
+      request.method = method
+      request.uri = uri
+      request.parameters = normalize_hash(parameters)
+      request.headers = {}
+      request.body = body.to_json
+      @client.execute(request)
+    end
+
+    def open
+      @transport.open unless open?
+      if block_given?
+        begin
+          yield self
+        ensure
+          close
+        end
+      end
+    end
+
+    def open?
+      @transport.open?
+    end
+
+    def close
+      @transport.close if open?
+    end
+
+    def normalize_hash(hash)
+      return hash if hash.empty?
+
+      result = {}
+      hash.each do |key, value|
+        result[key.to_s] = value.to_s
+      end
+      result
+    end
+  end
+
+  Response = Struct.new(:status, :headers, :body)
+end

data/lib/elastic_search_thrift/elasticsearch_constants.rb

@@ -0,0 +1,11 @@
+#
+# Autogenerated by Thrift Compiler (0.9.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'elastic_search_thrift/mini_thrift'
+require 'elastic_search_thrift/elasticsearch_types'
+
+module ElasticSearchThrift
+end

data/lib/elastic_search_thrift/elasticsearch_types.rb

@@ -0,0 +1,123 @@
+#
+# Autogenerated by Thrift Compiler (0.9.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'elastic_search_thrift/mini_thrift'
+
+module ElasticSearchThrift
+  module Method
+    GET = 0
+    PUT = 1
+    POST = 2
+    DELETE = 3
+    HEAD = 4
+    OPTIONS = 5
+    VALUE_MAP = {0 => "GET", 1 => "PUT", 2 => "POST", 3 => "DELETE", 4 => "HEAD", 5 => "OPTIONS"}
+    VALID_VALUES = Set.new([GET, PUT, POST, DELETE, HEAD, OPTIONS]).freeze
+  end
+
+  module Status
+    CONT = 100
+    SWITCHING_PROTOCOLS = 101
+    OK = 200
+    CREATED = 201
+    ACCEPTED = 202
+    NON_AUTHORITATIVE_INFORMATION = 203
+    NO_CONTENT = 204
+    RESET_CONTENT = 205
+    PARTIAL_CONTENT = 206
+    MULTI_STATUS = 207
+    MULTIPLE_CHOICES = 300
+    MOVED_PERMANENTLY = 301
+    FOUND = 302
+    SEE_OTHER = 303
+    NOT_MODIFIED = 304
+    USE_PROXY = 305
+    TEMPORARY_REDIRECT = 307
+    BAD_REQUEST = 400
+    UNAUTHORIZED = 401
+    PAYMENT_REQUIRED = 402
+    FORBIDDEN = 403
+    NOT_FOUND = 404
+    METHOD_NOT_ALLOWED = 405
+    NOT_ACCEPTABLE = 406
+    PROXY_AUTHENTICATION = 407
+    REQUEST_TIMEOUT = 408
+    CONFLICT = 409
+    GONE = 410
+    LENGTH_REQUIRED = 411
+    PRECONDITION_FAILED = 412
+    REQUEST_ENTITY_TOO_LARGE = 413
+    REQUEST_URI_TOO_LONG = 414
+    UNSUPPORTED_MEDIA_TYPE = 415
+    REQUESTED_RANGE_NOT_SATISFIED = 416
+    EXPECTATION_FAILED = 417
+    UNPROCESSABLE_ENTITY = 422
+    LOCKED = 423
+    FAILED_DEPENDENCY = 424
+    INTERNAL_SERVER_ERROR = 500
+    NOT_IMPLEMENTED = 501
+    BAD_GATEWAY = 502
+    SERVICE_UNAVAILABLE = 503
+    GATEWAY_TIMEOUT = 504
+    INSUFFICIENT_STORAGE = 506
+    VALUE_MAP = {100 => "CONT", 101 => "SWITCHING_PROTOCOLS", 200 => "OK", 201 => "CREATED", 202 => "ACCEPTED", 203 => "NON_AUTHORITATIVE_INFORMATION", 204 => "NO_CONTENT", 205 => "RESET_CONTENT", 206 => "PARTIAL_CONTENT", 207 => "MULTI_STATUS", 300 => "MULTIPLE_CHOICES", 301 => "MOVED_PERMANENTLY", 302 => "FOUND", 303 => "SEE_OTHER", 304 => "NOT_MODIFIED", 305 => "USE_PROXY", 307 => "TEMPORARY_REDIRECT", 400 => "BAD_REQUEST", 401 => "UNAUTHORIZED", 402 => "PAYMENT_REQUIRED", 403 => "FORBIDDEN", 404 => "NOT_FOUND", 405 => "METHOD_NOT_ALLOWED", 406 => "NOT_ACCEPTABLE", 407 => "PROXY_AUTHENTICATION", 408 => "REQUEST_TIMEOUT", 409 => "CONFLICT", 410 => "GONE", 411 => "LENGTH_REQUIRED", 412 => "PRECONDITION_FAILED", 413 => "REQUEST_ENTITY_TOO_LARGE", 414 => "REQUEST_URI_TOO_LONG", 415 => "UNSUPPORTED_MEDIA_TYPE", 416 => "REQUESTED_RANGE_NOT_SATISFIED", 417 => "EXPECTATION_FAILED", 422 => "UNPROCESSABLE_ENTITY", 423 => "LOCKED", 424 => "FAILED_DEPENDENCY", 500 => "INTERNAL_SERVER_ERROR", 501 => "NOT_IMPLEMENTED", 502 => "BAD_GATEWAY", 503 => "SERVICE_UNAVAILABLE", 504 => "GATEWAY_TIMEOUT", 506 => "INSUFFICIENT_STORAGE"}
+    VALID_VALUES = Set.new([CONT, SWITCHING_PROTOCOLS, OK, CREATED, ACCEPTED, NON_AUTHORITATIVE_INFORMATION, NO_CONTENT, RESET_CONTENT, PARTIAL_CONTENT, MULTI_STATUS, MULTIPLE_CHOICES, MOVED_PERMANENTLY, FOUND, SEE_OTHER, NOT_MODIFIED, USE_PROXY, TEMPORARY_REDIRECT, BAD_REQUEST, UNAUTHORIZED, PAYMENT_REQUIRED, FORBIDDEN, NOT_FOUND, METHOD_NOT_ALLOWED, NOT_ACCEPTABLE, PROXY_AUTHENTICATION, REQUEST_TIMEOUT, CONFLICT, GONE, LENGTH_REQUIRED, PRECONDITION_FAILED, REQUEST_ENTITY_TOO_LARGE, REQUEST_URI_TOO_LONG, UNSUPPORTED_MEDIA_TYPE, REQUESTED_RANGE_NOT_SATISFIED, EXPECTATION_FAILED, UNPROCESSABLE_ENTITY, LOCKED, FAILED_DEPENDENCY, INTERNAL_SERVER_ERROR, NOT_IMPLEMENTED, BAD_GATEWAY, SERVICE_UNAVAILABLE, GATEWAY_TIMEOUT, INSUFFICIENT_STORAGE]).freeze
+  end
+
+  class RestRequest
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    METHOD = 1
+    URI = 2
+    PARAMETERS = 3
+    HEADERS = 4
+    BODY = 5
+
+    FIELDS = {
+      METHOD => {:type => ::Thrift::Types::I32, :name => 'method', :enum_class => ::ElasticSearchThrift::Method},
+      URI => {:type => ::Thrift::Types::STRING, :name => 'uri'},
+      PARAMETERS => {:type => ::Thrift::Types::MAP, :name => 'parameters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}, :optional => true},
+      HEADERS => {:type => ::Thrift::Types::MAP, :name => 'headers', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}, :optional => true},
+      BODY => {:type => ::Thrift::Types::STRING, :name => 'body', :binary => true, :optional => true}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field method is unset!') unless @method
+      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field uri is unset!') unless @uri
+      unless @method.nil? || ::ElasticSearchThrift::Method::VALID_VALUES.include?(@method)
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field method!')
+      end
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class RestResponse
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    STATUS = 1
+    HEADERS = 2
+    BODY = 3
+
+    FIELDS = {
+      STATUS => {:type => ::Thrift::Types::I32, :name => 'status', :enum_class => ::ElasticSearchThrift::Status},
+      HEADERS => {:type => ::Thrift::Types::MAP, :name => 'headers', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}, :optional => true},
+      BODY => {:type => ::Thrift::Types::STRING, :name => 'body', :binary => true, :optional => true}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field status is unset!') unless @status
+      unless @status.nil? || ::ElasticSearchThrift::Status::VALID_VALUES.include?(@status)
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field status!')
+      end
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+end

data/lib/elastic_search_thrift/mini_thrift.rb

@@ -0,0 +1,40 @@
+# Require just the parts of the thrift library that we need.
+#
+# The thrift.rb in the thrift gem loads all sorts of stuff we don't use, and
+# depends on thin. This is brittle, but seems the lesser evil.
+
+require 'thrift/bytes'
+require 'thrift/core_ext'
+require 'thrift/exceptions'
+require 'thrift/types'
+require 'thrift/processor'
+require 'thrift/client'
+require 'thrift/struct'
+require 'thrift/union'
+require 'thrift/struct_union'
+
+# serializer
+require 'thrift/serializer/serializer'
+require 'thrift/serializer/deserializer'
+
+# protocol
+require 'thrift/protocol/base_protocol'
+require 'thrift/protocol/binary_protocol'
+require 'thrift/protocol/binary_protocol_accelerated'
+require 'thrift/protocol/compact_protocol'
+require 'thrift/protocol/json_protocol'
+
+# transport
+require 'thrift/transport/base_transport'
+require 'thrift/transport/base_server_transport'
+require 'thrift/transport/socket'
+require 'thrift/transport/server_socket'
+require 'thrift/transport/unix_socket'
+require 'thrift/transport/unix_server_socket'
+require 'thrift/transport/buffered_transport'
+require 'thrift/transport/framed_transport'
+require 'thrift/transport/http_client_transport'
+require 'thrift/transport/io_stream_transport'
+require 'thrift/transport/memory_buffer_transport'
+
+require 'thrift/thrift_native'

data/lib/elastic_search_thrift/rest.rb

@@ -0,0 +1,81 @@
+#
+# Autogenerated by Thrift Compiler (0.9.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'elastic_search_thrift/mini_thrift'
+require 'elastic_search_thrift/elasticsearch_types'
+
+module ElasticSearchThrift
+  module Rest
+    class Client
+      include ::Thrift::Client
+
+      def execute(request)
+        send_execute(request)
+        return recv_execute()
+      end
+
+      def send_execute(request)
+        send_message('execute', Execute_args, :request => request)
+      end
+
+      def recv_execute()
+        result = receive_message(Execute_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'execute failed: unknown result')
+      end
+
+    end
+
+    class Processor
+      include ::Thrift::Processor
+
+      def process_execute(seqid, iprot, oprot)
+        args = read_args(iprot, Execute_args)
+        result = Execute_result.new()
+        result.success = @handler.execute(args.request)
+        write_result(result, oprot, 'execute', seqid)
+      end
+
+    end
+
+    # HELPER FUNCTIONS AND STRUCTURES
+
+    class Execute_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      REQUEST = 1
+
+      FIELDS = {
+        REQUEST => {:type => ::Thrift::Types::STRUCT, :name => 'request', :class => ::ElasticSearchThrift::RestRequest}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field request is unset!') unless @request
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class Execute_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => ::ElasticSearchThrift::RestResponse}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+
+end

data/lib/elastic_search_thrift/thrift_extensions.rb

@@ -0,0 +1,9 @@
+module ElasticSearchThrift
+  module ResponseMixin
+    def data
+      @data ||= JSON.parse(body)
+    end
+  end
+
+  RestResponse.send :include, ResponseMixin
+end

data/lib/elastic_search_thrift/tire_client.rb

@@ -0,0 +1,69 @@
+require 'tire'
+require 'cgi'
+require 'uri'
+
+module ElasticSearchThrift
+  class TireClient
+    def self.install
+      Tire::Configuration.client(new)
+    end
+
+    def initialize
+      @client = ElasticSearchThrift.client
+      @client.open
+    end
+
+    attr_reader :client
+
+    def get(url, data = {})
+      execute :get, url, data
+    end
+
+    def post(url, data)
+      execute :post, url, data
+    end
+
+    def put(url, data)
+      execute :put, url, data
+    end
+
+    def delete(url)
+      execute :delete, url, nil
+    end
+
+    def head(url)
+      execute :head, url, nil
+    end
+
+    private
+
+    def execute(method, url, data)
+      path, params = parse_url(url)
+      path.chomp!('/')
+      response = client.send(method, path, params, normalize_data(data))
+      Tire::HTTP::Response.new(response.body, response.status, response.headers)
+    end
+
+    def parse_url(url)
+      uri = URI.parse(url)
+
+      params = {}
+      if uri.query
+        uri.query.split('&').each do |pair|
+          key, value = pair.split('=', 2)
+          params[CGI.unescape(key)] = CGI.unescape(value)
+        end
+      end
+
+      [uri.path, params]
+    end
+
+    def normalize_data(data)
+      if data.is_a?(String)
+        data =~ /\S/ ? JSON.parse(data) : {}
+      else
+        data || {}
+      end
+    end
+  end
+end

data/lib/elastic_search_thrift/version.rb

@@ -0,0 +1,11 @@
+module ElasticSearchThrift
+  VERSION = [0, 0, 1]
+
+  class << VERSION
+    include Comparable
+
+    def to_s
+      join('.')
+    end
+  end
+end

data/test/config.yml

@@ -0,0 +1,4 @@
+host: 127.0.0.1
+thrift_port: 9500
+http_port: 9200
+index: elastic_search_thrift

data/test/elastic_search_thrift/test_client.rb

@@ -0,0 +1,36 @@
+require_relative '../test_helper'
+
+describe ElasticSearchThrift::Client do
+  let(:client) { make_client }
+
+  it "executes requests and returns the response" do
+    with_thrift_client do |client|
+      response = client.request(ElasticSearchThrift::Method::GET, '/', {})
+      response.status.must_equal 200
+      response.headers.must_be_nil
+      response.data['version'].wont_be_nil
+    end
+  end
+
+  it "performs GET, POST, PUT, and DELETE requests properly" do
+    with_http_client do |http|
+      with_thrift_client do |thrift|
+        thrift.put("/#{INDEX}")
+
+        status = JSON.parse(http.get("/#{INDEX}/_status").body)
+        status['indices'].keys.must_include INDEX
+
+        thrift.post("/#{INDEX}/thing", {'refresh' => 'true'}, {name: 'a'})
+
+        response = thrift.get("/#{INDEX}/thing/_search")
+        response.data['hits']['total'].must_equal 1
+
+        id = response.data['hits']['hits'][0]['_id']
+        thrift.delete("/#{INDEX}/thing/#{id}", {'refresh' => 'true'})
+
+        response = thrift.get("/#{INDEX}/thing/_search")
+        response.data['hits']['total'].must_equal 0
+      end
+    end
+  end
+end

data/test/elastic_search_thrift/test_tire_client.rb

@@ -0,0 +1,56 @@
+require_relative '../test_helper'
+
+ElasticSearchThrift::TireClient.install
+
+describe ElasticSearchThrift::TireClient do
+  let(:client) { ElasticSearchThrift::TireClient.new }
+
+  def parse(json)
+    JSON.parse(json)
+  end
+
+  it "executes requests and returns the response" do
+    response = client.get('/', {})
+    parse(response.body)['status'].must_equal 200
+    parse(response.body)['version'].wont_be_nil
+  end
+
+  it "performs GET, POST, PUT, and DELETE requests properly" do
+    base_url = "http://#{CONFIG['host']}:#{CONFIG['port']}"
+    with_http_client do |http|
+      client.put("#{base_url}/#{INDEX}", {})
+      status = JSON.parse(http.get("/#{INDEX}/_status").body)
+      status['indices'].keys.must_include INDEX
+
+      client.post("#{base_url}/#{INDEX}/thing?refresh=true", {name: 'a'})
+      response = client.get("#{base_url}/#{INDEX}/thing/_search")
+      parse(response.body)['hits']['total'].must_equal 1
+
+      id = parse(response.body)['hits']['hits'][0]['_id']
+      client.delete("#{base_url}/#{INDEX}/thing/#{id}?refresh=true")
+      response = client.get("#{base_url}/#{INDEX}/thing/_search")
+      parse(response.body)['hits']['total'].must_equal 0
+    end
+  end
+
+  it "integrates correctly into the Tire DSL" do
+    with_http_client do |http|
+      index = Tire::Index.new(INDEX)
+      index.create.success?.must_equal true
+      status = JSON.parse(http.get("/#{INDEX}/_status").body)
+      status['indices'].keys.must_include INDEX
+
+      index.store(_type: 'thing', name: 'a')
+      index.refresh
+      response = client.get("/#{INDEX}/thing/_search")
+      parse(response.body)['hits']['total'].must_equal 1
+
+      id = parse(response.body)['hits']['hits'][0]['_id']
+      index.remove(_type: 'thing', id: id)
+      index.refresh
+
+      response = client.get("/#{INDEX}/thing/_search")
+      parse(response.body)['hits']['total'].must_equal 0
+    end
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,31 @@
+ROOT = File.expand_path('..', File.dirname(__FILE__))
+$:.unshift "#{ROOT}/lib"
+
+require 'minitest/spec'
+require 'yaml'
+require 'net/http'
+require 'elastic_search_thrift'
+
+require RUBY_VERSION < '2' ? 'debugger' : 'byebug'
+
+CONFIG = YAML.load_file("#{ROOT}/test/config.yml")
+INDEX = CONFIG.delete('index') || 'elastic_search_thrift'
+ElasticSearchThrift.configure(CONFIG)
+
+MiniTest::Spec.class_eval do
+  def with_thrift_client(&block)
+    client = ElasticSearchThrift::Client.new(host: CONFIG['host'], port: CONFIG['thrift_port'])
+    client.open(&block)
+  end
+
+  def with_http_client
+    http = Net::HTTP.new(CONFIG['host'], CONFIG['http_port'])
+    http.start do
+      begin
+        yield http
+      ensure
+        http.delete("/#{INDEX}")
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,125 @@
+--- !ruby/object:Gem::Specification
+name: elastic_search_thrift
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- George Ogata
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thrift
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.1
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.6.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: 1.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.6.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: 1.9.0
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Thrift client for ElasticSearch
+email:
+- george.ogata@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- CHANGELOG
+- Gemfile
+- LICENSE
+- README.markdown
+- Rakefile
+- config/elasticsearch.thrift
+- elastic_search_thrift.gemspec
+- lib/elastic_search_thrift.rb
+- lib/elastic_search_thrift/client.rb
+- lib/elastic_search_thrift/elasticsearch_constants.rb
+- lib/elastic_search_thrift/elasticsearch_types.rb
+- lib/elastic_search_thrift/mini_thrift.rb
+- lib/elastic_search_thrift/rest.rb
+- lib/elastic_search_thrift/thrift_extensions.rb
+- lib/elastic_search_thrift/tire_client.rb
+- lib/elastic_search_thrift/version.rb
+- test/config.yml
+- test/elastic_search_thrift/test_client.rb
+- test/elastic_search_thrift/test_tire_client.rb
+- test/test_helper.rb
+homepage: https://github.com/howaboutwe/elastic_search_thrift
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Thrift client for ElasticSearch
+test_files:
+- test/config.yml
+- test/elastic_search_thrift/test_client.rb
+- test/elastic_search_thrift/test_tire_client.rb
+- test/test_helper.rb
+has_rdoc: