data_sampler 0.0.6 → 1.0.0

data/lib/data_sampler/sample.rb

@@ -32,6 +32,11 @@ module DataSampler
         warn " Discovered #{new_dependencies} new dependencies" if new_dependencies > 0
       end while new_dependencies > 0
       warn "Referential integrity obtained"
+
+      warn "Final sample contains:"
+      @table_samples.values.each do |table_sample|
+        warn "  #{table_sample.size} row(s) from `#{table_sample.table_name}`"
+      end
       @computed = true
     end
 

data/lib/data_sampler/table_sample.rb

@@ -22,6 +22,10 @@ module DataSampler
       @sample
     end
 
+    def size
+      @sampled ? @sample.size : @size
+    end
+
     def fulfil(dependency)
       return 0 if fulfilled?(dependency)
       where = dependency.keys.collect { |col, val| "#{@connection.quote_column_name col} = #{@connection.quote val}" } * ' AND '
@@ -60,7 +64,7 @@ module DataSampler
       deps_in_progress = @pending_dependencies
       @pending_dependencies = Set.new
       deps_in_progress.each do |dependency|
-        raise "Table sample for #{dependency.table_name} not found" unless table_samples[dependency.table_name]
+        raise "Table sample for `#{dependency.table_name}` not found" unless table_samples[dependency.table_name]
         newly_added += table_samples[dependency.table_name].fulfil(dependency)
       end
       newly_added
@@ -96,6 +100,7 @@ module DataSampler
       sql += " ORDER BY #{@connection.quote_column_name pk} DESC" unless pk.nil?
       sql += " LIMIT #{count}"
       @connection.select_all(sql).each { |row| add(row) }
+      @sampled = true
     rescue ActiveRecord::StatementInvalid => e
       # Don't choke on unknown table engines, such as Sphinx
       []

data/lib/data_sampler/version.rb

@@ -1,3 +1,3 @@
 module DataSampler
-  VERSION = "0.0.6"
+  VERSION = "1.0.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: data_sampler
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 1.0.0
   prerelease: 
 platform: ruby
 authors: