graphshaper 0.2 → 0.2.1

data/bin/graphshaper

@@ -14,8 +14,10 @@ else
   edge_output_file = File.new "edges.csv", "w"
   logger = Graphshaper::LoggingAdapter.new vertex_output_file, edge_output_file
   
+  avocado_db = Graphshaper::AvocadoDbAdapter.new "vertices", "edges"
+  
   start_time = Time.now
-  graph = Graphshaper::UndirectedGraph.without_orphans_with_order_of inner_vertices, adapters: [logger]
+  graph = Graphshaper::UndirectedGraph.without_orphans_with_order_of inner_vertices, adapters: [logger, avocado_db]
   (number_of_vertices - inner_vertices).times do
     graph.add_vertex do |preferential_attachment|
       preferential_attachment > rand
@@ -29,10 +31,10 @@ else
   puts "#{graph.size} edges (saved to edges.csv)"
   
   if ellapsed_time < 2
-    puts "Generated in about one second"
+    puts "Generated and saved in about one second"
   elsif ellapsed_time < 60
-    puts "Generated in about #{ellapsed_time.round} seconds"
+    puts "Generated and saved in about #{ellapsed_time.round} seconds"
   else
-    puts "Generated in about #{ellapsed_time.round / 60} minutes and #{ellapsed_time.round % 60} seconds"
+    puts "Generated and saved in about #{ellapsed_time.round / 60} minutes and #{ellapsed_time.round % 60} seconds"
   end
 end

data/lib/graphshaper.rb

@@ -1,5 +1,6 @@
 require "graphshaper/version"
 require "graphshaper/undirected_graph"
+require "graphshaper/adapters/avocadodb_adapter"
 require "graphshaper/adapters/logging_adapter"
 
 module Graphshaper

data/lib/graphshaper/adapters/avocadodb_adapter.rb

@@ -0,0 +1,44 @@
+require "httparty"
+
+module Graphshaper
+  class AvocadoDbAdapter
+    include HTTParty
+    
+    base_uri 'localhost:8529'
+    format :json
+  
+    def initialize(vertex_collection_name, edge_collection_name)
+      @vertex_collection_name = vertex_collection_name
+      @edge_collection_name = edge_collection_name
+      @vertex_matching = []
+      @edge_matching = []
+      
+      [@vertex_collection_name, @edge_collection_name].each { |collection| drop_and_create_collection collection}
+    end
+    
+    # @return [String] id in the database
+    def add_vertex(id)
+      cmd = "/document?collection=#{@vertex_collection_name}"
+    	body = "{ \"id\" : \"#{id}\" }"
+    	response = self.class.post(cmd, :body => body)
+      @vertex_matching[id] = response.parsed_response["_id"]
+    end
+    
+    def add_edge(edge_id, from, to)
+      database_id_for_first_node = @vertex_matching[from]
+      database_id_for_second_node = @vertex_matching[to]
+      
+      cmd = "/edge?collection=#{@edge_collection_name}&from=#{database_id_for_first_node}&to=#{database_id_for_second_node}"
+    	body = "{ \"id\" : \"#{edge_id}\" }"
+    	response = self.class.post(cmd, :body => body)
+      @edge_matching[edge_id] = response.parsed_response["_id"]
+    end
+    
+    private
+    
+    def drop_and_create_collection(name)
+    self.class.delete "/_api/collection/#{name}"
+    self.class.post "/_api/collection", body: "{ \"name\" : \"#{name}\"}"
+  end  
+  end
+end

data/lib/graphshaper/version.rb

@@ -1,3 +1,3 @@
 module Graphshaper
-  VERSION = "0.2"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: graphshaper
 version: !ruby/object:Gem::Version
-  version: '0.2'
+  version: 0.2.1
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-04-04 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
-  requirement: &70304824409140 !ruby/object:Gem::Requirement
+  requirement: &70277557670180 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,10 +21,10 @@ dependencies:
         version: 0.8.1
   type: :runtime
   prerelease: false
-  version_requirements: *70304824409140
+  version_requirements: *70277557670180
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &70304824408640 !ruby/object:Gem::Requirement
+  requirement: &70277557669680 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -32,10 +32,10 @@ dependencies:
         version: 0.9.2.2
   type: :development
   prerelease: false
-  version_requirements: *70304824408640
+  version_requirements: *70277557669680
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70304824408180 !ruby/object:Gem::Requirement
+  requirement: &70277557669220 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -43,10 +43,10 @@ dependencies:
         version: 2.9.0
   type: :development
   prerelease: false
-  version_requirements: *70304824408180
+  version_requirements: *70277557669220
 - !ruby/object:Gem::Dependency
   name: yard
-  requirement: &70304824407720 !ruby/object:Gem::Requirement
+  requirement: &70277557668760 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -54,7 +54,7 @@ dependencies:
         version: 0.7.5
   type: :development
   prerelease: false
-  version_requirements: *70304824407720
+  version_requirements: *70277557668760
 description: Generate realistic graphs
 email:
 - me@moonglum.net
@@ -72,6 +72,7 @@ files:
 - bin/graphshaper
 - graphshaper.gemspec
 - lib/graphshaper.rb
+- lib/graphshaper/adapters/avocadodb_adapter.rb
 - lib/graphshaper/adapters/logging_adapter.rb
 - lib/graphshaper/undirected_graph.rb
 - lib/graphshaper/version.rb