mysqlknife 1.4.0 → 2.0.0

data/lib/mysqlknife/parameters.rb

@@ -0,0 +1,84 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Parameters
+    attr_writer :connection, :host, :tool
+    attr_reader :host, :tool
+
+    def initialize
+      @cnf = Configs.instance
+    end
+
+    def connections
+      if @connection.nil?
+        puts @cnf.connections
+      end
+    end
+
+    def connection_details
+      unless @connection.nil?
+        puts @cnf.show(@connection)
+      end
+    end
+
+    def hosts
+      if @connection.nil? == false &&
+         @host.nil?       == true
+        puts @cnf.hosts(@connection)
+      end
+    end
+
+    def slaves
+      if @connection.nil? == false &&
+         @host.nil?       == true
+        puts @cnf.slaves(@connection)
+      end
+    end
+
+    def tools
+      if @connection.nil? == false &&
+         @host.nil?       == false &&
+         @tool.nil?       == true
+        puts @cnf.tools(@connection)
+      end
+    end
+
+    def host_selected
+      if @connection.nil? == false &&
+         @host.nil?       == false
+        @cnf.hosts(@connection).each do |host|
+          return @host = host if host.include?(@host)
+        end
+
+        false
+      end
+    end
+
+    def slave_selected
+      if @connection.nil? == false &&
+         @host.nil?       == false
+        @cnf.slaves(@connection).each do |host|
+          return @host = host if host.include?(@host)
+        end
+
+        false
+      end
+    end
+
+    def tool_selected
+      if @connection.nil? == false &&
+         @host.nil?       == false &&
+         @tool.nil?       == false &&
+         @cnf.tools(@connection).include?(@tool)
+        @cnf.hosts(@connection).each do |host|
+          if host.include?(@host)
+            @host = host
+            @tool = @cnf.tool(@connection, @tool)
+
+            return @tool
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mysqlknife/ssh.rb

@@ -0,0 +1,18 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class SSH
+    def initialize
+      @cnf = Configs.instance
+    end
+
+    def execute(command)
+      %W[ssh
+         -t
+         -l #{@cnf.ssh_user}
+         -i #{@cnf.ssh_key}
+         #{@cnf.ssh_host}
+         "#{command}"].join(' ')
+    end
+  end
+end

data/lib/mysqlknife/version.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
 
 module Mysqlknife
-  VERSION = '1.4.0'
+  VERSION = '2.0.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mysqlknife
 version: !ruby/object:Gem::Version
-  version: 1.4.0
+  version: 2.0.0
 platform: ruby
 authors:
 - Nicola Strappazzon C.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-14 00:00:00.000000000 Z
+date: 2015-03-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -81,14 +81,18 @@ files:
 - README.md
 - bin/mysqlknife
 - lib/mysqlknife.rb
-- lib/mysqlknife/checksum.rb
 - lib/mysqlknife/cli.rb
-- lib/mysqlknife/config.rb
-- lib/mysqlknife/connection.rb
-- lib/mysqlknife/kill.rb
-- lib/mysqlknife/replica.rb
-- lib/mysqlknife/sql.rb
-- lib/mysqlknife/swap.rb
+- lib/mysqlknife/command.rb
+- lib/mysqlknife/configs.rb
+- lib/mysqlknife/iterm.rb
+- lib/mysqlknife/log.rb
+- lib/mysqlknife/mysql.rb
+- lib/mysqlknife/mysql/command.rb
+- lib/mysqlknife/mysql/kill.rb
+- lib/mysqlknife/mysql/replica.rb
+- lib/mysqlknife/mysql/sql.rb
+- lib/mysqlknife/parameters.rb
+- lib/mysqlknife/ssh.rb
 - lib/mysqlknife/version.rb
 homepage: https://github.com/nicola51980/mysqlknife
 licenses:

data/lib/mysqlknife/checksum.rb

@@ -1,26 +0,0 @@
-# encoding: utf-8
-
-module Mysqlknife
-  class Checksum
-    def initialize(options = {})
-      dsn_from  = Config.new.dsn(options[:from])
-      dsn_to    = Config.new.dsn(options[:to])
-
-      @sum_from = sum(dsn_from, options[:table])
-      @sum_to   = sum(dsn_to, options[:table])
-    end
-
-    def sum(dsn, table)
-      conn    = Connection.new(dsn)
-      sql     = Sql.new(dsn[:database], table)
-      columns = conn.execute(sql.columns)
-      sql     = sql.checksum(columns)
-      columns = conn.execute(sql)
-      columns.first['sum']
-    end
-
-    def equal?
-      (@sum_from == @sum_to)
-    end
-  end
-end

data/lib/mysqlknife/config.rb

@@ -1,60 +0,0 @@
-# encoding: utf-8
-
-module Mysqlknife
-  class Config
-    @configs = nil
-
-    def initialize(file = nil)
-      unless file.nil?
-        path     = File.expand_path(File.join(Dir.pwd, file))
-        @configs = YAML.load_file(path) if File.exist?(path)
-      end
-    end
-
-    def get
-      if !@configs.nil? && !ENV['ENV'].nil?
-        # Convert string keys to symbols in hash:
-        @configs = @configs[ENV['ENV']].reduce({}) do |config, (k, v)|
-          config[k.to_sym] = v
-          config
-        end
-
-        # Add defaults values:
-        defaults(@configs)
-      end
-    end
-
-    def symbolize_keys(options)
-      options.inject({}){|hash,(k,v)| hash[k.to_sym] = v; hash}
-    end
-
-    def defaults(options)
-      { host: 'localhost',
-        port: '3306',
-        user: 'root',
-        password: '' }
-      .merge(options)
-    end
-
-    def dsn(options)
-      options.split(',')
-             .map { |c| c.split('=', 2) }
-             .reduce({}) do |m, (key, value)|
-        case key.to_sym
-        when :h
-          key = :host
-        when :u
-          key = :user
-        when :p
-          key = :password
-        when :P
-          key = :port
-        when :d
-          key = :database
-        end
-        m[key] = value
-        m
-      end
-    end
-  end
-end

data/lib/mysqlknife/connection.rb

@@ -1,30 +0,0 @@
-# encoding: utf-8
-
-module Mysqlknife
-  class Connection
-    def initialize(options = {}, database = nil)
-      config    = Config.new
-      options   = config.symbolize_keys(options)
-      options   = config.defaults(options)
-
-      @host     = options[:host]
-      @port     = options[:port]
-      @username = options[:user]
-      @password = options[:password]
-      @database = options[:database] unless database
-    end
-
-    def execute(sql)
-      conn = Mysql2::Client.new(host: @host,
-                                port: @port,
-                                username: @username,
-                                password: @password,
-                                database: @database)
-      return conn.query(sql)
-    rescue => e
-      puts sql
-      puts e
-      exit(1)
-    end
-  end
-end

data/lib/mysqlknife/kill.rb

@@ -1,72 +0,0 @@
-# encoding: utf-8
-
-module Mysqlknife
-  class Kill
-    def initialize(options = {})
-      @conn  = Connection.new(options)
-      @sql   = Sql.new
-      @where = options[:where]
-    end
-
-    def clear
-      list.each do |process|
-        log(process)
-        kill(process['id'])
-      end
-    end
-
-    def check_user
-      sql    = @sql.current_user
-      result = @conn.execute(sql).first['CURRENT_USER'] =~ /^root\@/
-
-      return true if result
-    end
-
-    def check_grants
-      sql    = @sql.current_user_grants
-      result = @conn.execute(sql).first.values.to_s =~ /PROCESS/
-
-      return true if result
-    end
-
-    def check_privileges
-      check_user || check_grants
-    end
-
-    def list
-      @conn.execute(@sql.processlist(@where))
-    end
-
-    def show
-      unless list.first.nil?
-        table = Terminal::Table.new do |t|
-          t.add_row(list.first.keys)
-          t.add_separator
-          list.each do |row|
-            t.add_row(row.values)
-          end
-        end
-        table
-      end
-    end
-
-    def log(process)
-      puts %W(#{Time.now.getutc}
-              : MySQL Kill Process -
-              #{process.map { |k, v| "#{k}: #{v}" }.join(', ')}
-              ).join(' ')
-    end
-
-    def rds?
-      ! @conn.execute(@sql.show_procedure('rds_kill')).first
-    end
-
-    def kill(id)
-      if rds?
-        @conn.execute(@sql.mysql_kill(id))
-      else
-        @conn.execute(@sql.rds_kill(id))
-      end
-    end
-  end
-end

data/lib/mysqlknife/replica.rb

@@ -1,51 +0,0 @@
-# encoding: utf-8
-
-module Mysqlknife
-  class Replica
-    def initialize(options = {})
-      @conn   = Connection.new(options)
-      @sql    = Sql.new
-      @behind = options[:behind]
-    end
-
-    def check_status_negative(behind)
-      @behind < 0 && behind < 0 && behind <= @behind && behind != 0
-    end
-
-    def check_status_positive(behind)
-      @behind > 0 && behind > 0 && @behind <= behind && behind != 0
-    end
-
-    def skip
-      procedure_name  = @sql.show_procedure('rds_skip_repl_error')
-      exist_procedure = @conn.execute(procedure_name).first
-      slave_status    = @conn.execute(@sql.slave_status).first
-
-      return nil if slave_status.nil?
-      return nil unless slave_status.include? 'Seconds_Behind_Master'
-
-      if check_status_negative(slave_status['Seconds_Behind_Master'])
-        puts %W(#{Time.now.getutc} : MySQL Skip error replica -
-                #{slave_status.map { |k, v| "#{k}: #{v}" }.join(', ')}
-                ).join(' ')
-        if exist_procedure.nil?
-          @conn.execute(@sql.mysql_skip_repl_error)
-          while @conn.next_result
-            message = @conn.store_result
-            puts %W(#{Time.now.getutc} : MySQL Skip error replica -
-                    Message: #{message}).join(' ')
-            sleep(1)
-          end
-        else
-          message = @conn.execute(@sql.rds_skip_repl_error).first['Message']
-          puts %W(#{Time.now.getutc} : MySQL Skip error replica -
-                  Message: #{message}).join(' ')
-        end
-      # Resolv Positive Lag:
-      elsif check_status_positive(slave_status['Seconds_Behind_Master'])
-        puts %W(#{Time.now.getutc} : MySQL Skip error replica -
-                Not implemented.).join(' ')
-      end
-    end
-  end
-end

data/lib/mysqlknife/sql.rb

@@ -1,114 +0,0 @@
-# encoding: utf-8
-
-module Mysqlknife
-  class Sql
-    def initialize(database = nil, table = nil)
-      @database = database
-      @table    = table
-    end
-
-    def use
-      "USE `#{@database}`;"
-    end
-
-    def drop_view(view)
-      "DROP VIEW IF EXISTS `#{@database}`.`#{view}`;"
-    end
-
-    def create_view(view, select)
-      "CREATE OR REPLACE VIEW `#{@database}`.`#{view}` AS #{select};"
-    end
-
-    def swap_table(database, table)
-      %W(RENAME TABLE
-         `#{database}`.`#{table}` TO `#{@database}`.`_#{table}_tmp`,
-         `#{@database}`.`#{table}` TO `#{database}`.`#{table}`,
-         `#{@database}`.`_#{table}_tmp` TO `#{@database}`.`#{table}`;)
-      .join(' ')
-    end
-
-    def rename_table(database, table_old, table_new = nil)
-      table_new = table_old unless  table_old.nil?
-
-      %W(RENAME TABLE `#{@database}`.`#{table_old}`
-         TO `#{database}`.`#{table_new}`;).join(' ')
-    end
-
-    def tables
-      %W(SELECT table_name AS name
-         FROM information_schema.tables
-         WHERE table_schema = '#{@database}'
-         AND table_type = 'BASE TABLE'
-         ORDER BY table_name;).join(' ')
-    end
-
-    def views
-      %W(SELECT REPLACE(view_definition, '`#{@database}`.', '') AS code,
-         table_name AS name
-         FROM information_schema.views
-         WHERE table_schema = '#{@database}';).join(' ')
-    end
-
-    def 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_kill(pid)
-      "kill #{pid};"
-    end
-
-    def rds_kill(pid)
-      "CALL mysql.rds_kill(#{pid});"
-    end
-
-    def current_user
-      'SELECT CURRENT_USER;'
-    end
-
-    def current_user_grants
-      'SHOW GRANTS FOR CURRENT_USER;'
-    end
-
-    def columns
-      %W(SELECT COLUMN_NAME
-         FROM information_schema.columns
-         WHERE table_schema = '#{@database}'
-         AND table_name   = '#{@table}';).join(' ')
-    end
-
-    def checksum(columns)
-      md5 = columns.map do |column| %W(COALESCE(
-                                       #{column['COLUMN_NAME']},
-                                       '#{column['COLUMN_NAME']}')).join
-      end
-      "SELECT SUM(CRC32(CONCAT(#{md5.join(', ')}))) AS sum FROM #{@table};"
-    end
-
-    def slave_status
-      'SHOW SLAVE STATUS;'
-    end
-
-    def rds_skip_repl_error
-      'CALL mysql.rds_skip_repl_error;'
-    end
-
-    def mysql_skip_repl_error
-      %w(STOP SLAVE;
-         SET GLOBAL SQL_SLAVE_SKIP_COUNTER = 1;
-         START SLAVE;).join(' ')
-    end
-  end
-end