Commit 03f3350f authored by Ahmad Sherif's avatar Ahmad Sherif

Fetch commit signatures from Gitaly in batches

Closes gitaly#1046
parent 1c2b633b
...@@ -411,7 +411,7 @@ group :ed25519 do ...@@ -411,7 +411,7 @@ group :ed25519 do
end end
# Gitaly GRPC client # Gitaly GRPC client
gem 'gitaly-proto', '~> 0.85.0', require: 'gitaly' gem 'gitaly-proto', '~> 0.87.0', require: 'gitaly'
# Locked until https://github.com/google/protobuf/issues/4210 is closed # Locked until https://github.com/google/protobuf/issues/4210 is closed
gem 'google-protobuf', '= 3.5.1' gem 'google-protobuf', '= 3.5.1'
......
...@@ -285,7 +285,7 @@ GEM ...@@ -285,7 +285,7 @@ GEM
po_to_json (>= 1.0.0) po_to_json (>= 1.0.0)
rails (>= 3.2.0) rails (>= 3.2.0)
gherkin-ruby (0.3.2) gherkin-ruby (0.3.2)
gitaly-proto (0.85.0) gitaly-proto (0.87.0)
google-protobuf (~> 3.1) google-protobuf (~> 3.1)
grpc (~> 1.0) grpc (~> 1.0)
github-linguist (5.3.3) github-linguist (5.3.3)
...@@ -1057,7 +1057,7 @@ DEPENDENCIES ...@@ -1057,7 +1057,7 @@ DEPENDENCIES
gettext (~> 3.2.2) gettext (~> 3.2.2)
gettext_i18n_rails (~> 1.8.0) gettext_i18n_rails (~> 1.8.0)
gettext_i18n_rails_js (~> 1.2.0) gettext_i18n_rails_js (~> 1.2.0)
gitaly-proto (~> 0.85.0) gitaly-proto (~> 0.87.0)
github-linguist (~> 5.3.3) github-linguist (~> 5.3.3)
gitlab-flowdock-git-hook (~> 1.0.1) gitlab-flowdock-git-hook (~> 1.0.1)
gitlab-markup (~> 1.6.2) gitlab-markup (~> 1.6.2)
......
...@@ -14,8 +14,6 @@ class Projects::CommitsController < Projects::ApplicationController ...@@ -14,8 +14,6 @@ class Projects::CommitsController < Projects::ApplicationController
@merge_request = MergeRequestsFinder.new(current_user, project_id: @project.id).execute.opened @merge_request = MergeRequestsFinder.new(current_user, project_id: @project.id).execute.opened
.find_by(source_project: @project, source_branch: @ref, target_branch: @repository.root_ref) .find_by(source_project: @project, source_branch: @ref, target_branch: @repository.root_ref)
# https://gitlab.com/gitlab-org/gitaly/issues/931
Gitlab::GitalyClient.allow_n_plus_1_calls do
respond_to do |format| respond_to do |format|
format.html format.html
format.atom { render layout: 'xml.atom' } format.atom { render layout: 'xml.atom' }
...@@ -29,11 +27,8 @@ class Projects::CommitsController < Projects::ApplicationController ...@@ -29,11 +27,8 @@ class Projects::CommitsController < Projects::ApplicationController
end end
end end
end end
end
def signatures def signatures
# https://gitlab.com/gitlab-org/gitaly/issues/931
Gitlab::GitalyClient.allow_n_plus_1_calls do
respond_to do |format| respond_to do |format|
format.json do format.json do
render json: { render json: {
...@@ -47,7 +42,6 @@ class Projects::CommitsController < Projects::ApplicationController ...@@ -47,7 +42,6 @@ class Projects::CommitsController < Projects::ApplicationController
end end
end end
end end
end
private private
......
...@@ -19,6 +19,7 @@ class Commit ...@@ -19,6 +19,7 @@ class Commit
attr_accessor :project, :author attr_accessor :project, :author
attr_accessor :redacted_description_html attr_accessor :redacted_description_html
attr_accessor :redacted_title_html attr_accessor :redacted_title_html
attr_reader :gpg_commit
DIFF_SAFE_LINES = Gitlab::Git::DiffCollection::DEFAULT_LIMITS[:max_lines] DIFF_SAFE_LINES = Gitlab::Git::DiffCollection::DEFAULT_LIMITS[:max_lines]
...@@ -110,6 +111,7 @@ class Commit ...@@ -110,6 +111,7 @@ class Commit
@raw = raw_commit @raw = raw_commit
@project = project @project = project
@statuses = {} @statuses = {}
@gpg_commit = Gitlab::Gpg::Commit.new(self) if project
end end
def id def id
...@@ -452,8 +454,4 @@ class Commit ...@@ -452,8 +454,4 @@ class Commit
def merged_merge_request_no_cache(user) def merged_merge_request_no_cache(user)
MergeRequestsFinder.new(user, project_id: project.id).find_by(merge_commit_sha: id) if merge_commit? MergeRequestsFinder.new(user, project_id: project.id).find_by(merge_commit_sha: id) if merge_commit?
end end
def gpg_commit
@gpg_commit ||= Gitlab::Gpg::Commit.new(self)
end
end end
...@@ -250,6 +250,45 @@ module Gitlab ...@@ -250,6 +250,45 @@ module Gitlab
end end
end end
def extract_signature_lazily(repository, commit_id)
BatchLoader.for({ repository: repository, commit_id: commit_id }).batch do |items, loader|
items_by_repo = items.group_by { |i| i[:repository] }
items_by_repo.each do |repo, items|
commit_ids = items.map { |i| i[:commit_id] }
signatures = batch_signature_extraction(repository, commit_ids)
signatures.each do |commit_sha, signature_data|
loader.call({ repository: repository, commit_id: commit_sha }, signature_data)
end
end
end
end
def batch_signature_extraction(repository, commit_ids)
repository.gitaly_migrate(:extract_commit_signature_in_batch) do |is_enabled|
if is_enabled
gitaly_batch_signature_extraction(repository, commit_ids)
else
rugged_batch_signature_extraction(repository, commit_ids)
end
end
end
def gitaly_batch_signature_extraction(repository, commit_ids)
repository.gitaly_commit_client.get_commit_signatures(commit_ids)
end
def rugged_batch_signature_extraction(repository, commit_ids)
commit_ids.each_with_object({}) do |commit_id, signatures|
signature_data = rugged_extract_signature(repository, commit_id)
next unless signature_data
signatures[commit_id] = signature_data
end
end
def rugged_extract_signature(repository, commit_id) def rugged_extract_signature(repository, commit_id)
begin begin
Rugged::Commit.extract_signature(repository.rugged, commit_id) Rugged::Commit.extract_signature(repository.rugged, commit_id)
......
...@@ -319,6 +319,23 @@ module Gitlab ...@@ -319,6 +319,23 @@ module Gitlab
[signature, signed_text] [signature, signed_text]
end end
def get_commit_signatures(commit_ids)
request = Gitaly::GetCommitSignaturesRequest.new(repository: @gitaly_repo, commit_ids: commit_ids)
response = GitalyClient.call(@repository.storage, :commit_service, :get_commit_signatures, request)
signatures = Hash.new { |h, k| h[k] = [''.b, ''.b] }
current_commit_id = nil
response.each do |message|
current_commit_id = message.commit_id if message.commit_id.present?
signatures[current_commit_id].first << message.signature
signatures[current_commit_id].last << message.signed_text
end
signatures
end
private private
def call_commit_diff(request_params, options = {}) def call_commit_diff(request_params, options = {})
......
module Gitlab module Gitlab
module Gpg module Gpg
class Commit class Commit
include Gitlab::Utils::StrongMemoize
def initialize(commit) def initialize(commit)
@commit = commit @commit = commit
repo = commit.project.repository.raw_repository repo = commit.project.repository.raw_repository
@signature_text, @signed_text = Gitlab::Git::Commit.extract_signature(repo, commit.sha) @signature_data = Gitlab::Git::Commit.extract_signature_lazily(repo, commit.sha || commit.id)
end
def signature_text
strong_memoize(:signature_text) do
@signature_data&.itself && @signature_data[0]
end
end
def signed_text
strong_memoize(:signed_text) do
@signature_data&.itself && @signature_data[1]
end
end end
def has_signature? def has_signature?
!!(@signature_text && @signed_text) !!(signature_text && signed_text)
end end
def signature def signature
...@@ -53,7 +67,7 @@ module Gitlab ...@@ -53,7 +67,7 @@ module Gitlab
end end
def verified_signature def verified_signature
@verified_signature ||= GPGME::Crypto.new.verify(@signature_text, signed_text: @signed_text) do |verified_signature| @verified_signature ||= GPGME::Crypto.new.verify(signature_text, signed_text: signed_text) do |verified_signature|
break verified_signature break verified_signature
end end
end end
......
...@@ -393,10 +393,7 @@ describe Gitlab::Git::Commit, seed_helper: true do ...@@ -393,10 +393,7 @@ describe Gitlab::Git::Commit, seed_helper: true do
end end
end end
describe '.extract_signature' do shared_examples 'extracting commit signature' do
subject { described_class.extract_signature(repository, commit_id) }
shared_examples '.extract_signature' do
context 'when the commit is signed' do context 'when the commit is signed' do
let(:commit_id) { '0b4bc9a49b562e85de7cc9e834518ea6828729b9' } let(:commit_id) { '0b4bc9a49b562e85de7cc9e834518ea6828729b9' }
...@@ -462,12 +459,45 @@ describe Gitlab::Git::Commit, seed_helper: true do ...@@ -462,12 +459,45 @@ describe Gitlab::Git::Commit, seed_helper: true do
end end
end end
describe '.extract_signature_lazily' do
shared_examples 'loading signatures in batch once' do
it 'fetches signatures in batch once' do
commit_ids = %w[0b4bc9a49b562e85de7cc9e834518ea6828729b9 4b4918a572fa86f9771e5ba40fbd48e1eb03e2c6]
signatures = commit_ids.map do |commit_id|
described_class.extract_signature_lazily(repository, commit_id)
end
expect(described_class).to receive(:batch_signature_extraction)
.with(repository, commit_ids)
.once
.and_return({})
2.times { signatures.each(&:itself) }
end
end
subject { described_class.extract_signature_lazily(repository, commit_id).itself }
context 'with Gitaly extract_commit_signature_in_batch feature enabled' do
it_behaves_like 'extracting commit signature'
it_behaves_like 'loading signatures in batch once'
end
context 'with Gitaly extract_commit_signature_in_batch feature disabled', :disable_gitaly do
it_behaves_like 'extracting commit signature'
it_behaves_like 'loading signatures in batch once'
end
end
describe '.extract_signature' do
subject { described_class.extract_signature(repository, commit_id) }
context 'with gitaly' do context 'with gitaly' do
it_behaves_like '.extract_signature' it_behaves_like 'extracting commit signature'
end end
context 'without gitaly', :skip_gitaly_mock do context 'without gitaly', :disable_gitaly do
it_behaves_like '.extract_signature' it_behaves_like 'extracting commit signature'
end end
end end
end end
......
...@@ -38,7 +38,7 @@ describe Gitlab::Gpg::Commit do ...@@ -38,7 +38,7 @@ describe Gitlab::Gpg::Commit do
end end
before do before do
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return( .and_return(
[ [
...@@ -101,7 +101,7 @@ describe Gitlab::Gpg::Commit do ...@@ -101,7 +101,7 @@ describe Gitlab::Gpg::Commit do
end end
before do before do
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return( .and_return(
[ [
...@@ -140,7 +140,7 @@ describe Gitlab::Gpg::Commit do ...@@ -140,7 +140,7 @@ describe Gitlab::Gpg::Commit do
end end
before do before do
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return( .and_return(
[ [
...@@ -175,7 +175,7 @@ describe Gitlab::Gpg::Commit do ...@@ -175,7 +175,7 @@ describe Gitlab::Gpg::Commit do
end end
before do before do
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return( .and_return(
[ [
...@@ -211,7 +211,7 @@ describe Gitlab::Gpg::Commit do ...@@ -211,7 +211,7 @@ describe Gitlab::Gpg::Commit do
end end
before do before do
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return( .and_return(
[ [
...@@ -241,7 +241,7 @@ describe Gitlab::Gpg::Commit do ...@@ -241,7 +241,7 @@ describe Gitlab::Gpg::Commit do
let!(:commit) { create :commit, project: project, sha: commit_sha } let!(:commit) { create :commit, project: project, sha: commit_sha }
before do before do
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return( .and_return(
[ [
......
...@@ -26,7 +26,7 @@ RSpec.describe Gitlab::Gpg::InvalidGpgSignatureUpdater do ...@@ -26,7 +26,7 @@ RSpec.describe Gitlab::Gpg::InvalidGpgSignatureUpdater do
before do before do
allow_any_instance_of(Project).to receive(:commit).and_return(commit) allow_any_instance_of(Project).to receive(:commit).and_return(commit)
allow(Gitlab::Git::Commit).to receive(:extract_signature) allow(Gitlab::Git::Commit).to receive(:extract_signature_lazily)
.with(Gitlab::Git::Repository, commit_sha) .with(Gitlab::Git::Repository, commit_sha)
.and_return(signature) .and_return(signature)
end end
......
...@@ -15,8 +15,8 @@ describe MergeRequests::BuildService do ...@@ -15,8 +15,8 @@ describe MergeRequests::BuildService do
let(:target_branch) { 'master' } let(:target_branch) { 'master' }
let(:merge_request) { service.execute } let(:merge_request) { service.execute }
let(:compare) { double(:compare, commits: commits) } let(:compare) { double(:compare, commits: commits) }
let(:commit_1) { double(:commit_1, safe_message: "Initial commit\n\nCreate the app") } let(:commit_1) { double(:commit_1, sha: 'f00ba7', safe_message: "Initial commit\n\nCreate the app") }
let(:commit_2) { double(:commit_2, safe_message: 'This is a bad commit message!') } let(:commit_2) { double(:commit_2, sha: 'f00ba7', safe_message: 'This is a bad commit message!') }
let(:commits) { nil } let(:commits) { nil }
let(:service) do let(:service) do
......
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