From 51284b5243052dfd7951cad3276dbcc09d78fefe Mon Sep 17 00:00:00 2001
From: "Brian S. O'Neill" <bronee@gmail.com>
Date: Sat, 10 May 2008 22:02:17 +0000
Subject: Fix binding of filters which already contain some bound properties.

---
 .../amazon/carbonado/filter/BinaryOpFilter.java    |  2 +-
 .../java/com/amazon/carbonado/filter/Binder.java   | 76 +++++++++++++++++-----
 .../com/amazon/carbonado/qe/StandardQuery.java     |  6 +-
 3 files changed, 64 insertions(+), 20 deletions(-)

(limited to 'src')

diff --git a/src/main/java/com/amazon/carbonado/filter/BinaryOpFilter.java b/src/main/java/com/amazon/carbonado/filter/BinaryOpFilter.java
index 23797a2..b1cb970 100644
--- a/src/main/java/com/amazon/carbonado/filter/BinaryOpFilter.java
+++ b/src/main/java/com/amazon/carbonado/filter/BinaryOpFilter.java
@@ -67,7 +67,7 @@ public abstract class BinaryOpFilter<S extends Storable> extends Filter<S> {
         if (isBound()) {
             return this;
         }
-        return accept(new Binder<S>(), null);
+        return Binder.doBind(this);
     }
 
     @Override
diff --git a/src/main/java/com/amazon/carbonado/filter/Binder.java b/src/main/java/com/amazon/carbonado/filter/Binder.java
index 91a03e5..f29e8b3 100644
--- a/src/main/java/com/amazon/carbonado/filter/Binder.java
+++ b/src/main/java/com/amazon/carbonado/filter/Binder.java
@@ -31,16 +31,36 @@ import com.amazon.carbonado.Storable;
  * @author Brian S O'Neill
  */
 class Binder<S extends Storable> extends Visitor<S, Filter<S>, Object> {
-    // Maps PropertyFilter with bind ID zero to PropertyFilter with highest
-    // bind ID.
+    static <S extends Storable> Filter<S> doBind(Filter<S> filter) {
+        return doBind(filter, null);
+    }
+
+    private static <S extends Storable> Filter<S> doBind(Filter<S> filter, Binder<S> binder) {
+        binder = new Binder<S>(binder);
+        Filter<S> boundFilter = filter.accept(binder, null);
+        if (binder.isRebindNeeded()) {
+            binder = new Binder<S>(binder);
+            boundFilter = boundFilter.accept(binder, null);
+        }
+        return boundFilter;
+    }
+
+    // Maps PropertyFilter with bind ID zero to PropertyFilter with highest bind
+    // ID. All other mappings track which bindings have been created.
     private final Map<PropertyFilter<S>, PropertyFilter<S>> mBindMap;
 
-    Binder() {
-        mBindMap = new IdentityHashMap<PropertyFilter<S>, PropertyFilter<S>>();
+    private boolean mNeedsRebind;
+
+    private Binder(Binder<S> binder) {
+        if (binder == null) {
+            mBindMap = new IdentityHashMap<PropertyFilter<S>, PropertyFilter<S>>();
+        } else {
+            mBindMap = binder.mBindMap;
+        }
     }
 
-    private Binder(Map<PropertyFilter<S>, PropertyFilter<S>> bindMap) {
-        mBindMap = bindMap;
+    public boolean isRebindNeeded() {
+        return mNeedsRebind;
     }
 
     @Override
@@ -80,19 +100,42 @@ class Binder<S extends Storable> extends Visitor<S, Filter<S>, Object> {
     }
 
     @Override
-    public Filter<S> visit(PropertyFilter<S> filter, Object param) {
+    public Filter<S> visit(final PropertyFilter<S> filter, Object param) {
         if (filter.isBound()) {
+            if (mBindMap.containsKey(filter)) {
+                // Binding was created by this Binder.
+                return filter;
+            }
+            final PropertyFilter<S> zero = PropertyFilter.getCanonical(filter, 0);
+            PropertyFilter<S> highest = mBindMap.get(zero);
+            if (highest == null) {
+                mBindMap.put(zero, filter);
+            } else {
+                // Have already created bindings which clash with existing
+                // bindings.
+                mNeedsRebind = true;
+                if (filter.getBindID() > highest.getBindID()) {
+                    mBindMap.put(zero, filter);
+                }
+            }
             return filter;
-        }
-        filter = PropertyFilter.getCanonical(filter, 1);
-        PropertyFilter<S> highest = mBindMap.get(filter);
-        if (highest == null) {
-            highest = filter;
         } else {
-            highest = PropertyFilter.getCanonical(filter, highest.getBindID() + 1);
+            final PropertyFilter<S> zero;
+            if (filter.getBindID() == 0) {
+                zero = filter;
+            } else {
+                zero = PropertyFilter.getCanonical(filter, 0);
+            }
+            PropertyFilter<S> highest = mBindMap.get(zero);
+            if (highest == null) {
+                highest = PropertyFilter.getCanonical(filter, 1);
+            } else {
+                highest = PropertyFilter.getCanonical(filter, highest.getBindID() + 1);
+            }
+            mBindMap.put(zero, highest);
+            mBindMap.put(highest, highest);
+            return highest;
         }
-        mBindMap.put(filter, highest);
-        return highest;
     }
 
     @Override
@@ -100,8 +143,7 @@ class Binder<S extends Storable> extends Visitor<S, Filter<S>, Object> {
         if (filter.isBound()) {
             return filter;
         }
-        Filter<S> boundJoinedSubFilter =
-            filter.getJoinedSubFilter().accept(new Binder<S>(mBindMap), null);
+        Filter<S> boundJoinedSubFilter = doBind(filter.getJoinedSubFilter(), this);
         Filter<S>.NotJoined nj =
             boundJoinedSubFilter.notJoinedFromAny(filter.getChainedProperty());
         if (nj.getRemainderFilter() != null && !(nj.getRemainderFilter().isOpen())) {
diff --git a/src/main/java/com/amazon/carbonado/qe/StandardQuery.java b/src/main/java/com/amazon/carbonado/qe/StandardQuery.java
index 1bb03be..e2081d6 100644
--- a/src/main/java/com/amazon/carbonado/qe/StandardQuery.java
+++ b/src/main/java/com/amazon/carbonado/qe/StandardQuery.java
@@ -220,7 +220,8 @@ public abstract class StandardQuery<S extends Storable> extends AbstractQuery<S>
     private Query<S> buildAfter(S start, OrderingList<S> orderings) throws FetchException {
         Class<S> storableType = getStorableType();
         Filter<S> orderFilter = Filter.getClosedFilter(storableType);
-        Filter<S> lastSubFilter = Filter.getOpenFilter(storableType);
+        Filter<S> openFilter = Filter.getOpenFilter(storableType);
+        Filter<S> lastSubFilter = openFilter;
 
         Object[] values = new Object[orderings.size()];
 
@@ -240,7 +241,8 @@ public abstract class StandardQuery<S extends Storable> extends AbstractQuery<S>
                 break;
             }
 
-            lastSubFilter = lastSubFilter.and(propertyName, RelOp.EQ).bind();
+            Filter<S> propFilter = openFilter.and(propertyName, RelOp.EQ).bind();
+            lastSubFilter = lastSubFilter.and(propFilter);
         }
 
         Query<S> query = this.and(orderFilter);
-- 
cgit v1.2.3