rebat 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rvmrc

@@ -0,0 +1 @@
+rvm use 1.9.3-p125@rebatdb

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Omar Mekky
+
+MIT License
+
+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.md

@@ -0,0 +1,29 @@
+# Rebat
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rebat'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rebat
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/lib/rebat.rb

@@ -0,0 +1,23 @@
+require 'thrift'
+
+# thrift sources. load order is important.
+$LOAD_PATH << File.join(File.dirname(__FILE__), 'rebat', 'thrift')
+
+require 'rebat/thrift/rebat_d_b'
+require 'rebat/thrift/rebat_constants'
+require 'rebat/thrift/rebat_types'
+
+require 'rebat/version'
+
+module Rebat
+  autoload  :Selector,      'rebat/selector'
+  autoload  :Client,        'rebat/client'
+
+  class << self
+    attr_accessor :relations
+
+    def new(*args)
+      Rebat::Client.new(*args)
+    end
+  end
+end

data/lib/rebat/client.rb

@@ -0,0 +1,81 @@
+class Rebat::Client
+  def initialize(host, port, relations)
+    @transport = Thrift::BufferedTransport.new(Thrift::Socket.new(host, port))
+    @protocol = Thrift::BinaryProtocol.new(@transport)
+    @client = Rebat::Thrift::RebatDB::Client.new(@protocol)
+
+    @transport.open
+  end
+
+  def send(&block)
+    begin
+      @transport.open unless @transport.open?
+      yield @client
+    rescue Thrift::Exception => tx
+      print 'Thrift::Exception: ', tx.message, "\n"
+    end
+  end
+
+  def close
+    @transport.close
+  end
+
+  def add(from_entity_id, from_entity_type, to_entity_id, to_entity_type, weight, relation_id)
+    edge = Rebat::Thrift::Edge.new
+
+    edge.fromEntityId   = from_entity_id
+    edge.fromEntityType = from_entity_type
+    edge.toEntityId     = to_entity_id
+    edge.toEntityType   = to_entity_type
+    edge.weight         = weight
+    edge.relationId     = relation_id
+
+    self.send do |client|
+      client.addQuery(edge)
+    end
+  end
+
+  def updateWeight(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id, new_weight)
+    edge = Rebat::Thrift::Edge.new
+
+    edge.fromEntityId   = from_entity_id
+    edge.fromEntityType = from_entity_type
+    edge.toEntityId     = to_entity_id
+    edge.toEntityType   = to_entity_type
+    edge.relationId     = relation_id
+
+    self.send do |client|
+      client.updateWeightQuery(edge, new_weight)
+    end
+  end
+
+  def delete(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id)
+    edge = Rebat::Thrift::Edge.new
+
+    edge.fromEntityId   = from_entity_id
+    edge.fromEntityType = from_entity_type
+    edge.toEntityId     = to_entity_id
+    edge.toEntityType   = to_entity_type
+    edge.relationId     = relation_id
+
+    self.send do |client|
+      client.deleteQuery(edge)
+    end
+  end
+
+  def where(*args)
+    Rebat::Selector.new(self).where *args
+  end
+
+  def union(*args)
+    Rebat::Selector.new(self).union *args
+  end
+
+  def intersect(*args)
+    Rebat::Selector.new(self).intersect *args
+  end
+
+  def exclude(*args)
+    Rebat::Selector.new(self).exclude *args
+  end
+end

data/lib/rebat/selector.rb

@@ -0,0 +1,54 @@
+class Rebat::Selector
+  attr_reader :client
+
+  def initialize(client)
+    @client = client
+    @query_list = []
+  end
+
+  def where(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id)
+    @query_list << create_query(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id, Rebat::Thrift::QueryType::WHERE)
+    return self
+  end
+
+  def union(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id)
+    @query_list << create_query(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id, Rebat::Thrift::QueryType::UNION)
+    return self
+  end
+
+  def intersect(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id)
+    @query_list << create_query(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id, Rebat::Thrift::QueryType::INTERSECT)
+    return self
+  end
+
+  def exclude(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id)
+    @query_list << create_query(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id, Rebat::Thrift::QueryType::EXCLUDE)
+    return self
+  end
+
+  def entries
+    @client.send do |client|
+      return client.selectQuery(@query_list)
+    end
+  end
+
+  private
+  attr_reader :query_list
+
+  def create_query(from_entity_id, from_entity_type, to_entity_id, to_entity_type, relation_id, qtype)
+    edge = Rebat::Thrift::Edge.new
+
+    edge.fromEntityId   = from_entity_id    unless from_entity_id.nil?
+    edge.fromEntityType = from_entity_type  unless from_entity_type.nil?
+    edge.toEntityId     = to_entity_id      unless to_entity_id.nil?
+    edge.toEntityType   = to_entity_type    unless to_entity_type.nil?
+    edge.relationId     = relation_id
+
+    query = Rebat::Thrift::Query.new
+
+    query.edge = edge
+    query.qtype = qtype
+
+    return query
+  end
+end

data/lib/rebat/thrift/rebat_constants.rb

@@ -0,0 +1,12 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'rebat_types'
+
+    module Rebat
+      module Thrift
+  end
+end

data/lib/rebat/thrift/rebat_d_b.rb

@@ -0,0 +1,246 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'rebat_types'
+
+        module Rebat
+          module Thrift
+            module RebatDB
+              class Client
+                include ::Thrift::Client
+
+                def addQuery(edge)
+                  send_addQuery(edge)
+                  return recv_addQuery()
+                end
+
+                def send_addQuery(edge)
+                  send_message('addQuery', AddQuery_args, :edge => edge)
+                end
+
+                def recv_addQuery()
+                  result = receive_message(AddQuery_result)
+                  return result.success unless result.success.nil?
+                  raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'addQuery failed: unknown result')
+                end
+
+                def deleteQuery(edge)
+                  send_deleteQuery(edge)
+                  return recv_deleteQuery()
+                end
+
+                def send_deleteQuery(edge)
+                  send_message('deleteQuery', DeleteQuery_args, :edge => edge)
+                end
+
+                def recv_deleteQuery()
+                  result = receive_message(DeleteQuery_result)
+                  return result.success unless result.success.nil?
+                  raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'deleteQuery failed: unknown result')
+                end
+
+                def updateWeightQuery(edge, weight)
+                  send_updateWeightQuery(edge, weight)
+                  return recv_updateWeightQuery()
+                end
+
+                def send_updateWeightQuery(edge, weight)
+                  send_message('updateWeightQuery', UpdateWeightQuery_args, :edge => edge, :weight => weight)
+                end
+
+                def recv_updateWeightQuery()
+                  result = receive_message(UpdateWeightQuery_result)
+                  return result.success unless result.success.nil?
+                  raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'updateWeightQuery failed: unknown result')
+                end
+
+                def selectQuery(queryList)
+                  send_selectQuery(queryList)
+                  return recv_selectQuery()
+                end
+
+                def send_selectQuery(queryList)
+                  send_message('selectQuery', SelectQuery_args, :queryList => queryList)
+                end
+
+                def recv_selectQuery()
+                  result = receive_message(SelectQuery_result)
+                  return result.success unless result.success.nil?
+                  raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'selectQuery failed: unknown result')
+                end
+
+              end
+
+              class Processor
+                include ::Thrift::Processor
+
+                def process_addQuery(seqid, iprot, oprot)
+                  args = read_args(iprot, AddQuery_args)
+                  result = AddQuery_result.new()
+                  result.success = @handler.addQuery(args.edge)
+                  write_result(result, oprot, 'addQuery', seqid)
+                end
+
+                def process_deleteQuery(seqid, iprot, oprot)
+                  args = read_args(iprot, DeleteQuery_args)
+                  result = DeleteQuery_result.new()
+                  result.success = @handler.deleteQuery(args.edge)
+                  write_result(result, oprot, 'deleteQuery', seqid)
+                end
+
+                def process_updateWeightQuery(seqid, iprot, oprot)
+                  args = read_args(iprot, UpdateWeightQuery_args)
+                  result = UpdateWeightQuery_result.new()
+                  result.success = @handler.updateWeightQuery(args.edge, args.weight)
+                  write_result(result, oprot, 'updateWeightQuery', seqid)
+                end
+
+                def process_selectQuery(seqid, iprot, oprot)
+                  args = read_args(iprot, SelectQuery_args)
+                  result = SelectQuery_result.new()
+                  result.success = @handler.selectQuery(args.queryList)
+                  write_result(result, oprot, 'selectQuery', seqid)
+                end
+
+              end
+
+              # HELPER FUNCTIONS AND STRUCTURES
+
+              class AddQuery_args
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                EDGE = 1
+
+                FIELDS = {
+                  EDGE => {:type => ::Thrift::Types::STRUCT, :name => 'edge', :class => Rebat::Thrift::Edge}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class AddQuery_result
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                SUCCESS = 0
+
+                FIELDS = {
+                  SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class DeleteQuery_args
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                EDGE = 1
+
+                FIELDS = {
+                  EDGE => {:type => ::Thrift::Types::STRUCT, :name => 'edge', :class => Rebat::Thrift::Edge}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class DeleteQuery_result
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                SUCCESS = 0
+
+                FIELDS = {
+                  SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class UpdateWeightQuery_args
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                EDGE = 1
+                WEIGHT = 2
+
+                FIELDS = {
+                  EDGE => {:type => ::Thrift::Types::STRUCT, :name => 'edge', :class => Rebat::Thrift::Edge},
+                  WEIGHT => {:type => ::Thrift::Types::I64, :name => 'weight'}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class UpdateWeightQuery_result
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                SUCCESS = 0
+
+                FIELDS = {
+                  SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class SelectQuery_args
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                QUERYLIST = 1
+
+                FIELDS = {
+                  QUERYLIST => {:type => ::Thrift::Types::LIST, :name => 'queryList', :element => {:type => ::Thrift::Types::STRUCT, :class => Rebat::Thrift::Query}}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+              class SelectQuery_result
+                include ::Thrift::Struct, ::Thrift::Struct_Union
+                SUCCESS = 0
+
+                FIELDS = {
+                  SUCCESS => {:type => ::Thrift::Types::LIST, :name => 'success', :element => {:type => ::Thrift::Types::STRUCT, :class => Rebat::Thrift::Edge}}
+                }
+
+                def struct_fields; FIELDS; end
+
+                def validate
+                end
+
+                ::Thrift::Struct.generate_accessors self
+              end
+
+            end
+
+          end
+        end

data/lib/rebat/thrift/rebat_types.rb

@@ -0,0 +1,68 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+
+module Rebat
+  module Thrift
+        module QueryType
+          WHERE = 0
+          INTERSECT = 1
+          UNION = 2
+          NOT = 3
+          VALUE_MAP = {0 => "WHERE", 1 => "INTERSECT", 2 => "UNION", 3 => "NOT"}
+          VALID_VALUES = Set.new([WHERE, INTERSECT, UNION, NOT]).freeze
+        end
+
+        class Edge
+          include ::Thrift::Struct, ::Thrift::Struct_Union
+          FROMENTITYID = 1
+          FROMENTITYTYPE = 2
+          TOENTITYID = 3
+          TOENTITYTYPE = 4
+          WEIGHT = 5
+          RELATIONID = 6
+
+          FIELDS = {
+            FROMENTITYID => {:type => ::Thrift::Types::I64, :name => 'fromEntityId', :optional => true},
+            FROMENTITYTYPE => {:type => ::Thrift::Types::STRING, :name => 'fromEntityType', :optional => true},
+            TOENTITYID => {:type => ::Thrift::Types::I64, :name => 'toEntityId', :optional => true},
+            TOENTITYTYPE => {:type => ::Thrift::Types::STRING, :name => 'toEntityType', :optional => true},
+            WEIGHT => {:type => ::Thrift::Types::I64, :name => 'weight', :default => 0, :optional => true},
+            RELATIONID => {:type => ::Thrift::Types::I64, :name => 'relationId', :optional => true}
+          }
+
+          def struct_fields; FIELDS; end
+
+          def validate
+          end
+
+          ::Thrift::Struct.generate_accessors self
+        end
+
+        class Query
+          include ::Thrift::Struct, ::Thrift::Struct_Union
+          EDGE = 1
+          QTYPE = 2
+
+          FIELDS = {
+            EDGE => {:type => ::Thrift::Types::STRUCT, :name => 'edge', :class => Rebat::Thrift::Edge},
+            QTYPE => {:type => ::Thrift::Types::I32, :name => 'qtype', :default =>             0, :optional => true, :enum_class => Rebat::Thrift::QueryType}
+          }
+
+          def struct_fields; FIELDS; end
+
+          def validate
+            raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field edge is unset!') unless @edge
+            unless @qtype.nil? || Rebat::Thrift::QueryType::VALID_VALUES.include?(@qtype)
+              raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field qtype!')
+            end
+          end
+
+          ::Thrift::Struct.generate_accessors self
+        end
+
+      end
+    end

data/lib/rebat/version.rb

@@ -0,0 +1,3 @@
+module Rebat
+  VERSION = "0.0.1"
+end

data/rebat.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/rebat/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Omar Mekky", "Khaled Gomaa"]
+  gem.email         = ["omar.mekky@mashsolvents.com", "khaled.hassan@mashsolvents.com"]
+  gem.description   = %q{Rebat DB driver for ruby }
+  gem.summary       = %q{Rebat DB driver for ruby }
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "rebat"
+  gem.require_paths = ["lib"]
+  gem.version       = Rebat::VERSION
+
+  gem.add_dependency 'thrift', '>= 0.8.0'
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification
+name: rebat
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Omar Mekky
+- Khaled Gomaa
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thrift
+  requirement: &70286233997100 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70286233997100
+description: ! 'Rebat DB driver for ruby '
+email:
+- omar.mekky@mashsolvents.com
+- khaled.hassan@mashsolvents.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rvmrc
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/rebat.rb
+- lib/rebat/client.rb
+- lib/rebat/selector.rb
+- lib/rebat/thrift/rebat_constants.rb
+- lib/rebat/thrift/rebat_d_b.rb
+- lib/rebat/thrift/rebat_types.rb
+- lib/rebat/version.rb
+- rebat.gemspec
+homepage: ''
+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.17
+signing_key: 
+specification_version: 3
+summary: Rebat DB driver for ruby
+test_files: []