使用 Rails 5，如何在创建新用户后添加数据库记录？

Question

在我的应用程序中，我有帐户、用户和权限。创建新用户时，我已经自动创建了一个帐户记录，其中新用户外键设置为帐户表中的“所有者 ID”。在完成此操作的同时，我想在我的 Permissions 表中添加一条记录，在它们各自的列中设置新的 account_id 和新的 user_id，并将资源列设置为“帐户”，并将角色列设置为“所有者” ”。

这是我的模型的外观：

account.rb

class Account < ApplicationRecord
  belongs_to :owner, class_name: "User"
  has_many :permissions
end

user.rb

class User < ApplicationRecord  

  ...

  has_one :account, foreign_key: "owner_id"
  has_many :permissions

  after_initialize :set_account

  ...

  private

    def set_account
      build_account unless account.present?
    end

end

permissions.rb

class Permission < ApplicationRecord
  belongs_to :account
  belongs_to :user
end

我希望通过将“set_account”修改为以下内容至少可以填充 Permissions 表的 account_id 和 user_id 列，但我收到“未定义的局部变量或方法 'build_permission' 错误。

def set_account
  build_account and build_permission unless account.present?
end

这是我用来显示表/列的架构文件：

ActiveRecord::Schema.define(version: 2018_04_02_040606) do

  # These are extensions that must be enabled in order to support this database
  enable_extension "plpgsql"

  create_table "accounts", force: :cascade do |t|
    t.integer "hash_id"
    t.integer "owner_id"
    t.datetime "created_at", null: false
    t.datetime "updated_at", null: false
    t.index ["hash_id"], name: "index_accounts_on_hash_id", unique: true
    t.index ["owner_id"], name: "index_accounts_on_owner_id"
  end

  create_table "permissions", force: :cascade do |t|
    t.integer "account_id"
    t.integer "user_id"
    t.string "resource"
    t.string "role"
    t.datetime "created_at", null: false
    t.datetime "updated_at", null: false
    t.index ["account_id", "user_id"], name: "index_permissions_on_account_id_and_user_id", unique: true
  end

  create_table "users", force: :cascade do |t|
    t.string "email", default: "", null: false
    t.string "encrypted_password", default: "", null: false
    t.string "reset_password_token"
    t.datetime "reset_password_sent_at"
    t.datetime "remember_created_at"
    t.integer "sign_in_count", default: 0, null: false
    t.datetime "current_sign_in_at"
    t.datetime "last_sign_in_at"
    t.inet "current_sign_in_ip"
    t.inet "last_sign_in_ip"
    t.datetime "created_at", null: false
    t.datetime "updated_at", null: false
    t.jsonb "settings"
    t.index ["email"], name: "index_users_on_email", unique: true
    t.index ["reset_password_token"], name: "index_users_on_reset_password_token", unique: true
  end

  add_foreign_key "users"
end

我将不胜感激任何关于我可能遗漏的内容或我从这里到哪里让它按我想要的方式工作的见解。

Answer 1

除非创建了帐户对象，否则您无法在权限表中获取 account_id 来更新它。一种选择是让after_create 回调以设置帐户和权限。

class User < ApplicationRecord  

  ...

  has_one :account, foreign_key: "owner_id"
  has_many :permissions

  after_create :set_account_and_permission

  ...

  private

  def set_account_and_permission
    account = create_account
    permissions.create(role: 'owner', account_id: account.id, resource: 'Account')
  end

end

希望对你有帮助！