Commit 84f9311a authored by Michael Kozono's avatar Michael Kozono

Refactor Upload methods

Shorten names that are already unambiguous.
parent 9967fc58
...@@ -47,7 +47,7 @@ module Geo ...@@ -47,7 +47,7 @@ module Geo
# rubocop: disable CodeReuse/ActiveRecord # rubocop: disable CodeReuse/ActiveRecord
def find_unsynced(batch_size:, except_file_ids: []) def find_unsynced(batch_size:, except_file_ids: [])
attachments attachments
.missing_file_registry .missing_registry
.id_not_in(except_file_ids) .id_not_in(except_file_ids)
.limit(batch_size) .limit(batch_size)
end end
...@@ -56,7 +56,7 @@ module Geo ...@@ -56,7 +56,7 @@ module Geo
# rubocop: disable CodeReuse/ActiveRecord # rubocop: disable CodeReuse/ActiveRecord
def find_migrated_local(batch_size:, except_file_ids: []) def find_migrated_local(batch_size:, except_file_ids: [])
all_attachments all_attachments
.inner_join_file_registry .inner_join_registry
.with_files_stored_remotely .with_files_stored_remotely
.id_not_in(except_file_ids) .id_not_in(except_file_ids)
.limit(batch_size) .limit(batch_size)
...@@ -95,7 +95,7 @@ module Geo ...@@ -95,7 +95,7 @@ module Geo
end end
def registries_for_attachments def registries_for_attachments
attachments.inner_join_file_registry attachments.inner_join_registry
end end
end end
end end
...@@ -15,17 +15,17 @@ module Geo ...@@ -15,17 +15,17 @@ module Geo
class << self class << self
def for_model(model) def for_model(model)
inner_join_file_registry inner_join_registry
.where(model_id: model.id, model_type: model.class.name) .where(model_id: model.id, model_type: model.class.name)
end end
def inner_join_file_registry def inner_join_registry
joins(:registry) joins(:registry)
end end
def missing_file_registry def missing_registry
left_outer_join_file_registry left_outer_join_registry
.where(file_registry_table[:id].eq(nil)) .where(registry_table[:id].eq(nil))
end end
# Searches for a list of uploads based on the query given in `query`. # Searches for a list of uploads based on the query given in `query`.
...@@ -40,15 +40,15 @@ module Geo ...@@ -40,15 +40,15 @@ module Geo
private private
def file_registry_table def registry_table
Geo::UploadRegistry.arel_table Geo::UploadRegistry.arel_table
end end
def left_outer_join_file_registry def left_outer_join_registry
join_statement = join_statement =
arel_table arel_table
.join(file_registry_table, Arel::Nodes::OuterJoin) .join(registry_table, Arel::Nodes::OuterJoin)
.on(arel_table[:id].eq(file_registry_table[:file_id])) .on(arel_table[:id].eq(registry_table[:file_id]))
joins(join_statement.join_sources) joins(join_statement.join_sources)
end end
......
...@@ -17,8 +17,8 @@ module Gitlab ...@@ -17,8 +17,8 @@ module Gitlab
query query
.joins(fdw_inner_join_uploads) .joins(fdw_inner_join_uploads)
.where( .where(
fdw_upload_table[:model_id].eq(model.id) fdw_table[:model_id].eq(model.id)
.and(fdw_upload_table[:model_type].eq(model.class.name)) .and(fdw_table[:model_type].eq(model.class.name))
) )
) )
end end
...@@ -27,21 +27,21 @@ module Gitlab ...@@ -27,21 +27,21 @@ module Gitlab
private private
def base def base
::Geo::UploadRegistry.select(file_registry_table[Arel.star]) ::Geo::UploadRegistry.select(registry_table[Arel.star])
end end
def file_registry_table def registry_table
::Geo::UploadRegistry.arel_table ::Geo::UploadRegistry.arel_table
end end
def fdw_upload_table def fdw_table
::Geo::Fdw::Upload.arel_table ::Geo::Fdw::Upload.arel_table
end end
def fdw_inner_join_uploads def fdw_inner_join_uploads
file_registry_table registry_table
.join(fdw_upload_table, Arel::Nodes::InnerJoin) .join(fdw_table, Arel::Nodes::InnerJoin)
.on(file_registry_table[:file_id].eq(fdw_upload_table[:id])) .on(registry_table[:file_id].eq(fdw_table[:id]))
.join_sources .join_sources
end end
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment