1 files changed, 17 insertions, 0 deletions
diff --git a/config/initializers/relationship.rb b/config/initializers/relationship.rb
new file mode 100644
index 000000000..492aa627f
--- /dev/null
+++ b/config/initializers/relationship.rb
@@ -0,0 +1,17 @@
+if Rails.env.development?
+  require 'rails_erd/domain/relationship'
+
+  module RailsERD
+    class Domain
+      class Relationship
+        class << self
+          private
+
+          def association_identity(association)
+            Set[association_owner(association), association_target(association)]
+          end
+        end
+      end
+    end
+  end
+end