lib/einhorn/command.rb in einhorn-0.4.9 vs lib/einhorn/command.rb in einhorn-0.5.0

@@ -299,14 +299,14 @@
       reload
     end
 
     def self.upgrade_workers
       if Einhorn::State.upgrading
-        Einhorn.log_info("Currently upgrading (#{Einhorn::WorkerPool.ack_count} / #{Einhorn::WorkerPool.ack_target} ACKs; bumping version and starting over)...")
+        Einhorn.log_info("Currently upgrading (#{Einhorn::WorkerPool.ack_count} / #{Einhorn::WorkerPool.ack_target} ACKs; bumping version and starting over)...", :upgrade)
       else
         Einhorn::State.upgrading = true
-        Einhorn.log_info("Starting upgrade to #{Einhorn::State.version}...")
+        Einhorn.log_info("Starting upgrade from version #{Einhorn::State.version}...", :upgrade)
       end
 
       # Reset this, since we've just upgraded to a new universe (I'm
       # not positive this is the right behavior, but it's not
       # obviously wrong.)
@@ -321,10 +321,11 @@
       acked = Einhorn::WorkerPool.ack_count
       target = Einhorn::WorkerPool.ack_target
 
       if Einhorn::State.upgrading && acked >= target
         Einhorn::State.upgrading = false
-        Einhorn.log_info("Upgrade to version #{Einhorn::State.version} complete.")
+        Einhorn.log_info("Upgraded successfully to version #{Einhorn::State.version} (Einhorn #{Einhorn::VERSION}).", :upgrade)
+        Einhorn.send_tagged_message(:upgrade, "Upgrade done", true)
       end
 
       old_workers = Einhorn::WorkerPool.old_workers
       if !Einhorn::State.upgrading && old_workers.length > 0
         Einhorn.log_info("Killing off #{old_workers.length} old workers.")