여러 데이터베이스 객체 업데이트
여러 데이터베이스 객체를 하나 이상의 열에 대한 새로운 값으로 업데이트할 수 있습니다.
한 가지 방법은 Relation#update_all
을 사용하는 것입니다:
user.issues.open.update_all(due_date: 7.days.from_now) # (1)
user.issues.update_all('relative_position = relative_position + 1') # (2)
업데이트를 정적 값(1)이나 계산(2)으로 표현할 수 없는 경우,
UPDATE FROM
을 사용하여 단일 쿼리에서 여러 행을 서로 다른 값으로 업데이트해야 함을 표현하세요.
임시 테이블 또는 공통 테이블 표현식(CTE)을 생성하고 이를 업데이트의 소스로 사용하세요:
with updates(obj_id, new_title, new_weight) as (
values (1 :: integer, '매우 어려운 이슈' :: text, 8 :: integer),
(2, '매우 쉬운 이슈', 1)
)
update issues
set title = new_title, weight = new_weight
from updates
where id = obj_id
이것은 ActiveRecord로 표현할 수 없으며, Arel로 드롭다운할 수 없습니다.
왜냐하면 UpdateManager
는 update from
을 지원하지 않기 때문입니다.
그러나 우리는 이러한 종류의 업데이트를 생성하는 데 도움이 되는 추상화를 제공합니다: Gitlab::Database::BulkUpdate
.
이 추상화는 이전 예와 같은 쿼리를 구성하며 SQL 인젝션을 피하기 위해 바인딩 매개변수를 사용합니다.
사용법
Gitlab::Database::BulkUpdate
를 사용하기 위해 필요한 사항은 다음과 같습니다:
- 업데이트할 열 목록.
- 해당 객체(또는 ID)에서 설정할 새 값으로의 매핑.
- 각 객체의 테이블을 결정하는 방법.
예를 들어, object.class.table_name
을 호출하여 테이블을 결정하는 방식으로 예제 쿼리를 표현할 수 있습니다:
issue_a = Issue.find(..)
issue_b = Issue.find(..)
# 하나의 쿼리를 발행합니다:
::Gitlab::Database::BulkUpdate.execute(%i[title weight], {
issue_a => { title: '매우 어려운 이슈', weight: 8 },
issue_b => { title: '매우 쉬운 이슈', weight: 1 }
})
업데이트가 모두 이치에 맞다면 이질적인 객체 세트도 전달할 수 있습니다:
issue_a = Issue.find(..)
issue_b = Issue.find(..)
merge_request = MergeRequest.find(..)
# 두 개의 쿼리를 발행합니다
::Gitlab::Database::BulkUpdate.execute(%i[title], {
issue_a => { title: 'A' },
issue_b => { title: 'B' },
merge_request => { title: 'B' }
})
객체가 올바른 모델 클래스를 반환하지 않는 경우(예: 유니온의 일부인 경우) 블록에서 모델 클래스를 명시적으로 지정하세요:
bazzes = params
objects = Foo.from_union([
Foo.select("id, 'foo' as object_type").where(quux: true),
Bar.select("id, 'bar' as object_type").where(wibble: true)
])
# 이 시점에서 모든 객체는 Foo의 인스턴스입니다. 바 테이블의 객체도 포함됩니다.
mapping = objects.to_h { |obj| [obj, bazzes[obj.id]] }
# 최대 2개의 쿼리를 발행합니다
::Gitlab::Database::BulkUpdate.execute(%i[baz], mapping) do |obj|
obj.object_type.constantize
end
주의사항
이 도구는 매우 저수준이며, 원시 열 값에 직접 작동합니다. 이를 구현할 경우 다음 문제를 고려해야 합니다:
-
열거형 및 상태 필드는 그 기본 표현으로 변환되어야 합니다.
-
중첩된 연관관계는 지원되지 않습니다.
-
검증 또는 후크가 호출되지 않습니다.