rails generator to create the migration

Author: senny

lib/generators/queue_classic/install_generator.rb

@@ -0,0 +1,22 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+require 'active_record'
+
+module QC
+  class InstallGenerator < Rails::Generators::Base
+    include Rails::Generators::Migration
+
+    namespace "queue_classic:install"
+    self.source_paths << File.join(File.dirname(__FILE__), 'templates')
+    desc 'Generates (but does not run) a migration to add a queue_classic table.'
+
+    def self.next_migration_number(dirname)
+      next_migration_number = current_migration_number(dirname) + 1
+      ActiveRecord::Migration.next_migration_number(next_migration_number)
+    end
+
+    def create_migration_file
+      migration_template 'add_queue_classic.rb', 'db/migrate/add_queue_classic.rb'
+    end
+  end
+end

lib/generators/queue_classic/templates/add_queue_classic.rb

@@ -0,0 +1,9 @@
+class AddQueueClassic < ActiveRecord::Migration
+  def self.up
+    QC::Setup.create
+  end
+
+  def self.down
+    QC::Setup.drop
+  end
+end