From 56f2de870a7279e11575f6d0c6f2f9eea1374407 Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Tue, 20 Jul 2010 17:30:54 -0700
Subject: take advantage of inheritence for easier codes

---
 lib/arel/algebra/relations/writes.rb | 30 ++++++++++++++++--------------
 1 file changed, 16 insertions(+), 14 deletions(-)

(limited to 'lib/arel/algebra')

diff --git a/lib/arel/algebra/relations/writes.rb b/lib/arel/algebra/relations/writes.rb
index d344987915..17da2b172b 100644
--- a/lib/arel/algebra/relations/writes.rb
+++ b/lib/arel/algebra/relations/writes.rb
@@ -1,33 +1,35 @@
 module Arel
-  class Deletion < Compound
-    attributes :relation
-    deriving :initialize, :==
+  class Action < Compound
+    def == other
+      super || self.class === other && @relation == other.relation
+    end
+  end
 
+  class Deletion < Action
     def call
       engine.delete(self)
     end
   end
 
-  class Insert < Compound
-    attributes :relation, :record
-    deriving :==
+  class Insert < Action
+    attr_reader :record
 
     def initialize(relation, record)
-      @relation, @record = relation, record.bind(relation)
+      super(relation)
+      @record = record.bind(relation)
     end
 
     def call
       engine.create(self)
     end
-  end
 
-  class Update < Compound
-    attributes :relation, :assignments
-    deriving :==
-
-    def initialize(relation, assignments)
-      @relation, @assignments = relation, assignments.bind(relation)
+    def == other
+      super && @record == other.record
     end
+  end
+
+  class Update < Insert
+    alias :assignments :record
 
     def call
       engine.update(self)
-- 
cgit v1.2.3