mysqlknife 1.4.0 → 2.0.0

data/lib/mysqlknife/configs.rb

@@ -0,0 +1,123 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Configs
+    include Singleton
+
+    attr_writer :mysql_host,
+                :mysql_databases
+    attr_reader :path,
+                :ssh_host,
+                :ssh_port,
+                :ssh_user,
+                :ssh_password,
+                :ssh_key,
+                :ssh_color,
+                :ssh_use,
+                :connections,
+                :configs,
+                :name,
+                :mysql_host,
+                :mysql_username,
+                :mysql_password,
+                :mysql_port,
+                :mysql_databases
+
+    def initialize
+      # Define path for config file:
+      if ENV['ENV'] == 'test'
+        path = File.expand_path(File.join(Dir.pwd, '.db.yml'))
+      else
+        path = File.expand_path("#{Dir.home}/.db.yml")
+      end
+
+      # Load config file:
+      @configs = YAML.load_file(path) if File.exist?(path)
+
+      if @configs.nil?
+        puts "Not exist config file in #{path}."
+        exit 1
+      end
+
+      # Define generic variables:
+      @path         = path
+      @ssh_host     = @configs['ssh']['host']
+      @ssh_port     = @configs['ssh']['port']
+      @ssh_user     = @configs['ssh']['user']
+      @ssh_password = @configs['ssh']['password']
+      @ssh_key      = @configs['ssh']['keys']
+      @connections  = @configs['databases'].keys.sort
+    end
+
+    def method_missing(name, *args)
+      method = name.to_s.split(/_/)
+      if method.first == 'db'
+        param = method[1]
+        conn  = args[0]
+
+        if @configs['databases'].include?(conn)
+          @configs['databases'][conn][param]
+        else
+          puts "Not exist connecion name: #{conn}"
+          exit 1
+        end
+      end
+    end
+
+    def settings(name)
+      @name           = name
+      @ssh_color      = db_color(name)
+      @ssh_use        = db_ssh(name)
+      @mysql_port     = db_port(name)
+      @mysql_username = db_username(name)
+      @mysql_password = db_password(name)
+    end
+
+    def slaves(name)
+      slaves = db_slaves(name)
+
+      if slaves.nil?
+        []
+      else
+        slaves
+      end
+    end
+
+    def hosts(name)
+      hosts = []
+      hosts << db_host(name)
+      hosts.concat(slaves(name))
+      hosts.sort
+    end
+
+    def show(name)
+      host     = db_host(name)
+      port     = db_port(name)
+      username = db_username(name)
+      password = db_password(name)
+      slaves   = db_slaves(name)
+      slaves   = slaves.join(', ') unless slaves.nil?
+
+      %W[Master:\ #{host}
+         Slaves:\ #{slaves}
+         Port:\ #{port}
+         Username:\ #{username}
+         Password:\ #{password}].join("\n")
+    end
+
+    def tools(name)
+      unless db_tools(name).nil?
+        db_tools(name).map {|tool| tool.split.first(1).first }
+      else
+        puts "Please, add custom tools in config file."
+        exit 1
+      end
+    end
+
+    def tool(name, tool)
+      unless db_tools(name).nil?
+        db_tools(name).select { |command| /#{tool}/.match(command.to_s) }.first
+      end
+    end
+  end
+end

data/lib/mysqlknife/iterm.rb

@@ -0,0 +1,29 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Iterm
+    @@colors = { orange: [255,128,000],
+                 green:  [000,255,000],
+                 red:    [255,000,000] }
+
+    def name(name = nil)
+      print "\033]0;#{name}\007"
+    end
+
+    def color(name)
+      name  = name.to_sym
+
+      red   = @@colors[name][0]
+      green = @@colors[name][1]
+      blue  = @@colors[name][2]
+
+      print "\033]6;1;bg;red;brightness;#{red}\a"
+      print "\033]6;1;bg;green;brightness;#{green}\a"
+      print "\033]6;1;bg;blue;brightness;#{blue}\a"
+    end
+
+    def reset
+      print "\033]6;1;bg;*;default\a"
+    end
+  end
+end

data/lib/mysqlknife/log.rb

@@ -0,0 +1,34 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Log
+    attr_writer :verbose
+
+    def initialize
+      @log = Logger.new(STDOUT)
+    end
+
+    def kill(process)
+      process = process.map { |k, v| "#{k}: #{v}" }.join(', ')
+      @log.info("Kill MySQL Process: #{process}")
+    end
+
+    def sql(statement)
+      @log.debug("SQL Statement: #{statement}") if $DEBUG
+    end
+
+    def sql_error(error, sql)
+      @log.debug("SQL Error: #{error}")
+      @log.debug("SQL Statement: #{sql}")
+    end
+
+    def command(command)
+      @log.debug("Bash Command: #{command}") if $DEBUG
+    end
+
+    def skip_repl_error(status)
+      status = status.map { |k, v| "#{k}: #{v}" }.join(', ')
+      @log.info("MySQL Skip error replica: #{status}")
+    end
+  end
+end

data/lib/mysqlknife/mysql.rb

@@ -0,0 +1,24 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class MySQL
+    def initialize
+      @cnf = Configs.instance
+    end
+
+    def execute(sql)
+      conn = Mysql2::Client.new(host: @cnf.mysql_host,
+                                port: @cnf.mysql_port,
+                                username: @cnf.mysql_username,
+                                password: @cnf.mysql_password,
+                                database: @cnf.mysql_database)
+
+      Mysqlknife::Log.new.sql(sql)
+
+      return conn.query(sql)
+    rescue Mysql2::Error => error
+      Mysqlknife::Log.new.sql_error(error, sql)
+      exit(1)
+    end
+  end
+end

data/lib/mysqlknife/mysql/command.rb

@@ -0,0 +1,63 @@
+# encoding: utf-8
+
+module Mysqlknife
+  module Mysql
+    class Command
+      def initialize
+        @cnf = Configs.instance
+      end
+
+      def check
+        find_executable('mysql')
+        find_executable('mysqlshow')
+      end
+
+      def parse(command)
+        host     = @cnf.mysql_host
+        port     = @cnf.mysql_port
+        username = @cnf.mysql_username
+        password = password(@cnf.mysql_password)
+
+        eval("\"#{command}\"")
+      end
+
+      def prompt
+        if !!(@cnf.mysql_host =~ Resolv::IPv4::Regex)
+          @cnf.name
+        else
+          @cnf.mysql_host.partition('.').first
+        end
+      end
+
+      def execute(sentence)
+        "--execute='#{sentence}'" unless sentence.nil?
+      end
+
+      def password(password)
+        "-p#{password}" unless password.nil?
+      end
+
+      def console(sentence = nil)
+        %W[mysql
+           -h #{@cnf.mysql_host}
+           -P #{@cnf.mysql_port}
+           -u #{@cnf.mysql_username}
+           #{password(@cnf.mysql_password)}
+           --prompt='#{prompt}[\\d]> '
+           #{@cnf.mysql_database}
+           #{execute(sentence)}
+           ].join(' ')
+      end
+
+      def describe(*args)
+        %W[mysqlshow
+           -h #{@cnf.mysql_host}
+           -P #{@cnf.mysql_port}
+           -u #{@cnf.mysql_username}
+           #{password(@cnf.mysql_password)}
+           --keys
+           #{args.join(' ')}].join(' ')
+      end
+    end
+  end
+end

data/lib/mysqlknife/mysql/kill.rb

@@ -0,0 +1,73 @@
+# encoding: utf-8
+
+module Mysqlknife
+  module Mysql
+    class Kill
+      attr_writer :where
+
+      def initialize
+        @mysql     = MySQL.new
+        @mysql_sql = Mysql::SQL.new
+      end
+
+      def clear
+        list.each do |process|
+          kill(process['id'])
+
+          Mysqlknife::Log.new.kill(process)
+        end
+      end
+
+      def check_user
+        sql    = @mysql_sql.current_user
+        result = @mysql.execute(sql).first['CURRENT_USER'] =~ /^root\@/
+
+        return true if result
+      end
+
+      def check_grants
+        sql    = @mysql_sql.current_user_grants
+        result = @mysql.execute(sql).first.values.to_s =~ /PROCESS/
+
+        return true if result
+      end
+
+      def check_privileges
+        check_user || check_grants
+      end
+
+      def list
+        @mysql.execute(@mysql_sql.show_processlist(@where))
+      end
+
+      def show
+        process = list
+
+        unless process.first.nil?
+          table = Terminal::Table.new do |t|
+            t.add_row(process.first.keys)
+            t.add_separator
+            process.each do |row|
+              t.add_row(row.values)
+            end
+          end
+          puts table
+        end
+      end
+
+      def rds?
+        ! @mysql.execute(@mysql_sql.show_procedure('rds_kill')).first
+      end
+
+      def kill(id)
+        if check_privileges
+          if rds?
+            @mysql.execute(@mysql_sql.mysql_kill(id))
+          else
+            @mysql.execute(@mysql_sql.rds_kill(id))
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mysqlknife/mysql/replica.rb

@@ -0,0 +1,41 @@
+# encoding: utf-8
+
+module Mysqlknife
+  module Mysql
+    class Replica
+      attr_writer :behind
+
+      def initialize
+        @mysql     = MySQL.new
+        @mysql_cmd = Mysql::Command.new
+        @mysql_sql = Mysql::SQL.new
+      end
+
+      def check_status_negative(behind)
+        @behind < 0 && behind < 0 && behind <= @behind && behind != 0
+      end
+
+      def rds?
+        ! @mysql.execute(@mysql_sql.show_procedure('rds_skip_repl_error')).first
+      end
+
+      def slave_status
+        @mysql.execute(@mysql_sql.slave_status).first
+      end
+
+      def skip
+        status = slave_status
+
+        if check_status_negative(status['Seconds_Behind_Master'])
+          Mysqlknife::Log.new.skip_repl_error(status)
+
+          if rds?
+            @mysql.execute(@mysql_sql.rds_skip_repl_error)
+          else
+            @mysql.execute(@mysql_sql.mysql_skip_repl_error)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mysqlknife/mysql/sql.rb

@@ -0,0 +1,58 @@
+# encoding: utf-8
+
+module Mysqlknife
+  module Mysql
+    class SQL
+      def current_user
+        'SELECT CURRENT_USER;'
+      end
+
+      def current_user_grants
+        'SHOW GRANTS FOR CURRENT_USER;'
+      end
+
+      def replica_lag
+        'SHOW SLAVE STATUS\G'
+      end
+
+      def show_processlist(where = nil)
+        sql = %w(SELECT id, user, host, db, command, time, state, info
+                 FROM INFORMATION_SCHEMA.PROCESSLIST
+                 WHERE state NOT REGEXP '(slave|relay|event)'
+                 AND user NOT IN ('rdsadmin',
+                 'rdsrepladmin',
+                 'system user',
+                 'event_scheduler')
+                 AND id != CONNECTION_ID()
+                 AND command != 'Binlog Dump').join(' ')
+        sql << " AND #{where};" unless where.nil?
+      end
+
+      def show_procedure(name)
+        "SHOW PROCEDURE STATUS LIKE '#{name}';"
+      end
+
+      def mysql_skip_repl_error
+        %w(STOP SLAVE;
+           SET GLOBAL SQL_SLAVE_SKIP_COUNTER = 1;
+           START SLAVE;).join(' ')
+      end
+
+      def mysql_kill(pid)
+        "kill #{pid};"
+      end
+
+      def slave_status
+        'SHOW SLAVE STATUS;'
+      end
+
+      def rds_kill(pid)
+        "CALL mysql.rds_kill(#{pid});"
+      end
+
+      def rds_skip_repl_error
+        'CALL mysql.rds_skip_repl_error;'
+      end
+    end
+  end
+end