ar_dbcopy 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,5 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg

data/LICENSE

@@ -0,0 +1,22 @@
+The MIT License
+
+Copyright (c) 2009 Michael Siebert <siebertm85@googlemail.com>
+
+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.rdoc

@@ -0,0 +1,39 @@
+= AR_DBCopy
+
+Simple script to copy data from one database to another.
+This script can be used to migrate from one Database system
+to another one, e.q. when the SQL dumps are not compatible.
+Database agnosticism is achieved by using ActiveRecord.
+
+The script just defines 2 connections, looks at the tables,
+creates ActiveRecord model classes on-the-fly and copies over
+the data.
+
+Configure the databases using the database.yml.example as a
+template (the format is the same as Rails's database.yml)
+
+Then, run the script. The script only copies data, not structure,
+so make sure that you created the schema  and all tables in
+the destination database already (use rake db:schema:load)
+
+== Example
+
+    ARDBCopy.new("database.yml", :copy_schema => true).run!
+
+This software was written in less than an hour. So it may
+contain bugs!
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but
+  bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Author: Copyright (c) 2009 Michael Siebert <siebertm85@googlemail.com>

data/Rakefile

@@ -0,0 +1,60 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "ar_dbcopy"
+    gem.summary = %Q{copy data from one dbms to another via active_record}
+    gem.description = %Q{copy data from one dbms to another via active_record}
+    gem.email = "siebertm85@googlemail.com"
+    gem.homepage = "http://github.com/siebertm/ar_dbcopy"
+    gem.authors = ["Michael Siebert"]
+    gem.add_development_dependency "thoughtbot-shoulda"
+    gem.add_dependency "activerecord"
+    gem.require_paths = ['lib']
+    gem.executables = ["ar_dbcopy"]
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/*_test.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION')
+    version = File.read('VERSION')
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "ar_dbcopy #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/ar_dbcopy.gemspec

@@ -0,0 +1,60 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{ar_dbcopy}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Michael Siebert"]
+  s.date = %q{2009-10-13}
+  s.default_executable = %q{ar_dbcopy}
+  s.description = %q{copy data from one dbms to another via active_record}
+  s.email = %q{siebertm85@googlemail.com}
+  s.executables = ["ar_dbcopy"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "ar_dbcopy.gemspec",
+     "bin/ar_dbcopy",
+     "database.yml.example",
+     "lib/ar_dbcopy.rb",
+     "test/ar_dbcopy_test.rb",
+     "test/test_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/siebertm/ar_dbcopy}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{copy data from one dbms to another via active_record}
+  s.test_files = [
+    "test/ar_dbcopy_test.rb",
+     "test/test_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+      s.add_runtime_dependency(%q<activerecord>, [">= 0"])
+    else
+      s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+      s.add_dependency(%q<activerecord>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    s.add_dependency(%q<activerecord>, [">= 0"])
+  end
+end

data/bin/ar_dbcopy

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'ar_dbcopy'
+
+class ARDBCopy::Application
+  def self.run!
+    @options = {}
+
+    @opts = OptionParser.new do |opts|
+      opts.banner = "Usage: #{$0} [options] database.yml"
+
+      opts.separator "copy data between two databases defined in the given database_yaml"
+      opts.separator "uses the databases 'source' and 'target'"
+
+      opts.on "-c", "--with-schema", "also copy the schema to the target database" do |arg|
+        @options[:copy_schema] = arg
+      end
+
+      opts.on( '-h', '--help', 'Display this screen' ) do
+        puts opts
+        exit
+      end
+    end
+
+    @opts.parse!
+
+    unless ARGV[0]
+      puts @opts
+      exit 1
+    end
+
+    ARDBCopy.new(ARGV[0], @options).run!
+  end
+end
+
+ARDBCopy::Application.run!

data/database.yml.example

@@ -0,0 +1,13 @@
+source:
+  adapter: postgresql
+  host: localhost
+  username:
+  password:
+  database: test
+
+target:
+  adapter: mysql
+  host: localhost
+  username:
+  password:
+  database: test

data/lib/ar_dbcopy.rb

@@ -0,0 +1,90 @@
+#!/usr/bin/env ruby
+#
+# Simple script to copy data from one database to another.
+# This script can be used to migrate from one Database system
+# to another one, e.q. when the SQL dumps are not compatible.
+# Database agnosticism is achieved by using ActiveRecord.
+#
+# The script just defines 2 connections, looks at the tables,
+# creates ActiveRecord model classes on-the-fly and copies over
+# the data.
+#
+# Configure the databases using the database.yml.example as a
+# template (the format is the same as Rails's database.yml)
+#
+# Then, run the script. The script only copies data, not structure,
+# so make sure that you created the schema  and all tables in
+# the destination database already (use rake db:schema:load)
+#
+require 'active_record'
+
+class ARDBCopy
+  class SourceDB < ActiveRecord::Base; end
+  class TargetDB < ActiveRecord::Base; end
+
+  def initialize(config_file, opts={})
+    @copy_schema = opts[:copy_schema]
+
+    config = YAML.load_file(config_file)
+
+    ActiveRecord::Base.logger ||= Logger.new(nil)
+
+    SourceDB.establish_connection(config["source"])
+    ActiveRecord::Base.establish_connection(config["target"])
+
+    @tables = SourceDB.connection.tables.reject { |m| m == "schema_migrations" }
+  end
+
+  def run!
+    copy_schema if @copy_schema
+    copy_data
+  end
+
+  def copy_schema
+    io = StringIO.new
+
+    ActiveRecord::SchemaDumper.dump(SourceDB.connection, io) # dump the schema from the source database
+    io.rewind
+
+    eval(io.read)
+  end
+
+  def copy_data
+    @tables.each do |table_name|
+      source_model = Class.new(SourceDB) do
+        set_inheritance_column(:not_sti)
+        set_table_name table_name
+      end
+
+      dest_model = Class.new(TargetDB) do
+        set_inheritance_column(:not_sti)
+        set_table_name table_name
+      end
+
+      dest_model.delete_all
+
+      puts "Copying #{table_name} (#{source_model.count} lines)..."
+
+      i = 0
+      source_model.find_in_batches(:batch_size => 10_000) do |src_batch|
+        dest_model.transaction do
+          src_batch.each do |src_inst|
+            dst_inst = dest_model.new(src_inst.attributes)
+            dst_inst.id = src_inst.id
+            dst_inst.save!
+            i += 1
+          end
+
+          puts i
+        end
+      end
+    end
+  end
+end
+
+if __FILE__ == $0
+
+  copy = ARDBcopy.new(ARGV[0])
+  copy.copy_schema
+  copy.copy_data
+end

data/test/ar_dbcopy_test.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class ARDbcopyTest < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'ar-dbcopy'
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,87 @@
+--- !ruby/object:Gem::Specification 
+name: ar_dbcopy
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Michael Siebert
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-13 00:00:00 +02:00
+default_executable: ar_dbcopy
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: activerecord
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: copy data from one dbms to another via active_record
+email: siebertm85@googlemail.com
+executables: 
+- ar_dbcopy
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- ar_dbcopy.gemspec
+- bin/ar_dbcopy
+- database.yml.example
+- lib/ar_dbcopy.rb
+- test/ar_dbcopy_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: http://github.com/siebertm/ar_dbcopy
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: copy data from one dbms to another via active_record
+test_files: 
+- test/ar_dbcopy_test.rb
+- test/test_helper.rb