Commit 02ef45a2 authored by Heinrich Lee Yu's avatar Heinrich Lee Yu

Merge branch '325869-improve-maven-packages-group-level-api' into 'master'

Improvements to the Maven API files endpoints [RUN ALL RSPEC] [RUN AS-IF-FOSS]

See merge request gitlab-org/gitlab!57600
parents 743959dd 5fcc203d
...@@ -3,13 +3,15 @@ ...@@ -3,13 +3,15 @@
module Packages module Packages
module Maven module Maven
class PackageFinder class PackageFinder
attr_reader :path, :current_user, :project, :group include ::Packages::FinderHelper
include Gitlab::Utils::StrongMemoize
def initialize(path, current_user, project: nil, group: nil) def initialize(path, current_user, project: nil, group: nil, order_by_package_file: false)
@path = path @path = path
@current_user = current_user @current_user = current_user
@project = project @project = project
@group = group @group = group
@order_by_package_file = order_by_package_file
end end
def execute def execute
...@@ -23,9 +25,9 @@ module Packages ...@@ -23,9 +25,9 @@ module Packages
private private
def base def base
if project if @project
packages_for_a_single_project packages_for_a_single_project
elsif group elsif @group
packages_for_multiple_projects packages_for_multiple_projects
else else
::Packages::Package.none ::Packages::Package.none
...@@ -33,8 +35,13 @@ module Packages ...@@ -33,8 +35,13 @@ module Packages
end end
def packages_with_path def packages_with_path
matching_packages = base.only_maven_packages_with_path(path, use_cte: group.present?) matching_packages = base.only_maven_packages_with_path(@path, use_cte: @group.present?)
matching_packages = matching_packages.order_by_package_file if versionless_package?(matching_packages)
if group_level_improvements?
matching_packages = matching_packages.order_by_package_file if @order_by_package_file
else
matching_packages = matching_packages.order_by_package_file if versionless_package?(matching_packages)
end
matching_packages matching_packages
end end
...@@ -48,19 +55,29 @@ module Packages ...@@ -48,19 +55,29 @@ module Packages
# Produces a query that retrieves packages from a single project. # Produces a query that retrieves packages from a single project.
def packages_for_a_single_project def packages_for_a_single_project
project.packages @project.packages
end end
# Produces a query that retrieves packages from multiple projects that # Produces a query that retrieves packages from multiple projects that
# the current user can view within a group. # the current user can view within a group.
def packages_for_multiple_projects def packages_for_multiple_projects
::Packages::Package.for_projects(projects_visible_to_current_user) if group_level_improvements?
packages_visible_to_user(@current_user, within_group: @group)
else
::Packages::Package.for_projects(projects_visible_to_current_user)
end
end end
# Returns the projects that the current user can view within a group. # Returns the projects that the current user can view within a group.
def projects_visible_to_current_user def projects_visible_to_current_user
group.all_projects @group.all_projects
.public_or_visible_to_user(current_user) .public_or_visible_to_user(@current_user)
end
def group_level_improvements?
strong_memoize(:group_level_improvements) do
Feature.enabled?(:maven_packages_group_level_improvements)
end
end end
end end
end end
......
...@@ -136,7 +136,9 @@ class Packages::Package < ApplicationRecord ...@@ -136,7 +136,9 @@ class Packages::Package < ApplicationRecord
after_commit :update_composer_cache, on: :destroy, if: -> { composer? } after_commit :update_composer_cache, on: :destroy, if: -> { composer? }
def self.for_projects(projects) def self.for_projects(projects)
return none unless projects.any? unless Feature.enabled?(:maven_packages_group_level_improvements)
return none unless projects.any?
end
where(project_id: projects) where(project_id: projects)
end end
......
---
name: maven_packages_group_level_improvements
introduced_by_url: https://gitlab.com/gitlab-org/gitlab/-/merge_requests/57600
rollout_issue_url: https://gitlab.com/gitlab-org/gitlab/-/issues/326099
milestone: '13.11'
type: development
group: group::package
default_enabled: false
...@@ -77,6 +77,22 @@ module API ...@@ -77,6 +77,22 @@ module API
request.head? && request.head? &&
file.fog_credentials[:provider] == 'AWS' file.fog_credentials[:provider] == 'AWS'
end end
def fetch_package(file_name:, project: nil, group: nil)
order_by_package_file = false
if Feature.enabled?(:maven_packages_group_level_improvements)
order_by_package_file = file_name.include?(::Packages::Maven::Metadata.filename) &&
!params[:path].include?(::Packages::Maven::FindOrCreatePackageService::SNAPSHOT_TERM)
end
::Packages::Maven::PackageFinder.new(
params[:path],
current_user,
project: project,
group: group,
order_by_package_file: order_by_package_file
).execute!
end
end end
desc 'Download the maven package file at instance level' do desc 'Download the maven package file at instance level' do
...@@ -97,8 +113,7 @@ module API ...@@ -97,8 +113,7 @@ module API
authorize_read_package!(project) authorize_read_package!(project)
package = ::Packages::Maven::PackageFinder package = fetch_package(file_name: file_name, project: project)
.new(params[:path], current_user, project: project).execute!
package_file = ::Packages::PackageFileFinder package_file = ::Packages::PackageFileFinder
.new(package, file_name).execute! .new(package, file_name).execute!
...@@ -133,8 +148,7 @@ module API ...@@ -133,8 +148,7 @@ module API
not_found!('Group') unless can?(current_user, :read_group, group) not_found!('Group') unless can?(current_user, :read_group, group)
package = ::Packages::Maven::PackageFinder package = fetch_package(file_name: file_name, group: group)
.new(params[:path], current_user, group: group).execute!
authorize_read_package!(package.project) authorize_read_package!(package.project)
...@@ -171,8 +185,7 @@ module API ...@@ -171,8 +185,7 @@ module API
file_name, format = extract_format(params[:file_name]) file_name, format = extract_format(params[:file_name])
package = ::Packages::Maven::PackageFinder package = fetch_package(file_name: file_name, project: user_project)
.new(params[:path], current_user, project: user_project).execute!
package_file = ::Packages::PackageFileFinder package_file = ::Packages::PackageFileFinder
.new(package, file_name).execute! .new(package, file_name).execute!
......
...@@ -11,7 +11,8 @@ RSpec.describe ::Packages::Maven::PackageFinder do ...@@ -11,7 +11,8 @@ RSpec.describe ::Packages::Maven::PackageFinder do
let(:param_path) { nil } let(:param_path) { nil }
let(:param_project) { nil } let(:param_project) { nil }
let(:param_group) { nil } let(:param_group) { nil }
let(:finder) { described_class.new(param_path, user, project: param_project, group: param_group) } let(:param_order_by_package_file) { false }
let(:finder) { described_class.new(param_path, user, project: param_project, group: param_group, order_by_package_file: param_order_by_package_file) }
before do before do
group.add_developer(user) group.add_developer(user)
...@@ -46,7 +47,23 @@ RSpec.describe ::Packages::Maven::PackageFinder do ...@@ -46,7 +47,23 @@ RSpec.describe ::Packages::Maven::PackageFinder do
context 'within a group' do context 'within a group' do
let(:param_group) { group } let(:param_group) { group }
it_behaves_like 'handling valid and invalid paths' context 'with maven_packages_group_level_improvements enabled' do
before do
stub_feature_flags(maven_packages_group_level_improvements: true)
expect(finder).to receive(:packages_visible_to_user).with(user, within_group: group).and_call_original
end
it_behaves_like 'handling valid and invalid paths'
end
context 'with maven_packages_group_level_improvements disabled' do
before do
stub_feature_flags(maven_packages_group_level_improvements: false)
expect(finder).not_to receive(:packages_visible_to_user)
end
it_behaves_like 'handling valid and invalid paths'
end
end end
context 'across all projects' do context 'across all projects' do
...@@ -76,7 +93,39 @@ RSpec.describe ::Packages::Maven::PackageFinder do ...@@ -76,7 +93,39 @@ RSpec.describe ::Packages::Maven::PackageFinder do
create(:package_file, :xml, package: package2) create(:package_file, :xml, package: package2)
end end
it { is_expected.to eq(package2) } context 'with maven_packages_group_level_improvements enabled' do
before do
stub_feature_flags(maven_packages_group_level_improvements: true)
expect(finder).not_to receive(:versionless_package?)
end
context 'without order by package file' do
it { is_expected.to eq(package3) }
end
context 'with order by package file' do
let(:param_order_by_package_file) { true }
it { is_expected.to eq(package2) }
end
end
context 'with maven_packages_group_level_improvements disabled' do
before do
stub_feature_flags(maven_packages_group_level_improvements: false)
expect(finder).to receive(:versionless_package?).and_call_original
end
context 'without order by package file' do
it { is_expected.to eq(package2) }
end
context 'with order by package file' do
let(:param_order_by_package_file) { true }
it { is_expected.to eq(package2) }
end
end
end end
end end
end end
......
...@@ -99,6 +99,34 @@ RSpec.describe Packages::Package, type: :model do ...@@ -99,6 +99,34 @@ RSpec.describe Packages::Package, type: :model do
end end
end end
describe '.for_projects' do
let_it_be(:package1) { create(:maven_package) }
let_it_be(:package2) { create(:maven_package) }
let_it_be(:package3) { create(:maven_package) }
let(:projects) { ::Project.id_in([package1.project_id, package2.project_id]) }
subject { described_class.for_projects(projects.select(:id)) }
it 'returns package1 and package2' do
expect(projects).not_to receive(:any?)
expect(subject).to match_array([package1, package2])
end
context 'with maven_packages_group_level_improvements disabled' do
before do
stub_feature_flags(maven_packages_group_level_improvements: false)
end
it 'returns package1 and package2' do
expect(projects).to receive(:any?).and_call_original
expect(subject).to match_array([package1, package2])
end
end
end
describe 'validations' do describe 'validations' do
subject { build(:package) } subject { build(:package) }
......
This diff is collapsed.
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