debian-mirror-gitlab/lib/gitlab/gitaly_client/diff_stitcher.rb

37 lines
1 KiB
Ruby
Raw Normal View History

2018-12-23 12:14:25 +05:30
# frozen_string_literal: true
2017-09-10 17:25:29 +05:30
module Gitlab
module GitalyClient
class DiffStitcher
include Enumerable
def initialize(rpc_response)
@rpc_response = rpc_response
end
def each
current_diff = nil
@rpc_response.each do |diff_msg|
if current_diff.nil?
2018-03-17 18:26:18 +05:30
diff_params = diff_msg.to_h.slice(*GitalyClient::Diff::ATTRS)
2017-09-10 17:25:29 +05:30
# gRPC uses frozen strings by default, and we need to have an unfrozen string as it
# gets processed further down the line. So we unfreeze the first chunk of the patch
# in case it's the only chunk we receive for this diff.
diff_params[:patch] = diff_msg.raw_patch_data.dup
current_diff = GitalyClient::Diff.new(diff_params)
else
2018-12-23 12:14:25 +05:30
current_diff.patch = "#{current_diff.patch}#{diff_msg.raw_patch_data}"
2017-09-10 17:25:29 +05:30
end
if diff_msg.end_of_patch
yield current_diff
current_diff = nil
end
end
end
end
end
end