mode 0.0.9 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e7d69ae05d944f46182074adaa084e4b22454862
-  data.tar.gz: 111e422dca6963652e892f4543f6b91d13d4bf1e
+  metadata.gz: bafc28cae256c4ce4821bb00753799dadb4c468c
+  data.tar.gz: 7360d725309e1253b0cff15f9974942405fbd370
 SHA512:
-  metadata.gz: f33ca37d31a1966328a8dadaef02eb0337aac0fcd28be3473b8d21ee50f777023132cfa886641a96b97804bc6d18e640c1bc3f6f101b9d7cc6344158f30a3acd
-  data.tar.gz: df765cb61dd992ec67ce3cd3fe11a04a65d8d45927ca0cb43a6ffa338a487211e5421e2fd24525e53d8c962671597bb34280b54c8ddf2eb594e6b39a5777502d
+  metadata.gz: 1e6ffd7b71345345d3e8399c74a7b6b358353ae632839ac60f46085cb0cbfaed4c09f4e2a748429ece0df50ed802095977adf605b92812e462544b2a864c4e56
+  data.tar.gz: 9bd2f1c0f0f3fe7f788a83785d7d1b591d6e355231ea3507dcf70db113e44d1ffc5c44b08cda7a0d65bd3a469ffd481ae8b9f8e781f1536d06f748752ca0a4bd

data/lib/mode/connector/dataset.rb

@@ -2,10 +2,13 @@ module Mode
   module Connector
     class Dataset
       attr_reader :path
+
+      attr_reader :count
       attr_reader :column_types
 
-      def initialize(path, column_types)
+      def initialize(path, count, column_types)
         @path = path
+        @count = count
         @column_types = column_types
       end
     end

data/lib/mode/connector/selector.rb

@@ -5,16 +5,18 @@ module Mode
     class Selector
       attr_reader :query
       attr_reader :data_source
+
+      attr_reader :count
       attr_reader :column_types
 
       def initialize(query, data_source)
         @query, @data_source = query, data_source
-        @column_types = Mode::Connector::TypeMap.new
+        @count, @column_types = 0, Mode::Connector::TypeMap.new
       end
 
       def perform!
         File.open(path, 'w+') { |file| write(file) }
-        Mode::Connector::Dataset.new(path, column_types)
+        Mode::Connector::Dataset.new(path, count, column_types)
       end
 
       def path
@@ -27,6 +29,7 @@ module Mode
         index = 0
         file.write '['
         data_source.select(query) do |row|
+          @count += 1
           write_row(file, row, index += 1)
         end
         file.write ']'

data/lib/mode/connector/uploader.rb

@@ -14,6 +14,7 @@ module Mode
           path,
           :execution => {
             :dataset => {
+              :count   => count,
               :content => content,
               :columns => JSON.generate(columns)
             }
@@ -36,6 +37,10 @@ module Mode
 
       private
 
+      def count
+        dataset.count
+      end
+
       def columns
         dataset.column_types.map do |name, type|
           {:name => name, :type => type}

data/lib/mode/version.rb

@@ -1,3 +1,3 @@
 module Mode
-  VERSION = "0.0.9"
+  VERSION = "0.0.10"
 end

data/spec/connector/uploader_spec.rb

@@ -8,6 +8,7 @@ describe Mode::Connector::Uploader do
   let(:dataset) {
     double(:dataset,
       :path => outpath,
+      :count => 0,
       :column_types => {
         'int' => :integer,
         'float' => :number
@@ -31,6 +32,7 @@ describe Mode::Connector::Uploader do
     Mode::API::Request.should_receive(:put).with(execution_path,
       :execution => {
         :dataset => {
+          :count => 0,
           :columns => an_instance_of(String), 
           :content => an_instance_of(Faraday::UploadIO)
         }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mode
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors:
 - Mode Analytics