Commit 9ab20185 authored by Ash McKenzie's avatar Ash McKenzie

Merge branch 'mk/rename-methods' into 'master'

Geo: Small refactor renaming some methods

See merge request gitlab-org/gitlab!19712
parents 037dd58e 84f9311a
......@@ -47,7 +47,7 @@ module Geo
# rubocop: disable CodeReuse/ActiveRecord
def find_unsynced(batch_size:, except_file_ids: [])
attachments
.missing_file_registry
.missing_registry
.id_not_in(except_file_ids)
.limit(batch_size)
end
......@@ -56,7 +56,7 @@ module Geo
# rubocop: disable CodeReuse/ActiveRecord
def find_migrated_local(batch_size:, except_file_ids: [])
all_attachments
.inner_join_file_registry
.inner_join_registry
.with_files_stored_remotely
.id_not_in(except_file_ids)
.limit(batch_size)
......@@ -95,7 +95,7 @@ module Geo
end
def registries_for_attachments
attachments.inner_join_file_registry
attachments.inner_join_registry
end
end
end
......@@ -41,7 +41,7 @@ module Geo
# rubocop:disable CodeReuse/ActiveRecord
def find_unsynced(batch_size:, except_ids: [])
lfs_objects
.missing_file_registry
.missing_registry
.id_not_in(except_ids)
.limit(batch_size)
end
......@@ -50,7 +50,7 @@ module Geo
# rubocop:disable CodeReuse/ActiveRecord
def find_migrated_local(batch_size:, except_ids: [])
all_lfs_objects
.inner_join_lfs_object_registry
.inner_join_registry
.with_files_stored_remotely
.id_not_in(except_ids)
.limit(batch_size)
......
......@@ -17,45 +17,45 @@ module Geo
class << self
def failed
inner_join_lfs_object_registry
inner_join_registry
.merge(Geo::LfsObjectRegistry.failed)
end
def inner_join_lfs_object_registry
def inner_join_registry
join_statement =
arel_table
.join(lfs_object_registry_table, Arel::Nodes::InnerJoin)
.on(arel_table[:id].eq(lfs_object_registry_table[:lfs_object_id]))
.join(registry_table, Arel::Nodes::InnerJoin)
.on(arel_table[:id].eq(registry_table[:lfs_object_id]))
joins(join_statement.join_sources)
end
def missing_file_registry
left_outer_join_lfs_object_registry
.where(lfs_object_registry_table[:id].eq(nil))
def missing_registry
left_outer_join_registry
.where(registry_table[:id].eq(nil))
end
def missing_on_primary
inner_join_lfs_object_registry
inner_join_registry
.merge(Geo::LfsObjectRegistry.synced.missing_on_primary)
end
def synced
inner_join_lfs_object_registry
inner_join_registry
.merge(Geo::LfsObjectRegistry.synced)
end
private
def lfs_object_registry_table
def registry_table
Geo::LfsObjectRegistry.arel_table
end
def left_outer_join_lfs_object_registry
def left_outer_join_registry
join_statement =
arel_table
.join(lfs_object_registry_table, Arel::Nodes::OuterJoin)
.on(arel_table[:id].eq(lfs_object_registry_table[:lfs_object_id]))
.join(registry_table, Arel::Nodes::OuterJoin)
.on(arel_table[:id].eq(registry_table[:lfs_object_id]))
joins(join_statement.join_sources)
end
......
......@@ -15,17 +15,17 @@ module Geo
class << self
def for_model(model)
inner_join_file_registry
inner_join_registry
.where(model_id: model.id, model_type: model.class.name)
end
def inner_join_file_registry
def inner_join_registry
joins(:registry)
end
def missing_file_registry
left_outer_join_file_registry
.where(file_registry_table[:id].eq(nil))
def missing_registry
left_outer_join_registry
.where(registry_table[:id].eq(nil))
end
# Searches for a list of uploads based on the query given in `query`.
......@@ -40,15 +40,15 @@ module Geo
private
def file_registry_table
def registry_table
Geo::UploadRegistry.arel_table
end
def left_outer_join_file_registry
def left_outer_join_registry
join_statement =
arel_table
.join(file_registry_table, Arel::Nodes::OuterJoin)
.on(arel_table[:id].eq(file_registry_table[:file_id]))
.join(registry_table, Arel::Nodes::OuterJoin)
.on(arel_table[:id].eq(registry_table[:file_id]))
joins(join_statement.join_sources)
end
......
......@@ -14,10 +14,10 @@ module Gitlab
class Fdw
class LfsObjectRegistryQueryBuilder < BaseQueryBuilder
# rubocop:disable CodeReuse/ActiveRecord
def for_lfs_objects(lfs_object_ids)
def for_lfs_objects(ids)
query
.joins(fdw_inner_join_lfs_objects)
.lfs_object_id_in(lfs_object_ids)
.lfs_object_id_in(ids)
end
# rubocop:enable CodeReuse/ActiveRecord
......@@ -25,21 +25,21 @@ module Gitlab
def base
::Geo::LfsObjectRegistry
.select(lfs_registry_table[Arel.star])
.select(registry_table[Arel.star])
end
def lfs_registry_table
def registry_table
::Geo::LfsObjectRegistry.arel_table
end
def fdw_lfs_object_table
def fdw_table
::Geo::Fdw::LfsObject.arel_table
end
def fdw_inner_join_lfs_objects
lfs_registry_table
.join(fdw_lfs_object_table, Arel::Nodes::InnerJoin)
.on(lfs_registry_table[:lfs_object_id].eq(fdw_lfs_object_table[:id]))
registry_table
.join(fdw_table, Arel::Nodes::InnerJoin)
.on(registry_table[:lfs_object_id].eq(fdw_table[:id]))
.join_sources
end
end
......
......@@ -17,8 +17,8 @@ module Gitlab
query
.joins(fdw_inner_join_uploads)
.where(
fdw_upload_table[:model_id].eq(model.id)
.and(fdw_upload_table[:model_type].eq(model.class.name))
fdw_table[:model_id].eq(model.id)
.and(fdw_table[:model_type].eq(model.class.name))
)
)
end
......@@ -27,21 +27,21 @@ module Gitlab
private
def base
::Geo::UploadRegistry.select(file_registry_table[Arel.star])
::Geo::UploadRegistry.select(registry_table[Arel.star])
end
def file_registry_table
def registry_table
::Geo::UploadRegistry.arel_table
end
def fdw_upload_table
def fdw_table
::Geo::Fdw::Upload.arel_table
end
def fdw_inner_join_uploads
file_registry_table
.join(fdw_upload_table, Arel::Nodes::InnerJoin)
.on(file_registry_table[:file_id].eq(fdw_upload_table[:id]))
registry_table
.join(fdw_table, Arel::Nodes::InnerJoin)
.on(registry_table[:file_id].eq(fdw_table[:id]))
.join_sources
end
end
......
......@@ -8,7 +8,7 @@ RSpec.describe Geo::Fdw::LfsObject, :geo, type: :model do
it { is_expected.to have_many(:projects).class_name('Geo::Fdw::Project').through(:lfs_objects_projects) }
end
describe '.missing_file_registry', :geo_fdw do
describe '.missing_registry', :geo_fdw do
it "returns lfs objects that don't have a corresponding registry entry" do
missing_registry_entries = create_list(:lfs_object, 2)
......@@ -16,7 +16,7 @@ RSpec.describe Geo::Fdw::LfsObject, :geo, type: :model do
create(:geo_lfs_object_registry, lfs_object_id: lfs.id)
end
expect(described_class.missing_file_registry).to match_ids(missing_registry_entries)
expect(described_class.missing_registry).to match_ids(missing_registry_entries)
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