1

我有一个 Rails 5 应用程序，它对三个数据库进行了单独的迁移。我正在添加一个新数据库。我通常会做一个rails g model field1:type1 field2:type2 field3:type.... 来创建具有所需字段的数据库表。我想创建生成模型语句，它将为所需的数据库生成模型并将其放在相应的 db/migrate 文件夹中。

我为 lib/generators 中的其他数据库提供了自定义数据库迁移生成器。这是自定义迁移生成器的示例。

lib/generators/stats_migration_generator.rb

require 'rails/generators/active_record/migration/migration_generator'

class StatsMigrationGenerator < ActiveRecord::Generators::MigrationGenerator
  source_root File.join(File.dirname(ActiveRecord::Generators::MigrationGenerator.instance_method(:create_migration_file).source_location.first), "templates")

  def create_migration_file
    set_local_assigns!
    validate_file_name!
    migration_template @migration_template, "db_stats/migrate/#{file_name}.rb"
  end
end

我假设我需要为每个附加数据库创建 lib/generators/mydb_model_generator.rb 或任何正确的文件名结构。在查看 GitHub 中的model_generator.rb文件后，我想我可以做这样的事情。

lib/generators/stats_model_generator.rb

require 'rails/generators/active_record/model/model_generator'

class StatsModelGenerator < ActiveRecord::Generators::ModelGenerator
  source_root File.join(File.dirname(ActiveRecord::Generators::ModelGenerator.instance_method(:create_migration_file).source_location.first), "templates")

  def create_migration_file
    set_local_assigns!
    validate_file_name!
    migration_template @migration_template, "db_stats/migrate/#{file_name}.rb"
  end
end

以下是我在查看 Rails 的 git 存储库、关于创建和自定义生成器的 Rails 指南以及 Ruby on Rails API 文档后提出的问题：

  1. 我如何找到我需要的东西？
  2. 如何找到我需要覆盖 db/migrate 文件夹名称的方法？
  3. 如何找出如何为模型生成器命名 rb 文件？

我在尝试创建模型生成器时的假设是否正确？

4

1 回答 1

1

好吧，我没想到我会这么快找到解决方案。我又看了一遍 GitHub 上的model_generator.rb并有一个恍惚的时刻。我意识到我只需要用指向我想要的 db migrate 文件夹的版本替换默认生成器中的方法。这是我想出的解决方案。

require 'rails/generators/active_record/model/model_generator'

class MydbModelGenerator < ActiveRecord::Generators::ModelGenerator
  source_root File.join(File.dirname(ActiveRecord::Generators::ModelGenerator.instance_method(:create_migration_file).source_location.first), "templates")

  def create_migration_file
    return unless options[:migration] && options[:parent].nil?
    attributes.each { |a| a.attr_options.delete(:index) if a.reference? && !a.has_index? } if options[:indexes] == false
    migration_template "../../migration/templates/create_table_migration.rb", File.join("db_mydb/migrate", "create_#{table_name}.rb")
  end
end

现在我可以做rails g mydb_model field1:type1 field2:type2 field3:type，它会在正确的迁移文件夹中创建模型迁移文件。

于 2017-04-11T21:08:00.610 回答