mysqlknife 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 2d375284a62187872eddd91fba09f0da91bcc450
+  data.tar.gz: d556e33bf773b026f4024441949f4fa837760bee
+SHA512:
+  metadata.gz: 4a47e344f675bf8ff135010b824c85dd020275f68a962d2863ca710a430863a8a1ac6ce52064747f7f715bc9efa08880b85a334dde026d33f5a688ce6513a999
+  data.tar.gz: 2fda7a0ba3ad1c17317de5fa03cb3a2836aa28dfe612dae97fb46b3c168b4b361c8209c70501d22df21eadc1c8a5ef6011573564ba3d90f8f3de2cf9dc311407

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Nicola Strappazzon C.
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,53 @@
+# Mysqlknife
+
+TODO: Write a gem description
+
+## Installation
+
+Install this tool executing the following command:
+
+```Shell
+$ gem install mysqlkill
+```
+
+## Usage
+
+
+### Swap Databases
+
+Swap tables and views between two databases.
+
+You need to rename two databases, for example, foo and bar. You only need to run the
+following command and save the STDOUT output to a file and then execute DML
+statements.
+
+```Shell
+$ mysqlknife swap --host 127.0.0.1 --user root --password admin --databases from,to > swap.sql
+$ mysql -h 127.0.0.1 -u root -padmin < swap.sql
+```
+
+### Kill process
+
+Please, follow next instructions to kill process:
+
+```Shell
+$ mysqlknife kill --host 127.0.0.1 --user root --pass admin --where "time >= 200" --list
+```
+
+```Shell
+$ mysqlknife kill --host 127.0.0.1 --user root --pass admin --where "time >= 200"
+```
+
+## Warning
+
+1. Do not use this tool in production before testing it.
+2. Please, use when do you need.
+3. The author is NOT responsible for misuse use of this tool.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/bin/mysqlknife

@@ -0,0 +1,83 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+$LOAD_PATH << './lib'
+
+require 'commander'
+require 'commander/import'
+require 'terminal-table'
+require 'mysqlknife'
+
+program :name, 'MySQL knife tools'
+program :version, Mysqlknife::VERSION
+program :description, 'MySQL knife tools'
+program :help, 'Author', 'Nicola Strappazzon <nicola51980@gmail.com>'
+
+global_option('-H', '--host      STRING')
+global_option('-u', '--user      STRING')
+global_option('-p', '--password [STRING]')
+global_option('-P', '--port     [STRING]')
+
+command :swap do |c|
+  c.description = 'Swap two databases.'
+  c.syntax      = %w(mysqlknife swap
+                     --host 127.0.0.1
+                     --user root
+                     --password admin
+                     --databases from,to).join(' ')
+  c.option '--databases STRING', String, 'Databases to swap.'
+
+  c.action do |args, options|
+    if options.host.nil?     ||
+       options.user.nil?     ||
+       options.databases.nil?  ||
+       options.databases.split(',').count != 2
+      puts c.syntax
+    else
+      swap = Mysqlknife::Swap.new(options.__hash__).all
+
+      if swap
+        swap.each do |sql|
+          puts sql
+        end
+      end
+    end
+  end
+end
+
+command :kill do |c|
+  c.description = 'Kill all process.'
+  c.syntax      = %w(mysqlknife kill
+                     --host 127.0.0.1
+                     --user root
+                     --password admin
+                     --where "time >= 28800").join(' ')
+  c.option '--where STRING', String, 'Conditions to filter.'
+  c.option '--list', 'List match process to kill.'
+
+  c.action do |args, options|
+    if options.host.nil?  ||
+       options.user.nil?  ||
+       options.where.nil?
+      puts c.syntax
+    else
+      kill = Mysqlknife::Kill.new(options.__hash__)
+
+      if options.list.nil?
+        kill.clear unless kill.list.first.nil?
+      else
+        unless kill.list.first.nil?
+          table = Terminal::Table.new do |t|
+            t.add_row(kill.list.first.keys)
+            t.add_separator
+
+            kill.list.each do |row|
+              t.add_row(row.values)
+            end
+          end
+          puts table
+        end
+      end
+    end
+  end
+end

data/lib/mysqlknife.rb

@@ -0,0 +1,12 @@
+# encoding: utf-8
+
+require 'mysqlknife/config'
+require 'mysqlknife/connection'
+require 'mysqlknife/kill'
+require 'mysqlknife/sql'
+require 'mysqlknife/swap'
+require 'mysqlknife/version'
+
+module Mysqlknife
+  # Your code goes here...
+end

data/lib/mysqlknife/config.rb

@@ -0,0 +1,39 @@
+# encoding: utf-8
+
+require 'yaml'
+
+module Mysqlknife
+  class Config
+    @configs = nil
+
+    def initialize(file)
+      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 defaults(options)
+      options.delete_if { |k, v| v.nil? }
+
+      { host: 'localhost',
+        port: '3306',
+        user: 'root',
+        pass: '' }
+      .merge(options)
+    end
+  end
+end

data/lib/mysqlknife/connection.rb

@@ -0,0 +1,41 @@
+# encoding: utf-8
+
+require 'rubygems'
+require 'mysql2'
+
+module Mysqlknife
+  class Connection
+    def initialize(options = {}, database = nil)
+      options   = options(options)
+
+      @host     = options[:host]
+      @port     = options[:port]
+      @username = options[:user]
+      @password = options[:password]
+      @database = database
+    end
+
+    def options(options)
+      options.delete_if { |k, v| v.nil? }
+
+      { host: 'localhost',
+        port: '3306',
+        user: 'root',
+        password: '' }
+      .merge(options)
+    end
+
+    def execute(sql)
+      begin
+        conn = Mysql2::Client.new(host: @host,
+                                  port: @port,
+                                  username: @username,
+                                  password: @password,
+                                  database: @database)
+        return conn.query(sql)
+      rescue
+        exit(1)
+      end
+    end
+  end
+end

data/lib/mysqlknife/kill.rb

@@ -0,0 +1,38 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Kill
+    def initialize(options = {})
+      @conn  = Connection.new(options)
+      @sql   = Sql.new
+      @where = options[:where]
+    end
+
+    def check_privileges
+      sql_user   = @sql.current_user
+      sql_grants = @sql.current_user_grants
+      if @conn.execute(sql_user).first['CURRENT_USER'] =~ /^root\@/
+        return true
+      else
+
+        return true if @conn.execute(sql_grants).first.values.to_s =~ /PROCESS/
+      end
+    end
+
+    def list
+      @conn.execute(@sql.processlist(@where))
+    end
+
+    def clear
+      exist_procedure = @conn.execute(@sql.show_procedure('rds_kill'))
+
+      @conn.execute(@sql.processlist(@where)).each do |process|
+        if exist_procedure.first.nil?
+          @conn.execute(@sql.mysql_kill(process['id']))
+        else
+          @conn.execute(@sql.rds_kill(process['id']))
+        end
+      end
+    end
+  end
+end

data/lib/mysqlknife/sql.rb

@@ -0,0 +1,84 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Sql
+    def initialize(database = nil)
+      @database = database
+    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
+  end
+end

data/lib/mysqlknife/swap.rb

@@ -0,0 +1,63 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Swap
+    def initialize(options = {})
+      databases  = options[:databases].split(',')
+      @db_from   = databases[0]
+      @db_to     = databases[1]
+      @conn_from = Connection.new(options, @db_from)
+      @conn_to   = Connection.new(options, @db_to)
+      @from      = Sql.new(@db_from)
+      @to        = Sql.new(@db_to)
+    end
+
+    def all
+      sql = []
+
+      # Swap tables.
+      tables_from = @conn_from.execute(@from.tables)
+      tables_from = tables_from.map { |f| f['name'] } if tables_from
+      tables_to   = @conn_to.execute(@to.tables)
+      tables_to   = tables_to.map { |t| t['name'] } if tables_to
+
+      if tables_from && tables_to
+        swap      = tables_from & tables_to
+        rename    = (tables_from - tables_to) + (tables_to - tables_from)
+
+        unless swap.empty?
+          swap.map do | table |
+            sql << @to.swap_table(@db_from, table)
+          end
+        end
+
+        unless rename.empty?
+          rename.map do | table |
+            if tables_from.include?(table)
+              sql << @from.rename_table(@db_to, table)
+            else
+              sql << @to.rename_table(@db_from, table)
+            end
+          end
+        end
+      end
+
+      # Drop views and create new views in another DB.
+      sql.concat(swap_views(@conn_from, @from, @to))
+      sql.concat(swap_views(@conn_to, @to, @from))
+    end
+
+    def swap_views(conn, from, to)
+      sql   = []
+      views = conn.execute(to.views)
+      if views
+        sql << to.use
+        views.each do | view |
+          sql << to.drop_view(view['name'])
+          sql << from.create_view(view['name'], view['code'])
+        end
+      end
+      sql
+    end
+  end
+end

data/lib/mysqlknife/version.rb

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

metadata

@@ -0,0 +1,54 @@
+--- !ruby/object:Gem::Specification
+name: mysqlknife
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Nicola Strappazzon C.
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-03-22 00:00:00.000000000 Z
+dependencies: []
+description: MySQL knife tools
+email: nicola51980@gmail.com
+executables:
+- mysqlknife
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/mysqlknife
+- lib/mysqlknife/config.rb
+- lib/mysqlknife/connection.rb
+- lib/mysqlknife/kill.rb
+- lib/mysqlknife/sql.rb
+- lib/mysqlknife/swap.rb
+- lib/mysqlknife/version.rb
+- lib/mysqlknife.rb
+- LICENSE.txt
+- README.md
+homepage: https://github.com/nicola51980/mysqlknife
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.1.11
+signing_key: 
+specification_version: 4
+summary: ''
+test_files: []