lhm 1.0.0.rc5 → 1.0.0.rc6

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+# 1.0.0.rc6 (January 30, 2012)
+
+* added --confirm to kill script; fixes to kill script
+
 # 1.0.0.rc5 (January 30, 2012)
 
 * moved scripts into bin, renamed, added to gem binaries

data/bin/lhm-kill-queue

@@ -15,6 +15,7 @@ module Lhm
         opts.on("-p", "--password PASSWORD") { |v| @password = v }
         opts.on("-d", "--database DATABASE") { |v| @database = v }
         opts.on("-m", "--mode MODE") { |v| @mode = v.to_sym }
+        opts.on("-y", "--confirm") { |v| @confirm = true }
       end.parse!
 
       unless(@hostname && @username && @password && @database)
@@ -33,7 +34,7 @@ module Lhm
         kills queries on the given server after detecting 'lock table% -- lhm'.
         usage:
           lhm-kill-queue -h hostname -u username -p password -d database \\
-          (--kill | --master | --slave)
+          (-m kill | -m master | -m slave) [--confirm]
 
       desc
     end
@@ -48,17 +49,15 @@ module Lhm
 
     def kill
       lock = trip
-      puts "killing lock process #{ lock }."
       kill_process(lock)
     end
 
     def master
       lock = trip
       puts "starting to kill non lhm processes in 1 second"
-      sleep(1)
+      sleep(1.05)
 
       [list_non_lhm].flatten.each do |process|
-        puts "killing #{ select_statement(process) }"
         kill_process(process)
         sleep(0.05)
       end
@@ -71,7 +70,6 @@ module Lhm
 
       [list_non_lhm].flatten.each do |process|
         if(select?(process))
-          puts "killing #{ select_statement(process) }"
           kill_process(process)
           sleep(0.05)
         end
@@ -96,11 +94,11 @@ module Lhm
     end
 
     def list_non_lhm
-      select_process("info not like '% -- lhm' and time > 0 and command = 'Query'")
+      select_processes("info not like '% -- lhm' and time > 0 and command = 'Query'")
     end
 
     def trip
-      until res = select_process("info like 'lock table% -- lhm'")
+      until res = select_processes("info like 'lock table% -- lhm'")
         sleep 0.2
         print '.'
       end
@@ -109,12 +107,23 @@ module Lhm
     end
 
     def kill_process(process_id)
+      puts "killing #{ select_statement(process_id) }"
+
+      if(@confirm)
+        print "confirm ('y' to confirm): "
+
+        if(gets.strip != 'y')
+          puts "skipped."
+          return
+        end
+      end
+
       connection.execute("kill #{ process_id }")
+      puts "killed #{ process_id }"
     end
 
     def select?(process)
       if statement = select_statement(process)
-
         case statement
         when /delete/i then false
         when /update/i then false
@@ -133,8 +142,8 @@ module Lhm
       end
     end
 
-    def select_process(predicate)
-      value %Q(
+    def select_processes(predicate)
+      values %Q(
         select id from information_schema.processlist
          where db = '#{ @database }'
            and user = '#{ @username }'
@@ -145,9 +154,13 @@ module Lhm
     def value(statement)
       connection.select_value(statement)
     end
+
+    def values(statement)
+      connection.select_values(statement)
+    end
   end
 end
 
 killer = Lhm::KillQueue.new
 killer.run
-.
+

data/lib/lhm/version.rb

@@ -2,5 +2,5 @@
 # Schmidt
 
 module Lhm
-  VERSION = "1.0.0.rc5"
+  VERSION = "1.0.0.rc6"
 end

metadata

@@ -6,8 +6,8 @@ version: !ruby/object:Gem::Version
   - 1
   - 0
   - 0
-  - rc5
-  version: 1.0.0.rc5
+  - rc6
+  version: 1.0.0.rc6
 platform: ruby
 authors: 
 - SoundCloud