gitlab-org--gitlab-foss/lib/gitlab/gitaly_client/diff_stitcher.rb

35 lines
982 B
Ruby
Raw Normal View History

2017-05-30 15:30:05 -04:00
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?
diff_params = diff_msg.to_h.slice(*GitalyClient::Diff::ATTRS)
2017-06-21 19:51:46 -04:00
# 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
2017-05-30 15:30:05 -04:00
current_diff = GitalyClient::Diff.new(diff_params)
else
current_diff.patch += diff_msg.raw_patch_data
end
if diff_msg.end_of_patch
yield current_diff
current_diff = nil
end
end
end
end
end
end