mysql_rake_tasks 0.0.3 → 0.1.0

data/README.rdoc

@@ -2,13 +2,13 @@
 
 A collection of rails rake tasks for mysql
 1. create_users - creates localhost mysql user accounts for each database listing in the database.yml
-
+2. stats - database stats - table size
 
 == Install
 
 1. Add mysql_rake_tasks to your gem file
 
-  gem 'mysql_rake_tasks', '~> 0.0.1'
+  gem 'mysql_rake_tasks', '~> 0.1.0'
 
 2. Run the bundle command
 
@@ -28,6 +28,24 @@ You can also specify your root username and password on the command line:
 
   rake db:mysel:create_users[root,mypassword]
 
+=== stats
+
+To display database stats:
+
+  rake db:mysql:stats
+  +--------------------------------+---------------+-----------+----------+------------+
+  | Table Name                     |          Rows | Data Size | IDX Size | Total Size |
+  +--------------------------------+---------------+-----------+----------+------------+
+  | fish                           |  1.41 Million |   93.6 MB |  41.1 MB |     135 MB |
+  | birds                          |             0 |     16 KB |  0 Bytes |      16 KB |
+  | cats                           |            14 |     16 KB |  0 Bytes |      16 KB |
+  | schema_migrations              |             7 |     16 KB |  0 Bytes |      16 KB |
+  | users                          |             5 |     16 KB |    32 KB |      48 KB |
+  +--------------------------------+---------------+-----------+----------+------------+
+  |                                                                       |     135 MB |
+  +--------------------------------+---------------+-----------+----------+------------+
+  Database: mydb_development  MySQL Server Version: 5.1.58
+
 == License
 
 MIT

data/lib/mysql_rake_tasks/mysql_rake_tasks.rake

@@ -1,11 +1,15 @@
 namespace :db do
   namespace :mysql do
 
-    desc "Create MySQL users from database.yml (localhost only). Run without parameters for interactive mode." 
+    desc "Create MySQL users from database.yml (localhost only). Run without parameters for interactive mode."
     task :create_users, [:root_user, :pass] => [:environment] do |rake_task, args|
       MysqlRakeTasks::Tasks::create_users(args)
     end
 
+    desc "Display MySQL stats"
+    task :stats =>  [:environment] do |rake_task, args|
+      MysqlRakeTasks::Tasks::stats
+    end
   end
 end
 

data/lib/mysql_rake_tasks/tasks.rb

@@ -1,6 +1,12 @@
 require 'rails'
 require 'mysql2'
 
+# test unit seems to need these
+require 'active_support'
+require 'action_view'
+
+include ActionView::Helpers::NumberHelper
+
 module MysqlRakeTasks
   class Tasks
 
@@ -25,7 +31,7 @@ module MysqlRakeTasks
       {:root_user => root_user, :pass => root_pass}
     end
 
-    # creates user permissions for mysql database for localhost only 
+    # creates user permissions for mysql database for localhost only
     def self.create_users(args)
       args = self.get_input(args)
       @root_user = args[:root_user]
@@ -48,7 +54,7 @@ module MysqlRakeTasks
           $stdout.puts "Error code: #{e.errno}"
           $stdout.puts "Error message: #{e.error}"
           $stdout.puts "Error SQLSTATE: #{e.sqlstate}" if e.respond_to?("sqlstate")
-        ensure # disconnect from server 
+        ensure # disconnect from server
           db.close if db
         end
       end
@@ -72,6 +78,63 @@ module MysqlRakeTasks
       SQL
     end
 
+    def self.stats
+      config = Rails::configuration.database_configuration[Rails.env]
+
+      begin
+        dbh = Mysql2::Client.new( :host => config['host'], :username => config['username'], :password => config['password'])
+        sql = stats_query(config['database'])
+        result = dbh.query sql
+
+        print_header
+        db_total = 0
+        result.each  do |row|
+          printf "| %30s | %13s | %9s | %8s | %10s |\n",
+            row["table_name"].ljust(30),
+            number_to_human(row["rows"]).rjust(13),
+            number_to_human_size(row["data"]),
+            number_to_human_size(row["idx"]),
+            number_to_human_size(row["total_size"])
+
+	  db_total += row["total_size"].to_i
+        end
+
+        print_separator
+        printf "|%70s | %10s |\n",'',  number_to_human_size(db_total)
+        print_separator
+        puts "Database: #{config['database']}  MySQL Server Version: #{dbh.info[:version]}\n"
+        puts " "
+      rescue Mysql2::Error => e
+        puts "Error code: #{e.errno}"
+        puts "Error message: #{e.error}"
+        puts "Error SQLSTATE: #{e.sqlstate}" if e.respond_to?("sqlstate")
+      ensure
+        dbh.close if dbh
+      end
+    end
+
+    def self.print_separator
+      puts   "+--------------------------------+---------------+-----------+----------+------------+"
+    end
+
+    def self.print_header
+      print_separator
+      printf "| %30s | %13s | %9s | %8s | %8s |\n",
+        "Table Name".ljust(30), "Rows", "Data Size", "IDX Size", "Total Size"
+       print_separator
+    end
+
+    def self.stats_query(db_name)
+      sql = <<-SQL
+          SELECT table_name,
+          concat(table_rows) rows,
+          concat(data_length) data,
+          concat(index_length) idx,
+          concat(data_length+index_length) total_size
+          FROM information_schema.TABLES
+          WHERE table_schema LIKE '#{db_name}'
+          ORDER BY table_name;
+      SQL
+    end
   end
 end
-

data/lib/mysql_rake_tasks/version.rb

@@ -1,3 +1,3 @@
 module MysqlRakeTasks
-  VERSION = "0.0.3"
+  VERSION = "0.1.0"
 end

data/test/tasks_test.rb

@@ -5,6 +5,10 @@ require 'stringio'
 require 'mocha'
 
 class TasksTest < Test::Unit::TestCase
+  def setup
+    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate) 
+    Rails.configuration.stubs(:database_configuration).returns(stub_config)
+  end
 
   def test_two_args_returns_two_values
     output = MysqlRakeTasks::Tasks::get_input({:root_user => 'user', :pass => 'pass'})
@@ -17,21 +21,18 @@ class TasksTest < Test::Unit::TestCase
   def test_lack_of_args_invokes_cli_interface
     output = ""
     screen = io_mock do |input|
-               input.string = "root\nmypassword\n"
+               input.string = "root\nmyrootpass\n"
                output = MysqlRakeTasks::Tasks.get_input
              end
 
     assert_equal 'mysql user:', screen[0]
     assert_equal 'mysql password:', screen[1]
     assert_equal 'root', output[:root_user]
-    assert_equal 'mypassword', output[:pass]
+    assert_equal 'myrootpass', output[:pass]
   end
 
 
   def test_unsuccessful_authentication
-    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate) 
-    Rails.configuration.stubs(:database_configuration).returns(stub_config)
-
     screen = io_mock do |input|
       MysqlRakeTasks::Tasks.create_users(:root_user => 'root', :pass => 'wrong')
     end
@@ -45,9 +46,6 @@ class TasksTest < Test::Unit::TestCase
   end
 
   def test_successful_creation
-    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate)
-    Rails.configuration.stubs(:database_configuration).returns(stub_config)
-
     screen = io_mock do |input|
       # :pass needs to be set to mysql root in order to pass
       MysqlRakeTasks::Tasks.create_users(:root_user => 'root', :pass => 'myrootpass') 
@@ -62,7 +60,6 @@ class TasksTest < Test::Unit::TestCase
     config = stub_config
     config["development"].delete "username"
 
-    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate)
     Rails.configuration.stubs(:database_configuration).returns(config)
 
     screen = io_mock do |input|
@@ -74,6 +71,16 @@ class TasksTest < Test::Unit::TestCase
     assert_equal 'Error code: 1064', screen[1]
   end
 
+  def test_stats_outputs_header
+    screen = io_mock do |input|
+      MysqlRakeTasks::Tasks.stats
+    end
+
+    assert_equal  '+--------------------------------+---------------+-----------+----------+------------+', screen[0]
+    assert_equal  '| Table Name                     |          Rows | Data Size | IDX Size | Total Size |', screen[1]
+    assert_equal  '+--------------------------------+---------------+-----------+----------+------------+', screen[2]
+  end
+
   def stub_config
    {"development"=>{"adapter"=>"mysql2",
                     "encoding"=>"utf8",

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mysql_rake_tasks
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,12 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-09-05 00:00:00.000000000 -04:00
-default_executable: 
+date: 2012-01-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &22621080 !ruby/object:Gem::Requirement
+  requirement: &5896580 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -22,10 +21,10 @@ dependencies:
         version: 0.9.2
   type: :development
   prerelease: false
-  version_requirements: *22621080
+  version_requirements: *5896580
 - !ruby/object:Gem::Dependency
   name: mocha
-  requirement: &22582860 !ruby/object:Gem::Requirement
+  requirement: &5892460 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -33,10 +32,10 @@ dependencies:
         version: 0.9.12
   type: :development
   prerelease: false
-  version_requirements: *22582860
+  version_requirements: *5892460
 - !ruby/object:Gem::Dependency
   name: mysql2
-  requirement: &22582360 !ruby/object:Gem::Requirement
+  requirement: &5890820 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -44,7 +43,7 @@ dependencies:
         version: 0.2.7
   type: :runtime
   prerelease: false
-  version_requirements: *22582360
+  version_requirements: *5890820
 description: A collection of rails rake tasks for mysql.
 email:
 - gregory.ostermayr@gmail.com
@@ -66,7 +65,6 @@ files:
 - lib/mysql_rake_tasks/version.rb
 - mysql_rake_tasks.gemspec
 - test/tasks_test.rb
-has_rdoc: true
 homepage: https://github.com/gregors/mysql_rake_tasks
 licenses:
 - MIT
@@ -88,7 +86,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: mysql_rake_tasks
-rubygems_version: 1.6.2
+rubygems_version: 1.8.10
 signing_key: 
 specification_version: 3
 summary: Rake tasks for mysql