Rails tạo hay cập nhật phép thuật?

Tôi có một lớp được gọi là CachedObjectlưu trữ các đối tượng được tuần tự hóa chung được lập chỉ mục bằng khóa. Tôi muốn lớp này triển khai một create_or_updatephương thức. Nếu một đối tượng được tìm thấy, nó sẽ cập nhật nó, nếu không nó sẽ tạo một đối tượng mới.

Có cách nào để thực hiện việc này trong Rails hay tôi phải viết phương thức của riêng mình?

Đường ray 6

Rails 6 đã thêm một upsertvà upsert_allcác phương thức cung cấp chức năng này.

Model.upsert(column_name: value)

[upert] Nó không khởi tạo bất kỳ mô hình nào cũng như không kích hoạt các cuộc gọi lại hoặc xác nhận Active Record.

Đường ray 5, 4 và 3

Không phải nếu bạn đang tìm kiếm kiểu câu lệnh "upert" (nơi cơ sở dữ liệu thực hiện cập nhật hoặc câu lệnh chèn trong cùng một thao tác). Ngoài ra, Rails và ActiveRecord không có tính năng này. Bạn có thể sử dụng upsert đá quý, tuy nhiên.

Nếu không, bạn có thể sử dụng: find_or_initialize_byhoặc find_or_create_by, cung cấp chức năng tương tự, mặc dù với chi phí của một lần truy cập cơ sở dữ liệu bổ sung, mà trong hầu hết các trường hợp, hầu như không phải là một vấn đề. Vì vậy, trừ khi bạn có mối quan tâm nghiêm trọng về hiệu suất, tôi sẽ không sử dụng đá quý.

Ví dụ: nếu không tìm thấy người dùng nào có tên "Roger", một phiên bản người dùng mới sẽ được khởi tạo với bộ của nó namethành "Roger".

user = User.where(name: "Roger").first_or_initialize
user.email = "email@example.com"
user.save

Ngoài ra, bạn có thể sử dụng find_or_initialize_by.

user = User.find_or_initialize_by(name: "Roger")

Trong Rails 3.

user = User.find_or_initialize_by_name("Roger")
user.email = "email@example.com"
user.save

Bạn có thể sử dụng một khối, nhưng khối chỉ chạy nếu bản ghi là mới .

User.where(name: "Roger").first_or_initialize do |user|
  # this won't run if a user with name "Roger" is found
  user.save 
end

User.find_or_initialize_by(name: "Roger") do |user|
  # this also won't run if a user with name "Roger" is found
  user.save
end

Nếu bạn muốn sử dụng một khối bất kể độ bền của bản ghi, hãy sử dụng taptrên kết quả:

User.where(name: "Roger").first_or_initialize.tap do |user|
  user.email = "email@example.com"
  user.save
end

Trong Rails 4, bạn có thể thêm vào một mô hình cụ thể:

def self.update_or_create(attributes)
  assign_or_new(attributes).save
end

def self.assign_or_new(attributes)
  obj = first || new
  obj.assign_attributes(attributes)
  obj
end

và sử dụng nó như

User.where(email: "a@b.com").update_or_create(name: "Mr A Bbb")

Hoặc nếu bạn muốn thêm các phương thức này vào tất cả các mô hình được đặt trong bộ khởi tạo:

module ActiveRecordExtras
  module Relation
    extend ActiveSupport::Concern

    module ClassMethods
      def update_or_create(attributes)
        assign_or_new(attributes).save
      end

      def update_or_create!(attributes)
        assign_or_new(attributes).save!
      end

      def assign_or_new(attributes)
        obj = first || new
        obj.assign_attributes(attributes)
        obj
      end
    end
  end
end

ActiveRecord::Base.send :include, ActiveRecordExtras::Relation

Thêm cái này vào mô hình của bạn:

def self.update_or_create_by(args, attributes)
  obj = self.find_or_create_by(args)
  obj.update(attributes)
  return obj
end

Với điều đó, bạn có thể:

User.update_or_create_by({name: 'Joe'}, attributes)

Điều kỳ diệu bạn đang tìm kiếm đã được thêm vào Rails 6 Bây giờ bạn có thể nâng cấp (cập nhật hoặc chèn). Để sử dụng một bản ghi:

Model.upsert(column_name: value)

Đối với nhiều bản ghi, hãy sử dụng upsert_all :

Model.upsert_all(column_name: value, unique_by: :column_name)

Ghi chú :

• Cả hai phương pháp đều không kích hoạt các lệnh gọi lại hoặc xác nhận Active Record
• unique_by => Chỉ PostgreSQL và SQLite

Câu hỏi cũ nhưng ném giải pháp của tôi vào vòng cho hoàn chỉnh. Tôi cần cái này khi tôi cần một tìm cụ thể nhưng một bản tạo khác nếu nó không tồn tại.

def self.find_by_or_create_with(args, attributes) # READ CAREFULLY! args for finding, attributes for creating!
        obj = self.find_or_initialize_by(args)
        return obj if obj.persisted?
        return obj if obj.update_attributes(attributes) 
end

Bạn có thể làm điều đó trong một câu lệnh như sau:

CachedObject.where(key: "the given key").first_or_create! do |cached|
   cached.attribute1 = 'attribute value'
   cached.attribute2 = 'attribute value'
end

Phần tiếp theo đá quý thêm một phương thức update_or_create dường như thực hiện những gì bạn đang tìm kiếm.