mysqlknife 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a035195f41f18f9df67401eaf9aa6f9793945bf3
-  data.tar.gz: 33276a6c3dfe02dd97230bba521a0491c6a1295a
+  metadata.gz: 51955f4209a8173426bf5358df753a980cc13a44
+  data.tar.gz: deb6a001c669f47c3e988a0d7d96a70bb0ff1bef
 SHA512:
-  metadata.gz: 8402b7c4209d2fcf51aafbb12902492016840299083d9915a3b18e1b24d257d34474053b2fd8049d069420b541bbb7a4ca89bcef70ddc239d9f4cd83bace753e
-  data.tar.gz: b3831427b4a10f7868dcbe673063c4b296abd0e4c1aa6b6b50be5c5ef7f07a7ec95b3eea6490f907e0657222b6b016c42c50937200abcabefecc5d1a2c05af19
+  metadata.gz: cf03569b5bfe77afb7568cb9f5dac22579e74fdda74f5159b3da047db17bac7d6c6297ea6a789628ce049f8f1403b8905474d09dec146ae7abc52179d2efd37a
+  data.tar.gz: 62c6e2964b203bcb7183b994ace384743e109d0e1a042dd4b7600175bfc16be44c9c60dc3160aeadf5da1bd58e53521113e5624628e6ff7fc28a9fc2b29da4e7

data/bin/mysqlknife

@@ -18,6 +18,30 @@ global_option('-u', '--user      STRING')
 global_option('-p', '--password [STRING]')
 global_option('-P', '--port     [STRING]')
 
+command :checksum do |c|
+  c.description = 'Checksum table.'
+  c.syntax      = %w(mysqlknife checksum
+                     --from h=127.0.0.1,P=3306,u=root,p=admin,d=demo_from
+                     --to h=127.0.0.1,P=3306,u=root,p=admin,d=demo_to
+                     --table foo).join(' ')
+  c.option '--from STRING', String, 'DNS to connect to source database.'
+  c.option '--to STRING', String, 'DNS to connect to destination database.'
+  c.option '--table STRING', String, 'Table name to compare.'
+
+  c.action do |args, options|
+    if options.from.nil?  ||
+       options.to.nil?    ||
+       options.table.nil?
+      puts c.syntax
+    else
+      unless Mysqlknife::Checksum.new(options.__hash__).equal?
+        puts "-- Table '#{options.table}' is not equal data."
+        exit 1
+      end
+    end
+  end
+end
+
 command :swap do |c|
   c.description = 'Swap two databases.'
   c.syntax      = %w(mysqlknife swap

data/lib/mysqlknife.rb

@@ -2,6 +2,7 @@
 
 require 'mysqlknife/config'
 require 'mysqlknife/connection'
+require 'mysqlknife/checksum'
 require 'mysqlknife/kill'
 require 'mysqlknife/sql'
 require 'mysqlknife/swap'

data/lib/mysqlknife/checksum.rb

@@ -0,0 +1,26 @@
+# encoding: utf-8
+
+module Mysqlknife
+  class Checksum
+    def initialize(options = {})
+      dns_from  = Config.new.dns(options[:from])
+      dns_to    = Config.new.dns(options[:to])
+
+      @sum_from = sum(dns_from, options[:table])
+      @sum_to   = sum(dns_to, options[:table])
+    end
+
+    def sum(dns, table)
+      conn    = Connection.new(dns)
+      sql     = Sql.new(dns[: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

@@ -6,7 +6,7 @@ module Mysqlknife
   class Config
     @configs = nil
 
-    def initialize(file)
+    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)
@@ -32,8 +32,29 @@ module Mysqlknife
       { host: 'localhost',
         port: '3306',
         user: 'root',
-        pass: '' }
+        password: '' }
       .merge(options)
     end
+
+    def dns(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,28 +1,19 @@
 # encoding: utf-8
 
 require 'rubygems'
+require 'mysqlknife/config'
 require 'mysql2'
 
 module Mysqlknife
   class Connection
     def initialize(options = {}, database = nil)
-      options   = options(options)
+      options   = Config.new.defaults(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)
+      @database = options[:database] unless database
     end
 
     def execute(sql)

data/lib/mysqlknife/sql.rb

@@ -2,8 +2,9 @@
 
 module Mysqlknife
   class Sql
-    def initialize(database = nil)
+    def initialize(database = nil, table = nil)
       @database = database
+      @table    = table
     end
 
     def use
@@ -80,5 +81,20 @@ module Mysqlknife
     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
   end
 end

data/lib/mysqlknife/version.rb

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

metadata

@@ -1,15 +1,57 @@
 --- !ruby/object:Gem::Specification
 name: mysqlknife
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Nicola Strappazzon C.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-03-23 00:00:00.000000000 Z
-dependencies: []
+date: 2014-03-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: commander
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 4.1.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 4.1.6
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.4.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.4.5
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.5.5
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.5.5
 description: MySQL knife tools
 email: nicola51980@gmail.com
 executables:
@@ -18,6 +60,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/mysqlknife
+- lib/mysqlknife/checksum.rb
 - lib/mysqlknife/config.rb
 - lib/mysqlknife/connection.rb
 - lib/mysqlknife/kill.rb