Skip to content

Commit

Permalink
Forward map value extractors to HD indexes
Browse files Browse the repository at this point in the history
This change fixes two issues:

1. The consistency issue: queryable entries returned by HD indexes were
unaware of the map value extractors, so it was impossible to extract
values provided by the extractors from the returned entries.

2. The performance issue: Extractors.empty is a pretty expensive call
and it was performed for every returned entry. According to a slightly
modified version of HDIndexPerfTest the throughput is about 2 times
higher now.

Fixes: https://github.com/hazelcast/hazelcast-enterprise/issues/2109
  • Loading branch information
taburet authored and pveentjer committed Sep 17, 2018
1 parent 823ee87 commit 61c69be
Show file tree
Hide file tree
Showing 2 changed files with 111 additions and 2 deletions.
Expand Up @@ -39,22 +39,22 @@ public class IndexImpl implements Index {

protected final InternalSerializationService ss;
protected final IndexStore indexStore;
protected final Extractors extractors;
private final IndexCopyBehavior copyQueryResultOn;

private volatile TypeConverter converter;

private final String attributeName;
private final boolean ordered;
private final Extractors extractors;

public IndexImpl(String attributeName, boolean ordered, InternalSerializationService ss, Extractors extractors,
IndexCopyBehavior copyQueryResultOn) {
this.attributeName = attributeName;
this.ordered = ordered;
this.ss = ss;
this.extractors = extractors;
this.copyQueryResultOn = copyQueryResultOn;
this.indexStore = createIndexStore(ordered);
this.extractors = extractors;
}

public IndexStore createIndexStore(boolean ordered) {
Expand Down
@@ -0,0 +1,109 @@
/*
* 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.map.impl.query;

import com.hazelcast.config.Config;
import com.hazelcast.config.InMemoryFormat;
import com.hazelcast.config.MapAttributeConfig;
import com.hazelcast.config.MapIndexConfig;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.core.IMap;
import com.hazelcast.query.Predicate;
import com.hazelcast.query.Predicates;
import com.hazelcast.query.extractor.ValueCollector;
import com.hazelcast.query.extractor.ValueExtractor;
import com.hazelcast.spi.properties.GroupProperty;
import com.hazelcast.test.HazelcastParallelParametersRunnerFactory;
import com.hazelcast.test.HazelcastTestSupport;
import com.hazelcast.test.annotation.ParallelTest;
import com.hazelcast.test.annotation.QuickTest;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;

import java.io.Serializable;
import java.util.Collection;

import static com.hazelcast.query.Predicates.equal;
import static java.util.Arrays.asList;
import static org.junit.Assert.assertEquals;

@RunWith(Parameterized.class)
@Parameterized.UseParametersRunnerFactory(HazelcastParallelParametersRunnerFactory.class)
@Category({QuickTest.class, ParallelTest.class})
public class ExtractorsAndIndexesTest extends HazelcastTestSupport {

@Parameterized.Parameters(name = "format:{0}")
public static Collection<Object[]> parameters() {
return asList(new Object[][]{{InMemoryFormat.OBJECT}, {InMemoryFormat.BINARY}});
}

@Parameterized.Parameter
public InMemoryFormat inMemoryFormat;

@Test
public void testExtractorsAreRespectedByEntriesReturnedFromIndexes() {
String mapName = randomMapName();

Config config = new Config();
config.getMapConfig(mapName).setInMemoryFormat(inMemoryFormat).addMapIndexConfig(new MapIndexConfig("last", true))
.addMapAttributeConfig(new MapAttributeConfig("generated", Extractor.class.getName()));
config.getNativeMemoryConfig().setEnabled(true);

config.setProperty(GroupProperty.PARTITION_COUNT.getName(), "1");
HazelcastInstance instance = createHazelcastInstance(config);

IMap<Integer, Person> map = instance.getMap(mapName);
populateMap(map);

// this predicate queries the index
Predicate lastPredicate = equal("last", "last");

// this predicate is not indexed and acts on the entries returned from
// the index which must support extractors otherwise this test will fail
Predicate alwaysFirst = equal("generated", "first");

Predicate composed = Predicates.and(lastPredicate, alwaysFirst);

Collection<Person> values = map.values(composed);
assertEquals(100, values.size());
}

public static class Person implements Serializable {
public String first;
public String last;
}

public static class Extractor extends ValueExtractor<Person, Void> {
@SuppressWarnings("unchecked")
@Override
public void extract(Person person, Void aVoid, ValueCollector valueCollector) {
valueCollector.addObject("first");
}
}

private void populateMap(IMap<Integer, Person> map) {
for (int i = 0; i < 100; i++) {
Person p = new Person();
p.first = "first" + i;
p.last = "last";
map.put(i, p);
}
}

}

0 comments on commit 61c69be

Please sign in to comment.