tempest_db 0.12.1

data/lib/gen/tempest_types.rb

@@ -0,0 +1,93 @@
+#
+# Autogenerated by Thrift Compiler (0.9.3)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+
+class InvalidNodeIdException < ::Thrift::Exception
+  include ::Thrift::Struct, ::Thrift::Struct_Union
+  def initialize(message=nil)
+    super()
+    self.message = message
+  end
+
+  MESSAGE = 1
+
+  FIELDS = {
+    MESSAGE => {:type => ::Thrift::Types::STRING, :name => 'message'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+  ::Thrift::Struct.generate_accessors self
+end
+
+class InvalidIndexException < ::Thrift::Exception
+  include ::Thrift::Struct, ::Thrift::Struct_Union
+  def initialize(message=nil)
+    super()
+    self.message = message
+  end
+
+  MESSAGE = 1
+
+  FIELDS = {
+    MESSAGE => {:type => ::Thrift::Types::STRING, :name => 'message'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+  ::Thrift::Struct.generate_accessors self
+end
+
+class InvalidArgumentException < ::Thrift::Exception
+  include ::Thrift::Struct, ::Thrift::Struct_Union
+  def initialize(message=nil)
+    super()
+    self.message = message
+  end
+
+  MESSAGE = 1
+
+  FIELDS = {
+    MESSAGE => {:type => ::Thrift::Types::STRING, :name => 'message'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+  ::Thrift::Struct.generate_accessors self
+end
+
+class BidirectionalPPRParams
+  include ::Thrift::Struct, ::Thrift::Struct_Union
+  RELATIVEERROR = 1
+  RESETPROBABILITY = 2
+  MINPROBABILITY = 3
+
+  FIELDS = {
+    RELATIVEERROR => {:type => ::Thrift::Types::DOUBLE, :name => 'relativeError'},
+    RESETPROBABILITY => {:type => ::Thrift::Types::DOUBLE, :name => 'resetProbability'},
+    MINPROBABILITY => {:type => ::Thrift::Types::DOUBLE, :name => 'minProbability', :optional => true}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+    raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field relativeError is unset!') unless @relativeError
+    raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field resetProbability is unset!') unless @resetProbability
+  end
+
+  ::Thrift::Struct.generate_accessors self
+end
+

data/lib/tempest_db.rb

@@ -0,0 +1,82 @@
+# Copyright 2016 Teapot, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this
+# file except in compliance with the License. You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software distributed
+# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
+# CONDITIONS OF ANY KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations under the License.
+
+# This file defines a ruby TempestClient class for TempestDB.  The client wraps the methods
+# provided by the thrift generated client, and also makes some methods easier to use.
+# For example, it converts the types of attributes into the correct dynamic ruby type, and allows
+# optional arguments (which thrift 0.9.3 doesn't support) for some methods.
+
+$:.push(File.expand_path(File.join(File.dirname(__FILE__),"gen")))
+
+require 'tempest_d_b_service'
+require 'tempest_db_constants'
+require 'tempest_db_types'
+require 'base_thrift_client'
+
+# May also require the types and the constants for more complex services
+
+module Teapot
+  module TempestDB
+    class TempestClient < Teapot::BaseThriftClient
+      def init_executor(protocol)
+        Teapot::TempestDB::TempestDBService::Client.new(protocol)
+      end
+
+      def TempestClient.jsonToValue(json_attribute)
+        if json_attribute[0] == '"'
+          json_attribute[1...-1]
+        elsif json_attribute == "true"
+          true
+        elsif json_attribute == "false"
+          false
+        elsif json_attribute == "null"
+          nil
+        else
+          # If this isn't an int, the server made a mistake, and there isn't much the client can do.
+          json_attribute.to_i
+        end
+      end
+
+      def getMultiNodeAttribute(graphName, nodeIds, attributeName)
+        id_to_json = get_executor.getMultiNodeAttributeAsJSON(graphName, nodeIds, attributeName)
+        Hash[id_to_json.map{ |k,v| [k, TempestClient.jsonToValue(v)] }]
+      end
+
+      def getNodeAttribute(graphName, nodeId, attributeName)
+        self.getMultiNodeAttribute(graphName, [nodeId], attributeName)[nodeId]
+      end
+
+      def kStepOutNeighborsFiltered(edgeType, sourceId, k, sqlClause: "", degreeFilter: {}, alternating: true)
+        get_executor.kStepOutNeighborsFiltered(edgeType, sourceId, k, sqlClause, degreeFilter, alternating)
+      end
+      def kStepInNeighborsFiltered(edgeType, sourceId, k, sqlClause: "", degreeFilter: {}, alternating: true)
+        get_executor.kStepInNeighborsFiltered(edgeType, sourceId, k, sqlClause, degreeFilter, alternating)
+      end
+
+      # Delegate other methods to the thrift generated method
+      # Note: for autocomplete and documentation, we could improve usability by listing methods
+      # explicitly (as we do for python), but that would require manual changes whenever the thrift
+      # api changes, so for ruby we automatically delegate all other methods.
+      def method_missing(m, *args)
+        get_executor.send m, *args
+      end
+    end
+  end
+end
+
+def get_tempest_client(server = 'localhost', port = 10001)
+  Teapot::TempestDB::TempestClient.new(server, port)
+end
+
+def get_empty_filter()
+  Hash.new
+end

metadata

@@ -0,0 +1,55 @@
+--- !ruby/object:Gem::Specification
+name: tempest_db
+version: !ruby/object:Gem::Version
+  version: 0.12.1
+  prerelease: 
+platform: ruby
+authors:
+- Peter Lofgren
+- Pranav Dandekar
+- Ashish Goel
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-11-18 00:00:00.000000000 Z
+dependencies: []
+description: Client for Tempest Graph Database.
+email: peter@teapot.co
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/base_thrift_client.rb
+- lib/gen/tempest_constants.rb
+- lib/gen/tempest_d_b_service.rb
+- lib/gen/tempest_db_constants.rb
+- lib/gen/tempest_db_types.rb
+- lib/gen/tempest_graph_service.rb
+- lib/gen/tempest_types.rb
+- lib/tempest_db.rb
+homepage: http://teapot.co
+licenses:
+- Apache 2
+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.23
+signing_key: 
+specification_version: 3
+summary: Client for Tempest Graph Database
+test_files: []