data_sampler 0.0.4 → 0.0.5

data/lib/data_sampler/dependency.rb

@@ -3,10 +3,12 @@ module DataSampler
 
     attr_reader :table_name
     attr_reader :keys
+    attr_reader :referring_table_name
 
-    def initialize(table_name, keys)
+    def initialize(table_name, keys, referring_table_name)
       @table_name = table_name
       @keys = keys
+      @referring_table_name = referring_table_name
     end
 
     def eql? other
@@ -14,7 +16,7 @@ module DataSampler
     end
 
     def to_s
-      "#{keys} in table `#{table_name}`"
+      "#{keys} in table `#{table_name}` (referred from `#{referring_table_name}`)"
     end
 
   end

data/lib/data_sampler/table_sample.rb

@@ -27,7 +27,7 @@ module DataSampler
       where = dependency.keys.collect { |col, val| "#{@connection.quote_column_name col} = #{@connection.quote val}" } * ' AND '
       sql = "SELECT * FROM #{@connection.quote_table_name @table_name} WHERE " + where
       row = @connection.select_one(sql)
-      raise "Dependent row not found for #{dependency} (using SQL: #{sql})" if row.nil?
+      raise "Could not find dependent row: #{dependency} (using SQL: #{sql})" if row.nil?
       add row
     end
 
@@ -110,7 +110,7 @@ module DataSampler
         col = cols.shift
         ref[ref_col] = row[col] unless row[col].nil?
       end
-      Dependency.new(fk.references_table_name, ref) unless ref.empty?
+      Dependency.new(fk.references_table_name, ref, table_name) unless ref.empty?
     end
 
     def dependencies_for(row)

data/lib/data_sampler/version.rb

@@ -1,3 +1,3 @@
 module DataSampler
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: data_sampler
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-21 00:00:00.000000000 Z
+date: 2013-03-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: schema_plus