Add an attributes bag class as a GitalyClient helperfix/add-attributes-bag-class-to-gitaly-client

1 files changed, 31 insertions, 0 deletions

diff --git a/lib/gitlab/gitaly_client/attributes_bag.rb b/lib/gitlab/gitaly_client/attributes_bag.rb
new file mode 100644
index 00000000000..198a1de91c7
--- /dev/null
+++ b/lib/gitlab/gitaly_client/attributes_bag.rb
@@ -0,0 +1,31 @@
+module Gitlab
+  module GitalyClient
+    # This module expects an `ATTRS` const to be defined on the subclass
+    # See GitalyClient::WikiFile for an example
+    module AttributesBag
+      extend ActiveSupport::Concern
+
+      included do
+        attr_accessor(*const_get(:ATTRS))
+      end
+
+      def initialize(params)
+        params = params.with_indifferent_access
+
+        attributes.each do |attr|
+          instance_variable_set("@#{attr}", params[attr])
+        end
+      end
+
+      def ==(other)
+        attributes.all? do |field|
+          instance_variable_get("@#{field}") == other.instance_variable_get("@#{field}")
+        end
+      end
+
+      def attributes
+        self.class.const_get(:ATTRS)
+      end
+    end
+  end
+end