database_stalker 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 17057c858434c678b5bce9f6a0645f0ecc2ac02e
-  data.tar.gz: 1fbd37277e38e96aba343154281f48c96cfb1706
+  metadata.gz: 7971413c47361657c7bf2e049c210f4308af2e0c
+  data.tar.gz: 346327937f09e485bcb5ce0610a9e67fab735cb3
 SHA512:
-  metadata.gz: 6fd2553c35173b03ff83aa35ec5247b069d801527a2771720dba769cca7ffa45641dc2db7c353cc4a1b1757f237db215fc9687b42e39e8f3ca19a068ec5bac91
-  data.tar.gz: fa7bae7622552804509119c7a1781d4b8ce08faba22e1b14e05725f6e9f9398e7ac5ab57b3084fd19566a19e695b00f5cb956e3c6a533bb6dba09c1a9112a2e0
+  metadata.gz: b54cde30f9f7842f90a5134efb81a6719cc1df5a8d9807bb6298833bcfd63aeddc01f52f4b0dff311ee59a589b369a916bc80de295429e1d548398f01e25df06
+  data.tar.gz: 9c506778ac4f519d1519512aa2fb3480ae79b248661e7a4a2c845fc8d7cab0c065106c609ed391783c5b0c65612229520272334f50748ee495521591c697f1d9

data/lib/database_stalker/log_stalker.rb

@@ -0,0 +1,46 @@
+require 'database_stalker/util'
+
+module DatabaseStalker
+  class LogStalker
+
+    def initialize(stalked_file_path, stalking_result_path)
+      @stalked_file_path = stalked_file_path
+      @stalking_result_path = stalking_result_path
+    end
+
+    def run
+      @runned_tails = Util.runned_tail_pids
+      spawn("tail -f -n 0 #{@stalked_file_path} > #{@stalking_result_path}")
+      wait_for_tail_process_runninng
+    end
+
+    def stop
+      wait_for_tail_process_output
+      current_runned_tails = Util.runned_tail_pids
+      (current_runned_tails - @runned_tails).each do |pid|
+        Process.kill('KILL', pid)
+      end
+    end
+
+    def result
+      result = []
+      File.open(@stalking_result_path) do |file|
+        file.each_line do |line|
+          result << line
+        end
+      end
+      result
+    end
+
+
+    private
+
+      def wait_for_tail_process_runninng
+        sleep(0.01)
+      end
+
+      def wait_for_tail_process_output
+        sleep(0.01)
+      end
+  end
+end

data/lib/database_stalker/parser.rb

@@ -1,15 +1,13 @@
 class DatabaseStalker::Parser
-  def initialize(log_file)
-    @log_file = log_file
+  def initialize(log)
+    @log = log
   end
 
   def table_names
     tables = []
-    File.open(@log_file, 'r') do |f|
-      f.each_line do |line|
-        matched = line.match(/INSERT\ INTO\ `(.+)` \(/)
-        tables << matched[1] unless matched.nil?
-      end
+    @log.each do |line|
+      matched = line.match(/INSERT\ INTO\ `(.+)` \(/)
+      tables << matched[1] unless matched.nil?
     end
     tables.uniq
   end

data/lib/database_stalker/util.rb

@@ -0,0 +1,18 @@
+module DatabaseStalker::Util
+
+  module_function
+
+  def runned_tail_pids
+    pids = ''
+    IO.popen("ps x | grep 'tail -f -n 0' | grep -v grep | awk '{print $1}'") do |io|
+      while true
+        buffer = io.gets
+        break if buffer.nil?
+        pids += buffer
+      end
+    end
+    pids.split("\n").map do |pid|
+      pid.to_i
+    end
+  end
+end

data/lib/database_stalker/version.rb

@@ -1,3 +1,3 @@
 module DatabaseStalker
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/lib/database_stalker.rb

@@ -1,25 +1,62 @@
-require "database_stalker/version"
-require "database_stalker/parser"
+require 'database_stalker/version'
+require 'database_stalker/parser'
+require 'database_stalker/log_stalker'
 
 module DatabaseStalker
 
   DEFAULT_LOG_FILE = 'log/test.log'
   DEFAULT_TABLE_LOG_FILE = 'log/table_names.log'
+  DEFAULT_STALKING_LOG_FILE = 'log/stalking_log.log'
+  DEFAULT_STALKING_LOG_PER_TEST_FILE = 'log/stalking_log_per_test.log'
+  DEFAULT_STALKING_LOG_PER_TEST_TEMPORARY_FILE = 'log/stalking_log_per_test_temporary.log'
 
   class  << self
 
-    def start(log_file: DEFAULT_LOG_FILE, table_log_file: DEFAULT_TABLE_LOG_FILE)
-      clean_up_file(log_file) if File.exist?(log_file)
-      Process.fork do
+    def set_up(
+      test_log: DEFAULT_LOG_FILE,
+      table_log: DEFAULT_TABLE_LOG_FILE,
+      stalking_log: DEFAULT_STALKING_LOG_FILE,
+      stalking_log_per_test: DEFAULT_STALKING_LOG_PER_TEST_FILE,
+      stalking_log_per_test_temporary: DEFAULT_STALKING_LOG_PER_TEST_TEMPORARY_FILE)
+      @test_log = test_log
+      @table_log = table_log
+      @stalking_log = stalking_log
+      @stalking_log_per_test = stalking_log_per_test
+      @stalking_log_per_test_temporary = stalking_log_per_test_temporary
+      File.delete(@stalking_log_per_test_temporary) if File.exist?(@stalking_log_per_test_temporary)
+      FileUtils.touch(@stalking_log_per_test_temporary)
+    end
+
+    def stalk
+      fork do
+        log_stalker = LogStalker.new(@test_log, @stalking_log)
+        log_stalker.run
         watch_test_process
-        save_stalked_tables(log_file, table_log_file)
+        log_stalker.stop
+        used_log = []
+        File.open(@stalking_log_per_test_temporary, 'r') do |f|
+          f.each_line do |line|
+            used_log << line
+          end
+        end
+        all_log = log_stalker.result
+        parser = Parser.new(all_log.slice(used_log.size .. all_log.size - 1))
+        File.open(@table_log, 'w') do |file|
+          parser.table_names.each do |table_name|
+            file.write("#{table_name}\n")
+          end
+        end
       end
+      wait_for_log_stalker
     end
 
-    def read_table_names(table_log_file: DEFAULT_TABLE_LOG_FILE)
-      return [] if not File.exist?(table_log_file)
+    def wait_for_log_stalker
+      sleep(0.1)
+    end
+
+    def table_names
       result = []
-      File.open(table_log_file, 'r') do |f|
+      File.open(@table_log, 'r') do |f|
         f.each_line do |line|
           result << line.strip
         end
@@ -27,17 +64,23 @@ module DatabaseStalker
       result
     end
 
-    def clean_up_file(file)
-      File.open(file,'w'){ |f| f = nil }
+    def stalk_per_test
+      @log_stalker = LogStalker.new(@test_log, @stalking_log_per_test)
+      @log_stalker.run
+    end
+
+    def table_names_per_test
+      @log_stalker.stop
+      appended_log = @log_stalker.result
+      parser = Parser.new(appended_log)
+      parser.table_names
     end
 
-    def save_stalked_tables(log_file, table_log_file)
-      File.open(table_log_file, 'w') do |f|
-        if File.exist?(log_file)
-          parser = Parser.new(log_file)
-          parser.table_names.each { |table| f.puts table }
-        else
-          f = nil
+    def notify_table_deletion
+      File.open(@stalking_log_per_test_temporary, 'a') do |file|
+        appended_log = @log_stalker.result
+        appended_log.each do |line|
+          file.puts("#{line}")
         end
       end
     end
@@ -49,5 +92,5 @@ module DatabaseStalker
     end
   end
 
-  private_class_method :save_stalked_tables, :watch_test_process, :clean_up_file
+  private_class_method :watch_test_process, :wait_for_log_stalker
 end

data/rails_test/spec/database_stalker_usage_spec.rb

@@ -2,10 +2,15 @@ require 'spec_helper'
 require 'database_stalker'
 
 describe 'use database_stakler' do
+
   it do
-    DatabaseStalker.start('./log/test.log', './spec/tables.log')
+    DatabaseStalker.set_up
+    DatabaseStalker.stalk
+    DatabaseStalker.stalk_per_test
     Sample1.create!
     Sample2.create!
+    DatabaseStalker.table_names_per_test
+    DatabaseStalker.notify_table_deletion
     Sample3.create!
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: database_stalker
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - tetsutaroendo
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-11-13 00:00:00.000000000 Z
+date: 2017-11-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -73,7 +73,9 @@ files:
 - bin/setup
 - database_stalker.gemspec
 - lib/database_stalker.rb
+- lib/database_stalker/log_stalker.rb
 - lib/database_stalker/parser.rb
+- lib/database_stalker/util.rb
 - lib/database_stalker/version.rb
 - rails_test/.gitignore
 - rails_test/.rspec