com.hazelcast.query.impl.AndResultSet Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2018, Hazelcast, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.query.impl;
import com.hazelcast.query.Predicate;
import java.util.AbstractSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Set;
import static com.hazelcast.util.Preconditions.isNotNull;
/**
* And Result set for Predicates.
*/
public class AndResultSet extends AbstractSet {
private static final int SIZE_UNINITIALIZED = -1;
private final Set setSmallest;
private final List> otherIndexedResults;
private final List lsNoIndexPredicates;
private int cachedSize;
public AndResultSet(Set setSmallest, List> otherIndexedResults,
List lsNoIndexPredicates) {
this.setSmallest = isNotNull(setSmallest, "setSmallest");
this.otherIndexedResults = otherIndexedResults;
this.lsNoIndexPredicates = lsNoIndexPredicates;
this.cachedSize = SIZE_UNINITIALIZED;
}
@Override
public boolean contains(Object o) {
if (!setSmallest.contains(o)) {
return false;
}
if (otherIndexedResults != null) {
for (Set otherIndexedResult : otherIndexedResults) {
if (!otherIndexedResult.contains(o)) {
return false;
}
}
}
if (lsNoIndexPredicates != null) {
for (Predicate noIndexPredicate : lsNoIndexPredicates) {
if (!noIndexPredicate.apply((Map.Entry) o)) {
return false;
}
}
}
return true;
}
@Override
public Iterator iterator() {
return new It();
}
class It implements Iterator {
QueryableEntry currentEntry;
final Iterator it = setSmallest.iterator();
@Override
public boolean hasNext() {
if (currentEntry != null) {
return true;
}
while (it.hasNext()) {
QueryableEntry entry = it.next();
if (checkOtherIndexedResults(entry) && checkNoIndexPredicates(entry)) {
currentEntry = entry;
return true;
}
}
return false;
}
private boolean checkNoIndexPredicates(QueryableEntry currentEntry) {
if (lsNoIndexPredicates == null) {
return true;
}
for (Predicate noIndexPredicate : lsNoIndexPredicates) {
if (!noIndexPredicate.apply(currentEntry)) {
return false;
}
}
return true;
}
private boolean checkOtherIndexedResults(QueryableEntry currentEntry) {
if (otherIndexedResults == null) {
return true;
}
for (Set otherIndexedResult : otherIndexedResults) {
if (!otherIndexedResult.contains(currentEntry)) {
return false;
}
}
return true;
}
@Override
public QueryableEntry next() {
if (!hasNext()) {
throw new NoSuchElementException();
}
QueryableEntry result = currentEntry;
currentEntry = null;
return result;
}
@Override
public void remove() {
throw new UnsupportedOperationException();
}
}
@Override
public int size() {
if (cachedSize == SIZE_UNINITIALIZED) {
int calculatedSize = 0;
for (Iterator it = iterator(); it.hasNext(); it.next()) {
calculatedSize++;
}
cachedSize = calculatedSize;
}
return cachedSize;
}
/**
* @return returns estimated size without calculating the full result set in full-result scan.
*/
public int estimatedSize() {
if (cachedSize == SIZE_UNINITIALIZED) {
if (setSmallest == null) {
return 0;
} else {
return setSmallest.size();
}
}
return cachedSize;
}
}