snapback 0.0.1

data/lib/snapback/dsl.rb

@@ -0,0 +1,208 @@
+require "colorize"
+require "open4"
+
+LOCKED_FILES = ['.', '..', 'lost+found']
+
+# DSL: on_rollback
+# Add a callback if a method (somewhere) fails
+def on_rollback method
+  Snapback::Transaction.instance.add_rollback_method method
+end
+
+def run description, command
+  puts "#{command}".yellow if $options[:verbose]
+  debug "#{description}".ljust(72)
+
+  begin
+    err = ""
+    status = Open4::popen4(command) do |pid, stdin, stdout, stderr|
+      err = stderr.read
+    end
+
+    if status != 0 then
+      raise err
+    end
+  rescue
+    show_failed
+    raise $!
+  end
+
+  show_ok
+end
+
+def check description, command
+  print description.ljust(72)
+  result = command.call
+  if result == true then
+    show_ok
+  else
+    show_no
+  end
+
+  result
+end
+
+def debug *args
+  print "#{args}"
+end
+
+def show_ok
+  puts "[#{"  OK  ".green}]"
+end
+
+def show_no
+  puts "[#{"  NO  ".red}]"
+end
+
+def show_failed
+  puts "[#{"FAILED".red}]"
+end
+
+def ask_int question, max
+  number = nil
+
+  while true
+    print "#{question}: "
+
+    number = $stdin.gets.chomp
+
+    if !Integer(number) then
+      puts "The value you entered is not a number."
+      puts ""
+      next
+    end
+
+    number = number.to_i
+
+    if number < 1 || number > max then
+      puts "Please enter a number between 1 and #{max}."
+      puts ""
+      next
+    end
+
+    break
+  end
+
+  number
+end
+
+def ask_string question
+  print "#{question}: "
+  $stdin.gets.chomp
+end
+
+
+# Move MySQL Files
+def move_mysql_files from_dir, to_dir
+  Dir.foreach(from_dir) do |filename|
+    if !LOCKED_FILES.include?(filename) then
+      run "Moving #{from_dir}/#{filename} to #{to_dir}",
+        "mv #{from_dir}/#{filename} #{to_dir}"
+    end
+  end
+end
+
+def get_mount_dir database
+  "#{$config['filesystem']['mount']}/#{$config['lvm']['prefix_database']}-#{database}"
+end
+
+# Flush table
+def exec_flush
+  check "Flushing database to file system with read lock", lambda {
+    $database.flush_lock
+  }
+
+  on_rollback lambda {
+    check "Removing read lock", lambda { $database.unlock }
+  }
+end
+
+# Mount the logical volume
+def exec_mount device, directory
+  # If it's already mounted, don't worry
+  mount_status = Open4::popen4("mountpoint -q #{directory}") do |pid, stdin, stdout, stderr|
+  end
+
+  if mount_status == 0 then
+    return
+  end
+
+  # Run command and set rollback
+  run "Mounting logical volume", 
+    "mount #{device} #{directory}"
+
+  on_rollback lambda {
+    run "Unmounting logical volume",
+      "umount #{directory}"
+  }
+end
+
+# Unmount
+def exec_unmount device, directory
+  # If it's not already mounted, don't worry
+  mount_status = Open4::popen4("mountpoint -q #{directory}") do |pid, stdin, stdout, stderr|
+  end
+
+  if mount_status != 0 then
+    return
+  end
+
+  # Run command and set rollback
+  run "Unmounting logical volume",
+    "umount #{directory}"
+
+  on_rollback lambda {
+    run "Mounting logical volume", 
+      "mount #{device} #{directory}"
+  }
+end
+
+# Symbolic-link
+def exec_link entry, exit
+  run "Linking symbolic directory",
+    "ln -s #{exit} #{entry}"
+
+  on_rollback lambda {
+    run "Unlinking symbolic directory",
+      "unlink #{entry}"
+  }
+end
+
+# Remove symbolic link
+def exec_unlink entry, exit
+  run "Unlinking symbolic directory",
+    "unlink #{entry}"
+
+  on_rollback lambda {
+    run "Linking symbolic directory",
+      "ln -s #{exit} #{entry}"
+  }
+end
+
+# Activate logical volume
+def exec_activate device
+  run "Activating logical volume",
+    "lvchange -ay #{device}"
+
+  on_rollback lambda {
+    run "Deactivate logical volume",
+      "lvchange -an #{device}"
+  }
+end
+
+# Activate logical volume
+def exec_deactivate device
+  run "Deactivate logical volume",
+    "lvchange -an #{device}"
+
+  on_rollback lambda {
+    run "Activating logical volume",
+      "lvchange -ay #{device}"
+  }
+end
+
+# Take ownership
+def exec_chown directory
+  run "Changing ownership",
+    "chown -R mysql:mysql #{directory}"
+end

data/lib/snapback/options.rb

@@ -0,0 +1,85 @@
+require 'optparse'
+
+module Snapback
+  class Options
+    def self.parse(args)
+      options = {}
+      
+      options[:command]  = nil
+      options[:database] = nil
+
+      options[:config]  = "#{File.expand_path('~')}/.snapback.yml"
+      options[:verbose] = false
+      options[:size]    = nil
+
+      opts = OptionParser.new do |opts|
+        opts.banner = "Usage: sudo #{$0} [command] [options]"
+
+        opts.separator "Create database snapshots using logical volume management (LVM)"
+        opts.separator ""
+        opts.separator "Commands:"
+        opts.separator "    install".ljust(37)                 + "Check the environment is sane and setup"
+        opts.separator "    create DBNAME -s SIZE".ljust(37)   + "Create a new snapback compliant database"
+        opts.separator "    snapshot DBNAME -s SIZE".ljust(37) + "Take a snapshot of the current position"
+        opts.separator "    commit DBNAME".ljust(37)           + "Commit the working copy"
+        opts.separator "    rollback DBNAME".ljust(37)         + "Rollback the working copy to the snapshot"
+        opts.separator "    drop DBNAME".ljust(37)             + "Drop an existing database"
+        opts.separator "    mount DBNAME".ljust(37)            + "Mount an existing database"
+        opts.separator "    unmount DBNAME".ljust(37)          + "Unmount an existing database"
+
+        opts.separator ""
+        opts.separator "Options:"
+
+        opts.on("-s", "--size SIZE",
+                "Allocate a size parameter (e.g.: 1M, 1G, 1T)") do |q|
+          options[:size] = q
+        end
+
+        opts.on("-v", "--verbose",
+                "Show detailed transaction information") do
+          options[:verbose] = true
+        end
+
+        opts.on("-h", "--help",
+                "Show this help screen") do
+          puts opts
+          exit
+        end
+
+        opts.on("-c", "--config",
+                "Configuration file") do |q|
+          options[:config] = q
+        end
+
+        opts.separator ""
+      end
+
+      opts.parse!(args)
+
+      case args[0]
+      when "install", "create", "snapshot", "commit", "rollback", "drop", "mount", "unmount"
+        options[:command] = args[0]
+      when nil
+        puts opts
+        exit
+      else
+        $stderr.puts "command '#{args[0]}' not recognised"
+        puts opts
+        exit
+      end
+
+      case options[:command]
+      when "create", "snapshot", "commit", "rollback", "drop", "mount", "unmount"
+        if args[1].nil? then
+          $stderr.puts "you must specify a database with command '#{options[:command]}'"
+          puts opts
+          exit
+        end
+
+        options[:database] = args[1]
+      end
+
+      options
+    end
+  end
+end

data/lib/snapback/transaction.rb

@@ -0,0 +1,19 @@
+require 'singleton'
+
+module Snapback
+  class Transaction
+    include Singleton
+    
+    @@rollback_commands = []
+
+    def add_rollback_method method
+      @@rollback_commands.push method
+    end
+
+    def do_rollback
+      while rollback_command = @@rollback_commands.pop do
+        rollback_command.call
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification 
+name: snapback
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Bashkim Isai
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-06-19 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mysql
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.9.1
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: open4
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.3.0
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: ruby-lvm
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.1.1
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: colorize
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.5.8
+  type: :runtime
+  version_requirements: *id004
+description: A simple logical volume database manager
+email: 
+executables: 
+- snapback
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/snapback/app/commit.rb
+- lib/snapback/app/create.rb
+- lib/snapback/app/drop.rb
+- lib/snapback/app/install.rb
+- lib/snapback/app/mount.rb
+- lib/snapback/app/rollback.rb
+- lib/snapback/app/snapshot.rb
+- lib/snapback/app/unmount.rb
+- lib/snapback/database.rb
+- lib/snapback/dsl.rb
+- lib/snapback/options.rb
+- lib/snapback/transaction.rb
+- bin/snapback
+- README.md
+homepage: http://github.com/bashaus/snapback
+licenses: 
+- MIT
+metadata: {}
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - &id005 
+    - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - *id005
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Snapback
+test_files: []
+