redis-copy 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+# encoding: utf-8
+source 'https://rubygems.org'
+
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Ryan Biesemeyer
+
+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,29 @@
+# Redis::Copy
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'redis-copy'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install redis-copy
+
+## Usage
+
+TODO: Write usage instructions here
+
+## 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/Rakefile

@@ -0,0 +1,9 @@
+# encoding: utf-8
+
+require 'bundler/gem_tasks'
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+  spec.verbose = true
+end

data/bin/redis-copy

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+$LOAD_PATH.unshift(File.expand_path('../../lib', __FILE__))
+require 'redis-copy/cli'
+
+RedisCopy::CLI.new(ARGV).run!

data/lib/redis-copy.rb

@@ -0,0 +1,79 @@
+# encoding: utf-8
+
+require 'redis'
+require 'active_support/inflector'
+require 'active_support/core_ext/string/strip' # String#strip_heredoc
+
+require 'redis-copy/version'
+require 'redis-copy/ui'
+require 'redis-copy/strategy'
+require 'redis-copy/key-emitter'
+
+module RedisCopy
+  class << self
+    # @param source [String]
+    # @param destination [String]
+    # @options options [Hash<Symbol,Object>]
+    def copy(source, destination, options = {})
+      puts options.inspect
+      ui = UI.load(options)
+
+      source = redis_from(source)
+      destination = redis_from(destination)
+
+      ui.abort('source cannot equal destination!') if same_redis?(source, destination)
+
+      key_emitter = KeyEmitter.load(source, ui, options)
+      strategem = Strategy.load(source, destination, ui, options)
+
+      dest_empty = !(destination.randomkey) # randomkey returns string unless db empty.
+
+      return false unless ui.confirm? <<-EODESC.strip_heredoc
+        Source:      #{source.client.id}
+        Destination: #{destination.client.id} (#{dest_empty ? '' : 'NOT '}empty)
+        Key Emitter: #{key_emitter}
+        Strategy:    #{strategem}
+      EODESC
+
+      ui.abort('Destination not empty!') unless dest_empty
+
+      key_emitter.keys.each_with_object(Hash.new {0}) do |key, stats|
+        success = strategem.copy(key)
+        stats[success ? :success : :failure] += 1
+        stats[:attempt] += 1
+
+        unless success
+          ui.notify("FAIL: #{key.dump}")
+          ui.abort if options[:fail_fast]
+        end
+        ui.notify(stats.inspect) if (stats[:attempt] % 1000).zero?
+      end.tap do |stats|
+        ui.notify("DONE: #{stats.inspect}")
+      end
+    end
+
+    private
+
+    def same_redis?(redis_a, redis_b)
+      # Redis::Client#id returns the connection uri
+      # e.g. 'redis://localhost:6379/0'
+      redis_a.client.id == redis_b.client.id
+    end
+
+    def redis_from(connection_string)
+      require 'uri'
+      connection_string = "redis://#{connection_string}" unless connection_string.start_with?("redis://")
+      uri = URI(connection_string)
+      ret = {uri: uri}
+
+      # Require the URL to have at least a host
+      raise ArgumentError, "invalid url: #{connection_string}" unless uri.host
+
+      host = uri.host
+      port = uri.port if uri.port
+      db = uri.path ? uri.path[1..-1].to_i : 0
+
+      Redis.new(host: host, port: port, db: db)
+    end
+  end
+end

data/lib/redis-copy/cli.rb

@@ -0,0 +1,90 @@
+# encoding: utf-8
+
+require 'redis-copy'
+require 'optparse'
+
+module RedisCopy
+  class CLI
+    REDIS_URI = (/\A(?:redis:\/\/)?([a-z0-9\-.]+)(:[0-9]{1,5})?(\/(?:(?:1[0-5])|[0-9]))?\z/i).freeze
+    DEFAULTS = {
+      ui:           :command_line,
+      key_emitter:  :default,
+      strategy:     :auto,
+      fail_fast:    false,
+      yes:          false,
+    }.freeze unless defined?(DEFAULTS)
+
+    def initialize(argv = ARGV)
+      argv = argv.dup
+      options = {}
+
+      OptionParser.new do |opts|
+        opts.banner = "Usage: #{opts.program_name} [options] <source> <destination>"
+        opts.version = RedisCopy::VERSION
+
+        indent_desc = proc do |desc|
+          desc.split("\n").join("\n#{opts.summary_indent}#{' '*opts.summary_width} ")
+        end
+
+        opts.separator "    <source> and <destination> must be redis connection uris"
+        opts.separator "    like [redis://]<hostname>[:<port>][/<db>]"
+        opts.separator ''
+        opts.separator "Specific options:"
+
+        opts.on('--strategy STRATEGY', [:auto, :new, :classic],
+          indent_desc.(
+            "Select strategy (auto, new, classic) (default #{DEFAULTS[:strategy]})\n" +
+            "  auto:    uses new if available, otherwise fallback\n" +
+            "  new:     use redis DUMP and RESTORE commands (faster)\n" +
+            "  classic: migrates via multiple type-specific commands"
+          )
+        ) do |strategy|
+          options[:strategy] = strategy
+        end
+
+        opts.on('--[no-]dry-run', 'Output configuration and exit') do |d|
+          options[:dry_run] = true
+        end
+
+        opts.on('-d', '--[no-]debug', 'Write debug output') do |debug|
+          options[:debug] = debug
+        end
+
+        opts.on('-t', '--[no-]trace', 'Enable backtrace on failure') do |trace|
+          options[:trace] = trace
+        end
+
+        opts.on('-f', '--[no-]fail-fast', 'Abort on first failure') do |ff|
+          options[:fail_fast] = ff
+        end
+
+        opts.on('-y', '--yes', 'Automatically accept any prompts') do
+          options[:yes] = true
+        end
+
+        opts.parse!(argv)
+        unless argv.size == 2
+          opts.abort "Source and Destination must be specified\n\n" +
+                            opts.help
+        end
+        @source = argv.shift
+        @destination = argv.shift
+
+        opts.abort "source is not valid URI" unless @source =~ REDIS_URI
+        opts.abort "destination is not valid URI" unless @destination =~ REDIS_URI
+      end
+
+      @config = DEFAULTS.merge(options)
+    end
+
+    def run!
+      (puts self.inspect; exit 1) if @config.delete(:dry_run)
+
+      RedisCopy::copy(@source, @destination, @config)
+    rescue => exception
+      $stderr.puts exception.message
+      $stderr.puts exception.backtrace if @config[:trace]
+      exit 1
+    end
+  end
+end

data/lib/redis-copy/core_ext.rb

@@ -0,0 +1,3 @@
+# encoding: utf-8
+
+require_relative 'core_ext/string'

data/lib/redis-copy/key-emitter.rb

@@ -0,0 +1,45 @@
+# encoding: utf-8
+
+module RedisCopy
+  # A Key emitter emits keys.
+  # This is built to be an abstraction on top of
+  # redis.keys('*') (implemented by RedisCopy::KeyEmitter::Default),
+  # but should allow smarter implementations to be built that can handle
+  # billion-key dbs without blocking on IO.
+  module KeyEmitter
+    def self.load(redis, ui, options = {})
+      key_emitter = options.fetch(:key_emitter, :default)
+      const_name = key_emitter.to_s.camelize
+      require "redis-copy/key-emitter/#{key_emitter}" unless const_defined?(const_name)
+      const_get(const_name).new(redis, ui, options)
+    end
+
+    # @param redis [Redis]
+    # @param options [Hash<Symbol:String>]
+    def initialize(redis, ui, options = {})
+      @redis    = redis
+      @ui       = ui
+      @options  = options
+    end
+
+    # @return [Enumerable<String>]
+    def keys
+      return super if defined?(super)
+      raise NotImplementedError
+    end
+
+    def to_s
+      self.class.name.demodulize.humanize
+    end
+
+    # The default strategy blindly uses `redis.keys('*')`
+    class Default
+      include KeyEmitter
+
+      def keys
+        @ui.debug "REDIS: #{@redis.client.id} KEYS *"
+        @redis.keys('*').to_enum
+      end
+    end
+  end
+end

data/lib/redis-copy/strategy.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+
+require_relative 'strategy/new'
+require_relative 'strategy/classic'
+
+module RedisCopy
+  module Strategy
+    # @param source [Redis]
+    # @param destination [Redis]
+    def self.load(source, destination, ui, options = {})
+      strategy = options.fetch(:strategy, :auto).to_sym
+      new_compatible = [source, destination].all?(&New.method(:compatible?))
+      copierklass = case strategy
+                    when :classic then Classic
+                    when :new
+                      raise ArgumentError unless new_compatible
+                      New
+                    when :auto
+                      new_compatible ? New : Classic
+                    end
+      copierklass.new(source, destination, ui, options)
+    end
+
+    # @param source [Redis]
+    # @param destination [Redis]
+    def initialize(source, destination, ui, options = {})
+      @src = source
+      @dst = destination
+      @ui  = ui
+      @opt = options.dup
+    end
+
+    def to_s
+      self.class.name.demodulize.humanize
+    end
+
+    # @param key [String]
+    # @return [Boolean]
+    def copy(key)
+      return super if defined? super
+      raise NotImplementedError
+    end
+  end
+end

data/lib/redis-copy/strategy/classic.rb

@@ -0,0 +1,55 @@
+# encoding: utf-8
+
+module RedisCopy
+  module Strategy
+    class Classic
+      include Strategy
+
+      def copy(key)
+        vtype = @src.type(key)
+        ttl = @src.ttl(key)
+
+        case vtype
+        when 'string'
+          string = @src.get(key)
+          @dst.set(key, string)
+        when "list"
+          list = @src.lrange(key, 0, -1)
+          if list.length == 0
+            # Empty list special case
+            @dst.lpush(key, '')
+            @dst.lpop(key)
+          else
+            list.each do |ele|
+              @dst.rpush(key, ele)
+            end
+          end
+        when "set"
+          set = @src.smembers(key)
+          if set.length == 0
+            # Empty set special case
+            @dst.sadd(key, '')
+            @dst.srem(key, '')
+          else
+            set.each do |ele|
+              @dst.sadd(key,ele)
+            end
+          end
+        when 'hash'
+          hash = @src.hgetall(key)
+          @dst.mapped_hmset(key, hash)
+        when 'zset'
+          vs_zset = @src.zrange(key, 0, -1, :with_scores => true)
+          sv_zset = vs_zset.map(&:reverse)
+          @dst.zadd(key, sv_zset)
+        else
+          return false
+        end
+
+        @dst.expire(key, ttl) unless ttl < 0 || vtype == 'none'
+
+        return true
+      end
+    end
+  end
+end

data/lib/redis-copy/strategy/new.rb

@@ -0,0 +1,32 @@
+# encoding: utf-8
+
+module RedisCopy
+  module Strategy
+    class New
+      include Strategy
+
+      def copy(key)
+        ttl = @src.ttl(key)
+        # TTL returns seconds, -1 means none set
+        # RESTORE ttl is in miliseconds, 0 means none set
+        translated_ttl = (ttl && ttl > 0) ? (ttl * 1000) : 0
+
+        dumped_value = @src.dump(key)
+        @dst.restore(key, translated_ttl, dumped_value)
+
+        return true
+      rescue Redis::CommandError => error
+        @ui.debug("ERROR: #{error}")
+        return false
+      end
+
+      def self.compatible?(redis)
+        maj, min, *_ = redis.info['redis_version'].split('.').map(&:to_i)
+        return false unless maj >= 2
+        return false unless min >= 6
+
+        return true
+      end
+    end
+  end
+end

data/lib/redis-copy/ui.rb

@@ -0,0 +1,35 @@
+# encoding: utf-8
+
+module RedisCopy
+  module UI
+    def self.load(options = {})
+      ui = options.fetch(:ui, :auto_run)
+      const_name = ui.to_s.camelize
+      require "redis-copy/ui/#{ui}" unless const_defined?(const_name)
+      const_get(const_name).new(options)
+    end
+
+    def initialize(options)
+      @options = options
+    end
+
+    def confirm?(prompt)
+      return super if defined?(super)
+      raise NotImplementedError
+    end
+
+    def abort(message = nil)
+      return super if defined?(super)
+      raise NotImplementedError
+    end
+
+    def notify(message)
+      return super if defined?(super)
+      raise NotImplementedError
+    end
+
+    def debug(message)
+      notify(message) if @options[:debug]
+    end
+  end
+end

data/lib/redis-copy/ui/auto_run.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+module RedisCopy
+  module UI
+    class AutoRun
+      include UI
+
+      def confirm?(prompt)
+        $stderr.puts(prompt)
+        true
+      end
+      def abort(message = nil)
+        raise RuntimeError, message
+      end
+      def notify(message)
+        $stderr.puts(message)
+      end
+    end
+  end
+end

data/lib/redis-copy/ui/command_line.rb

@@ -0,0 +1,26 @@
+# encoding: utf-8
+
+module RedisCopy
+  module UI
+    class CommandLine
+      include UI
+
+      def confirm?(prompt)
+        $stderr.puts(prompt)
+        return true if @options[:yes]
+        $stderr.puts("Continue? [yN]")
+        abort unless $stdin.gets.chomp =~ /y/i
+        true
+      end
+
+      def abort(message = nil)
+        notify(['ABORTED',message].compact.join(': '))
+        exit 1
+      end
+
+      def notify(message)
+        $stderr.puts(message)
+      end
+    end
+  end
+end

data/lib/redis-copy/version.rb

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

data/redis-copy.gemspec

@@ -0,0 +1,33 @@
+# encoding: utf-8
+
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'redis-copy/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'redis-copy'
+  spec.version       = RedisCopy::VERSION
+
+  authors_and_emails = [['Ryan Biesemeyer', 'ryan@yaauie.com']]
+  # authors_and_emails = (`sh git shortlog -sne`).lines.map do |l|
+  #   (/(?<=\t)(.+) <(.+)>\z/).match(l.chomp).last(2)
+  # end.compact.map(&:to_a)
+
+  spec.authors       = authors_and_emails.map(&:first)
+  spec.email         = authors_and_emails.map(&:last)
+  spec.summary       = 'Copy the contents of one redis db to another'
+  spec.homepage      = 'https://github.com/yaauie/redis-copy'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(/^bin\//) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(/^(test|spec|features)\//)
+  spec.require_paths = ['lib']
+
+  spec.add_development_dependency 'bundler', '~> 1.3'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'rspec',   '~> 2.14'
+
+  spec.add_runtime_dependency     'redis'
+  spec.add_runtime_dependency     'activesupport'
+end

data/spec/redis-copy/key-emitter_spec.rb

@@ -0,0 +1,33 @@
+# encoding: utf-8
+require 'redis-copy'
+
+describe RedisCopy::KeyEmitter::Default do
+  let(:redis) { double }
+  let(:ui) { double }
+  let(:instance) { RedisCopy::KeyEmitter::Default.new(redis, ui)}
+  let(:connection_uri) { 'redis://12.34.56.78:9000/15' }
+
+  before(:each) do
+    redis.stub_chain('client.id').and_return(connection_uri)
+    ui.stub(:debug).with(anything)
+  end
+
+  context '#keys' do
+    let(:mock_return) { ['foo:bar', 'asdf:qwer'] }
+    before(:each) do
+      redis.should_receive(:keys).with('*').exactly(:once).and_return(mock_return)
+    end
+    context 'the result' do
+      subject { instance.keys }
+      its(:to_a) { should eq mock_return }
+    end
+    context 'the supplied ui' do
+      it 'should get a debug message' do
+        ui.should_receive(:debug).
+          with(/#{Regexp.escape(connection_uri)} KEYS \*/).
+          exactly(:once)
+        instance.keys
+      end
+    end
+  end
+end

data/spec/redis-copy/strategy_spec.rb

@@ -0,0 +1,343 @@
+# encoding: utf-8
+class RedisMultiplex < Struct.new(:source, :destination)
+  ResponseError = Class.new(RuntimeError)
+
+  def ensure_same!(&blk)
+    responses = {
+      source:      capture_result(source, &blk),
+      destination: capture_result(destination, &blk)
+    }
+    unless responses[:source] == responses[:destination]
+      raise ResponseError.new(responses.to_s)
+    end
+    case responses[:destination].first
+    when :raised then raise responses[:destination].last
+    when :returned then return responses[:destination].last
+    end
+  end
+  alias_method :both!, :ensure_same!
+
+  def both(&blk)
+    both!(&blk)
+    true
+  rescue ResponseError
+    false
+  end
+
+  def capture_result(redis, &block)
+    return [:returned, block.call(redis)]
+  rescue Object => exception
+    return [:raised, exception]
+  end
+end
+
+shared_examples_for(RedisCopy::Strategy) do
+  let(:ui) { double.as_null_object }
+  let(:strategy) { strategy_class.new(source, destination, ui)}
+  let(:source) { Redis.new(db: 14) }
+  let(:destination) { Redis.new(db: 15) }
+  let(:multiplex) { RedisMultiplex.new(source, destination) }
+  let(:key) { rand(16**128).to_s(16) }
+  after(:each) { multiplex.both { |redis| redis.del(key) } }
+
+  context '#copy' do
+    context 'string' do
+      let(:source_string) { rand(16**256).to_s(16) }
+      before(:each) { source.set(key, source_string) }
+      [true,false].each do |with_expiry|
+        context "with_expiry(#{with_expiry})" do
+          before(:each) { source.expire(key, 100) } if with_expiry
+          context 'before' do
+            context 'source' do
+              subject { source.get(key) }
+              it { should_not be_nil }
+              it { should eq source_string }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.get(key) }
+              it { should be_nil }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq -1 }
+              end
+            end
+          end
+
+          context 'after' do
+            before(:each) { strategy.copy(key) }
+            context 'source' do
+              subject { source.get(key) }
+              it { should_not be_nil }
+              it { should eq source_string }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.get(key) }
+              it { should_not be_nil }
+              it { should eq source_string }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+          end
+        end
+      end
+    end
+
+    context 'list' do
+      let(:source_list) do
+        %w(foo bar baz buz bingo jango)
+      end
+      before(:each) { source_list.each{|x| source.rpush(key, x)} }
+      [true,false].each do |with_expiry|
+        context "with_expiry(#{with_expiry})" do
+          before(:each) { source.expire(key, 100) } if with_expiry
+          context 'before' do
+            context 'source' do
+              subject { source.lrange(key, 0, -1) }
+              it { should_not be_empty }
+              it { should eq source_list }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.lrange(key, 0, -1) }
+              it { should be_empty }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq -1 }
+              end
+            end
+          end
+
+          context 'after' do
+            before(:each) { strategy.copy(key) }
+            context 'source' do
+              subject { source.lrange(key, 0, -1) }
+              it { should_not be_empty }
+              it { should eq source_list }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.lrange(key, 0, -1) }
+              it { should_not be_empty }
+              it { should eq source_list }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+          end
+        end
+      end
+    end
+
+    context 'set' do
+      let(:source_list) do
+        %w(foo bar baz buz bingo jango)
+      end
+      before(:each) { source_list.each{|x| source.sadd(key, x)} }
+      [true,false].each do |with_expiry|
+        context "with_expiry(#{with_expiry})" do
+          before(:each) { source.expire(key, 100) } if with_expiry
+          context 'before' do
+            context 'source' do
+              subject { source.smembers(key) }
+              it { should_not be_empty }
+              it { should =~ source_list }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.smembers(key) }
+              it { should be_empty }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq -1 }
+              end
+            end
+          end
+
+          context 'after' do
+            before(:each) { strategy.copy(key) }
+            context 'source' do
+              subject { source.smembers(key) }
+              it { should_not be_empty }
+              it { should =~ source_list }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.smembers(key) }
+              it { should_not be_empty }
+              it { should =~ source_list }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+          end
+        end
+      end
+    end
+
+    context 'hash' do
+      let(:source_hash) do
+        {
+          'foo' => 'bar',
+          'baz' => 'buz'
+        }
+      end
+      before(:each) { source.mapped_hmset(key, source_hash) }
+      [true,false].each do |with_expiry|
+        context "with_expiry(#{with_expiry})" do
+          before(:each) { source.expire(key, 100) } if with_expiry
+          context 'before' do
+            context 'source' do
+              subject { source.hgetall(key) }
+              it { should_not be_empty }
+              it { should eq source_hash }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.hgetall(key) }
+              it { should be_empty }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq -1 }
+              end
+            end
+          end
+
+          context 'after' do
+            before(:each) { strategy.copy(key) }
+            context 'source' do
+              subject { source.hgetall(key) }
+              it { should_not be_empty }
+              it { should eq source_hash }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.hgetall(key) }
+              it { should_not be_empty }
+              it { should eq source_hash }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+          end
+        end
+      end
+    end
+
+    context 'zset' do
+      let(:source_zset) do
+        {
+          'foo' => 1.0,
+          'baz' => 2.5,
+          'bar' => 1.1,
+          'buz' => 2.7
+        }
+      end
+      let(:vs_source_zset) { source_zset.to_a }
+      let(:sv_source_zset) { vs_source_zset.map(&:reverse) }
+      before(:each) { source.zadd(key, sv_source_zset) }
+      [true,false].each do |with_expiry|
+        context "with_expiry(#{with_expiry})" do
+          before(:each) { source.expire(key, 100) } if with_expiry
+          context 'before' do
+            context 'source' do
+              subject { source.zrange(key, 0, -1, :with_scores => true) }
+              it { should_not be_empty }
+              it { should =~ vs_source_zset }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.zrange(key, 0, -1, :with_scores => true) }
+              it { should be_empty }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq -1 }
+              end
+            end
+          end
+
+          context 'after' do
+            before(:each) { strategy.copy(key) }
+            context 'source' do
+              subject { source.zrange(key, 0, -1, :with_scores => true) }
+              it { should_not be_empty }
+              it { should =~ vs_source_zset }
+              context 'ttl' do
+                subject { source.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+            context 'destination' do
+              subject { destination.zrange(key, 0, -1, :with_scores => true) }
+              it { should_not be_empty }
+              it { should =~ vs_source_zset }
+              context 'ttl' do
+                subject { destination.ttl(key) }
+                it { should eq 100 } if with_expiry
+                it { should eq -1 } unless with_expiry
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+describe RedisCopy::Strategy do
+  describe :New do
+    let(:strategy_class) { RedisCopy::Strategy::New }
+    it_should_behave_like RedisCopy::Strategy
+  end
+  describe :Classic do
+    let(:strategy_class) { RedisCopy::Strategy::Classic }
+    it_should_behave_like RedisCopy::Strategy
+  end
+end

metadata

@@ -0,0 +1,151 @@
+--- !ruby/object:Gem::Specification
+name: redis-copy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Ryan Biesemeyer
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  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'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.14'
+- !ruby/object:Gem::Dependency
+  name: redis
+  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: activesupport
+  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'
+description: 
+email:
+- ryan@yaauie.com
+executables:
+- redis-copy
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/redis-copy
+- lib/redis-copy.rb
+- lib/redis-copy/cli.rb
+- lib/redis-copy/core_ext.rb
+- lib/redis-copy/key-emitter.rb
+- lib/redis-copy/strategy.rb
+- lib/redis-copy/strategy/classic.rb
+- lib/redis-copy/strategy/new.rb
+- lib/redis-copy/ui.rb
+- lib/redis-copy/ui/auto_run.rb
+- lib/redis-copy/ui/command_line.rb
+- lib/redis-copy/version.rb
+- redis-copy.gemspec
+- redis-copy_spec.rb
+- spec/redis-copy/key-emitter_spec.rb
+- spec/redis-copy/strategy_spec.rb
+homepage: https://github.com/yaauie/redis-copy
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Copy the contents of one redis db to another
+test_files:
+- spec/redis-copy/key-emitter_spec.rb
+- spec/redis-copy/strategy_spec.rb
+has_rdoc: