themigrator 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ef4bbd72db7e1b8cc560597c5791a29fa7ab4a60
-  data.tar.gz: c46be0bb730681e4da0bf54dfda0b873f4f10148
+  metadata.gz: d8dbb153310083d3935db37f8662511c357c24c4
+  data.tar.gz: 1f33da2750d76e79976c7d1c7e3c53e595dde1c3
 SHA512:
-  metadata.gz: 8a197db7f24a848c9728964e8826a06d319008c4139d5cf0fc442cf34c441681df7a1076d1849965e20a7a7b1a491263db1e47a338f34460402e574f3435b448
-  data.tar.gz: be2a459d55a032ff92813c0bd7a7a27736d2f59ebd85db705834ae8394247ce15f7a190e54de434c581b398a3aee1d7b1a8f2ef3618c11b36f00b01d0ac1325f
+  metadata.gz: 7ab7b20ff28c285ff0717580a94466dea113d3ed4522f019dbe08f0cb6b4d2b00265139f9eca4af870afca9a56330a796f571d1032f22cf9076fdf0b009a9500
+  data.tar.gz: 3ea69f65eaee4f40143d6ac4a4749a758d40d5e8af8296aea79d084ecc2453ef39ed91523e33e1a157b8bb25259305ac7be05f16ebe0122089d47a5722ce0b13

data/lib/themigrator/migration_progress.rb

@@ -1,29 +1,114 @@
-
+require 'time'
+require 'themigrator/migration'
 
 module Themigrator
+  # MigrationProgress takes the main log fails and parse it.
   class MigrationProgress
     include Logger
+
+    attr_reader :roles, :run_id, :start_time, :pid, :status, :current_action, :actions
     
     
-    def initialize(dir, run_id)
+    def initialize(dir, run_id, has_roles_cbk: nil, start_action_cbk: nil ) 
       @dir, @run_id = dir, run_id
       @log_file_path = main_log_path(dir, run_id)
+      @roles = []
+      @current_action = nil
+      @cbks = {
+	has_roles: has_roles_cbk,
+	start_action: start_action_cbk
+      }
+      @actions = Hash.new do |h,k| 
+	h[k] = {
+	  start_time: nil,
+	  roles: Hash.new {|rh,rk| rh[rk] = {} },
+	  end_time: nil}
+      end
+
       start_log_parser
     end
 
     private
 
     def start_log_parser
+      IO.foreach(@log_file_path) do |line|
+	process_line(line)
+      end
+    rescue EOFError
+    end
 
-      File.open(@log_file_path, "r") do |f|
-f.readline
-raise	File.read(@log_file_path)
 
+    def call_cbk(name, *args)
+      cbk = @cbks[name]
+      cbk.call(self) if cbk
+    end
 
+
+    def process_line(line)
+      line.strip!
+      case line
+      when /^# Logfile created on ([0-9-]+) ([0-9:]+) /
+	@start_time = Time.parse("#{$1} #{$2}")
+      else
+	process_log_entry(line)
       end
+      call_cbk(:new_line, line)
+    end
 
-    rescue EOFError
+    def process_log_entry(line)
+      time, line = extract_time(line)
+      scripts = Themigrator::Migration::SCRIPTS.join("|")
+      case line
+      when /start PID=(\d+)/
+	@pid = $1.to_i
+      when /roles: (.+)/
+	@roles = $1.strip.split.sort
+      when /(#{scripts}) => start/
+	action = $1.to_sym
+	@current_action = action
+	@actions[action][:start_time] = time
+      when /(#{scripts}) => (done|fail) \(([0-9\.]+) ms\)/
+	action = $1.to_sym
+	success = to_success($2)
+	run_time = $3.to_f
+
+	@actions[action][:end_time] = time
+	@actions[action][:run_time] = run_time
+	@actions[action][:success] = success
+	@current_action = nil
+
+      when /(#{scripts}) \| (.*) => start/
+	action = $1.to_sym
+	role = $2
+	@actions[action][:roles][role][:start_time] = time
+      when /(#{scripts}) \| (.*) => exited with (\d+) \(([0-9\.]+) ms\)/
+	action = $1.to_sym
+	role = $2
+	exitcode = $3.to_i
+	success = exitcode == 0
+	run_time = $4.to_f
+	@actions[action][:roles][role][:end_time] = time
+	@actions[action][:roles][role][:exitcode] = exitcode
+	@actions[action][:roles][role][:success] = success
+	@actions[action][:roles][role][:run_time] = run_time
+      else 
+	raise line
+      end
+    end
+
+    def extract_time(line)
+      line =~ /^([0-9-]+) ([0-9:]+) (.*)/
+      time = Time.parse("#{$1} #{$2}")
+      line = $3
+      return time, line.strip
+    end
+
+    def rollback?
+      @actions.keys.include?("rollback")
+    end
 
+    def to_success(val)
+      val == "done" ? true : false
     end
 
   end

data/lib/themigrator/migrator.rb

@@ -12,7 +12,8 @@ module Themigrator
     def initialize(dir)
       @dir = dir
       @runned_roles = []
-      @run_id = Time.now.strftime("%Y-%m-%d-%H:%M:%S")
+      @start_time = Time.now
+      @run_id = @start_time.strftime("%Y-%m-%d-%H:%M:%S")
     end
 
     def migrate!
@@ -27,12 +28,16 @@ module Themigrator
       each_action do |action|
 	if !run_action_and_wait(action) 
 	  did_it_work = false
-#	  run_action_and_wait("rollback")
+	  run_action_and_wait("rollback")
 	  return -1
 	end
       end
       close_logger
       did_it_work
+    ensure
+      @end_time = Time.now
+      msg = did_it_work ? "done" : "fail"
+      log "end => #{msg} (#{run_time * 1000} ms)"
     end
 
     private
@@ -79,6 +84,10 @@ module Themigrator
     def find_script(role, action)
       File.join(@dir, role, action)
     end
+
+    def run_time
+      @end_time - @start_time if @end_time
+    end
   end
 
 end

data/lib/themigrator/version.rb

@@ -1,3 +1,3 @@
 module Themigrator
-  VERSION = "0.1.7"
+  VERSION = "0.1.9"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: themigrator
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.9
 platform: ruby
 authors:
 - Guillermo Álvarez
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-08-26 00:00:00.000000000 Z
+date: 2016-08-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler