summaryrefslogtreecommitdiff
path: root/lib/gitlab/gitaly_client/diff_stitcher.rb
blob: d84e8d752dc36a77ef70e3d73652cbaa023b8a3b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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::FIELDS)
            diff_params[:patch] = diff_msg.raw_patch_data

            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