cassandra_backup 0.0.1

data/.gitignore

@@ -0,0 +1 @@
+Gemfile.lock

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+
+gemspec

data/README.rdoc

@@ -0,0 +1,21 @@
+= CassandraBackup
+
+CassandraBackup is a command line tool for dumping and restoring a Cassandra column family.
+
+== Dumping a Keyspace
+
+Run cassandra_dump, specifying the keyspace:
+
+  cassandra_dump depot_production > backup.json
+
+== Restoring a Keyspace
+
+Run cassandra_import, specifying the keyspace:
+
+  cassandra_import depot_production < backup.json
+
+== Options
+
+  --servers:: A comma delimited list of servers. Defaults to 127.0.0.1:9160. (--servers 127.0.0.14:9169,127.0.1.19:9100)
+  --version:: Set cassandra version. Defaults to 1.0. (--version 0.6)
+  --columns:: The column families to dump. Defaults to all. (--columns Widgets --columns People)

data/bin/cassandra_dump

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+require File.expand_path('../../lib/cassandra_backup',  __FILE__)
+CassandraBackup::Dumper.run!

data/bin/cassandra_import

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+require File.expand_path('../../lib/cassandra_backup',  __FILE__)
+CassandraBackup::Importer.run!

data/cassandra_backup.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = 'cassandra_backup'
+  s.version = '0.0.1'
+  s.description = 'Cassandra Backup'
+  s.summary = 'Cassandra Backup'
+  s.authors = ['Infogroup']
+  s.email = 'gems@gotime.com'
+  s.homepage = 'http://github.com/data-axle/cassandra_backup'
+
+  s.required_ruby_version     = '>= 1.8.7'
+  s.required_rubygems_version = '>= 1.8.0'  
+  s.executables   = ['cassandra_dump', 'cassandra_import']
+  s.extra_rdoc_files = ['README.rdoc']
+  s.files       = `git ls-files`.split("\n")
+  s.test_files  = `git ls-files -- {test}/*`.split("\n")
+
+  s.add_runtime_dependency('mcmire-cassandra', ">= 0.12.3")
+  s.add_runtime_dependency('yajl-ruby')
+  s.add_development_dependency('bundler')
+end

data/lib/cassandra_backup.rb

@@ -0,0 +1,7 @@
+require 'bundler/setup'
+Bundler.require
+
+require 'yajl'
+
+require 'cassandra_backup/dumper'
+require 'cassandra_backup/importer'

data/lib/cassandra_backup/backup_process.rb

@@ -0,0 +1,30 @@
+require 'cassandra_backup/command'
+
+module CassandraBackup
+  class BackupProcess
+    def self.run!
+      new(CassandraBackup::Command.new(ARGV)).run
+    end
+
+    attr_reader :command
+    def initialize(command)
+      @command = command
+    end
+
+    def run
+      raise "inheriting class must implement"
+    end
+
+    def connection
+      @connection ||= Cassandra.new(command.keyspace, command.servers)
+    end
+
+    def input_io
+      $stdin
+    end
+
+    def output_io
+      $stdout
+    end
+  end
+end

data/lib/cassandra_backup/command.rb

@@ -0,0 +1,60 @@
+module CassandraBackup
+  class Command
+
+    attr_accessor :options
+    def initialize(args)
+      self.options = {}
+
+      if args.first
+        options[:keyspace] = args.first
+      else
+        args.push '-h'
+      end
+
+      OptionParser.new do |opts|
+        opts.banner = "Usage: keyspace [options]"
+        opts.on('-s', '--servers SERVERS', 'Set server list. Default is 127.0.0.1:9160.') do |v|
+          options[:servers] = v.split(/\s|,/)
+        end
+        opts.on('-v', '--version VERSION', 'Set cassandra version. Default is 1.0.') do |v|
+          options[:version] = v
+        end
+        opts.on('-c', '--columns Columns', 'Set column families. Defaults to all') do |v|
+          (options[:columns] ||= []).concat v.split(/\s|,/)
+        end
+        opts.on('-h', 'Show this help message.') do
+          $stdout.puts opts; exit
+        end
+        opts.parse!(args)
+      end
+
+      require required_version
+    end
+
+    def keyspace
+      options[:keyspace]
+    end
+
+    def servers
+      if options[:servers]
+        options[:servers]
+      else
+        ['127.0.0.1:9160']
+      end
+    end
+
+    def columns
+      if options[:columns]
+        options[:columns]
+      end
+    end
+
+    def required_version
+      if options[:version]
+        "cassandra/#{options[:version]}"
+      else
+        "cassandra/1.0"
+      end
+    end
+  end
+end

data/lib/cassandra_backup/dumper.rb

@@ -0,0 +1,26 @@
+require 'cassandra_backup/backup_process'
+
+module CassandraBackup
+  class Dumper < BackupProcess
+    def run
+      column_family_names.each do |column_family|
+        connection.each(column_family) do |key, attributes|
+          output_io.puts encoder.encode([column_family, key, attributes])
+        end
+      end
+    end
+
+    private
+      def column_family_names
+        if command.columns
+          command.columns
+        else
+          connection.column_families.keys - ['schema_migrations']
+        end
+      end
+
+      def encoder
+        @encoder ||= Yajl::Encoder.new
+      end
+  end
+end

data/lib/cassandra_backup/importer.rb

@@ -0,0 +1,17 @@
+require 'cassandra_backup/backup_process'
+
+module CassandraBackup
+  class Importer < BackupProcess
+    def run
+      parser.parse(input_io) do |column_family, key, attributes|
+        p "inserting"
+        connection.insert column_family, key, attributes
+      end
+    end
+
+    private
+      def parser
+        @parser ||= Yajl::Parser.new
+      end
+  end
+end

metadata

@@ -0,0 +1,106 @@
+--- !ruby/object:Gem::Specification
+name: cassandra_backup
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Infogroup
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mcmire-cassandra
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.12.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.12.3
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Cassandra Backup
+email: gems@gotime.com
+executables:
+- cassandra_dump
+- cassandra_import
+extensions: []
+extra_rdoc_files:
+- README.rdoc
+files:
+- .gitignore
+- Gemfile
+- README.rdoc
+- bin/cassandra_dump
+- bin/cassandra_import
+- cassandra_backup.gemspec
+- lib/cassandra_backup.rb
+- lib/cassandra_backup/backup_process.rb
+- lib/cassandra_backup/command.rb
+- lib/cassandra_backup/dumper.rb
+- lib/cassandra_backup/importer.rb
+homepage: http://github.com/data-axle/cassandra_backup
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.8.7
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.8.0
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Cassandra Backup
+test_files: []