Add ID type knowledge to LREngine

SFDO-Community · Nov 23, 2015 · 7584ca2 · 7584ca2
1 parent 4bc42b6
commit 7584ca2
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/rolluptool/src/classes/LREngine.cls b/rolluptool/src/classes/LREngine.cls
@@ -372,6 +372,8 @@ public class LREngine {
         public boolean isMasterTypeCurrency;
         public boolean isMasterTypeText;
         public boolean isDetailTypeText;
+        public boolean isMasterTypeId;
+        public boolean isDetailTypeId;
 
         public RollupSummaryField(Schema.Describefieldresult m, 
                                          Schema.Describefieldresult d, RollupOperation op) {
@@ -396,6 +398,8 @@ public class LREngine {
             this.isMasterTypeCurrency = isCurrency(master.getType());
             this.isMasterTypeText = isText(master.getType());
             this.isDetailTypeText = isText(detail.getType());
+			this.isMasterTypeId = isIdOrReference(master.getType());
+            this.isDetailTypeId = isIdOrReference(detail.getType());
             // validate if field is good to work on later 
             validate();
         }   
@@ -414,8 +418,9 @@ public class LREngine {
             if (operation == RollupOperation.First ||
                 operation == RollupOperation.Last) {
                 if ( (this.master.getType() != this.detail.getType()) &&
-                     (!isDetailTypeText && !isMasterTypeText) ) {
-                    throw new BadRollUpSummaryStateException('Master and detail fields must be the same field type (or text based) for First or Last operations');
+                     ((!isDetailTypeText && !isDetailTypeId) || 
+                     (!isMasterTypeText && !isMasterTypeId))) {
+                    throw new BadRollUpSummaryStateException('Master and detail fields must be the same field type (or text/Id based) for First or Last operations');
                 }
             }
 
@@ -440,6 +445,11 @@ public class LREngine {
                 dt == Schema.Displaytype.Picklist ||
                 dt == Schema.Displaytype.MultiPicklist;
         }
+
+        boolean isIdOrReference (Schema.Displaytype dt) {
+            return dt == Schema.DisplayType.ID ||
+                   dt == Schema.DisplayType.REFERENCE;
+        }
 
         boolean isNumber (Schema.Displaytype dt) {
             return dt == Schema.Displaytype.Currency 
@@ -592,4 +602,4 @@ public class LREngine {
         }
     }
 
-}
+}