1 files changed, 2 insertions, 14 deletions
diff --git a/lib/gitlab/gitaly_client/diff.rb b/lib/gitlab/gitaly_client/diff.rb
index 1e117b7e74a..d98a0ce988f 100644
--- a/lib/gitlab/gitaly_client/diff.rb
+++ b/lib/gitlab/gitaly_client/diff.rb
@@ -1,21 +1,9 @@
 module Gitlab
   module GitalyClient
     class Diff
-      FIELDS = %i(from_path to_path old_mode new_mode from_id to_id patch).freeze
+      ATTRS = %i(from_path to_path old_mode new_mode from_id to_id patch overflow_marker collapsed).freeze
 
-      attr_accessor(*FIELDS)
-
-      def initialize(params)
-        params.each do |key, val|
-          public_send(:"#{key}=", val)
-        end
-      end
-
-      def ==(other)
-        FIELDS.all? do |field|
-          public_send(field) == other.public_send(field)
-        end
-      end
+      include AttributesBag
     end
   end
 end