samidare 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e82d6b46459095203aa62a7333fa25ab91230431
-  data.tar.gz: 12e5f597f31938450a506bcc887dba4e3ef1842f
+  metadata.gz: e667f045b627932973bb7818c2c3fc6a3f7450a1
+  data.tar.gz: 9bc577d40f8abe15dc27136b7add5b3813b93e8c
 SHA512:
-  metadata.gz: 04ec157dbc55d7329a11b3c45e0906cce182f58b7cecd4225f94cb0e345a00aeadd0e315f9ce95df18c7c27ddcd51f1e7dc59484c6d10c728c7c96da228bdfb4
-  data.tar.gz: 3216bf771d840636770338661da646139b81d624a729ee09c463a353095f33739930884039b646183e2e580183700460d9944602ac3def5869aaf12a313eff6d
+  metadata.gz: 0e52080a119b57b13a4badf68056ffe6e92c31b7e67e9d240b8891cad7f3583d0463e790ef1d9e5dbef26f700247b3468c9ed160d86bb9b854ac8ae0738e43b1
+  data.tar.gz: b22a32d057449f8688228c33148475049d2219e2a02d7ce9399ba682fd6481ee7c93a8e8444b4471fa4aa988f1b5a3e4b7038bfce40718f1f9d22e7a4b462db1

data/lib/samidare.rb

@@ -7,19 +7,21 @@ module Samidare
       Samidare::EmbulkUtility::ConfigGenerator.new(config).generate_config
     end
 
-    def run(config)
+    def run(config, target_tables = [])
       db_infos = Samidare::EmbulkUtility::DBInfo.generate_db_infos
       table_infos = Samidare::EmbulkUtility::TableInfo.generate_table_infos
       db_infos.keys.each do |db_name|
-        embulk_run(db_name, db_infos[db_name]['bq_dataset'], table_infos[db_name], config)
+        embulk_run(db_name, db_infos[db_name]['bq_dataset'], table_infos[db_name], config, target_tables)
       end
     end
 
     private
-    def embulk_run(db_name, bq_dataset, tables, config)
+    def embulk_run(db_name, bq_dataset, tables, config, target_tables)
       process_times = []
       big_query = Samidare::BigQueryUtility.new(config)
       tables.each do |table|
+        next unless target_table?(table.name, target_tables)
+
         start_time = Time.now
         log "table: #{table.name} - start"
 
@@ -44,6 +46,11 @@ module Samidare
       process_times.each { |process_time| log process_time }
     end
 
+    def target_table?(table, target_tables)
+      return true if target_tables.empty?
+      target_tables.include?(table)
+    end
+
     def log(message)
       puts "[#{Time.now.strftime("%Y-%m-%d %H:%M:%S")}] #{message}"
     end

data/lib/samidare/version.rb

@@ -1,3 +1,3 @@
 module Samidare
-  VERSION = "0.0.6"
+  VERSION = "0.0.7"
 end

data/spec/samidare_spec.rb

@@ -4,4 +4,27 @@ describe Samidare do
   it 'has a version number' do
     expect(Samidare::VERSION).not_to be nil
   end
+
+  describe Samidare::EmbulkClient do
+    describe '#target_table?' do
+      subject { Samidare::EmbulkClient.new.send(:target_table?, table, target_tables) }
+      context 'target_tables is empty' do
+        let(:table) { 'hoge' }
+        let(:target_tables) { [] }
+        it { expect(subject).to be_truthy }
+      end
+
+      context 'is included' do
+        let(:table) { 'hoge' }
+        let(:target_tables) { ['hoge'] }
+        it { expect(subject).to be_truthy }
+      end
+
+      context 'is not included' do
+        let(:table) { 'hoge' }
+        let(:target_tables) { ['fuga'] }
+        it { expect(subject).to be_falsy }
+      end
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: samidare
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Ryoji Kobori